ปลั๊กอินโครมา

ปลั๊กอิน 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 เริ่มต้น คุณสามารถระบุ Chroma ChromaClientParams 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 });

ดูการอภิปรายทั่วไปเกี่ยวกับเครื่องมือจัดทําดัชนีและเครื่องมือดึงข้อมูลได้ในหน้าการสร้างที่เพิ่มการดึงข้อมูล