graphiti-core 0.9.5__py3-none-any.whl → 0.9.6__py3-none-any.whl

graphiti_core/search/search_config.py

@@ -26,7 +26,7 @@ from graphiti_core.search.search_utils import (
     MAX_SEARCH_DEPTH,
 )
 
-DEFAULT_SEARCH_LIMIT = 20
+DEFAULT_SEARCH_LIMIT = 10
 
 
 class EdgeSearchMethod(Enum):

graphiti_core/search/search_helpers.py

@@ -14,6 +14,8 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
+import json
+
 from graphiti_core.edges import EntityEdge
 from graphiti_core.search.search_config import SearchResults
 
@@ -25,23 +27,37 @@ def format_edge_date_range(edge: EntityEdge) -> str:
 
 def search_results_to_context_string(search_results: SearchResults) -> str:
     """Reformats a set of SearchResults into a single string to pass directly to an LLM as context"""
-    context_string = """FACTS and ENTITIES represent relevant context to the current conversation.
-                        COMMUNITIES represent a cluster of closely related entities.
-
-                        # These are the most relevant facts and their valid date ranges
-                        # format: FACT (Date range: from - to)
-                    """
-    context_string += '<FACTS>\n'
-    for edge in search_results.edges:
-        context_string += f'- {edge.fact} ({format_edge_date_range(edge)})\n'
-    context_string += '</FACTS>\n'
-    context_string += '<ENTITIES>\n'
-    for node in search_results.nodes:
-        context_string += f'- {node.name}: {node.summary}\n'
-    context_string += '</ENTITIES>\n'
-    context_string += '<COMMUNITIES>\n'
-    for community in search_results.communities:
-        context_string += f'- {community.name}: {community.summary}\n'
-    context_string += '</COMMUNITIES>\n'
+    fact_json = [
+        {
+            'fact': edge.fact,
+            'valid_at': str(edge.valid_at),
+            'invalid_at': str(edge.invalid_at or 'Present'),
+        }
+        for edge in search_results.edges
+    ]
+    entity_json = [
+        {'entity_name': node.name, 'summary': node.summary} for node in search_results.nodes
+    ]
+    community_json = [
+        {'community_name': community.name, 'summary': community.summary}
+        for community in search_results.communities
+    ]
+
+    context_string = f"""
+    FACTS and ENTITIES represent relevant context to the current conversation.
+    COMMUNITIES represent a cluster of closely related entities.
+
+    These are the most relevant facts and their valid and invalid dates. Facts are considered valid
+    between their valid_at and invalid_at dates. Facts with an invalid_at date of "Present" are considered valid.
+    <FACTS>
+    {json.dumps(fact_json, indent=12)}
+    </FACTS>
+    <ENTITIES>
+    {json.dumps(entity_json, indent=12)}
+    </ENTITIES>
+    <COMMUNITIES>
+    {json.dumps(community_json, indent=12)}
+    </COMMUNITIES>
+"""
 
     return context_string

{graphiti_core-0.9.5.dist-info → graphiti_core-0.9.6.dist-info}/METADATA

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

{graphiti_core-0.9.5.dist-info → graphiti_core-0.9.6.dist-info}/RECORD

@@ -43,10 +43,10 @@ graphiti_core/prompts/summarize_nodes.py,sha256=PeA1Taov5KBNNBKgrCPeF1tLg4_SMgT-
 graphiti_core/py.typed,sha256=vlmmzQOt7bmeQl9L3XJP4W6Ry0iiELepnOrinKz5KQg,79
 graphiti_core/search/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 graphiti_core/search/search.py,sha256=DX-tcIa0SiKI2HX-b_WdjGE74A8RLWQor4p90dJluUA,12643
-graphiti_core/search/search_config.py,sha256=Yn_p1cnMwtZAdIOga19iaCoS7FmA95-GLN7KmHd-7N0,3117
+graphiti_core/search/search_config.py,sha256=UZN8jFA4pBlw2O5N1cuhVRBdTwMLR9N3Oyo6sQ4MDVw,3117
 graphiti_core/search/search_config_recipes.py,sha256=yUqiLnn9vFg39M8eVwjVKfBCL_ptGrfDMQ47m_Blb0g,6885
 graphiti_core/search/search_filters.py,sha256=JkP7NbM4Dor27dne5vAuxbJic12dIJDtWJxNqmVuRec,5884
-graphiti_core/search/search_helpers.py,sha256=Tt5rRUfblBlFqV7r8XuGZiT5P_wjlh01xS0feELZjXg,2173
+graphiti_core/search/search_helpers.py,sha256=7hHmhAbuo88CkKFy-U6lRtObzw89hzJIV4rdF1bd134,2367
 graphiti_core/search/search_utils.py,sha256=4i5Uwp6iNz849njIf89jOgI2NDtSFHQ_grnZpLSNv0I,25771
 graphiti_core/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 graphiti_core/utils/bulk_utils.py,sha256=P4LKO46Yle4tBdNcQ3hDHcSQFaR8UBLfoL-z1M2Wua0,14690
@@ -59,7 +59,7 @@ graphiti_core/utils/maintenance/node_operations.py,sha256=WhZQixx05dAFFQAd5KTXJ8
 graphiti_core/utils/maintenance/temporal_operations.py,sha256=RdNtubCyYhOVrvcOIq2WppHls1Q-BEjtsN8r38l-Rtc,3691
 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.9.5.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
-graphiti_core-0.9.5.dist-info/METADATA,sha256=ptj44vyxZwC9xQmIsnEeKGTpUniDD4fLz7gRKMAXqI4,14394
-graphiti_core-0.9.5.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-graphiti_core-0.9.5.dist-info/RECORD,,
+graphiti_core-0.9.6.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
+graphiti_core-0.9.6.dist-info/METADATA,sha256=Op_DuEb-dSaqcyI5d1yIxwG7Belqdu1_cmb7kimVWs0,14394
+graphiti_core-0.9.6.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+graphiti_core-0.9.6.dist-info/RECORD,,