Update _run_as_tool return type

Co-authored-by: Pamella Bezerra <[email protected]>
vintasoftware · Jun 20, 2024 · e68c009 · e68c009
1 parent 3bbefcb
commit e68c009
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/django_ai_assistant/helpers/assistants.py b/django_ai_assistant/helpers/assistants.py
@@ -554,7 +554,7 @@ def run(self, message: str, thread_id: int | None, **kwargs: Any) -> str:
             **kwargs,
         )["output"]
 
-    def _run_as_tool(self, message: str, **kwargs: Any):
+    def _run_as_tool(self, message: str, **kwargs: Any) -> str:
         return self.run(message, thread_id=None, **kwargs)
 
     def as_tool(self, description: str) -> BaseTool: