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

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/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/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/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/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.dist-info → graphiti_core-0.9.4.dist-info}/METADATA

@@ -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.dist-info → graphiti_core-0.9.4.dist-info}/RECORD

@@ -8,7 +8,7 @@ graphiti_core/embedder/__init__.py,sha256=EL564ZuE-DZjcuKNUK_exMn_XHXm2LdO9fzdXe
 graphiti_core/embedder/client.py,sha256=HKIlpPLnzFT81jurPkry6z8F8nxfZVfejdcfxHVUSFU,995
 graphiti_core/embedder/gemini.py,sha256=nE0XH8wYVGcPSO7DaNQ7kdsQLFSoH4FQOu2HMQUy2ss,2200
 graphiti_core/embedder/openai.py,sha256=fcU63koSRI-OjDuEcBfUKgXu8XV_-8EF6HpVrYa1_8I,1880
-graphiti_core/embedder/voyage.py,sha256=7kqrLG75J3Q6cdA2Nlx1JSYtpk2141ckdl3OtDDw0vU,1882
+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/helpers.py,sha256=7BQzUBFmoBDA2OIDdFtoN4W-vXOhPRIsF0uDb7PsNi0,2913
@@ -30,11 +30,11 @@ graphiti_core/models/nodes/node_db_queries.py,sha256=AQgRGVO-GgFWfLq1G6k8s86WItw
 graphiti_core/nodes.py,sha256=L_sAXuS-Hbj1I_kmln_y3dBiu_UDMKu64oWTB3ecFss,16938
 graphiti_core/prompts/__init__.py,sha256=EA-x9xUki9l8wnu2l8ek_oNf75-do5tq5hVq7Zbv8Kw,101
 graphiti_core/prompts/dedupe_edges.py,sha256=GrLKEHPrEsjK31wQf7AnMJDXaVCrCuJWaNlVAxEd4ks,3543
-graphiti_core/prompts/dedupe_nodes.py,sha256=YTJPsbeldHDVbBG1p9JTRZMKQxXa72EZl44YUL9gkbs,4632
+graphiti_core/prompts/dedupe_nodes.py,sha256=VXSb3chBo4l7qzzWxT-sNHJhjY3jNtUBAndnALUcqws,4632
 graphiti_core/prompts/eval.py,sha256=csW494kKBMvWSm2SYLIRuGgNghhwNR3YwGn3veo3g_Y,3691
-graphiti_core/prompts/extract_edge_dates.py,sha256=hcQ2rUyrshExIlTWxg8RNeso3FOPRlKZ27_TcqcXDh8,4080
-graphiti_core/prompts/extract_edges.py,sha256=vyEdW7JAPOT_eLWUi6nRmxbvucyVoyoYX2SxXfknRUg,3467
-graphiti_core/prompts/extract_nodes.py,sha256=fSi07hGnlo7L5t9c0Q5zcBxDSki7QQKWhex9DFsVWW8,7048
+graphiti_core/prompts/extract_edge_dates.py,sha256=wBM80uGIpUuDQs-5nwSieu3JcEg8PzKVrAFx08qUH80,4080
+graphiti_core/prompts/extract_edges.py,sha256=6-MkMkQp4QSpbdZ-gB3PkMcs7q3E07dAWYM5oY1FZdQ,3467
+graphiti_core/prompts/extract_nodes.py,sha256=I8CO8D7rMaKQv2biBxToYrAMD1RNCe1RMG9tGNv1Tm4,7048
 graphiti_core/prompts/invalidate_edges.py,sha256=DV2mEyIhhjc0hdKEMFLQMeG0FiUCkv_X0ctCliYjQ2c,3577
 graphiti_core/prompts/lib.py,sha256=DCyHePM4_q-CptTpEXGO_dBv9k7xDtclEaB1dGu7EcI,4092
 graphiti_core/prompts/models.py,sha256=NgxdbPHJpBEcpbXovKyScgpBc73Q-GIW-CBDlBtDjto,894
@@ -58,7 +58,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.3.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
-graphiti_core-0.9.3.dist-info/METADATA,sha256=Aa5QrNkSRsXnuYShlNnCi3L83ySgNABW4QgCZQ2coG4,14224
-graphiti_core-0.9.3.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-graphiti_core-0.9.3.dist-info/RECORD,,
+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,,