Dark-Alex-17/llm-functions
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: update web_search_aichat.sh (#136)

Browse Source
This commit is contained in:
sigoden
2024-12-10 12:56:53 +08:00
committed by GitHub
parent 0f0ae607d3
commit 22a87a8753
1 changed files with 2 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/web_search_aichat.sh
+2 -8
View File
@@ -11,7 +11,6 @@ set -e
# @env WEB_SEARCH_MODEL! The model for web-searching. # @env WEB_SEARCH_MODEL! The model for web-searching.
# #
# supported aichat models: # supported aichat models:
# - cohere:*
# - vertexai:gemini-* # - vertexai:gemini-*
# - perplexity:*-online # - perplexity:*-online
# - ernie:* # - ernie:*
@@ -20,14 +19,9 @@ set -e
main() { main() {
client="${WEB_SEARCH_MODEL%%:*}" client="${WEB_SEARCH_MODEL%%:*}"
case "$client" in if [[ "$client" == "vertexai" ]]; then
cohere)
export AICHAT_PATCH_COHERE_CHAT_COMPLETIONS='{".*":{"body":{"connectors":[{"id":"web-search"}]}}}'
;;
vertexai)
export AICHAT_PATCH_VERTEXAI_CHAT_COMPLETIONS='{"gemini-.*":{"body":{"tools":[{"googleSearchRetrieval":{}}]}}}' export AICHAT_PATCH_VERTEXAI_CHAT_COMPLETIONS='{"gemini-.*":{"body":{"tools":[{"googleSearchRetrieval":{}}]}}}'
;; fi
esac
aichat -m "$WEB_SEARCH_MODEL" "$argc_query" >> "$LLM_OUTPUT" aichat -m "$WEB_SEARCH_MODEL" "$argc_query" >> "$LLM_OUTPUT"
} }