graphiti-core 0.11.6rc4__py3-none-any.whl → 0.11.6rc6__py3-none-any.whl

graphiti_core/helpers.py

@@ -18,6 +18,7 @@ import asyncio
 import os
 from collections.abc import Coroutine
 from datetime import datetime
+from typing import Any
 
 import numpy as np
 from dotenv import load_dotenv
@@ -91,11 +92,27 @@ def normalize_l2(embedding: list[float]):
 
 
 # Use this instead of asyncio.gather() to bound coroutines
-async def semaphore_gather(*coroutines: Coroutine, max_coroutines: int = SEMAPHORE_LIMIT):
+async def semaphore_gather(
+    *coroutines: Coroutine,
+    max_coroutines: int = SEMAPHORE_LIMIT,
+):
     semaphore = asyncio.Semaphore(max_coroutines)
 
-    async def _wrap_coroutine(coroutine):
+    async def _wrap(coro: Coroutine) -> Any:
         async with semaphore:
-            return await coroutine
-
-    return await asyncio.gather(*(_wrap_coroutine(coroutine) for coroutine in coroutines))
+            return await coro
+
+    results = []
+    batch = []
+    for coroutine in coroutines:
+        batch.append(_wrap(coroutine))
+        # once we hit max_coroutines, gather and clear the batch
+        if len(batch) >= max_coroutines:
+            results.extend(await asyncio.gather(*batch))
+            batch.clear()
+
+    # gather any remaining coroutines in the final batch
+    if batch:
+        results.extend(await asyncio.gather(*batch))
+
+    return results

graphiti_core/utils/maintenance/edge_operations.py

@@ -399,7 +399,7 @@ async def dedupe_extracted_edge(
     )
 
     if duplicate_fact_id >= 0 and episode is not None:
-        edge.episodes += episode.uuid
+        edge.episodes.append(episode.uuid)
 
     end = time()
     logger.debug(

{graphiti_core-0.11.6rc4.dist-info → graphiti_core-0.11.6rc6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: graphiti-core
-Version: 0.11.6rc4
+Version: 0.11.6rc6
 Summary: A temporal graph building library
 License: Apache-2.0
 Author: Paul Paliychuk

{graphiti_core-0.11.6rc4.dist-info → graphiti_core-0.11.6rc6.dist-info}/RECORD

@@ -12,7 +12,7 @@ graphiti_core/embedder/voyage.py,sha256=gQhdcz2IYPSyOcDn3w8aHToVS3KQhyZrUBm4vqr3
 graphiti_core/errors.py,sha256=Nib1uQx2cO_VOizupmRjpFfmuRg-hFAVqTtZAuBehR8,2405
 graphiti_core/graphiti.py,sha256=niRU1sZ2hf3a8WUPQAEIOsg8ixR_r_NUau5famfe1uM,27090
 graphiti_core/graphiti_types.py,sha256=46ueysKPwUCpxkMePHdCJLspfTImoZN7JiRwpz7cqd0,1013
-graphiti_core/helpers.py,sha256=Rj_ayM7hphlyY4v0ln_9vjFtr76aTjLVenYny0Wntwo,3086
+graphiti_core/helpers.py,sha256=iRN1gFMrYa07C5MyaQkefiD5t0itfNddczyLmLIxRs0,3465
 graphiti_core/llm_client/__init__.py,sha256=PA80TSMeX-sUXITXEAxMDEt3gtfZgcJrGJUcyds1mSo,207
 graphiti_core/llm_client/anthropic_client.py,sha256=392rtkH_I7yOJUlQvjoOnS8Lz14WBP8egQ3OfRH0nFs,12481
 graphiti_core/llm_client/client.py,sha256=v_w5TBbDJYYADCXSs2r287g5Ami2Urma-GGEbHSI_Jg,5826
@@ -54,13 +54,13 @@ graphiti_core/utils/bulk_utils.py,sha256=_56TZ_gyOt3V5UAmiGhSNtcWOVgZdLZ9rAYCkvk
 graphiti_core/utils/datetime_utils.py,sha256=Ti-2tnrDFRzBsbfblzsHybsM3jaDLP4-VT2t0VhpIzU,1357
 graphiti_core/utils/maintenance/__init__.py,sha256=vW4H1KyapTl-OOz578uZABYcpND4wPx3Vt6aAPaXh78,301
 graphiti_core/utils/maintenance/community_operations.py,sha256=TF-4eHuvMe_jMqvWg3swxK80zLLtOR0t1pmUUQlNulM,10067
-graphiti_core/utils/maintenance/edge_operations.py,sha256=mnm-9Qps6poV7ix0BtCYI3rACVb7b1JyA2gusiManQU,14751
+graphiti_core/utils/maintenance/edge_operations.py,sha256=vjeCUt_WlKZ3SRIIgaF9pqc8Na6ajLOzUIo2pWo7NyU,14756
 graphiti_core/utils/maintenance/graph_data_operations.py,sha256=BIJKc8tbvU4IjWxLgeotw57b1eE3Iw8YtV74j6eo4RQ,7493
 graphiti_core/utils/maintenance/node_operations.py,sha256=VMiDrzXryUV7-_btDVw87K3OawYeC9-9UqNQbfC5Bbo,15316
 graphiti_core/utils/maintenance/temporal_operations.py,sha256=mJkw9xLB4W2BsLfC5POr0r-PHWL9SIfNj_l_xu0B5ug,3410
 graphiti_core/utils/maintenance/utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 graphiti_core/utils/ontology_utils/entity_types_utils.py,sha256=QJX5cG0GSSNF_Mm_yrldr69wjVAbN_MxLhOSznz85Hk,1279
-graphiti_core-0.11.6rc4.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
-graphiti_core-0.11.6rc4.dist-info/METADATA,sha256=Q7NixQjg9eI5oOvT5faMjZcVDU8_H1xxV6u5rqgtyus,15301
-graphiti_core-0.11.6rc4.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-graphiti_core-0.11.6rc4.dist-info/RECORD,,
+graphiti_core-0.11.6rc6.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
+graphiti_core-0.11.6rc6.dist-info/METADATA,sha256=kpR2OKGYOpTt1EnVoD6cWhLOI0vXOGyWdeY9JgjQn7g,15301
+graphiti_core-0.11.6rc6.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+graphiti_core-0.11.6rc6.dist-info/RECORD,,