Make text-embedding-3-large the default embedding model (#775)

* add openai to models list in ui update default models in backend * make text-embedding-3-large the default model
obot-platform · Dec 5, 2024 · ab0b558 · ab0b558
1 parent a4a5b61
commit ab0b558
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/pkg/controller/data/default-model-aliases.yaml b/pkg/controller/data/default-model-aliases.yaml
@@ -25,7 +25,7 @@ items:
     spec:
       manifest:
         alias: text-embedding
-        model: text-embedding-3-small
+        model: text-embedding-3-large
   - apiVersion: otto.otto8.ai/v1
     kind: DefaultModelAlias
     metadata:

diff --git a/pkg/controller/data/default-models.yaml b/pkg/controller/data/default-models.yaml
@@ -32,6 +32,17 @@ items:
         modelProvider: openai-model-provider
         active: true
         usage: text-embedding
+  - apiVersion: otto.otto8.ai/v1
+    kind: Model
+    metadata:
+      name: text-embedding-3-large
+      namespace: default
+    spec:
+      manifest:
+        targetModel: text-embedding-3-large
+        modelProvider: openai-model-provider
+        active: true
+        usage: text-embedding
   - apiVersion: otto.otto8.ai/v1
     kind: Model
     metadata:

diff --git a/pkg/controller/handlers/knowledgeset/knowledgeset.go b/pkg/controller/handlers/knowledgeset/knowledgeset.go
@@ -158,7 +158,7 @@ func (h *Handler) SetEmbeddingModel(req router.Request, _ router.Response) error
 	if err := req.Get(&defaultEmbeddingModel, req.Namespace, string(types.DefaultModelAliasTypeTextEmbedding)); err == nil {
 		ks.Status.TextEmbeddingModel = defaultEmbeddingModel.Spec.Manifest.Model
 	} else if apierrors.IsNotFound(err) {
-		ks.Status.TextEmbeddingModel = "text-embedding-3-small"
+		ks.Status.TextEmbeddingModel = "text-embedding-3-large"
 	} else if err != nil {
 		return err
 	}