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

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

@@ -1,11 +1,11 @@
 Metadata-Version: 2.3
 Name: graphiti-core
-Version: 0.9.3
+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
+Requires-Python: >=3.10,<4
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
@@ -262,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

{graphiti_core-0.9.3 → 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

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

@@ -1,7 +1,7 @@
 [project]
 name = "graphiti-core"
 description = "A temporal graph building library"
-version = "0.9.3"
+version = "0.9.4"
 authors = [
     {"name" = "Paul Paliychuk", "email" = "paul@getzep.com"},
     {"name" = "Preston Rasmussen", "email" = "preston@getzep.com"},
@@ -9,7 +9,7 @@ authors = [
 ]
 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",
@@ -21,7 +21,6 @@ dependencies = [
     "python-dotenv>=1.0.1",
 ]
 
-
 [project.urls]
 Homepage = "https://help.getzep.com/graphiti/graphiti/overview"
 Repository = "https://github.com/getzep/graphiti"
@@ -31,27 +30,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"]