From 4c9504bc123914b65ac4d15eb8fec76db01f4781 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 1 Aug 2024 15:48:06 +0800
Subject: refactor: improve llm-based web_search tools (#87)

---
 tools/web_search_cohere.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'tools/web_search_cohere.sh')

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"}]
 }
-- 
cgit v1.2.3