gmakstutis/graphiti
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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(