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 index 1ca593a8a..135558dce 100644 --- 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 index e9549c876..36074a7ae 100644 --- 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 index 09509a524..f8dd42244 100644 --- 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 index e5071c70b..74e5d05d3 100644 --- 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 index b72becfc2..48d6d042c 100644 --- 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(), )