Upload tools.py

tools.py

@@ -9,10 +9,20 @@ import io
 import os
 import time
 import datetime
-from typing import List, Dict
+from typing import TypedDict, List, Union
 
 from agents import function_tool
 
+class DuckDuckGoResult(TypedDict):
+    title: str
+    href: str
+    body: str
+
+class SpreadsheetRow(TypedDict):
+    # If you don't know the columns, leave this empty,
+    # but ideally, define them.
+    pass
+
 def log(msg):
     print(f"[{datetime.datetime.now():%Y-%m-%d %H:%M:%S}] {msg}")
 
@@ -56,7 +66,7 @@ def python_run(code: str) -> str:
 # 2. --------------------------------------------------------------------
 @function_tool
 @log_tool_call
-def load_spreadsheet(path: str, sheet: str | int | None = None) -> list[dict]:
+def load_spreadsheet(path: str, sheet: Union[str, int, None] = None) -> List[SpreadsheetRow]:
     """Read .csv, .xls or .xlsx from disk and return rows as list of dictionaries.
 
     Args:
@@ -144,7 +154,7 @@ def image_ocr(path: str) -> str:
 # 6. --------------------------------------------------------------------
 @function_tool
 @log_tool_call
-def duckduckgo_search(query: str, max_results: int = 5) -> list[dict]:
+def duckduckgo_search(query: str, max_results: int = 5) -> List[DuckDuckGoResult]:
     """Search DuckDuckGo and return a list of result dicts with title, href and body.
 
     Args: