From 22a87a8753ca1ad23ebe0a816c230302edfd7416 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Tue, 10 Dec 2024 12:56:53 +0800
Subject: refactor: update web_search_aichat.sh (#136)

---
 tools/web_search_aichat.sh | 10 ++--------
 1 file changed, 2 insertions(+), 8 deletions(-)

(limited to 'tools')

diff --git a/tools/web_search_aichat.sh b/tools/web_search_aichat.sh
index 6a6528e..a20bd35 100755
--- a/tools/web_search_aichat.sh
+++ b/tools/web_search_aichat.sh
@@ -11,7 +11,6 @@ set -e
 # @env WEB_SEARCH_MODEL! The model for web-searching.
 #
 # supported aichat models:
-#   - cohere:*
 #   - vertexai:gemini-*
 #   - perplexity:*-online
 #   - ernie:*
@@ -20,14 +19,9 @@ set -e
 
 main() {
     client="${WEB_SEARCH_MODEL%%:*}"
-    case "$client" in
-    cohere)
-        export AICHAT_PATCH_COHERE_CHAT_COMPLETIONS='{".*":{"body":{"connectors":[{"id":"web-search"}]}}}'
-        ;;
-    vertexai)
+    if [[ "$client" == "vertexai" ]]; then
         export AICHAT_PATCH_VERTEXAI_CHAT_COMPLETIONS='{"gemini-.*":{"body":{"tools":[{"googleSearchRetrieval":{}}]}}}'
-        ;;
-    esac
+    fi
     aichat -m "$WEB_SEARCH_MODEL" "$argc_query" >> "$LLM_OUTPUT"
 }
 
-- 
cgit v1.2.3