diff options
Diffstat (limited to 'tools')
| -rwxr-xr-x | tools/web_search_cohere.sh | 3 | ||||
| -rwxr-xr-x | tools/web_search_perplexity.sh | 3 |
2 files changed, 4 insertions, 2 deletions
diff --git a/tools/web_search_cohere.sh b/tools/web_search_cohere.sh index 5744b7c..dd4dbad 100755 --- a/tools/web_search_cohere.sh +++ b/tools/web_search_cohere.sh @@ -5,6 +5,7 @@ set -e # Use this when you need current information or feel a search could provide a better answer. # @env COHERE_API_KEY! The api key +# @env COHERE_WEB_SEARCH_MODEL=command-r The LLM model for web search # @option --query! The query to search for. main() { @@ -14,7 +15,7 @@ main() { -H "content-type: application/json" \ --data ' { - "model": "command-r", + "model": "'"$COHERE_WEB_SEARCH_MODEL"'", "message": "'"$argc_query"'", "connectors": [{"id": "web-search"}] } diff --git a/tools/web_search_perplexity.sh b/tools/web_search_perplexity.sh index 95b3d27..c5b7ee6 100755 --- a/tools/web_search_perplexity.sh +++ b/tools/web_search_perplexity.sh @@ -5,6 +5,7 @@ set -e # Use this when you need current information or feel a search could provide a better answer. # @env PERPLEXITY_API_KEY! The api key +# @env PERPLEXITY_WEB_SEARCH_MODEL=llama-3.1-sonar-small-128k-online The LLM model for web search # @option --query! The query to search for. main() { @@ -14,7 +15,7 @@ main() { -H "content-type: application/json" \ --data ' { - "model": "llama-3-sonar-small-32k-online", + "model": "'"$PERPLEXITY_WEB_SEARCH_MODEL"'", "messages": [ { "role": "user", |