OpenAI Chroma 嵌入函数

此软件包为Chroma提供了OpenAI嵌入提供商.

安装

npm install @chroma-core/openai

使用量

import { ChromaClient } from 'chromadb';
import { OpenAIEmbeddingFunction } from '@chroma-core/openai';

// Initialize the embedder
const embedder = new OpenAIEmbeddingFunction({
  apiKey: 'your-api-key', // Or set OPENAI_API_KEY env var
  modelName: 'text-embedding-3-small',
  // Optional: specify dimensions for supported models
  dimensions: 512,
  // Optional: specify organization ID
  organizationId: 'your-org-id',
  // Optional: specify API base (e.g. for Azure OpenAI)
  apiBase: 'your-api-base'
});

// Create a new ChromaClient
const client = new ChromaClient({
  path: 'http://localhost:8000',
});

// Create a collection with the embedder
const collection = await client.createCollection({
  name: 'my-collection',
  embeddingFunction: embedder,
});

// Add documents
await collection.add({
  ids: ["1", "2", "3"],
  documents: ["Document 1", "Document 2", "Document 3"],
});

// Query documents
const results = await collection.query({
  queryTexts: ["Sample query"],
  nResults: 2,
});

配置

将您的 OpenAI API 密钥设定为环境变量 :

export OPENAI_API_KEY=your-api-key

或者直接传给建筑师.

支持的模式

• text-embedding-3-small (默认)
• text-embedding-3-large
• text-embedding-ada-002

对于支持它的模型(如 text-embedding-3-small 财务报告和审定财务报表 text-embedding-3-large),可以指定自定义维度来缩小嵌入大小.