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

refactor: improve llm-based web_search tools (#87)

Browse Source
This commit is contained in:
sigoden
2024-08-01 15:48:06 +08:00
committed by GitHub
parent 1cb74d4910
commit 4c9504bc12
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/web_search_cohere.sh
+2 -1
View File
@@ -5,6 +5,7 @@ set -e
# Use this when you need current information or feel a search could provide a better answer. # 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_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. # @option --query! The query to search for.
main() { main() {
@@ -14,7 +15,7 @@ main() {
-H "content-type: application/json" \ -H "content-type: application/json" \
--data ' --data '
{ {
"model": "command-r", "model": "'"$COHERE_WEB_SEARCH_MODEL"'",
"message": "'"$argc_query"'", "message": "'"$argc_query"'",
"connectors": [{"id": "web-search"}] "connectors": [{"id": "web-search"}]
} }
tools/web_search_perplexity.sh
+2 -1
View File
@@ -5,6 +5,7 @@ set -e
# Use this when you need current information or feel a search could provide a better answer. # 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_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. # @option --query! The query to search for.
main() { main() {
@@ -14,7 +15,7 @@ main() {
-H "content-type: application/json" \ -H "content-type: application/json" \
--data ' --data '
{ {
"model": "llama-3-sonar-small-32k-online", "model": "'"$PERPLEXITY_WEB_SEARCH_MODEL"'",
"messages": [ "messages": [
{ {
"role": "user", "role": "user",