Merge pull request #750 from dcSpark/fix/save-embeddings-in-tool_data

Fix/save embeddings in tool data
dcSpark · Dec 23, 2024 · 664c399 · 664c399
2 parents 096ad6f + 5ce8af3
commit 664c399
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/shinkai-libs/shinkai-sqlite/src/shinkai_tool_manager.rs b/shinkai-libs/shinkai-sqlite/src/shinkai_tool_manager.rs
@@ -4,6 +4,7 @@ use keyphrases::KeyPhraseExtractor;
 use rusqlite::{params, Result};
 use shinkai_tools_primitives::tools::shinkai_tool::{ShinkaiTool, ShinkaiToolHeader};
 use std::collections::HashSet;
+use shinkai_vector_resources::embeddings::Embedding;
 
 impl SqliteManager {
     // Adds a ShinkaiTool entry to the shinkai_tools table
@@ -44,6 +45,7 @@ impl SqliteManager {
 
         // Clone the tool to make it mutable
         let mut tool_clone = tool.clone();
+        tool_clone.set_embedding(Embedding::new("", embedding.clone()));
 
         // Determine if the tool can be enabled
         let is_enabled = tool_clone.is_enabled() && tool_clone.can_be_enabled();