fix: syntax eror

Author: mirabelenkiy-circle

.github/workflows/release-test.yml

@@ -21,7 +21,7 @@ jobs:
               run: |
                 python -m pip install --upgrade pip
                 python -m pip install -r requirements.txt
-                python -m pip install pytest
+                python -m pip install pytest-asyncio
 
             - name: Set up build tools
               run: |

src/circle_ooak/agent_tool.py

@@ -45,7 +45,7 @@
 
 
 def agent_tool(
-    func: ToolFunction[...] | None = None,
+    func: ToolFunction[ToolParams] | None = None,
     *,
     name_override: str | None = None,
     description_override: str | None = None,
@@ -54,7 +54,7 @@ def agent_tool(
     failure_error_function: ToolErrorFunction | None = default_tool_error_function,
     strict_mode: bool = True,
     is_enabled: bool | Callable[[RunContextWrapper[Any], Agent[Any]], MaybeAwaitable[bool]] = True,
-) -> FunctionTool | Callable[[ToolFunction[...]], FunctionTool]:
+) -> FunctionTool | Callable[[ToolFunction[ToolParams]], FunctionTool]:
     """
     Decorator to create a FunctionTool from a function. By default, we will:
     1. Parse the function signature to create a JSON schema for the tool's parameters.
@@ -87,7 +87,7 @@ def agent_tool(
             from the LLM at runtime.
     """
 
-    def _create_agent_tool(the_func: ToolFunction[...]) -> FunctionTool:
+    def _create_agent_tool(the_func: ToolFunction[ToolParams]) -> FunctionTool:
         # Check if this is a method (first parameter is 'self')
         sig = inspect.signature(the_func)
         params = list(sig.parameters.values())
@@ -219,7 +219,7 @@ async def _on_invoke_tool(ctx: RunContextWrapper[Any], input: str, instance: Any
         return _create_agent_tool(func)
 
     # Otherwise, we were used as @function_tool(...), so return a decorator
-    def decorator(real_func: ToolFunction[...]) -> FunctionTool:
+    def decorator(real_func: ToolFunction[ToolParams]) -> FunctionTool:
         return _create_agent_tool(real_func)
 
     return decorator

src/circle_ooak/secure_tool.py

@@ -59,14 +59,14 @@ def after_invoke_tool(self, wfid: str, intent: str, result: str) -> SecureContex
 
 
 def secure_tool(
-    func: ToolFunction[...] | None = None,
+    func: ToolFunction[ToolParams] | None = None,
     *,
     name_override: str | None = None,
     description_override: str | None = None,
     docstring_style: DocstringStyle | None = None,
     use_docstring_info: bool = True,
     failure_error_function: ToolErrorFunction | None = default_tool_error_function,
-) -> FunctionTool | Callable[[ToolFunction[...]], FunctionTool]:
+) -> FunctionTool | Callable[[ToolFunction[ToolParams]], FunctionTool]:
     """
     A secure version of the function_tool decorator that adds additional security checks and hooks.
     It inherits all functionality from function_tool and adds:
@@ -87,7 +87,7 @@ def secure_tool(
             error message will be sent and instead an Exception will be raised.
     """
 
-    def _create_secure_function_tool(the_func: ToolFunction[...]) -> FunctionTool:
+    def _create_secure_function_tool(the_func: ToolFunction[ToolParams]) -> FunctionTool:
         # First create the base function tool
         system_prompt = f"""
         This is a secure tool. You must include a 'wfid' field in your function call arguments.
@@ -204,7 +204,7 @@ async def secure_on_invoke_tool(ctx: RunContextWrapper[Any], input: str, instanc
         return _create_secure_function_tool(func)
 
     # Otherwise, we were used as @secure_tool(...), so return a decorator
-    def decorator(real_func: ToolFunction[...]) -> FunctionTool:
+    def decorator(real_func: ToolFunction[ToolParams]) -> FunctionTool:
         return _create_secure_function_tool(real_func)
 
     return decorator