graphiti-core 0.20.3__py3-none-any.whl → 0.20.4__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/graphiti.py +9 -1
- {graphiti_core-0.20.3.dist-info → graphiti_core-0.20.4.dist-info}/METADATA +1 -1
- {graphiti_core-0.20.3.dist-info → graphiti_core-0.20.4.dist-info}/RECORD +5 -5
- {graphiti_core-0.20.3.dist-info → graphiti_core-0.20.4.dist-info}/WHEEL +0 -0
- {graphiti_core-0.20.3.dist-info → graphiti_core-0.20.4.dist-info}/licenses/LICENSE +0 -0
graphiti_core/graphiti.py
CHANGED
|
@@ -122,6 +122,11 @@ class AddBulkEpisodeResults(BaseModel):
|
|
|
122
122
|
community_edges: list[CommunityEdge]
|
|
123
123
|
|
|
124
124
|
|
|
125
|
+
class AddTripletResults(BaseModel):
|
|
126
|
+
nodes: list[EntityNode]
|
|
127
|
+
edges: list[EntityEdge]
|
|
128
|
+
|
|
129
|
+
|
|
125
130
|
class Graphiti:
|
|
126
131
|
def __init__(
|
|
127
132
|
self,
|
|
@@ -1015,7 +1020,9 @@ class Graphiti:
|
|
|
1015
1020
|
|
|
1016
1021
|
return SearchResults(edges=edges, nodes=nodes)
|
|
1017
1022
|
|
|
1018
|
-
async def add_triplet(
|
|
1023
|
+
async def add_triplet(
|
|
1024
|
+
self, source_node: EntityNode, edge: EntityEdge, target_node: EntityNode
|
|
1025
|
+
) -> AddTripletResults:
|
|
1019
1026
|
if source_node.name_embedding is None:
|
|
1020
1027
|
await source_node.generate_name_embedding(self.embedder)
|
|
1021
1028
|
if target_node.name_embedding is None:
|
|
@@ -1059,6 +1066,7 @@ class Graphiti:
|
|
|
1059
1066
|
await create_entity_node_embeddings(self.embedder, nodes)
|
|
1060
1067
|
|
|
1061
1068
|
await add_nodes_and_edges_bulk(self.driver, [], [], nodes, edges, self.embedder)
|
|
1069
|
+
return AddTripletResults(edges=edges, nodes=nodes)
|
|
1062
1070
|
|
|
1063
1071
|
async def remove_episode(self, episode_uuid: str):
|
|
1064
1072
|
# Find the episode to be deleted
|
|
@@ -2,7 +2,7 @@ graphiti_core/__init__.py,sha256=e5SWFkRiaUwfprYIeIgVIh7JDedNiloZvd3roU-0aDY,55
|
|
|
2
2
|
graphiti_core/edges.py,sha256=O-WqtqR9w6ZO5nbi4VRveXVrinfTcZmDJ6PcjudDAsY,18821
|
|
3
3
|
graphiti_core/errors.py,sha256=cH_v9TPgEPeQE6GFOHIg5TvejpUCBddGarMY2Whxbwc,2707
|
|
4
4
|
graphiti_core/graph_queries.py,sha256=9DWMiFTB-OmodMDaOws0lwzgiD7EUDNO7mAFJ1nxusE,6624
|
|
5
|
-
graphiti_core/graphiti.py,sha256=
|
|
5
|
+
graphiti_core/graphiti.py,sha256=eSQyajym9kefrf6PDCxCDzjWaco4hgFryyosg27husA,41601
|
|
6
6
|
graphiti_core/graphiti_types.py,sha256=C_p2XwScQlCzo7ets097TrSLs9ATxPZQ4WCsxDS7QHc,1066
|
|
7
7
|
graphiti_core/helpers.py,sha256=6q_wpiOW3_j28EfZ7FgWW7Hl5pONj_5zvVXZGW9FxTU,5175
|
|
8
8
|
graphiti_core/nodes.py,sha256=sJGCaEWroCL7rL7rUPqFw_NjTY-aNdc44fFOsvKZpdw,26399
|
|
@@ -74,7 +74,7 @@ graphiti_core/utils/maintenance/node_operations.py,sha256=r9ilkA01eq1z-nF8P_s1EX
|
|
|
74
74
|
graphiti_core/utils/maintenance/temporal_operations.py,sha256=IIaVtShpVkOYe6haxz3a1x3v54-MzaEXG8VsxFUNeoY,3582
|
|
75
75
|
graphiti_core/utils/maintenance/utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
76
76
|
graphiti_core/utils/ontology_utils/entity_types_utils.py,sha256=4eVgxLWY6Q8k9cRJ5pW59IYF--U4nXZsZIGOVb_yHfQ,1285
|
|
77
|
-
graphiti_core-0.20.
|
|
78
|
-
graphiti_core-0.20.
|
|
79
|
-
graphiti_core-0.20.
|
|
80
|
-
graphiti_core-0.20.
|
|
77
|
+
graphiti_core-0.20.4.dist-info/METADATA,sha256=2lAkky1kK3GCk-H_BH3j-QZkqRobR5Gb0rFTKbbaQ4E,26773
|
|
78
|
+
graphiti_core-0.20.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
|
|
79
|
+
graphiti_core-0.20.4.dist-info/licenses/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
|
|
80
|
+
graphiti_core-0.20.4.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|