Update tools.py

tools.py

@@ -16,73 +16,6 @@ import re
 import time
 from typing import Optional, List, Dict, Any
 
-# class WebSearchTool:
-#     def __init__(self):
-#         self.last_request_time = 0
-#         self.min_request_interval = 2.0
-#         self.max_retries = 5
-
-#     def _do_search(self, query: str, domain: Optional[str] = None) -> str:
-#         base_url = "https://html.duckduckgo.com/html"
-#         params = {"q": query}
-#         if domain:
-#             params["q"] += f" site:{domain}"
-
-#         headers = {"User-Agent": "Mozilla/5.0"}
-#         response = requests.get(base_url, params=params, headers=headers, timeout=10)
-#         response.raise_for_status()
-
-#         if response.status_code == 202:
-#             raise Exception("202 Ratelimit")
-
-#         soup = BeautifulSoup(response.text, 'html.parser')
-#         results = []
-#         for result in soup.find_all('div', {'class': 'result'}):
-#             title = result.find('a', {'class': 'result__a'})
-#             snippet = result.find('a', {'class': 'result__snippet'})
-#             if title and snippet:
-#                 results.append({
-#                     'title': title.get_text(),
-#                     'snippet': snippet.get_text(),
-#                     'url': title.get('href')
-#                 })
-
-#         if not results:
-#             return "No results found."
-
-#         # Format results with title, URL, and snippet
-#         formatted = "\n\n".join(
-#             f"**{r['title']}**\n{r['url']}\n{r['snippet']}" for r in results[:5]
-#         )
-#         return formatted
-
-#     def search(self, query: str, domain: Optional[str] = None) -> str:
-#         for attempt in range(self.max_retries):
-#             current_time = time.time()
-#             time_since_last = current_time - self.last_request_time
-#             if time_since_last < self.min_request_interval:
-#                 time.sleep(self.min_request_interval - time_since_last)
-#             try:
-#                 result = self._do_search(query, domain)
-#                 self.last_request_time = time.time()
-#                 return result
-#             except Exception as e:
-#                 time.sleep(2 ** attempt)
-#         return "Search failed after retries."
-
-# web_search_tool = WebSearchTool()
-
-# @tool
-# def search_web(query: str) -> str:
-#     """
-#     Perform a full web search using DuckDuckGo HTML scraping. 
-#     Returns multiple top results as text snippets.
-#     Use this when Wikipedia or standard search is not enough or fails to return numeric/statistical data.
-#     Example: "How many albums did Mercedes Sosa release between 2000 and 2009?"
-#     """
-#     print(f"🛠️ search_web called with: {query}")
-#     return web_search_tool.search(query)
-
 @tool
 def extract_number_from_text(text: str) -> str:
     """
@@ -98,32 +31,6 @@ def extract_number_from_text(text: str) -> str:
     except Exception as e:
         return f"Error extracting number: {e}"
 
-# @tool
-# def search_and_extract_number(query: str) -> str:
-#     """
-#     Search the web and extract a number from the top results.
-#     Best for questions like: 'How many X did Y do?' or 'What year was Z?'
-#     Example: "Mercedes Sosa studio albums 2000–2009"
-#     """
-#     print(f"🛠️ search_and_extract_number called with: {query}")
-#     raw = web_search_tool.search(query)
-#     return extract_number_from_text.invoke(raw)
-
-# @tool
-# def search_duckduckgo(query: str) -> str:
-#     """
-#     Perform a general-purpose web search using DuckDuckGo. 
-#     Use this when no other tool applies or you need open-ended facts from the internet.
-#     Returns the top result snippet.
-#     """
-#     print(f"🛠️ search_duckduckgo called with: {query}")
-#     with DDGS() as ddgs:
-#         results = ddgs.text(query, safesearch="Moderate")
-#         if results:
-#             return results[0]["body"]
-#         else:
-#             return "No results found."
-
 @tool
 def search_wikipedia(query: str) -> str:
     """
@@ -141,113 +48,6 @@ def search_wikipedia(query: str) -> str:
     except Exception as e:
         return f"Error: {e}"
 
-# @tool
-# def classify_vegetable(item: str) -> str:
-#     """
-#     Determine if the food item is a botanical vegetable or fruit.
-#     Use this for tasks involving scientific food classification.
-#     Returns 'vegetable' or 'fruit'.
-#     """
-#     print(f"🛠️ classify_vegetable called with: {item}")
-#     fruits = {"sweet potatoes", "plums", "bell pepper", "corn", "green beans"}
-#     if item.lower() in fruits:
-#         return "fruit"
-#     else:
-#         return "vegetable"
-
-# @tool
-# def solve_chess_position(fen: str) -> str:
-#     """
-#     Analyze a chess position in FEN format using Stockfish engine.
-#     Returns the best move for the current player.
-#     Use this for decision-making or evaluation in chess scenarios.
-#     """
-#     print(f"🛠️ solve_chess_position called with: {fen}")
-#     try:
-#         board = chess.Board(fen)
-#         with chess.engine.SimpleEngine.popen_uci("/usr/bin/stockfish") as engine:
-#             result = engine.play(board, chess.engine.Limit(time=0.1))
-#             return board.san(result.move)
-#     except Exception as e:
-#         return f"Error solving chess: {e}"
-
-# @tool
-# def math_solver(expression: str) -> str:
-#     """
-#     Solve or simplify a symbolic math expression.
-#     Use this for algebraic simplifications or symbolic math steps.
-#     """
-#     print(f"🛠️ math_solver called with: {expression}")
-#     try:
-#         result = sympy.simplify(expression)
-#         return str(result)
-#     except Exception as e:
-#         return f"Error solving math: {e}"
-
-# @tool
-# def document_search(filepath: str, query: str) -> str:
-#     """
-#     Search for a specific keyword or sentence inside a PDF document.
-#     Use this for information retrieval from uploaded documents.
-#     Returns the page text where the query was found.
-#     """
-#     print(f"🛠️ document_search called with: {filepath}, query={query}")
-#     try:
-#         doc = fitz.open(filepath)
-#         for page in doc:
-#             text = page.get_text()
-#             if query.lower() in text.lower():
-#                 return text
-#         return "Not found."
-#     except Exception as e:
-#         return f"Error searching document: {e}"
-
-# @tool
-# def transcribe_audio(file_path: str) -> str:
-#     """
-#     Transcribe an audio file into text using Whisper model.
-#     Use this when the user provides audio recordings.
-#     Returns the full transcript.
-#     """
-#     print(f"🛠️ transcribe_audio called with: {file_path}")
-#     try:
-#         model = whisper.load_model("small")
-#         result = model.transcribe(file_path)
-#         return result["text"]
-#     except Exception as e:
-#         return f"Error transcribing audio: {e}"
-
-# @tool
-# def search_imdb(title: str) -> str:
-#     """
-#     Look up a movie or show title on IMDb.
-#     Use this to identify media titles, actors, or related entertainment metadata.
-#     Returns the best matching title.
-#     """
-#     print(f"🛠️ search_imdb called with: {title}")
-#     try:
-#         ia = IMDb()
-#         results = ia.search_movie(title)
-#         if results:
-#             return results[0]['title']
-#         return "Not found."
-#     except Exception as e:
-#         return f"Error searching IMDb: {e}"
-
-# @tool
-# def run_python_code(code: str) -> str:
-#     """
-#     Safely execute a block of Python code.
-#     Use this for running basic computations or logic blocks shared in string form.
-#     Returns the result of 'result' variable if defined.
-#     """
-#     print(f"🛠️ run_python_code called with code snippet: {code[:60]}...")
-#     try:
-#         exec_globals = {}
-#         exec(code, exec_globals)
-#         return str(exec_globals.get("result", "Executed"))
-#     except Exception as e:
-#         return f"Error executing code: {e}"
 
 @tool
 def analyze_youtube_video(url: str) -> str:
@@ -314,115 +114,10 @@ def fetch_video_metadata(video_id: str) -> str:
     except Exception as e:
         return f"❌ Error fetching metadata: {e}"
         
-# @tool
-# def search_baseball_stats(player: str) -> str:
-#     """
-#     Simulate a lookup of a baseball player's stats on Baseball Reference.
-#     Use this for sports questions involving MLB players.
-#     """
-#     print(f"🛠️ search_baseball_stats called with: {player}")
-#     try:
-#         url = f"https://www.baseball-reference.com/search/search.fcgi?search={player}"
-#         r = requests.get(url)
-#         return "Player found on Baseball Reference." if r.status_code == 200 else "Player not found."
-#     except Exception as e:
-#         return f"Error searching baseball stats: {e}"
-
-# @tool
-# def scrape_document(url: str) -> str:
-#     """
-#     Download and extract raw text from a web page.
-#     Use this when needing the full text content of an online article.
-#     """
-#     print(f"🛠️ scrape_document called with: {url}")
-#     try:
-#         r = requests.get(url)
-#         soup = BeautifulSoup(r.text, 'html.parser')
-#         return soup.get_text()
-#     except Exception as e:
-#         return f"Error scraping document: {e}"
-
-# @tool
-# def search_zoological_database(query: str) -> str:
-#     """
-#     Simulate a search in a zoological records database.
-#     Use for biological or species-related queries when the user references animal data.
-#     """
-#     print(f"🛠️ search_zoological_database called with: {query}")
-#     return f"Simulated search result for {query} in zoological database."
-
-# @tool
-# def search_olympic_database(country: str) -> str:
-#     """
-#     Simulate a country search in an Olympics dataset.
-#     Use this for questions about Olympic history, participation, or medal stats by country.
-#     """
-#     print(f"🛠️ search_olympic_database called with: {country}")
-#     return f"Simulated search for {country} in Olympics database."
-
-# @tool
-# def search_baseball_roster(player_number: str) -> str:
-#     """
-#     Simulate a roster lookup for a baseball player by jersey number.
-#     Use for lineup, identity, or number-matching tasks.
-#     """
-#     print(f"🛠️ search_baseball_roster called with: {player_number}")
-#     return f"Simulated search result for player number {player_number}."
-
-# @tool
-# def process_excel(file_path: str) -> str:
-#     """
-#     Analyze an Excel sheet to calculate total food sales.
-#     Use this for business, finance, or menu data-related tasks.
-#     Returns total food sales as formatted currency.
-#     """
-#     print(f"🛠️ process_excel called with: {file_path}")
-#     try:
-#         df = pd.read_excel(file_path)
-#         food_sales = df[df['Category'] == 'Food']['Sales'].sum()
-#         return f"{food_sales:.2f} USD"
-#     except Exception as e:
-#         return f"Error processing Excel file: {e}"
-
-# @tool
-# def search_classical_music_database(query: str) -> str:
-#     """
-#     Simulate a lookup in a classical music competition or archive database.
-#     Use for questions about award winners, orchestras, or composers.
-#     """
-#     print(f"🛠️ search_classical_music_database called with: {query}")
-#     return f"Simulated search result for {query} in classical music competition database."
-
-# @tool
-# def reverse_sentence(sentence: str) -> str:
-#     """
-#     Reverse each word in a sentence individually.
-#     Use this for reversing letter order in each word, not the sentence order.
-#     """
-#     print(f"🛠️ reverse_sentence called with: {sentence}")
-#     return " ".join(word[::-1] for word in sentence.split())
 
 # List of all tools
 all_tools = [
-    # search_and_extract_number,
-    # search_web,
     extract_number_from_text,
-    # search_duckduckgo,
     search_wikipedia,
     analyze_youtube_video,
-    # reverse_sentence,
-    # solve_chess_position,
-    # math_solver,
-    # document_search,
-    # classify_vegetable,
-    # transcribe_audio,
-    # search_imdb,
-    # run_python_code,
-    # search_baseball_stats,
-    # scrape_document,
-    # search_zoological_database,
-    # search_olympic_database,
-    # search_baseball_roster,
-    # process_excel,
-    # search_classical_music_database,
 ]