Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -456,22 +456,28 @@ def process_tool_request(tool_request_data):
|
|
| 456 |
tool_name = tool_request_data["name"]
|
| 457 |
tool_parameters = tool_request_data["parameters"]
|
| 458 |
tool_return = None
|
|
|
|
| 459 |
if tool_name == arxiv_tool.json_name:
|
| 460 |
query = tool_parameters["query"]
|
| 461 |
max_results = tool_parameters.get("max_results", 5)
|
| 462 |
sort_by = tool_parameters.get("sort_by", "relevance")
|
| 463 |
search_results = arxiv_tool.actual_function(query=query, max_results=max_results, sort_by=sort_by)
|
| 464 |
tool_return = {"name": arxiv_tool.json_name, "return": search_results}
|
|
|
|
| 465 |
elif tool_name == nih_ref_snp_tool.json_name:
|
| 466 |
snp = tool_parameters["snp"]
|
| 467 |
search_results = nih_ref_snp_tool.actual_function(snp=snp)
|
| 468 |
tool_return = {"name": nih_ref_snp_tool.json_name, "return": search_results}
|
| 469 |
-
|
| 470 |
-
|
| 471 |
-
|
| 472 |
-
|
|
|
|
|
|
|
|
|
|
| 473 |
else:
|
| 474 |
raise ValueError(f"Unknown tool name: {tool_name}")
|
|
|
|
| 475 |
lgs("TOOL: " + str(tool_return))
|
| 476 |
return tool_return
|
| 477 |
|
|
|
|
| 456 |
tool_name = tool_request_data["name"]
|
| 457 |
tool_parameters = tool_request_data["parameters"]
|
| 458 |
tool_return = None
|
| 459 |
+
|
| 460 |
if tool_name == arxiv_tool.json_name:
|
| 461 |
query = tool_parameters["query"]
|
| 462 |
max_results = tool_parameters.get("max_results", 5)
|
| 463 |
sort_by = tool_parameters.get("sort_by", "relevance")
|
| 464 |
search_results = arxiv_tool.actual_function(query=query, max_results=max_results, sort_by=sort_by)
|
| 465 |
tool_return = {"name": arxiv_tool.json_name, "return": search_results}
|
| 466 |
+
|
| 467 |
elif tool_name == nih_ref_snp_tool.json_name:
|
| 468 |
snp = tool_parameters["snp"]
|
| 469 |
search_results = nih_ref_snp_tool.actual_function(snp=snp)
|
| 470 |
tool_return = {"name": nih_ref_snp_tool.json_name, "return": search_results}
|
| 471 |
+
|
| 472 |
+
elif tool_name == perplexity_tool.programmatic_name:
|
| 473 |
+
query = tool_parameters["query"]
|
| 474 |
+
api_key = tool_parameters["api_key"]
|
| 475 |
+
search_results = perplexity_tool.actual_function(query=query, api_key=api_key)
|
| 476 |
+
tool_return = {"name": perplexity_tool.programmatic_name, "return": search_results}
|
| 477 |
+
|
| 478 |
else:
|
| 479 |
raise ValueError(f"Unknown tool name: {tool_name}")
|
| 480 |
+
|
| 481 |
lgs("TOOL: " + str(tool_return))
|
| 482 |
return tool_return
|
| 483 |
|