Merge pull request #381 from naynaly10/patch-4

Fix OpenAIEmbeddingFunction in JS to use passed model name
chroma-core · Apr 22, 2023 · b96fc38 · b96fc38
2 parents a8d9800 + 893b5d8
commit b96fc38
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/clients/js/src/index.ts b/clients/js/src/index.ts
@@ -51,7 +51,7 @@ export class OpenAIEmbeddingFunction {
     const openai = new OpenAIApi.OpenAIApi(configuration);
     const embeddings = [];
     const response = await openai.createEmbedding({
-      model: "text-embedding-ada-002",
+      model: this.model,
       input: texts,
     });
     const data = response.data['data'];
@@ -436,4 +436,4 @@ export class ChromaClient {
     return response
   }
 
-}
+}