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

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

@@ -1,26 +1,32 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: graphiti-core
-Version: 0.9.1
+Version: 0.9.3
 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
 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">
@@ -286,8 +292,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.1 → graphiti_core-0.9.3}/README.md

@@ -261,8 +261,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.1 → graphiti_core-0.9.3}/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.1 → graphiti_core-0.9.3}/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.1 → graphiti_core-0.9.3}/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.1 → graphiti_core-0.9.3}/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.1 → graphiti_core-0.9.3}/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.1 → graphiti_core-0.9.3}/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.1 → graphiti_core-0.9.3}/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",
     )

{graphiti_core-0.9.1 → graphiti_core-0.9.3}/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.',
     )

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

@@ -0,0 +1,89 @@
+[project]
+name = "graphiti-core"
+description = "A temporal graph building library"
+version = "0.9.3"
+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"
+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"]
+
+[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",
+]
+
+[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.1/pyproject.toml

@@ -1,79 +0,0 @@
-[tool.poetry]
-name = "graphiti-core"
-version = "0.9.1"
-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"
-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