Update app.py

app.py

@@ -1,7 +1,8 @@
 import os, re, requests, pandas as pd, gradio as gr
 from langchain_huggingface.llms import HuggingFacePipeline
-from langchain_core.tools import tool
-from langchain_core.agents import AgentExecutor, JsonOutputParser
+from langchain.tools import tool
+from langchain_core,output_parsers import JsonOutputParser
+from langchain.agents import AgentExecutor
 from youtube_transcript_api import YouTubeTranscriptApi
 import chess, chess.engine
 from bs4 import BeautifulSoup
@@ -12,12 +13,7 @@ from SPARQLWrapper import SPARQLWrapper, JSON
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 HF_TOKEN = os.getenv("HF_TOKEN", None)
 
-@tool(
-    name="wiki_get_page",
-    description="Fetch raw wikitext for a given Wikipedia page title",
-    inputs={"title": "string"},
-    output_type="string",
-)
+@tool
 def wiki_get_page(title: str) -> str:
     API = "https://en.wikipedia.org/w/api.php"
     params = {"action": "query", "format": "json", "prop": "revisions", "rvprop": "content", "rvslots": "*", "titles": title}
@@ -25,31 +21,16 @@ def wiki_get_page(title: str) -> str:
     page = next(iter(data["query"]["pages"].values()))
     return page["revisions"][0]["slots"]["main"]["*"]
 
-@tool(
-    name="youtube_transcript",
-    description="Retrieve transcript for a YouTube video ID",
-    inputs={"video_id": "string"},
-    output_type="string",
-)
+@tool
 def youtube_transcript(video_id: str) -> str:
     transcript = YouTubeTranscriptApi().fetch_transcript(video_id)
     return " ".join(t["text"] for t in transcript)
 
-@tool(
-    name="reverse_text",
-    description="Reverse the input string",
-    inputs={"text": "string"},
-    output_type="string",
-)
+@tool
 def reverse_text(text: str) -> str:
     return text[::-1]
 
-@tool(
-    name="chess_best_move",
-    description="Return best move in UCI notation for given FEN",
-    inputs={"fen": "string", "time_limit": "float"},
-    output_type="string",
-)
+@tool
 def chess_best_move(fen: str, time_limit: float = 0.1) -> str:
     board = chess.Board(fen)
     engine = chess.engine.SimpleEngine.popen_uci("/usr/bin/stockfish")
@@ -57,12 +38,7 @@ def chess_best_move(fen: str, time_limit: float = 0.1) -> str:
     engine.quit()
     return result.move.uci()
 
-@tool(
-    name="find_non_commutative",
-    description="Find elements involved in non-commutativity from operation table",
-    inputs={"table": "dict"},
-    output_type="list[string]",
-)
+@tool
 def find_non_commutative(table: dict) -> list:
     elems = set(x for x,_ in table.keys())
     bad = set()
@@ -72,56 +48,31 @@ def find_non_commutative(table: dict) -> list:
                 bad.update([x,y])
     return sorted(bad)
 
-@tool(
-    name="libretext_extract",
-    description="Extract text from LibreTexts URL using CSS selector",
-    inputs={"url": "string", "selector": "string"},
-    output_type="string",
-)
+@tool
 def libretext_extract(url: str, selector: str) -> str:
     r = requests.get(url, timeout=10)
     soup = BeautifulSoup(r.text, "html.parser")
     return soup.select_one(selector).get_text(strip=True)
 
-@tool(
-    name="classify_vegetables",
-    description="Return alphabetized list of vegetables from input list",
-    inputs={"items": "list[string]"},
-    output_type="list[string]",
-)
+@tool
 def classify_vegetables(items: list) -> list:
     VEGETABLE_SET = {"bell pepper","broccoli","celery","green beans","lettuce","zucchini","sweet potatoes"}
     return sorted([i for i in items if i in VEGETABLE_SET])
 
-@tool(
-    name="execute_code",
-    description="Execute Python code snippet and return 'output' variable",
-    inputs={"code": "string"},
-    output_type="string",
-)
+@tool
 def execute_code(code: str) -> str:
     local_ns = {}
     exec(code, {"__builtins__": {}}, local_ns)
     return str(local_ns.get("output", ""))
 
-@tool(
-    name="yankee_at_bats_most_walks",
-    description="Return at bats for Yankee with most walks in given season",
-    inputs={"year": "int"},
-    output_type="int",
-)
+@tool
 def yankee_at_bats_most_walks(year: int) -> int:
     leaders = statsapi.team_leaders("walks", season=year, team=147)
     pid = leaders[0]["id"]
     stats = statsapi.player_stats(pid, "hitting", "season", season=year)
     return stats["batting"][0]["atBats"]
 
-@tool(
-    name="least_athletes_olympics",
-    description="Return IOC code of country with least athletes in given Olympics year",
-    inputs={"year": "int"},
-    output_type="string",
-)
+@tool
 def least_athletes_olympics(year: int) -> str:
     url = f"https://en.wikipedia.org/wiki/{year}_Summer_Olympics"
     r = requests.get(url)
@@ -133,12 +84,7 @@ def least_athletes_olympics(year: int) -> str:
     candidates = sorted([code for code,count in data if count==min_val])
     return candidates[0]
 
-@tool(
-    name="get_nasa_award_number",
-    description="Get NASA award number for a Wikidata QID",
-    inputs={"qid": "string"},
-    output_type="string",
-)
+@tool
 def get_nasa_award_number(qid: str) -> str:
     sparql = SPARQLWrapper("https://query.wikidata.org/sparql")
     sparql.setQuery(f'SELECT ?award WHERE {{ wd:{qid} wdt:P496 ?award. }}')