bugfix: set embedding model name

devflowinc · Apr 9, 2024 · 087cbf9 · 087cbf9
1 parent bb0e35d
commit 087cbf9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/dashboard/src/components/NewDatasetModal.tsx b/dashboard/src/components/NewDatasetModal.tsx
@@ -228,6 +228,8 @@ export const NewDatasetModal = (props: NewDatasetModalProps) => {
                               return {
                                 ...prev,
                                 EMBEDDING_SIZE: embeddingSize,
+                                EMBEDDING_MODEL_NAME:
+                                  selectedModel?.id ?? "text-embedding-3-small",
                                 EMBEDDING_BASE_URL:
                                   selectedModel?.url ??
                                   "https://api.openai.com/v1",

diff --git a/dashboard/src/types/apiTypes.ts b/dashboard/src/types/apiTypes.ts
@@ -253,6 +253,12 @@ export const availableEmbeddingModels = [
     url: "https://api.openai.com/v1",
     dimension: 1536,
   },
+  {
+    id: "text-embedding-3-large",
+    name: "text-embedding-3-large (hosted by OpenAI)",
+    url: "https://api.openai.com/v1",
+    dimension: 3072,
+  },
 ];
 
 export interface EventDTO {