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

graphiti_core/graphiti.py

@@ -33,6 +33,7 @@ from graphiti_core.nodes import CommunityNode, EntityNode, EpisodeType, Episodic
 from graphiti_core.search.search import SearchConfig, search
 from graphiti_core.search.search_config import DEFAULT_SEARCH_LIMIT, SearchResults
 from graphiti_core.search.search_config_recipes import (
+    COMBINED_HYBRID_SEARCH_CROSS_ENCODER,
     EDGE_HYBRID_SEARCH_NODE_DISTANCE,
     EDGE_HYBRID_SEARCH_RRF,
 )
@@ -647,7 +648,10 @@ class Graphiti:
         Perform a hybrid search on the knowledge graph.
 
         This method executes a search query on the graph, combining vector and
-        text-based search techniques to retrieve relevant facts.
+        text-based search techniques to retrieve relevant facts, returning the edges as a string.
+
+        This is our basic out-of-the-box search, for more robust results we recommend using our more advanced
+        search method graphiti.search_().
 
         Parameters
         ----------
@@ -668,8 +672,7 @@ class Graphiti:
         Notes
         -----
         This method uses a SearchConfig with num_episodes set to 0 and
-        num_results set to the provided num_results parameter. It then calls
-        the hybrid_search function to perform the actual search operation.
+        num_results set to the provided num_results parameter.
 
         The search is performed using the current date and time as the reference
         point for temporal relevance.
@@ -703,6 +706,27 @@ class Graphiti:
         bfs_origin_node_uuids: list[str] | None = None,
         search_filter: SearchFilters | None = None,
     ) -> SearchResults:
+        """DEPRECATED"""
+        return await self.search_(
+            query, config, group_ids, center_node_uuid, bfs_origin_node_uuids, search_filter
+        )
+
+    async def search_(
+        self,
+        query: str,
+        config: SearchConfig = COMBINED_HYBRID_SEARCH_CROSS_ENCODER,
+        group_ids: list[str] | None = None,
+        center_node_uuid: str | None = None,
+        bfs_origin_node_uuids: list[str] | None = None,
+        search_filter: SearchFilters | None = None,
+    ) -> SearchResults:
+        """search_ (replaces _search) is our advanced search method that returns Graph objects (nodes and edges) rather
+        than a list of facts. This endpoint allows the end user to utilize more advanced features such as filters and
+        different search and reranker methodologies across different layers in the graph.
+
+        For different config recipes refer to search/search_config_recipes.
+        """
+
         return await search(
             self.driver,
             self.embedder,

graphiti_core/search/search_config.py

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

graphiti_core/search/search_helpers.py

@@ -0,0 +1,47 @@
+"""
+Copyright 2024, Zep Software, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+from graphiti_core.edges import EntityEdge
+from graphiti_core.search.search_config import SearchResults
+
+
+def format_edge_date_range(edge: EntityEdge) -> str:
+    # return f"{datetime(edge.valid_at).strftime('%Y-%m-%d %H:%M:%S') if edge.valid_at else 'date unknown'} - {(edge.invalid_at.strftime('%Y-%m-%d %H:%M:%S') if edge.invalid_at else 'present')}"
+    return f'{edge.valid_at if edge.valid_at else "date unknown"} - {(edge.invalid_at if edge.invalid_at else "present")}'
+
+
+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'
+
+    return context_string

graphiti_core/search/search_utils.py

@@ -229,8 +229,8 @@ async def edge_similarity_search(
 
     query: LiteralString = (
         """
-                                                                                MATCH (n:Entity)-[r:RELATES_TO]->(m:Entity)
-                                                                                """
+                                                                                                MATCH (n:Entity)-[r:RELATES_TO]->(m:Entity)
+                                                                                                """
         + group_filter_query
         + filter_query
         + """\nWITH DISTINCT r, vector.similarity.cosine(r.fact_embedding, $search_vector) AS score

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: graphiti-core
-Version: 0.9.4
+Version: 0.9.5
 Summary: A temporal graph building library
 License: Apache-2.0
 Author: Paul Paliychuk
@@ -18,6 +18,7 @@ Provides-Extra: groq
 Requires-Dist: anthropic (>=0.49.0) ; extra == "anthropic"
 Requires-Dist: diskcache (>=5.6.3)
 Requires-Dist: google-genai (>=1.8.0) ; extra == "google-genai"
+Requires-Dist: graph-service (>=1.0.0.7,<2.0.0.0)
 Requires-Dist: groq (>=0.2.0) ; extra == "groq"
 Requires-Dist: neo4j (>=5.23.0)
 Requires-Dist: numpy (>=1.0.0)

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

@@ -10,7 +10,7 @@ graphiti_core/embedder/gemini.py,sha256=nE0XH8wYVGcPSO7DaNQ7kdsQLFSoH4FQOu2HMQUy
 graphiti_core/embedder/openai.py,sha256=fcU63koSRI-OjDuEcBfUKgXu8XV_-8EF6HpVrYa1_8I,1880
 graphiti_core/embedder/voyage.py,sha256=DZsH1nSTfP1vqCinNIIwSyEzv7jsyur2tKxlBv-ZZ_E,1902
 graphiti_core/errors.py,sha256=Nib1uQx2cO_VOizupmRjpFfmuRg-hFAVqTtZAuBehR8,2405
-graphiti_core/graphiti.py,sha256=Jztk1PGsr15FkJhf-mFYgFGF40tTMQHPUq96kdWz6c8,29726
+graphiti_core/graphiti.py,sha256=ykV9YARsGFHRYg6wxSHzqGKV8Gm7ZU8P5opJxYDAV8I,30794
 graphiti_core/helpers.py,sha256=7BQzUBFmoBDA2OIDdFtoN4W-vXOhPRIsF0uDb7PsNi0,2913
 graphiti_core/llm_client/__init__.py,sha256=PA80TSMeX-sUXITXEAxMDEt3gtfZgcJrGJUcyds1mSo,207
 graphiti_core/llm_client/anthropic_client.py,sha256=dTM8rKhk9TZAU4O-0jFMivOwJvWM-gHpp5gLmuJHiGQ,2723
@@ -43,10 +43,11 @@ 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=UZN8jFA4pBlw2O5N1cuhVRBdTwMLR9N3Oyo6sQ4MDVw,3117
+graphiti_core/search/search_config.py,sha256=Yn_p1cnMwtZAdIOga19iaCoS7FmA95-GLN7KmHd-7N0,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_utils.py,sha256=Bywp7trqP_gEwqH8I-JHuj3Mljw9P2K6_XooXI75jHI,25739
+graphiti_core/search/search_helpers.py,sha256=Tt5rRUfblBlFqV7r8XuGZiT5P_wjlh01xS0feELZjXg,2173
+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
 graphiti_core/utils/datetime_utils.py,sha256=Ti-2tnrDFRzBsbfblzsHybsM3jaDLP4-VT2t0VhpIzU,1357
@@ -58,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.4.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
-graphiti_core-0.9.4.dist-info/METADATA,sha256=wl7sIqZBszdCc_7go_qGyGTJCktv6xikxjlrWRedtn8,14344
-graphiti_core-0.9.4.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-graphiti_core-0.9.4.dist-info/RECORD,,
+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,,