updated embeddings to openai

ChatWithPDF · Nov 7, 2023 · 1da591f · 1da591f
1 parent 1a74632
commit 1da591f
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/prisma/migrations/20231107060620_update_embeddings_vector_size/migration.sql b/prisma/migrations/20231107060620_update_embeddings_vector_size/migration.sql
@@ -0,0 +1,12 @@
+-- This is an empty migration.
+ALTER TABLE "prompt_history"
+DROP COLUMN "embedding",
+ADD COLUMN  "embedding" vector(1536);
+
+ALTER TABLE "document"
+DROP COLUMN "contentEmbedding",
+ADD COLUMN  "contentEmbedding" vector(1536),
+DROP COLUMN "headingEmbedding",
+ADD COLUMN  "headingEmbedding" vector(1536),
+DROP COLUMN "summaryEmbedding",
+ADD COLUMN  "summaryEmbedding" vector(1536);
diff --git a/src/modules/aiTools/ai-tools.service.ts b/src/modules/aiTools/ai-tools.service.ts
@@ -198,7 +198,7 @@ export class AiToolsService {
     let embeddings;
     try {
       embeddings = await fetch(
-        `${this.configService.get("AI_TOOLS_BASE_URL")}/embeddings/instructor/local/`,
+        `${this.configService.get("AI_TOOLS_BASE_URL")}/embeddings/openai/remote/`,
         requestOptions
       )
       embeddings = await embeddings.json()
@@ -241,7 +241,7 @@ export class AiToolsService {
     formdata.append('file', fs.createReadStream(csvFilePath), 'data.csv');
     try{
       const response = await axios.post(
-        `${this.configService.get("AI_TOOLS_BASE_URL")}/embeddings/instructor/local/`,
+        `${this.configService.get("AI_TOOLS_BASE_URL")}/embeddings/openai/remote/`,
         formdata, {
         headers: formdata.getHeaders(),
       });