aboutsummaryrefslogtreecommitdiff
path: root/apps/cf-ai-backend/src/OpenAIEmbedder.ts
diff options
context:
space:
mode:
Diffstat (limited to 'apps/cf-ai-backend/src/OpenAIEmbedder.ts')
-rw-r--r--apps/cf-ai-backend/src/OpenAIEmbedder.ts16
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;