diff options
Diffstat (limited to 'apps/cf-ai-backend/src/OpenAIEmbedder.ts')
| -rw-r--r-- | apps/cf-ai-backend/src/OpenAIEmbedder.ts | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/apps/cf-ai-backend/src/OpenAIEmbedder.ts b/apps/cf-ai-backend/src/OpenAIEmbedder.ts index 35c36c74..e227d1e3 100644 --- a/apps/cf-ai-backend/src/OpenAIEmbedder.ts +++ b/apps/cf-ai-backend/src/OpenAIEmbedder.ts @@ -1,4 +1,4 @@ -import { AiTextGenerationOutput } from "@cloudflare/ai/dist/ai/tasks/text-generation"; +import { AiTextGenerationOutput } from '@cloudflare/ai/dist/ai/tasks/text-generation'; interface OpenAIEmbeddingsParams { apiKey: string; @@ -15,7 +15,7 @@ export class OpenAIEmbeddings { } async embedDocuments(texts: string[]): Promise<number[][]> { - const responses = await Promise.all(texts.map(text => this.embedQuery(text))); + const responses = await Promise.all(texts.map((text) => this.embedQuery(text))); return responses; } @@ -24,18 +24,18 @@ export class OpenAIEmbeddings { method: 'POST', headers: { 'Content-Type': 'application/json', - 'Authorization': `Bearer ${this.apiKey}` + Authorization: `Bearer ${this.apiKey}`, }, body: JSON.stringify({ input: text, - model: this.modelName - }) + model: this.modelName, + }), }); - const data = await response.json() as { + const data = (await response.json()) as { data: { - embedding: number[] - }[] + embedding: number[]; + }[]; }; return data.data[0].embedding; |