铬嵌入物

这个软件包为Chroma的云嵌入服务所托管的Quen模型家族提供了嵌入功能.

安装

npm install @chroma-core/chroma-cloud-qwen

使用量

import { ChromaClient } from "chromadb";
import {
  ChromaCloudQwenEmbeddingFunction,
  ChromaCloudQwenEmbeddingModel,
  ChromaCloudQwenEmbeddingTask,
} from "@chroma-core/chroma-cloud-qwen";

// Initialize the embedder
const embedder = new ChromaCloudQwenEmbeddingFunction({
  model: ChromaCloudQwenEmbeddingModel.QWEN3_EMBEDDING_0p6B,
  task: ChromaCloudQwenEmbeddingTask.CODE_TO_CODE,
  apiKeyEnvVar: "CHROMA_API_KEY",
});

// 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,
});

配置

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

export CHROMA_API_KEY=your-api-key

获取您的 API 密钥 克罗玛的仪表板。 。 。 .

配置选项

• 模式: 嵌入所用的模型
• 任务: 正在生成嵌入的任务
• 指令( d): 用于自定义 Quen 模型的字典绘图任务和目标(文档/queries)
• apiKeyEnvVar 密码API 密钥的环境变量名( 默认 ): CHROMA_API_KEY(中文(简体) )

支持的模式

• Qwen/Qwen3-Embedding-0.6B