From 74789b5a72cd8e3261f196f4e7197ddba92947d0 Mon Sep 17 00:00:00 2001 From: sigoden Date: Fri, 5 Jul 2024 19:49:26 +0800 Subject: feat: add search tools (#58) --- tools/search_tavily.sh | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) (limited to 'tools/search_tavily.sh') 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" "$@")" -- cgit v1.2.3