graphiti-core 0.9.3__tar.gz → 0.9.5__tar.gz

{graphiti_core-0.9.3 → graphiti_core-0.9.5}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.3
 Name: graphiti-core
-Version: 0.9.3
+Version: 0.9.5
 Summary: A temporal graph building library
 License: Apache-2.0
 Author: Paul Paliychuk
 Author-email: paul@getzep.com
-Requires-Python: >=3.10
+Requires-Python: >=3.10,<4
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
@@ -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)
@@ -262,6 +263,16 @@ Make sure to replace the placeholder values with your actual Azure OpenAI creden
 
 Graphiti supports Google's Gemini models for both LLM inference and embeddings. To use Gemini, you'll need to configure both the LLM client and embedder with your Google API key.
 
+Install Graphiti:
+
+```bash
+poetry add "graphiti-core[google-genai]"
+
+# or
+
+uv add "graphiti-core[google-genai]"
+```
+
 ```python
 from graphiti_core import Graphiti
 from graphiti_core.llm_client.gemini_client import GeminiClient, LLMConfig

{graphiti_core-0.9.3 → graphiti_core-0.9.5}/README.md

@@ -231,6 +231,16 @@ Make sure to replace the placeholder values with your actual Azure OpenAI creden
 
 Graphiti supports Google's Gemini models for both LLM inference and embeddings. To use Gemini, you'll need to configure both the LLM client and embedder with your Google API key.
 
+Install Graphiti:
+
+```bash
+poetry add "graphiti-core[google-genai]"
+
+# or
+
+uv add "graphiti-core[google-genai]"
+```
+
 ```python
 from graphiti_core import Graphiti
 from graphiti_core.llm_client.gemini_client import GeminiClient, LLMConfig

{graphiti_core-0.9.3 → graphiti_core-0.9.5}/graphiti_core/embedder/voyage.py

@@ -55,4 +55,4 @@ class VoyageAIEmbedder(EmbedderClient):
             return []
 
         result = await self.client.embed(input_list, model=self.config.embedding_model)
-        return result.embeddings[0][: self.config.embedding_dim]
+        return [float(x) for x in result.embeddings[0][: self.config.embedding_dim]]

{graphiti_core-0.9.3 → graphiti_core-0.9.5}/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-0.9.3 → graphiti_core-0.9.5}/graphiti_core/prompts/dedupe_nodes.py

@@ -57,7 +57,7 @@ def node(context: dict[str, Any]) -> list[Message]:
         {json.dumps([ep for ep in context['previous_episodes']], indent=2)}
         </PREVIOUS MESSAGES>
         <CURRENT MESSAGE>
-        {context["episode_content"]}
+        {context['episode_content']}
         </CURRENT MESSAGE>
 
         <EXISTING NODES>

{graphiti_core-0.9.3 → graphiti_core-0.9.5}/graphiti_core/prompts/extract_edge_dates.py

@@ -53,7 +53,7 @@ def v1(context: dict[str, Any]) -> list[Message]:
             {context['previous_episodes']}
             </PREVIOUS MESSAGES>
             <CURRENT MESSAGE>
-            {context["current_episode"]}
+            {context['current_episode']}
             </CURRENT MESSAGE>
             <REFERENCE TIMESTAMP>
             {context['reference_timestamp']}

{graphiti_core-0.9.3 → graphiti_core-0.9.5}/graphiti_core/prompts/extract_edges.py

@@ -60,11 +60,11 @@ def edge(context: dict[str, Any]) -> list[Message]:
         {json.dumps([ep for ep in context['previous_episodes']], indent=2)}
         </PREVIOUS MESSAGES>
         <CURRENT MESSAGE>
-        {context["episode_content"]}
+        {context['episode_content']}
         </CURRENT MESSAGE>
         
         <ENTITIES>
-        {context["nodes"]}
+        {context['nodes']}
         </ENTITIES>
         
         {context['custom_prompt']}
@@ -90,15 +90,15 @@ def reflexion(context: dict[str, Any]) -> list[Message]:
 {json.dumps([ep for ep in context['previous_episodes']], indent=2)}
 </PREVIOUS MESSAGES>
 <CURRENT MESSAGE>
-{context["episode_content"]}
+{context['episode_content']}
 </CURRENT MESSAGE>
 
 <EXTRACTED ENTITIES>
-{context["nodes"]}
+{context['nodes']}
 </EXTRACTED ENTITIES>
 
 <EXTRACTED FACTS>
-{context["extracted_facts"]}
+{context['extracted_facts']}
 </EXTRACTED FACTS>
 
 Given the above MESSAGES, list of EXTRACTED ENTITIES entities, and list of EXTRACTED FACTS; 

{graphiti_core-0.9.3 → graphiti_core-0.9.5}/graphiti_core/prompts/extract_nodes.py

@@ -68,7 +68,7 @@ def extract_message(context: dict[str, Any]) -> list[Message]:
 {json.dumps([ep for ep in context['previous_episodes']], indent=2)}
 </PREVIOUS MESSAGES>
 <CURRENT MESSAGE>
-{context["episode_content"]}
+{context['episode_content']}
 </CURRENT MESSAGE>
 
 {context['custom_prompt']}
@@ -96,10 +96,10 @@ def extract_json(context: dict[str, Any]) -> list[Message]:
 
     user_prompt = f"""
 <SOURCE DESCRIPTION>:
-{context["source_description"]}
+{context['source_description']}
 </SOURCE DESCRIPTION>
 <JSON>
-{context["episode_content"]}
+{context['episode_content']}
 </JSON>
 
 {context['custom_prompt']}
@@ -121,7 +121,7 @@ def extract_text(context: dict[str, Any]) -> list[Message]:
 
     user_prompt = f"""
 <TEXT>
-{context["episode_content"]}
+{context['episode_content']}
 </TEXT>
 
 {context['custom_prompt']}
@@ -148,11 +148,11 @@ def reflexion(context: dict[str, Any]) -> list[Message]:
 {json.dumps([ep for ep in context['previous_episodes']], indent=2)}
 </PREVIOUS MESSAGES>
 <CURRENT MESSAGE>
-{context["episode_content"]}
+{context['episode_content']}
 </CURRENT MESSAGE>
 
 <EXTRACTED ENTITIES>
-{context["extracted_entities"]}
+{context['extracted_entities']}
 </EXTRACTED ENTITIES>
 
 Given the above previous messages, current message, and list of extracted entities; determine if any entities haven't been
@@ -172,7 +172,7 @@ def classify_nodes(context: dict[str, Any]) -> list[Message]:
     {json.dumps([ep for ep in context['previous_episodes']], indent=2)}
     </PREVIOUS MESSAGES>
     <CURRENT MESSAGE>
-    {context["episode_content"]}
+    {context['episode_content']}
     </CURRENT MESSAGE>
     
     <EXTRACTED ENTITIES>

{graphiti_core-0.9.3 → graphiti_core-0.9.5}/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-0.9.5/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-0.9.3 → graphiti_core-0.9.5}/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.3 → graphiti_core-0.9.5}/pyproject.toml

@@ -1,15 +1,15 @@
 [project]
 name = "graphiti-core"
 description = "A temporal graph building library"
-version = "0.9.3"
+version = "0.9.5"
 authors = [
-    {"name" = "Paul Paliychuk", "email" = "paul@getzep.com"},
-    {"name" = "Preston Rasmussen", "email" = "preston@getzep.com"},
-    {"name" = "Daniel Chalef", "email" = "daniel@getzep.com"},
+    { "name" = "Paul Paliychuk", "email" = "paul@getzep.com" },
+    { "name" = "Preston Rasmussen", "email" = "preston@getzep.com" },
+    { "name" = "Daniel Chalef", "email" = "daniel@getzep.com" },
 ]
 readme = "README.md"
 license = "Apache-2.0"
-requires-python = ">=3.10"
+requires-python = ">=3.10,<4"
 packages = [{ include = "graphiti_core", from = "." }]
 dependencies = [
     "pydantic>=2.8.2",
@@ -19,9 +19,9 @@ dependencies = [
     "tenacity>=9.0.0",
     "numpy>=1.0.0",
     "python-dotenv>=1.0.1",
+    "graph-service (>=1.0.0.7,<2.0.0.0)",
 ]
 
-
 [project.urls]
 Homepage = "https://help.getzep.com/graphiti/graphiti/overview"
 Repository = "https://github.com/getzep/graphiti"
@@ -31,27 +31,25 @@ anthropic = ["anthropic>=0.49.0"]
 groq = ["groq>=0.2.0"]
 google-genai = ["google-genai>=1.8.0"]
 
-[dependency-groups]
-dev = [
-    "mypy>=1.11.1",
-    "groq>=0.2.0",
-    "anthropic>=0.49.0",
-    "google-genai>=1.8.0",
-    "ipykernel>=6.29.5",
-    "jupyterlab>=4.2.4",
-    "diskcache-stubs>=5.6.3.6.20240818",
-    "langgraph>=0.2.15",
-    "langchain-anthropic>=0.2.4",
-    "langsmith>=0.1.108",
-    "langchain-openai>=0.2.6",
-    "sentence-transformers>=3.2.1",
-    "transformers>=4.45.2",
-    "voyageai>=0.2.3",
-    "pytest>=8.3.3",
-    "pytest-asyncio>=0.24.0",
-    "pytest-xdist>=3.6.1",
-    "ruff>=0.7.1",
-]
+[tool.poetry.group.dev.dependencies]
+mypy = ">=1.11.1"
+groq = ">=0.2.0"
+anthropic = ">=0.49.0"
+google-genai = ">=1.8.0"
+ipykernel = ">=6.29.5"
+jupyterlab = ">=4.2.4"
+diskcache-stubs = ">=5.6.3.6.20240818"
+langgraph = ">=0.2.15"
+langchain-anthropic = ">=0.2.4"
+langsmith = ">=0.1.108"
+langchain-openai = ">=0.2.6"
+sentence-transformers = ">=3.2.1"
+transformers = ">=4.45.2"
+voyageai = ">=0.2.3"
+pytest = ">=8.3.3"
+pytest-asyncio = ">=0.24.0"
+pytest-xdist = ">=3.6.1"
+ruff = ">=0.7.1"
 
 [build-system]
 requires = ["poetry-core"]