人工智能服务
这个 Python 脚本使用 API 密钥和命令行参数与 Jina AI 服务进行交互。它支持两个主要任务：从 URL 获取内容和执行搜索查询。脚本从环境变量中检索 Jina API 密钥，确保安全访问服务。它使用 requests 库发送 HTTP 请求，并使用 base64 解码搜索查询。然后，脚本打印来自 Jina AI 服务的响应。
import os
import requests
from dotenv import load_dotenv
import argparse
import base64
load_dotenv()
api_key = os.environ.get("JINA_API_KEY")
if not api_key:
raise ValueError("JINA_API_KEY 环境变量未设置。")
parser = argparse.ArgumentParser()
parser.add_argument("--job", type=str, choices=['url', 'search'], help="要执行的任务（url 或 search）", required=True)
parser.add_argument("--input", type=str, help="任务的输入", required=True)
args = parser.parse_args()
if args.job == 'url':
url = f'https://r.jina.ai/{args.input}'
headers = {'Authorization': f'Bearer {api_key}'}
print(f"URL: {url}")
print(f"Headers: {headers}")
response = requests.get(url, headers=headers)
print(response.text)
elif args.job == 'search':
question = base64.b64decode(args.input).decode('utf-8', errors='ignore')
url = f'https://s.jina.ai/{question}'
headers = {
'Authorization': f'Bearer {api_key}',
'X-Engine': 'direct',
'X-Retain-Images': 'none'
}
print(f"URL: {url}")
print(f"Headers: {headers}")
response = requests.get(url, headers=headers)
print(response.text)
else:
print("请指定 --job url 或 --job search")