Add support for Fireworks embedding model***

n4ze3m · Feb 19, 2024 · 08a13b2 · 08a13b2
1 parent 0b68717
commit 08a13b2
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/server/prisma/seed.ts b/server/prisma/seed.ts
@@ -341,6 +341,12 @@ const EMBEDDING_MODELS: {
     name: "text-embedding-3-large (OpenAI)",
     model_type: "embedding",
     model_provider: "OpenAI",
+  },
+  {
+    model_provider: "Fireworks",
+    model_type: "embedding",
+    model_id: "dialoqbase_eb_nomic-ai/nomic-embed-text-v1.5",
+    name: "nomic-ai/nomic-embed-text-v1.5 (Fireworks)",
   }
 ];
 

diff --git a/server/src/utils/embeddings.ts b/server/src/utils/embeddings.ts
@@ -80,6 +80,15 @@ export const embeddings = (
           },
         },
       });
+    case "fireworks":
+      return new OpenAIEmbeddings({
+        modelName,
+        openAIApiKey: process.env.FIREWORKS_API_KEY!,
+        configuration: {
+          baseURL: "https://api.fireworks.ai/inference/v1",
+          apiKey: process.env.FIREWORKS_API_KEY!,
+        },
+      });
     default:
       console.log("Using Default Embeddings");
       return new OpenAIEmbeddings();
@@ -96,4 +105,5 @@ export const supportedEmbeddings = [
   "supabase",
   "jina",
   "google",
+  "fireworks",
 ];
diff --git a/server/src/utils/models.ts b/server/src/utils/models.ts
@@ -65,7 +65,7 @@ export const chatModelProvider = (
       return new ChatOpenAI({
         modelName: modelName,
         temperature: temperature,
-        openAIApiKey: otherFields.apiKey || process.env.OPENAI_API_KEY,
+        openAIApiKey: otherFields.apiKey || process.env.OPENAI_API_KEY ,
         ...otherFields,
         configuration: {
           baseURL: otherFields.baseURL,