diff options
| author | sigoden <sigoden@gmail.com> | 2024-08-01 15:48:06 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-08-01 15:48:06 +0800 |
| commit | 4c9504bc123914b65ac4d15eb8fec76db01f4781 (patch) | |
| tree | 460c0624f15a2cdd691c57ef0976a24dc1334239 | |
| parent | 1cb74d4910e3f720c4be1b62c5a4bcb2ffe0ab55 (diff) | |
| download | llm-functions-docker-4c9504bc123914b65ac4d15eb8fec76db01f4781.tar.gz | |
refactor: improve llm-based web_search tools (#87)
| -rwxr-xr-x | tools/web_search_cohere.sh | 3 | ||||
| -rwxr-xr-x | tools/web_search_perplexity.sh | 3 |
2 files changed, 4 insertions, 2 deletions
diff --git a/tools/web_search_cohere.sh b/tools/web_search_cohere.sh index 5744b7c..dd4dbad 100755 --- a/tools/web_search_cohere.sh +++ b/tools/web_search_cohere.sh @@ -5,6 +5,7 @@ set -e # Use this when you need current information or feel a search could provide a better answer. # @env COHERE_API_KEY! The api key +# @env COHERE_WEB_SEARCH_MODEL=command-r The LLM model for web search # @option --query! The query to search for. main() { @@ -14,7 +15,7 @@ main() { -H "content-type: application/json" \ --data ' { - "model": "command-r", + "model": "'"$COHERE_WEB_SEARCH_MODEL"'", "message": "'"$argc_query"'", "connectors": [{"id": "web-search"}] } diff --git a/tools/web_search_perplexity.sh b/tools/web_search_perplexity.sh index 95b3d27..c5b7ee6 100755 --- a/tools/web_search_perplexity.sh +++ b/tools/web_search_perplexity.sh @@ -5,6 +5,7 @@ set -e # Use this when you need current information or feel a search could provide a better answer. # @env PERPLEXITY_API_KEY! The api key +# @env PERPLEXITY_WEB_SEARCH_MODEL=llama-3.1-sonar-small-128k-online The LLM model for web search # @option --query! The query to search for. main() { @@ -14,7 +15,7 @@ main() { -H "content-type: application/json" \ --data ' { - "model": "llama-3-sonar-small-32k-online", + "model": "'"$PERPLEXITY_WEB_SEARCH_MODEL"'", "messages": [ { "role": "user", |