From c7e027c6c3176a7684f4e901b8ddb9a0916c5cd8 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 4 Jul 2024 10:46:34 +0800
Subject: feat: add tool search_tavily.sh (#57)

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

(limited to 'tools/search_tavily.sh')

diff --git a/tools/search_tavily.sh b/tools/search_tavily.sh
new file mode 100755
index 0000000..068615d
--- /dev/null
+++ b/tools/search_tavily.sh
@@ -0,0 +1,19 @@
+#!/usr/bin/env bash
+set -e
+
+# @describe Perform a web search using Tavily 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 max results to return.
+# @env TAVILY_MAX_RESULTS=5 The max results to return.
+# @option --query! The query to search for.
+
+main() {
+   curl -fsSL -X POST \
+    -H 'content-type: application/json' \
+    -d '{"api_key":"'"$TAVILY_API_KEY"'","query":"'"$argc_query"'","search_depth":"advanced","max_results":"'"$TAVILY_MAX_RESULTS"'"}' \
+    https://api.tavily.com/search
+}
+
+eval "$(argc --argc-eval "$0" "$@")"
+
-- 
cgit v1.2.3