Spaces:
Sleeping
Sleeping
Update tools.py
Browse files
tools.py
CHANGED
@@ -132,6 +132,7 @@ def parse_excel_tool(args: dict) -> str:
|
|
132 |
• excel_sheet_name – optional sheet to load
|
133 |
returns: stringified list of records from the specified sheet
|
134 |
"""
|
|
|
135 |
task_id = args.get("task_id", "")
|
136 |
sheet = args.get("excel_sheet_name", "")
|
137 |
|
@@ -160,10 +161,7 @@ def audio_transcriber_tool(args: dict) -> str:
|
|
160 |
• A Task ID (e.g. "abc123"), in which case we try downloading
|
161 |
GET {DEFAULT_API_URL}/files/{task_id} with .mp3, .wav, .m4a extensions.
|
162 |
Returns:
|
163 |
-
|
164 |
-
"audio_path": None,
|
165 |
-
"transcript": "<text or error message>"
|
166 |
-
}
|
167 |
Always attempts to download the file for the given path or task ID.
|
168 |
"""
|
169 |
print("reached audio_transcriber_tool")
|
@@ -210,11 +208,8 @@ def wikipedia_search_tool(args: dict) -> str:
|
|
210 |
"""
|
211 |
LangGraph wrapper for searching Wikipedia.
|
212 |
Expects: state["wiki_query"] to be a non‐empty string.
|
213 |
-
Returns:
|
214 |
-
|
215 |
-
"wiki_query": None,
|
216 |
-
"wiki_result": "<text summary of first matching page or an error message>"
|
217 |
-
}
|
218 |
If no valid wiki_query is provided, returns {}.
|
219 |
"""
|
220 |
print("reached wikipedia search tool")
|
@@ -238,12 +233,14 @@ def wikipedia_search_tool(args: dict) -> str:
|
|
238 |
search_results = search_data.get("query", {}).get("search", [])
|
239 |
# print("wikipedia: search_results",search_results)
|
240 |
if not search_results:
|
241 |
-
|
|
|
242 |
|
243 |
# 2) Take the first search result's title
|
244 |
first_title = search_results[0].get("title", "")
|
245 |
if not first_title:
|
246 |
-
|
|
|
247 |
|
248 |
# 3) Fetch the page summary for that title via the REST summary endpoint
|
249 |
title_for_url = requests.utils.requote_uri(first_title)
|
|
|
132 |
• excel_sheet_name – optional sheet to load
|
133 |
returns: stringified list of records from the specified sheet
|
134 |
"""
|
135 |
+
print("reached parse_excel_tool")
|
136 |
task_id = args.get("task_id", "")
|
137 |
sheet = args.get("excel_sheet_name", "")
|
138 |
|
|
|
161 |
• A Task ID (e.g. "abc123"), in which case we try downloading
|
162 |
GET {DEFAULT_API_URL}/files/{task_id} with .mp3, .wav, .m4a extensions.
|
163 |
Returns:
|
164 |
+
"<text or error message>"
|
|
|
|
|
|
|
165 |
Always attempts to download the file for the given path or task ID.
|
166 |
"""
|
167 |
print("reached audio_transcriber_tool")
|
|
|
208 |
"""
|
209 |
LangGraph wrapper for searching Wikipedia.
|
210 |
Expects: state["wiki_query"] to be a non‐empty string.
|
211 |
+
Returns: text summary of first matching page or an error message>"
|
212 |
+
|
|
|
|
|
|
|
213 |
If no valid wiki_query is provided, returns {}.
|
214 |
"""
|
215 |
print("reached wikipedia search tool")
|
|
|
233 |
search_results = search_data.get("query", {}).get("search", [])
|
234 |
# print("wikipedia: search_results",search_results)
|
235 |
if not search_results:
|
236 |
+
print(f"No Wikipedia page found for '{query}'.")
|
237 |
+
return f"No Wikipedia page found for '{query}'."
|
238 |
|
239 |
# 2) Take the first search result's title
|
240 |
first_title = search_results[0].get("title", "")
|
241 |
if not first_title:
|
242 |
+
print("Unexpected format from Wikipedia search.")
|
243 |
+
return "Unexpected format from Wikipedia search."
|
244 |
|
245 |
# 3) Fetch the page summary for that title via the REST summary endpoint
|
246 |
title_for_url = requests.utils.requote_uri(first_title)
|