fix handler to support old tools

datastax · Aug 21, 2024 · e2be7f9 · e2be7f9
1 parent 59ece47
commit e2be7f9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/client/astra_assistants/astra_assistants_event_handler.py b/client/astra_assistants/astra_assistants_event_handler.py
@@ -63,6 +63,8 @@ def run_tool(self, tool_call):
                 model: BaseModel = tool.get_model()
                 if issubclass(model, BaseModel):
                     self.arguments = model(**arguments)
+                else:
+                    self.arguments = arguments
                 results = tool.call(self.arguments)
                 return results
             except Exception as e: