ปลั๊กอิน Chroma มีการติดตั้งใช้งานเครื่องมือจัดทำดัชนีและเครื่องมือดึงข้อมูลที่ใช้ฐานข้อมูลเวกเตอร์ Chroma ในโหมดไคลเอ็นต์/เซิร์ฟเวอร์
การติดตั้ง
npm i --save genkitx-chromadb
การกำหนดค่า
หากต้องการใช้ปลั๊กอินนี้ ให้ระบุเมื่อเริ่มต้น Genkit
import { genkit } from 'genkit';
import { chroma } from 'genkitx-chromadb';
const ai = genkit({
plugins: [
chroma([
{
collectionName: 'bob_collection',
embedder: textEmbedding004,
},
]),
],
});
คุณต้องระบุคอลเล็กชัน Chroma และรูปแบบการฝังที่ต้องการใช้ นอกจากนี้ พารามิเตอร์ที่ไม่บังคับอีก 2 รายการ ได้แก่
clientParams
: หากคุณไม่ได้เรียกใช้เซิร์ฟเวอร์ Chroma ในเครื่องเดียวกับเวิร์กโฟลว์ Genkit คุณจะต้องระบุตัวเลือกการตรวจสอบสิทธิ์ หรือหากไม่ได้ใช้การกำหนดค่าเซิร์ฟเวอร์ Chroma เริ่มต้น คุณสามารถระบุ ChromaChromaClientParams object
เพื่อส่งไปยังไคลเอ็นต์ Chroma ได้โดยทำดังนี้clientParams: { path: "http://192.168.10.42:8000", }
embedderOptions
: ใช้พารามิเตอร์นี้เพื่อส่งตัวเลือกไปยังผู้ฝังembedderOptions: { taskType: 'RETRIEVAL_DOCUMENT' },
การใช้งาน
นําเข้าข้อมูลอ้างอิงเครื่องมือดึงข้อมูลและเครื่องมือจัดทำดัชนีดังนี้
import { chromaRetrieverRef } from 'genkitx-chromadb';
import { chromaIndexerRef } from 'genkitx-chromadb';
จากนั้นใช้ข้อมูลอ้างอิงด้วย ai.retrieve()
และ ai.index()
// To use the index you configured when you loaded the plugin:
let docs = await ai.retrieve({ retriever: chromaRetrieverRef, query });
// To specify an index:
export const bobFactsRetriever = chromaRetrieverRef({
collectionName: 'bob-facts',
});
docs = await ai.retrieve({ retriever: bobFactsRetriever, query });
// To use the index you configured when you loaded the plugin:
await ai.index({ indexer: chromaIndexerRef, documents });
// To specify an index:
export const bobFactsIndexer = chromaIndexerRef({
collectionName: 'bob-facts',
});
await ai.index({ indexer: bobFactsIndexer, documents });
ดูการอภิปรายทั่วไปเกี่ยวกับเครื่องมือจัดทําดัชนีและเครื่องมือดึงข้อมูลได้ในหน้าการสร้างที่เพิ่มการดึงข้อมูล