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_perplexity.sh | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100755 tools/web_search_perplexity.sh

(limited to 'tools/web_search_perplexity.sh')

diff --git a/tools/web_search_perplexity.sh b/tools/web_search_perplexity.sh
new file mode 100755
index 0000000..95b3d27
--- /dev/null
+++ b/tools/web_search_perplexity.sh
@@ -0,0 +1,30 @@
+#!/usr/bin/env bash
+set -e
+
+# @describe Perform a web search using Perplexity 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 PERPLEXITY_API_KEY! The api key
+# @option --query! The query to search for.
+
+main() {
+    curl -fsS -X POST https://api.perplexity.ai/chat/completions \
+     -H "authorization: Bearer $PERPLEXITY_API_KEY" \
+     -H "accept: application/json" \
+     -H "content-type: application/json" \
+     --data '
+{
+  "model": "llama-3-sonar-small-32k-online",
+  "messages": [
+    {
+      "role": "user",
+      "content": "'"$argc_query"'"
+    }
+  ]
+}
+'  | \
+        jq -r '.choices[0].message.content' \
+        >> "$LLM_OUTPUT"
+}
+
+eval "$(argc --argc-eval "$0" "$@")"
-- 
cgit v1.2.3