aboutsummaryrefslogtreecommitdiffstats
path: root/tools/web_search_vertexai.sh
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-11-02 20:42:42 +0800
committerGitHub <noreply@github.com>2024-11-02 20:42:42 +0800
commit3573a43978afbb519a8793a3cd70dd5ec14dcdfa (patch)
treeb5c0543a9f0d643140666f61f442ab4111c8e23a /tools/web_search_vertexai.sh
parent68b1f57a4b3b932247e14d05933aee3aa6c022a4 (diff)
downloadllm-functions-docker-3573a43978afbb519a8793a3cd70dd5ec14dcdfa.tar.gz
feat(tool): add web_search_aichat.sh (#120)
Diffstat (limited to 'tools/web_search_vertexai.sh')
-rwxr-xr-xtools/web_search_vertexai.sh51
1 files changed, 0 insertions, 51 deletions
diff --git a/tools/web_search_vertexai.sh b/tools/web_search_vertexai.sh
deleted file mode 100755
index a1477d7..0000000
--- a/tools/web_search_vertexai.sh
+++ /dev/null
@@ -1,51 +0,0 @@
-#!/usr/bin/env bash
-set -e
-
-# @describe Perform a web search using VertexAI Gemini API 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.
-
-# @env VERTEXAI_PROJECT_ID! The project id
-# @env VERTEXAI_LOCATION! The location
-# @env VERTEXAI_WEB_SEARCH_MODEL=gemini-1.5-pro-001 The LLM model for web search
-# @option --query! The query to search for.
-# @meta require-tools gcloud
-
-main() {
- curl -fsSL https://$VERTEXAI_LOCATION-aiplatform.googleapis.com/v1beta1/projects/$VERTEXAI_PROJECT_ID/locations/$VERTEXAI_LOCATION/publishers/google/models/$VERTEXAI_WEB_SEARCH_MODEL:generateContent \
- -X POST \
- -H "Authorization: Bearer $(gcloud auth application-default print-access-token)" \
- -H "Content-Type: application/json" \
- -d '
-{
- "contents": [{
- "role": "user",
- "parts": [{
- "text": "'"$argc_query"'"
- }]
- }],
- "safetySettings": [
- {
- "category": "HARM_CATEGORY_HARASSMENT",
- "threshold": "BLOCK_ONLY_HIGH"
- },
- {
- "category": "HARM_CATEGORY_HATE_SPEECH",
- "threshold": "BLOCK_ONLY_HIGH"
- },
- {
- "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
- "threshold": "BLOCK_ONLY_HIGH"
- },
- {
- "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
- "threshold": "BLOCK_ONLY_HIGH"
- }
- ],
- "tools": [{
- "googleSearchRetrieval": {}
- }]
- }' | \
- jq -r '.candidates[0].content.parts[0].text' >> "$LLM_OUTPUT"
-}
-
-eval "$(argc --argc-eval "$0" "$@")"