diff options
| author | sigoden <sigoden@gmail.com> | 2024-11-02 20:42:42 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-11-02 20:42:42 +0800 |
| commit | 3573a43978afbb519a8793a3cd70dd5ec14dcdfa (patch) | |
| tree | b5c0543a9f0d643140666f61f442ab4111c8e23a /tools/web_search_vertexai.sh | |
| parent | 68b1f57a4b3b932247e14d05933aee3aa6c022a4 (diff) | |
| download | llm-functions-docker-3573a43978afbb519a8793a3cd70dd5ec14dcdfa.tar.gz | |
feat(tool): add web_search_aichat.sh (#120)
Diffstat (limited to 'tools/web_search_vertexai.sh')
| -rwxr-xr-x | tools/web_search_vertexai.sh | 51 |
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" "$@")" |
