shuiyu1123
dc77dfd304
- 新增 operit_runtime.py: 跨平台 Python 运行时 - 实现 Tools.System/Files/Net/UI/Memory/FFmpeg/Workflow/Chat/SoftwareSettings/Tasker - 实现 OkHttp 链式 HTTP 客户端兼容层 - 实现 complete()/exports/console 全局对象 - 实现 OperitScriptLoader 脚本加载器 - 支持 Windows/macOS/Linux 三平台 - 新增 operit_py/ 目录: 47 个 Python 翻译脚本 - 搜索类: duckduckgo, google_search, tavily, zhipu_search, various_search, crossref - 绘图类: openai_draw, qwen_draw, xai_draw, minimax_draw, siliconflow_draw, zhipu_draw, nanobanana_draw - 媒体类: ffmpeg, file_converter, douyin_download, bilibili_tools, jmcomic - 文件类: extended_file_tools, hex_editor, reader, beeimg_image_uploader - 网络类: extended_http_tools, network_test - 自动化类: browser, automatic_ui_base, automatic_ui_subagent, automatic_bilibili_assistant, automatic_xiaohongshu_assistant, automatic_baidu_map_assistant - 系统类: system_tools, super_admin, daily_life, operit_editor - 对话类: ai_chat, extended_chat, qq_intelligent - 工作流类: workflow, vflow_trigger, tasker - 工具类: time, code_runner, quick_start, extended_memory_tools, baidu_map, 12306, pdf_vision_parser - 测试结果: 47/47 语法检查通过, 38 脚本加载成功, 280 个工具注册 - Android 专属功能已降级处理(保留接口,返回不支持提示)
200 lines
11 KiB
Python
200 lines
11 KiB
Python
# METADATA
|
|
# {
|
|
# "name": "tavily",
|
|
#
|
|
# "display_name": {
|
|
# "zh": "Tavily 搜索",
|
|
# "en": "Tavily Search"
|
|
# },
|
|
# "description": {
|
|
# "zh": "使用Tavily API执行高级网络搜索、内容提取、网站爬取和站点地图生成。",
|
|
# "en": "Use the Tavily API for advanced web search, content extraction, website crawling, and sitemap generation."
|
|
# },
|
|
# "env": [
|
|
# "TAVILY_API_KEY"
|
|
# ],
|
|
# "category": "Search",
|
|
# "tools": [
|
|
# {
|
|
# "name": "search",
|
|
# "description": { "zh": "一个强大的网络搜索工具，使用Tavily的AI搜索引擎提供全面、实时的结果。", "en": "Powerful web search using Tavily's AI search engine for comprehensive, up-to-date results." },
|
|
# "parameters": [
|
|
# { "name": "query", "description": { "zh": "搜索查询", "en": "Search query" }, "type": "string", "required": true },
|
|
# { "name": "search_depth", "description": { "zh": "搜索深度，可以是 'basic' 或 'advanced'", "en": "Search depth: 'basic' or 'advanced'" }, "type": "string", "required": false, "default": "basic" },
|
|
# { "name": "topic", "description": { "zh": "搜索类别，可以是 'general' 或 'news'", "en": "Search topic: 'general' or 'news'" }, "type": "string", "required": false, "default": "general" },
|
|
# { "name": "days", "description": { "zh": "包含在搜索结果中的过去天数", "en": "Number of past days to include in results" }, "type": "number", "required": false },
|
|
# { "name": "max_results", "description": { "zh": "返回的最大搜索结果数", "en": "Maximum number of search results" }, "type": "number", "required": false, "default": 10 },
|
|
# { "name": "include_images", "description": { "zh": "在响应中包含与查询相关的图像列表", "en": "Include a list of images related to the query in the response" }, "type": "boolean", "required": false, "default": false },
|
|
# { "name": "include_raw_content", "description": { "zh": "包含每个搜索结果的已清理和解析的HTML内容", "en": "Include cleaned and parsed HTML content for each search result" }, "type": "boolean", "required": false, "default": false },
|
|
# { "name": "include_domains", "description": { "zh": "要专门包含在搜索结果中的域名列表", "en": "Domain list to specifically include in results" }, "type": "array", "required": false, "default": [] },
|
|
# { "name": "exclude_domains", "description": { "zh": "要专门排除的域名列表", "en": "Domain list to specifically exclude from results" }, "type": "array", "required": false, "default": [] }
|
|
# ]
|
|
# },
|
|
# {
|
|
# "name": "extract",
|
|
# "description": { "zh": "一个强大的网页内容提取工具，从指定的URL检索和处理原始内容。", "en": "Extract web content from specified URLs and process the raw content." },
|
|
# "parameters": [
|
|
# { "name": "urls", "description": { "zh": "要提取内容的URL列表", "en": "List of URLs to extract content from" }, "type": "array", "required": true },
|
|
# { "name": "extract_depth", "description": { "zh": "提取深度 - 'basic' 或 'advanced'", "en": "Extraction depth: 'basic' or 'advanced'" }, "type": "string", "required": false, "default": "basic" },
|
|
# { "name": "include_images", "description": { "zh": "在响应中包含从URL中提取的图像列表", "en": "Include a list of images extracted from the URL(s)" }, "type": "boolean", "required": false, "default": false },
|
|
# { "name": "format", "description": { "zh": "提取内容的格式，'markdown' 或 'text'", "en": "Output format: 'markdown' or 'text'" }, "type": "string", "required": false, "default": "markdown" }
|
|
# ]
|
|
# },
|
|
# {
|
|
# "name": "crawl",
|
|
# "description": { "zh": "一个强大的网络爬虫，从指定的基URL开始结构化地爬取网站。", "en": "Structured website crawler starting from a base URL." },
|
|
# "parameters": [
|
|
# { "name": "url", "description": { "zh": "开始爬取的根URL", "en": "Root URL to start crawling from" }, "type": "string", "required": true },
|
|
# { "name": "max_depth", "description": { "zh": "爬取的最大深度", "en": "Maximum crawl depth" }, "type": "number", "required": false, "default": 1 },
|
|
# { "name": "max_breadth", "description": { "zh": "每层要跟随的最大链接数", "en": "Maximum number of links to follow per depth level" }, "type": "number", "required": false, "default": 20 },
|
|
# { "name": "limit", "description": { "zh": "爬虫将处理的总链接数", "en": "Total number of links the crawler will process" }, "type": "number", "required": false, "default": 50 },
|
|
# { "name": "instructions", "description": { "zh": "给爬虫的自然语言指令", "en": "Natural-language instructions for the crawler" }, "type": "string", "required": false },
|
|
# { "name": "allow_external", "description": { "zh": "是否允许跟随外部域名的链接", "en": "Whether to allow following links to external domains" }, "type": "boolean", "required": false, "default": false },
|
|
# { "name": "format", "description": { "zh": "提取内容的格式，'markdown' 或 'text'", "en": "Output format: 'markdown' or 'text'" }, "type": "string", "required": false, "default": "markdown" }
|
|
# ]
|
|
# },
|
|
# {
|
|
# "name": "map",
|
|
# "description": { "zh": "一个强大的网站地图工具，创建网站URL的结构化地图。", "en": "Generate a structured sitemap of website URLs." },
|
|
# "parameters": [
|
|
# { "name": "url", "description": { "zh": "开始映射的根URL", "en": "Root URL to start mapping from" }, "type": "string", "required": true },
|
|
# { "name": "max_depth", "description": { "zh": "映射的最大深度", "en": "Maximum mapping depth" }, "type": "number", "required": false, "default": 1 },
|
|
# { "name": "max_breadth", "description": { "zh": "每层要跟随的最大链接数", "en": "Maximum number of links to follow per depth level" }, "type": "number", "required": false, "default": 20 },
|
|
# { "name": "limit", "description": { "zh": "爬虫将处理的总链接数", "en": "Total number of links the crawler will process" }, "type": "number", "required": false, "default": 50 }
|
|
# ]
|
|
# }
|
|
# ]
|
|
# }
|
|
|
|
import json
|
|
import traceback
|
|
|
|
client = OkHttp.newClient()
|
|
BASE_URLS = {
|
|
"search": "https://api.tavily.com/search",
|
|
"extract": "https://api.tavily.com/extract",
|
|
"crawl": "https://api.tavily.com/crawl",
|
|
"map": "https://api.tavily.com/map"
|
|
}
|
|
|
|
|
|
async def makeTavilyRequest(endpoint, params):
|
|
apiKey = getEnv("TAVILY_API_KEY")
|
|
if not apiKey:
|
|
raise Exception("Tavily API key is not set. Please configure it in the environment variables.")
|
|
|
|
requestBody = {**params}
|
|
|
|
headers = {
|
|
"accept": "application/json",
|
|
"content-type": "application/json",
|
|
"Authorization": f"Bearer {apiKey}"
|
|
}
|
|
|
|
try:
|
|
request = client.newRequest()
|
|
request = request.url(endpoint)
|
|
request = request.method("POST")
|
|
request = request.headers(headers)
|
|
request = request.body(json.dumps(requestBody), "json")
|
|
|
|
response = await request.build().execute()
|
|
|
|
if not response.isSuccessful():
|
|
raise Exception(f"Tavily API Error: {response.statusCode} - {response.content}")
|
|
return json.loads(response.content)
|
|
except Exception as error:
|
|
console.error(f"Tavily request failed: {error}")
|
|
raise error
|
|
|
|
|
|
def formatSearchResults(response):
|
|
output = []
|
|
if response.get("answer"):
|
|
output.append(f"Answer: {response['answer']}")
|
|
output.append("Detailed Results:")
|
|
for result in response.get("results", []):
|
|
output.append(f"\nTitle: {result.get('title')}")
|
|
output.append(f"URL: {result.get('url')}")
|
|
output.append(f"Content: {result.get('content')}")
|
|
if result.get("raw_content"):
|
|
output.append(f"Raw Content: {result['raw_content']}")
|
|
return "\n".join(output)
|
|
|
|
|
|
def formatCrawlResults(response):
|
|
output = []
|
|
output.append("Crawl Results:")
|
|
output.append(f"Base URL: {response.get('base_url')}")
|
|
output.append("\nCrawled Pages:")
|
|
for index, page in enumerate(response.get("results", [])):
|
|
output.append(f"\n[{index + 1}] URL: {page.get('url')}")
|
|
if page.get("raw_content"):
|
|
raw = page["raw_content"]
|
|
contentPreview = (raw[:200] + "...") if len(raw) > 200 else raw
|
|
output.append(f"Content: {contentPreview}")
|
|
return "\n".join(output)
|
|
|
|
|
|
def formatMapResults(response):
|
|
output = []
|
|
output.append("Site Map Results:")
|
|
output.append(f"Base URL: {response.get('base_url')}")
|
|
output.append("\nMapped Pages:")
|
|
for index, page in enumerate(response.get("results", [])):
|
|
output.append(f"\n[{index + 1}] URL: {page}")
|
|
return "\n".join(output)
|
|
|
|
|
|
async def search(params):
|
|
response = await makeTavilyRequest(BASE_URLS["search"], params)
|
|
return formatSearchResults(response)
|
|
|
|
|
|
async def extract(params):
|
|
response = await makeTavilyRequest(BASE_URLS["extract"], params)
|
|
# Extract uses the same format as search
|
|
return formatSearchResults(response)
|
|
|
|
|
|
async def crawl(params):
|
|
response = await makeTavilyRequest(BASE_URLS["crawl"], params)
|
|
return formatCrawlResults(response)
|
|
|
|
|
|
async def map(params):
|
|
response = await makeTavilyRequest(BASE_URLS["map"], params)
|
|
return formatMapResults(response)
|
|
|
|
|
|
async def wrap(func, params, successMessage, failMessage):
|
|
try:
|
|
result = await func(params)
|
|
complete({"success": True, "message": successMessage, "data": result})
|
|
except Exception as error:
|
|
console.error(f"Function {func.__name__} failed! Error: {error}")
|
|
msg = str(getattr(error, "message", error))
|
|
complete({"success": False, "message": f"{failMessage}: {msg}", "error_stack": traceback.format_exc()})
|
|
|
|
|
|
async def search_wrapper(p):
|
|
await wrap(search, p, "搜索成功", "搜索失败")
|
|
|
|
|
|
async def extract_wrapper(p):
|
|
await wrap(extract, p, "提取成功", "提取失败")
|
|
|
|
|
|
async def crawl_wrapper(p):
|
|
await wrap(crawl, p, "爬取成功", "爬取失败")
|
|
|
|
|
|
async def map_wrapper(p):
|
|
await wrap(map, p, "映射成功", "映射失败")
|
|
|
|
|
|
exports.search = search_wrapper
|
|
exports.extract = extract_wrapper
|
|
exports.crawl = crawl_wrapper
|
|
exports.map = map_wrapper
|