From 3efad4c2253d492f615f0ec90a1cc6bed5e19134 Mon Sep 17 00:00:00 2001
From: paulpaliychuk <pavlo.paliychuk.ca@gmail.com>
Date: Thu, 26 Sep 2024 16:20:01 -0400
Subject: [PATCH] chore: Pass embedding model in search utils

---
 graphiti_core/search/search.py | 28 +++++++++++++++++++++++++---
 1 file changed, 25 insertions(+), 3 deletions(-)

diff --git a/graphiti_core/search/search.py b/graphiti_core/search/search.py
index 0fda25c3..070efcb3 100644
--- a/graphiti_core/search/search.py
+++ b/graphiti_core/search/search.py
@@ -68,12 +68,34 @@ async def search(
     group_ids = group_ids if group_ids else None
     edges, nodes, communities = await asyncio.gather(
         edge_search(
-            driver, embedder, query, group_ids, config.edge_config, center_node_uuid, config.limit
+            driver,
+            embedder,
+            query,
+            group_ids,
+            config.edge_config,
+            center_node_uuid,
+            config.limit,
+            config.embedding_model,
         ),
         node_search(
-            driver, embedder, query, group_ids, config.node_config, center_node_uuid, config.limit
+            driver,
+            embedder,
+            query,
+            group_ids,
+            config.node_config,
+            center_node_uuid,
+            config.limit,
+            config.embedding_model,
+        ),
+        community_search(
+            driver,
+            embedder,
+            query,
+            group_ids,
+            config.community_config,
+            config.limit,
+            config.embedding_model,
         ),
-        community_search(driver, embedder, query, group_ids, config.community_config, config.limit),
     )
 
     results = SearchResults(