bugfix: jina-code embedding model correct name

devflowinc · May 14, 2024 · c7050c9 · c7050c9
1 parent 06f29c4
commit c7050c9
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/server/src/operators/model_operator.rs b/server/src/operators/model_operator.rs
@@ -71,6 +71,12 @@ pub async fn create_embedding(
                 .filter(|s| !s.is_empty())
                 .unwrap_or("https://embedding.trieve.ai/jinaai-code".to_string())
         }
+        "https://embedding.trieve.ai/jina-code" => {
+            std::env::var("EMBEDDING_SERVER_ORIGIN_JINA_CODE")
+                .ok()
+                .filter(|s| !s.is_empty())
+                .unwrap_or("https://embedding.trieve.ai/jina-code".to_string())
+        }
         _ => config_embedding_base_url,
     };