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 ++-
 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",
-- 
cgit v1.2.3