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 |
|