renaming exa tool

This commit is contained in:
João Moura
2024-05-02 02:50:54 -03:00
parent ffd5942b31
commit 768bb74a2c
4 changed files with 33 additions and 1 deletions

View File

@@ -0,0 +1,24 @@
import requests
from typing import Any
from .exa_base_tool import EXABaseTool
class EXASearchTool(EXABaseTool):
def _run(
self,
**kwargs: Any,
) -> Any:
search_query = kwargs.get('search_query')
if search_query is None:
search_query = kwargs.get('query')
payload = {
"query": search_query,
}
response = requests.post(self.search_url, json=payload, headers=self.headers)
results = response.json()
if 'results' in results:
results = super()._parse_results(results['results'])
else:
return results