Merge 100a43b214 into 5e593dd096
This commit is contained in:
commit
fbab6b45ad
1 changed files with 5 additions and 1 deletions
|
|
@ -778,13 +778,17 @@ class Graphiti:
|
||||||
communities = []
|
communities = []
|
||||||
community_edges = []
|
community_edges = []
|
||||||
if update_communities:
|
if update_communities:
|
||||||
communities, community_edges = await semaphore_gather(
|
results = await semaphore_gather(
|
||||||
*[
|
*[
|
||||||
update_community(self.driver, self.llm_client, self.embedder, node)
|
update_community(self.driver, self.llm_client, self.embedder, node)
|
||||||
for node in nodes
|
for node in nodes
|
||||||
],
|
],
|
||||||
max_coroutines=self.max_coroutines,
|
max_coroutines=self.max_coroutines,
|
||||||
)
|
)
|
||||||
|
# Flatten results from multiple update_community calls
|
||||||
|
for community_list, edge_list in results:
|
||||||
|
communities.extend(community_list)
|
||||||
|
community_edges.extend(edge_list)
|
||||||
|
|
||||||
end = time()
|
end = time()
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue