yujunjun/graphiti
1
0
Fork 0
mirror of https://github.com/getzep/graphiti.git synced 2025-06-27 02:00:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: Pass embedding model in search utils

Browse Source
This commit is contained in:
paulpaliychuk 2024-09-26 16:20:01 -04:00
parent e4bc756c31
commit 3efad4c225
1 changed files with 25 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
graphiti_core/search/search.py
View File

@ -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(