diff options
Diffstat (limited to 'tools/web_search_cohere.sh')
| -rwxr-xr-x | tools/web_search_cohere.sh | 3 |
1 files changed, 2 insertions, 1 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"}] } |