fix: Handle empty tools

julep-ai · Sep 13, 2024 · 9118d86 · 9118d86
1 parent 6db5e4c
commit 9118d86
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/agents-api/agents_api/common/protocol/sessions.py b/agents-api/agents_api/common/protocol/sessions.py
@@ -96,7 +96,9 @@ def get_active_tools(self) -> list[Tool]:
         active_agent = self.get_active_agent()
         try:
             active_toolset = next(
-                toolset for toolset in self.toolsets if toolset.agent_id == active_agent.id
+                toolset
+                for toolset in self.toolsets
+                if toolset.agent_id == active_agent.id
             )
         except StopIteration:
             return []

diff --git a/agents-api/agents_api/routers/sessions/chat.py b/agents-api/agents_api/routers/sessions/chat.py
@@ -70,7 +70,7 @@ async def chat(
     # Get the response from the model
     model_response = await litellm.acompletion(
         messages=messages,
-        tools=tools,
+        tools=tools or None,
         user=str(developer.id),  # For tracking usage
         tags=developer.tags,  # For filtering models in litellm
         **settings,