graphiti-core 0.9.2__tar.gz → 0.9.4__tar.gz

{graphiti_core-0.9.2 → graphiti_core-0.9.4}/PKG-INFO

@@ -1,26 +1,32 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: graphiti-core
-Version: 0.9.2
+Version: 0.9.4
 Summary: A temporal graph building library
 License: Apache-2.0
 Author: Paul Paliychuk
 Author-email: paul@getzep.com
-Requires-Python: >=3.10,<4.0
+Requires-Python: >=3.10,<4
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Provides-Extra: anthropic
 Provides-Extra: google-genai
 Provides-Extra: groq
-Requires-Dist: diskcache (>=5.6.3,<6.0.0)
-Requires-Dist: neo4j (>=5.23.0,<6.0.0)
+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: groq (>=0.2.0) ; extra == "groq"
+Requires-Dist: neo4j (>=5.23.0)
 Requires-Dist: numpy (>=1.0.0)
-Requires-Dist: openai (>=1.53.0,<2.0.0)
-Requires-Dist: pydantic (>=2.8.2,<3.0.0)
-Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
-Requires-Dist: tenacity (==9.0.0)
+Requires-Dist: openai (>=1.53.0)
+Requires-Dist: pydantic (>=2.8.2)
+Requires-Dist: python-dotenv (>=1.0.1)
+Requires-Dist: tenacity (>=9.0.0)
+Project-URL: Homepage, https://help.getzep.com/graphiti/graphiti/overview
+Project-URL: Repository, https://github.com/getzep/graphiti
 Description-Content-Type: text/markdown
 
 <p align="center">
@@ -256,6 +262,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
@@ -286,8 +302,6 @@ graphiti = Graphiti(
 # Now you can use Graphiti with Google Gemini
 ```
 
-Make sure to replace the placeholder value with your actual Google API key. You can find more details in the example file at `examples/gemini_example.py`.
-
 ## Documentation
 
 - [Guides and API documentation](https://help.getzep.com/graphiti).

{graphiti_core-0.9.2 → graphiti_core-0.9.4}/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
@@ -261,8 +271,6 @@ graphiti = Graphiti(
 # Now you can use Graphiti with Google Gemini
 ```
 
-Make sure to replace the placeholder value with your actual Google API key. You can find more details in the example file at `examples/gemini_example.py`.
-
 ## Documentation
 
 - [Guides and API documentation](https://help.getzep.com/graphiti).

{graphiti_core-0.9.2 → graphiti_core-0.9.4}/graphiti_core/cross_encoder/bge_reranker_client.py

@@ -36,7 +36,7 @@ class BGERerankerClient(CrossEncoderClient):
         scores = await loop.run_in_executor(None, self.model.predict, input_pairs)
 
         ranked_passages = sorted(
-            [(passage, float(score)) for passage, score in zip(passages, scores)],
+            [(passage, float(score)) for passage, score in zip(passages, scores, strict=False)],
             key=lambda x: x[1],
             reverse=True,
         )

{graphiti_core-0.9.2 → graphiti_core-0.9.4}/graphiti_core/cross_encoder/openai_reranker_client.py

@@ -111,7 +111,7 @@ class OpenAIRerankerClient(CrossEncoderClient):
                     if bool(logprob.token):
                         scores.append(logprob.logprob)
 
-            results = [(passage, score) for passage, score in zip(passages, scores)]
+            results = [(passage, score) for passage, score in zip(passages, scores, strict=False)]
             results.sort(reverse=True, key=lambda x: x[1])
             return results
         except openai.RateLimitError as e:

{graphiti_core-0.9.2 → graphiti_core-0.9.4}/graphiti_core/embedder/openai.py

@@ -15,7 +15,6 @@ limitations under the License.
 """
 
 from collections.abc import Iterable
-from typing import Union
 
 from openai import AsyncAzureOpenAI, AsyncOpenAI
 from openai.types import EmbeddingModel
@@ -41,7 +40,7 @@ class OpenAIEmbedder(EmbedderClient):
     def __init__(
         self,
         config: OpenAIEmbedderConfig | None = None,
-        client: Union[AsyncOpenAI, AsyncAzureOpenAI, None] = None,
+        client: AsyncOpenAI | AsyncAzureOpenAI | None = None,
     ):
         if config is None:
             config = OpenAIEmbedderConfig()

{graphiti_core-0.9.2 → graphiti_core-0.9.4}/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.2 → graphiti_core-0.9.4}/graphiti_core/graphiti.py

@@ -451,7 +451,7 @@ class Graphiti:
             existing_edges_list: list[list[EntityEdge]] = [
                 source_lst + target_lst
                 for source_lst, target_lst in zip(
-                    existing_source_edges_list, existing_target_edges_list
+                    existing_source_edges_list, existing_target_edges_list, strict=False
                 )
             ]
 

{graphiti_core-0.9.2 → graphiti_core-0.9.4}/graphiti_core/llm_client/client.py

@@ -36,7 +36,7 @@ logger = logging.getLogger(__name__)
 
 
 def is_server_or_retry_error(exception):
-    if isinstance(exception, (RateLimitError, json.decoder.JSONDecodeError)):
+    if isinstance(exception, RateLimitError | json.decoder.JSONDecodeError):
         return True
 
     return (

{graphiti_core-0.9.2 → graphiti_core-0.9.4}/graphiti_core/prompts/dedupe_edges.py

@@ -15,7 +15,7 @@ limitations under the License.
 """
 
 import json
-from typing import Any, Optional, Protocol, TypedDict
+from typing import Any, Protocol, TypedDict
 
 from pydantic import BaseModel, Field
 
@@ -24,7 +24,7 @@ from .models import Message, PromptFunction, PromptVersion
 
 class EdgeDuplicate(BaseModel):
     is_duplicate: bool = Field(..., description='true or false')
-    uuid: Optional[str] = Field(
+    uuid: str | None = Field(
         None,
         description="uuid of the existing edge like '5d643020624c42fa9de13f97b1b3fa39' or null",
     )

{graphiti_core-0.9.2 → graphiti_core-0.9.4}/graphiti_core/prompts/dedupe_nodes.py

@@ -15,7 +15,7 @@ limitations under the License.
 """
 
 import json
-from typing import Any, Optional, Protocol, TypedDict
+from typing import Any, Protocol, TypedDict
 
 from pydantic import BaseModel, Field
 
@@ -24,7 +24,7 @@ from .models import Message, PromptFunction, PromptVersion
 
 class NodeDuplicate(BaseModel):
     is_duplicate: bool = Field(..., description='true or false')
-    uuid: Optional[str] = Field(
+    uuid: str | None = Field(
         None,
         description="uuid of the existing node like '5d643020624c42fa9de13f97b1b3fa39' or null",
     )
@@ -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.2 → graphiti_core-0.9.4}/graphiti_core/prompts/extract_edge_dates.py

@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from typing import Any, Optional, Protocol, TypedDict
+from typing import Any, Protocol, TypedDict
 
 from pydantic import BaseModel, Field
 
@@ -22,11 +22,11 @@ from .models import Message, PromptFunction, PromptVersion
 
 
 class EdgeDates(BaseModel):
-    valid_at: Optional[str] = Field(
+    valid_at: str | None = Field(
         None,
         description='The date and time when the relationship described by the edge fact became true or was established. YYYY-MM-DDTHH:MM:SS.SSSSSSZ or null.',
     )
-    invalid_at: Optional[str] = Field(
+    invalid_at: str | None = Field(
         None,
         description='The date and time when the relationship described by the edge fact stopped being true or ended. YYYY-MM-DDTHH:MM:SS.SSSSSSZ or null.',
     )
@@ -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.2 → graphiti_core-0.9.4}/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.2 → graphiti_core-0.9.4}/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.2 → graphiti_core-0.9.4}/graphiti_core/prompts/models.py

@@ -14,7 +14,8 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from typing import Any, Callable, Protocol
+from collections.abc import Callable
+from typing import Any, Protocol
 
 from pydantic import BaseModel
 

{graphiti_core-0.9.2 → graphiti_core-0.9.4}/graphiti_core/utils/maintenance/community_operations.py

@@ -164,7 +164,7 @@ async def build_community(
                 *[
                     summarize_pair(llm_client, (str(left_summary), str(right_summary)))
                     for left_summary, right_summary in zip(
-                        summaries[: int(length / 2)], summaries[int(length / 2) :]
+                        summaries[: int(length / 2)], summaries[int(length / 2) :], strict=False
                     )
                 ]
             )

{graphiti_core-0.9.2 → graphiti_core-0.9.4}/graphiti_core/utils/maintenance/edge_operations.py

@@ -213,7 +213,7 @@ async def resolve_extracted_edges(
                     previous_episodes,
                 )
                 for extracted_edge, related_edges, existing_edges in zip(
-                    extracted_edges, related_edges_lists, existing_edges_lists
+                    extracted_edges, related_edges_lists, existing_edges_lists, strict=False
                 )
             ]
         )

{graphiti_core-0.9.2 → graphiti_core-0.9.4}/graphiti_core/utils/maintenance/node_operations.py

@@ -279,7 +279,9 @@ async def resolve_extracted_nodes(
                     previous_episodes,
                     entity_types,
                 )
-                for extracted_node, existing_nodes in zip(extracted_nodes, existing_nodes_lists)
+                for extracted_node, existing_nodes in zip(
+                    extracted_nodes, existing_nodes_lists, strict=False
+                )
             ]
         )
     )

graphiti_core-0.9.4/pyproject.toml

@@ -0,0 +1,86 @@
+[project]
+name = "graphiti-core"
+description = "A temporal graph building library"
+version = "0.9.4"
+authors = [
+    {"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,<4"
+packages = [{ include = "graphiti_core", from = "." }]
+dependencies = [
+    "pydantic>=2.8.2",
+    "neo4j>=5.23.0",
+    "diskcache>=5.6.3",
+    "openai>=1.53.0",
+    "tenacity>=9.0.0",
+    "numpy>=1.0.0",
+    "python-dotenv>=1.0.1",
+]
+
+[project.urls]
+Homepage = "https://help.getzep.com/graphiti/graphiti/overview"
+Repository = "https://github.com/getzep/graphiti"
+
+[project.optional-dependencies]
+anthropic = ["anthropic>=0.49.0"]
+groq = ["groq>=0.2.0"]
+google-genai = ["google-genai>=1.8.0"]
+
+[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"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.poetry]
+requires-poetry = ">=2.0"
+
+[tool.pytest.ini_options]
+pythonpath = ["."]
+
+[tool.ruff]
+line-length = 100
+
+[tool.ruff.lint]
+select = [
+    # pycodestyle
+    "E",
+    # Pyflakes
+    "F",
+    # pyupgrade
+    "UP",
+    # flake8-bugbear
+    "B",
+    # flake8-simplify
+    "SIM",
+    # isort
+    "I",
+]
+ignore = ["E501"]
+
+[tool.ruff.format]
+quote-style = "single"
+indent-style = "space"
+docstring-code-format = true

graphiti_core-0.9.2/pyproject.toml

@@ -1,80 +0,0 @@
-[tool.poetry]
-name = "graphiti-core"
-version = "0.9.2"
-description = "A temporal graph building library"
-authors = [
-    "Paul Paliychuk <paul@getzep.com>",
-    "Preston Rasmussen <preston@getzep.com>",
-    "Daniel Chalef <daniel@getzep.com>",
-]
-readme = "README.md"
-license = "Apache-2.0"
-
-packages = [{ include = "graphiti_core", from = "." }]
-
-[tool.poetry.dependencies]
-python = "^3.10"
-pydantic = "^2.8.2"
-neo4j = "^5.23.0"
-diskcache = "^5.6.3"
-openai = "^1.53.0"
-tenacity = "9.0.0"
-numpy = ">=1.0.0"
-python-dotenv = "^1.0.1"
-
-[tool.poetry.extras]
-anthropic = ["anthropic"]
-groq = ["groq"]
-google-genai = ["google-genai"]
-
-[tool.poetry.group.dev.dependencies]
-mypy = "^1.11.1"
-groq = ">=0.9,<0.12"
-anthropic = "~0.49.0"
-google-genai = ">=0.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"]
-build-backend = "poetry.core.masonry.api"
-
-[tool.pytest.ini_options]
-pythonpath = ["."]
-
-[tool.ruff]
-line-length = 100
-
-[tool.ruff.lint]
-select = [
-    # pycodestyle
-    "E",
-    # Pyflakes
-    "F",
-    # pyupgrade
-    "UP",
-    # flake8-bugbear
-    "B",
-    # flake8-simplify
-    "SIM",
-    # isort
-    "I",
-]
-ignore = ["E501"]
-
-[tool.ruff.format]
-quote-style = "single"
-indent-style = "space"
-docstring-code-format = true