Update ai_tool.py

ai_tool.py

@@ -14,7 +14,7 @@ import random
 from tavily import TavilyClient
 from playwright.sync_api import sync_playwright
 from bs4 import BeautifulSoup
-
+import os
 # Tool definitions
 def tool(func: Callable) -> Callable:
     """Decorator to mark functions as tools."""