ProximileAdmin commited on
Commit
3499ff6
·
verified ·
1 Parent(s): cc3986d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -4
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
- elif tool_name == weather_tool.json_name:
470
- location = tool_parameters["location"]
471
- search_results = weather_tool.actual_function(location=location)
472
- tool_return = {"name": weather_tool.json_name, "return": search_results}
 
 
 
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