style(mypy): Add explicit future type

Libertai · Feb 26, 2025 · ceaa989 · ceaa989
1 parent 17c163a
commit ceaa989
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/libertai_agents/libertai_agents/agents.py b/libertai_agents/libertai_agents/agents.py
@@ -3,6 +3,7 @@
 import json
 import time
 import weakref
+from asyncio import Future
 from http import HTTPStatus
 from typing import Any, AsyncIterable, Awaitable
 
@@ -255,7 +256,7 @@ def __execute_tool_calls(
             function_name = call.function.name
             tool = find(lambda x: x.name == function_name, self.tools)
             if tool is None:
-                future = asyncio.Future()
+                future: Future = asyncio.Future()
                 future.set_result(None)
                 executed_calls.append(future)
                 continue