poc

dcSpark · Jul 10, 2024 · dafb06e · dafb06e
1 parent a2de4b9
commit dafb06e
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 4 deletions.
diff --git a/shinkai-bin/shinkai-node/src/llm_provider/execution/job_vector_search.rs b/shinkai-bin/shinkai-node/src/llm_provider/execution/job_vector_search.rs
@@ -172,6 +172,7 @@ impl JobManager {
             }
         }
 
+        // TODO: add this back under a flag
         // println!(
         //     "\n\n\nDone Vector Searching: {}\n------------------------------------------------",
         //     query_text

diff --git a/shinkai-bin/shinkai-node/src/vector_fs/vector_fs_internals.rs b/shinkai-bin/shinkai-node/src/vector_fs/vector_fs_internals.rs
@@ -52,7 +52,7 @@ impl VectorFSInternals {
     pub async fn new_empty() -> Self {
         let node_name = ShinkaiName::from_node_name("@@node1_test.shinkai".to_string()).unwrap();
         let default_embedding_model =
-            EmbeddingModelType::OllamaTextEmbeddingsInference(OllamaTextEmbeddingsInference::SnowflakeArcticEmbed_M);
+            EmbeddingModelType::OllamaTextEmbeddingsInference(OllamaTextEmbeddingsInference::JinaEmbeddingsV2BaseEs);
         let supported_embedding_models = vec![default_embedding_model.clone()];
         Self::new(node_name, default_embedding_model, supported_embedding_models).await
     }

diff --git a/shinkai-libs/shinkai-vector-resources/src/embedding_generator.rs b/shinkai-libs/shinkai-vector-resources/src/embedding_generator.rs
@@ -227,7 +227,7 @@ impl RemoteEmbeddingGenerator {
     /// Create a RemoteEmbeddingGenerator that uses the default model and server
     pub fn new_default() -> RemoteEmbeddingGenerator {
         let model_architecture =
-            EmbeddingModelType::OllamaTextEmbeddingsInference(OllamaTextEmbeddingsInference::SnowflakeArcticEmbed_M);
+            EmbeddingModelType::OllamaTextEmbeddingsInference(OllamaTextEmbeddingsInference::JinaEmbeddingsV2BaseEs);
         RemoteEmbeddingGenerator {
             model_type: model_architecture,
             api_url: DEFAULT_EMBEDDINGS_SERVER_URL.to_string(),

diff --git a/shinkai-libs/shinkai-vector-resources/src/vector_resource/document_resource.rs b/shinkai-libs/shinkai-vector-resources/src/vector_resource/document_resource.rs
@@ -582,7 +582,7 @@ impl DocumentVectorResource {
             Embedding::new(&String::new(), vec![]),
             Vec::new(),
             Vec::new(),
-            EmbeddingModelType::OllamaTextEmbeddingsInference(OllamaTextEmbeddingsInference::SnowflakeArcticEmbed_M),
+            EmbeddingModelType::OllamaTextEmbeddingsInference(OllamaTextEmbeddingsInference::JinaEmbeddingsV2BaseEs),
             is_merkelized,
             DistributionInfo::new_empty(),
         )

diff --git a/shinkai-libs/shinkai-vector-resources/src/vector_resource/map_resource.rs b/shinkai-libs/shinkai-vector-resources/src/vector_resource/map_resource.rs
@@ -435,7 +435,7 @@ impl MapVectorResource {
             Embedding::new(&String::new(), vec![]),
             HashMap::new(),
             HashMap::new(),
-            EmbeddingModelType::OllamaTextEmbeddingsInference(OllamaTextEmbeddingsInference::SnowflakeArcticEmbed_M),
+            EmbeddingModelType::OllamaTextEmbeddingsInference(OllamaTextEmbeddingsInference::JinaEmbeddingsV2BaseEs),
             is_merkelized,
             DistributionInfo::new_empty(),
         )