Update app.py

app.py

@@ -15,6 +15,7 @@ HF_TOKEN = os.getenv("HF_TOKEN", None)
 
 @tool
 def wiki_get_page(title: str) -> str:
+    """Fetch raw wikitext for a given Wikipedia page title"""
     API = "https://en.wikipedia.org/w/api.php"
     params = {"action": "query", "format": "json", "prop": "revisions", "rvprop": "content", "rvslots": "*", "titles": title}
     data = requests.get(API, params=params, timeout=10).json()
@@ -23,15 +24,18 @@ def wiki_get_page(title: str) -> str:
 
 @tool
 def youtube_transcript(video_id: str) -> str:
+    """Retrieve transcript for a YouTube video ID"""
     transcript = YouTubeTranscriptApi().fetch_transcript(video_id)
     return " ".join(t["text"] for t in transcript)
 
 @tool
 def reverse_text(text: str) -> str:
+    """Reverse the input string"""
     return text[::-1]
 
 @tool
 def chess_best_move(fen: str, time_limit: float = 0.1) -> str:
+    """Best move in UCI notation for given FEN"""
     board = chess.Board(fen)
     engine = chess.engine.SimpleEngine.popen_uci("/usr/bin/stockfish")
     result = engine.play(board, chess.engine.Limit(time=time_limit))
@@ -40,6 +44,7 @@ def chess_best_move(fen: str, time_limit: float = 0.1) -> str:
 
 @tool
 def find_non_commutative(table: dict) -> list:
+    """Elements involved in non-commutativity"""
     elems = set(x for x,_ in table.keys())
     bad = set()
     for x in elems:
@@ -50,30 +55,27 @@ def find_non_commutative(table: dict) -> list:
 
 @tool
 def libretext_extract(url: str, selector: str) -> str:
+    """Extract text via CSS selector"""
     r = requests.get(url, timeout=10)
     soup = BeautifulSoup(r.text, "html.parser")
     return soup.select_one(selector).get_text(strip=True)
 
 @tool
 def classify_vegetables(items: list) -> list:
+    """Alphabetize true vegetables"""
     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
 def execute_code(code: str) -> str:
+    """Execute code and return `output`"""
     local_ns = {}
     exec(code, {"__builtins__": {}}, local_ns)
     return str(local_ns.get("output", ""))
 
-@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
 def least_athletes_olympics(year: int) -> str:
+    """IOC code of least-athlete country"""
     url = f"https://en.wikipedia.org/wiki/{year}_Summer_Olympics"
     r = requests.get(url)
     soup = BeautifulSoup(r.text,"html.parser")
@@ -86,6 +88,7 @@ def least_athletes_olympics(year: int) -> str:
 
 @tool
 def get_nasa_award_number(qid: str) -> str:
+    """NASA award number for Wikidata QID"""
     sparql = SPARQLWrapper("https://query.wikidata.org/sparql")
     sparql.setQuery(f'SELECT ?award WHERE {{ wd:{qid} wdt:P496 ?award. }}')
     sparql.setReturnFormat(JSON)