From a7342a1cbf0ae558db9005870ddb20a539daac68 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 29 Jul 2024 07:32:26 +0800
Subject: feat: add web_search_* tools, rm some search_* tools (#78)

---
 tools/web_search_tavily.sh | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100755 tools/web_search_tavily.sh

(limited to 'tools/web_search_tavily.sh')

diff --git a/tools/web_search_tavily.sh b/tools/web_search_tavily.sh
new file mode 100755
index 0000000..d80e5dd
--- /dev/null
+++ b/tools/web_search_tavily.sh
@@ -0,0 +1,22 @@
+#!/usr/bin/env bash
+set -e
+
+# @describe Perform a web search using EXA 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 TAVILY_API_KEY! The api key
+# @option --query! The query to search for.
+
+main() {
+    curl -fsSL -X POST https://api.tavily.com/search \
+        -H "content-type: application/json" \
+        -d '
+{
+    "api_key": "'"$TAVILY_API_KEY"'",
+    "query": "'"$argc_query"'",
+    "include_answer": true
+}' | \
+    jq -r '.answer' >> "$LLM_OUTPUT"
+}
+
+eval "$(argc --argc-eval "$0" "$@")"
-- 
cgit v1.2.3