graphiti-core 0.7.5__py3-none-any.whl → 0.7.6__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of graphiti-core might be problematic. Click here for more details.
- graphiti_core/utils/maintenance/node_operations.py +11 -7
- {graphiti_core-0.7.5.dist-info → graphiti_core-0.7.6.dist-info}/METADATA +1 -1
- {graphiti_core-0.7.5.dist-info → graphiti_core-0.7.6.dist-info}/RECORD +5 -5
- {graphiti_core-0.7.5.dist-info → graphiti_core-0.7.6.dist-info}/LICENSE +0 -0
- {graphiti_core-0.7.5.dist-info → graphiti_core-0.7.6.dist-info}/WHEEL +0 -0
|
@@ -158,13 +158,17 @@ async def extract_nodes(
|
|
|
158
158
|
node_classifications: dict[str, str | None] = {}
|
|
159
159
|
|
|
160
160
|
if entity_types is not None:
|
|
161
|
-
|
|
162
|
-
|
|
163
|
-
|
|
164
|
-
|
|
165
|
-
|
|
166
|
-
|
|
167
|
-
|
|
161
|
+
try:
|
|
162
|
+
llm_response = await llm_client.generate_response(
|
|
163
|
+
prompt_library.extract_nodes.classify_nodes(node_classification_context),
|
|
164
|
+
response_model=EntityClassification,
|
|
165
|
+
)
|
|
166
|
+
entities = llm_response.get('entities', [])
|
|
167
|
+
entity_classifications = llm_response.get('entity_classifications', [])
|
|
168
|
+
node_classifications.update(dict(zip(entities, entity_classifications)))
|
|
169
|
+
# catch classification errors and continue if we can't classify
|
|
170
|
+
except Exception as e:
|
|
171
|
+
logger.exception(e)
|
|
168
172
|
|
|
169
173
|
end = time()
|
|
170
174
|
logger.debug(f'Extracted new nodes: {extracted_node_names} in {(end - start) * 1000} ms')
|
|
@@ -52,10 +52,10 @@ graphiti_core/utils/maintenance/__init__.py,sha256=TRY3wWWu5kn3Oahk_KKhltrWnh0NA
|
|
|
52
52
|
graphiti_core/utils/maintenance/community_operations.py,sha256=gIw1M5HGgc2c3TXag5ygPPpAv5WsG-yoC8Lhmfr6FMs,10011
|
|
53
53
|
graphiti_core/utils/maintenance/edge_operations.py,sha256=tNw56vN586JYZMgie6RLRTiHZ680-kWzDIxW8ucL6SU,12780
|
|
54
54
|
graphiti_core/utils/maintenance/graph_data_operations.py,sha256=qds9ALk9PhpQs1CNZTZGpi70mqJ93Y2KhIh9X2r8MUI,6533
|
|
55
|
-
graphiti_core/utils/maintenance/node_operations.py,sha256=
|
|
55
|
+
graphiti_core/utils/maintenance/node_operations.py,sha256=ZorGKps8sijOVS1V0Dv1oe3yWzq9s4u8stbe-4IY0vQ,14468
|
|
56
56
|
graphiti_core/utils/maintenance/temporal_operations.py,sha256=RdNtubCyYhOVrvcOIq2WppHls1Q-BEjtsN8r38l-Rtc,3691
|
|
57
57
|
graphiti_core/utils/maintenance/utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
58
|
-
graphiti_core-0.7.
|
|
59
|
-
graphiti_core-0.7.
|
|
60
|
-
graphiti_core-0.7.
|
|
61
|
-
graphiti_core-0.7.
|
|
58
|
+
graphiti_core-0.7.6.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
|
|
59
|
+
graphiti_core-0.7.6.dist-info/METADATA,sha256=BnBzvu8YGhYy8aqvVvPH288bebp-kjUQf5YdoxbNSEY,10541
|
|
60
|
+
graphiti_core-0.7.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
|
|
61
|
+
graphiti_core-0.7.6.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|