diff options
| author | sigoden <sigoden@gmail.com> | 2024-07-05 19:49:26 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-07-05 19:49:26 +0800 |
| commit | 74789b5a72cd8e3261f196f4e7197ddba92947d0 (patch) | |
| tree | d9029079d8368e4a35a8556aeea50ba40c9939e3 /tools/search_tavily.sh | |
| parent | c7e027c6c3176a7684f4e901b8ddb9a0916c5cd8 (diff) | |
| download | llm-functions-docker-74789b5a72cd8e3261f196f4e7197ddba92947d0.tar.gz | |
feat: add search tools (#58)
Diffstat (limited to 'tools/search_tavily.sh')
| -rwxr-xr-x | tools/search_tavily.sh | 18 |
1 files changed, 12 insertions, 6 deletions
diff --git a/tools/search_tavily.sh b/tools/search_tavily.sh index 068615d..5e99aba 100755 --- a/tools/search_tavily.sh +++ b/tools/search_tavily.sh @@ -1,18 +1,24 @@ #!/usr/bin/env bash set -e -# @describe Perform a web search using Tavily API to get up-to-date information or additional context. +# @describe Perform a web search using EXA API to get up-to-date information or additional context. # Use this when you need current information or feel a search could provide a better answer. -# @env TAVILY_API_KEY! The max results to return. +# @env TAVILY_API_KEY! The api key # @env TAVILY_MAX_RESULTS=5 The max results to return. # @option --query! The query to search for. main() { - curl -fsSL -X POST \ - -H 'content-type: application/json' \ - -d '{"api_key":"'"$TAVILY_API_KEY"'","query":"'"$argc_query"'","search_depth":"advanced","max_results":"'"$TAVILY_MAX_RESULTS"'"}' \ - https://api.tavily.com/search + curl -fsSL -X POST https://api.tavily.com/search \ + -H "content-type: application/json" \ + -d ' +{ + "api_key": "'"$TAVILY_API_KEY"'", + "query": "'"$argc_query"'", + "search_depth": "advanced", + "max_results": "'"$TAVILY_MAX_RESULTS"'" +}' | \ + jq '[.results[] | {title: .title, url: .url, content: .content}]' } eval "$(argc --argc-eval "$0" "$@")" |
