55 lines
1.9 KiB
TypeScript
55 lines
1.9 KiB
TypeScript
/**
|
||
* 由于 agentic 暂时不支持 AI SDK v5，因此使用自定义的 Serper 工具。
|
||
* @see https://docs.agentic.so/marketplace/ts-sdks/ai-sdk
|
||
*/
|
||
import { tool } from 'ai';
|
||
import { z } from 'zod';
|
||
|
||
const API_BASE_URL = 'https://google.serper.dev';
|
||
|
||
const searchParamsSchema = z.object({
|
||
q: z.string().describe('搜索查询词'),
|
||
autocorrect: z.boolean().optional().default(true).describe('是否自动纠正拼写错误'),
|
||
gl: z.string().optional().default('us').describe('地理位置代码，如"us"表示美国'),
|
||
hl: z.string().optional().default('en').describe('语言代码，如"en"表示英语'),
|
||
page: z.number().optional().default(1).describe('页码'),
|
||
num: z.number().optional().default(10).describe('结果数量'),
|
||
type: z
|
||
.enum(['search', 'images', 'videos', 'places', 'news', 'shopping'])
|
||
.optional()
|
||
.default('search')
|
||
.describe('搜索类型'),
|
||
});
|
||
|
||
export const serperTool = tool({
|
||
description: '使用Google搜索获取最新信息。适用于查找新闻、事实、数据和当前事件等实时信息。',
|
||
inputSchema: searchParamsSchema,
|
||
execute: async ({ q, ...params }) => {
|
||
const apiKey = process.env.SERPER_API_KEY || '';
|
||
if (!apiKey) {
|
||
throw new Error('Missing SERPER_API_KEY');
|
||
}
|
||
|
||
try {
|
||
const response = await fetch(`${API_BASE_URL}/search`, {
|
||
method: 'POST',
|
||
headers: {
|
||
'X-API-KEY': apiKey,
|
||
'Content-Type': 'application/json',
|
||
},
|
||
body: JSON.stringify({ q, ...params }),
|
||
});
|
||
|
||
if (!response.ok) {
|
||
throw new Error(`Serper API responded with status: ${response.status}`);
|
||
}
|
||
|
||
return await response.json();
|
||
} catch (error: unknown) {
|
||
console.error('Serper API error:', error);
|
||
const errorMessage = error instanceof Error ? error.message : '未知错误';
|
||
throw new Error(`搜索失败: ${errorMessage}`);
|
||
}
|
||
},
|
||
});
|