Fix: Add some await missing

Author: lambda-science

CHANGELOG.md

@@ -1,5 +1,8 @@
-# v0.0.3
+# v0.0.4
+
+- Fix some non-async calls
 
+# v0.0.3
 - Async Query of Mistral API
 
 # v0.0.2

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "opensymbiose"
-version = "0.0.3"
+version = "0.0.4"
 description = "OpenSymbiose is an open-source biotechnology / biology research AI agent designed to support researcher."
 readme = "README.md"
 requires-python = ">=3.10"

requirements.txt

@@ -1,5 +1,5 @@
 # For HuggingFace Space Gradio App
-opensymbiose
+opensymbiose>=0.0.4
 mistralai
 python-dotenv
 gradio

src/opensymbiose/agents/agent.py

@@ -47,7 +47,7 @@ async def add_handoff(self, agent_id: str, client: Mistral) -> None:
         """
         if agent_id not in self.handoffs:
             self.handoffs.append(agent_id)
-            updated_agent = client.beta.agents.update_async(
+            updated_agent = await client.beta.agents.update_async(
                 agent_id=self.id, handoffs=self.handoffs
             )
             # Update the raw data with the updated agent
@@ -63,7 +63,7 @@ async def remove_handoff(self, agent_id: str, client: Mistral) -> None:
         """
         if agent_id in self.handoffs:
             self.handoffs.remove(agent_id)
-            updated_agent = client.beta.agents.update_async(
+            updated_agent = await client.beta.agents.update_async(
                 agent_id=self.id, handoffs=self.handoffs
             )
             # Update the raw data with the updated agent