From 3573a43978afbb519a8793a3cd70dd5ec14dcdfa Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sat, 2 Nov 2024 20:42:42 +0800
Subject: feat(tool): add web_search_aichat.sh (#120)

---
 tools/web_search_aichat.sh | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100755 tools/web_search_aichat.sh

(limited to 'tools/web_search_aichat.sh')

diff --git a/tools/web_search_aichat.sh b/tools/web_search_aichat.sh
new file mode 100755
index 0000000..e35948f
--- /dev/null
+++ b/tools/web_search_aichat.sh
@@ -0,0 +1,32 @@
+#!/usr/bin/env bash
+set -e
+
+# @describe Perform a web search to get up-to-date information or additional context.
+# Use this when you need current information or feel a search could provide a better answer.
+
+# @option --query! The query to search for.
+
+# @meta require-tools aichat
+# @env WEB_SEARCH_MODEL! The model for web-searching.
+#
+# supported aichat models:
+#   - cohere:*
+#   - vertexai:gemini-*
+#   - perplexity:*-online
+#   - ernie:*
+#   - lingyiwanwu:yi-large-rag
+
+main() {
+    client="${WEB_SEARCH_MODEL%%:*}"
+    case "$client" in
+    cohere)
+        export AICHAT_PATCH_COHERE_CHAT_COMPLETIONS='{".*":{"body":{"connectors":[{"id":"web-search"}]}}}'
+        ;;
+    vertexai)
+        export AICHAT_PATCH_VERTEXAI_CHAT_COMPLETIONS='{"gemini-.*":{"body":{"tools":[{"googleSearchRetrieval":{}}]}}}'
+        ;;
+    esac
+    aichat -m "$WEB_SEARCH_MODEL" "$argc_query" >> "$LLM_OUTPUT"
+}
+
+eval "$(argc --argc-eval "$0" "$@")"
-- 
cgit v1.2.3