chore: update dependencies and refactor type hinting (camel-ai#339)

* Bump version from 0.9.0 to 0.9.1 in pyproject.toml and update google-genai dependency to >=0.1.0

* Bump version from 0.9.1 to 0.9.2 in pyproject.toml

* Update google-genai dependency version to >=0.8.0 in pyproject.toml

* loc file

* Update pyproject.toml to version 0.9.3, restructure dependencies, and modify author format. Remove outdated Google API key note from README.md.

* upgrade poetry and ruff

* Update README.md to include installation instructions for Graphiti with Google Gemini support

* fix to deps since peotry doesn't fully implement PEP 735

* Refactor string formatting in various files to use single quotes for consistency and improve readability. This includes updates in agent.ipynb, quickstart.py, multiple prompt files, and ingest.py and retrieve.py modules.

* Remove optional dependencies from pyproject.toml to streamline project requirements.

Author: danielchalef

examples/langgraph-agent/agent.ipynb

@@ -363,7 +363,7 @@
     "    asyncio.create_task(\n",
     "        client.add_episode(\n",
     "            name='Chatbot Response',\n",
-    "            episode_body=f\"{state['user_name']}: {state['messages'][-1]}\\nSalesBot: {response.content}\",\n",
+    "            episode_body=f'{state[\"user_name\"]}: {state[\"messages\"][-1]}\\nSalesBot: {response.content}',\n",
     "            source=EpisodeType.message,\n",
     "            reference_time=datetime.now(timezone.utc),\n",
     "            source_description='Chatbot',\n",

examples/quickstart/quickstart.py

@@ -217,7 +217,7 @@ async def main():
             print(f'Node Name: {node.name}')
             node_summary = node.summary[:100] + '...' if len(node.summary) > 100 else node.summary
             print(f'Content Summary: {node_summary}')
-            print(f"Node Labels: {', '.join(node.labels)}")
+            print(f'Node Labels: {", ".join(node.labels)}')
             print(f'Created At: {node.created_at}')
             if hasattr(node, 'attributes') and node.attributes:
                 print('Attributes:')

graphiti_core/embedder/voyage.py

@@ -55,4 +55,4 @@ async def create(
             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>