feat: 为行程规划 agent 添加 get_travel_time 工具,调用高德驾车 API 验证交通时间
agent 之前完全靠 LLM 猜测地理距离,现在可以查询真实驾车时间,超过 30 分钟会换更近地点。 同时修复 dev 端口未生效的问题(next dev 加 --port 3721)。
This commit is contained in:
@@ -195,6 +195,7 @@ export interface PlanGenResult {
|
||||
const AGENT_SYSTEM_PROMPT = `你是一个周末行程规划 Agent。你有以下工具可以使用:
|
||||
- list_ideas: 查看想法池中的所有活动
|
||||
- search_poi: 在地图上搜索地点(支持品牌名、地名、品类搜索)
|
||||
- get_travel_time: 查询两点间驾车时间和距离
|
||||
- finalize_plan: 提交最终行程方案
|
||||
|
||||
规划流程:
|
||||
@@ -202,8 +203,9 @@ const AGENT_SYSTEM_PROMPT = `你是一个周末行程规划 Agent。你有以下
|
||||
2. 根据时间、多样性、强度平衡选出合适的活动组合
|
||||
3. 为每个活动 search_poi 找到具体地点
|
||||
4. 如果搜索结果不理想(0结果或不相关),尝试换关键词重搜
|
||||
5. 综合地理位置、时间安排、活动特点,规划最优路线
|
||||
6. 确认无误后 finalize_plan 提交
|
||||
5. 用 get_travel_time 检查关键路段的交通时间,如果某段超过 30 分钟考虑换更近的地点
|
||||
6. 综合地理位置、时间安排、活动特点,规划最优路线
|
||||
7. 确认无误后 finalize_plan 提交
|
||||
|
||||
规划原则:
|
||||
- 地理位置相近,最小化移动距离
|
||||
@@ -297,6 +299,53 @@ function buildAgentTools(
|
||||
},
|
||||
};
|
||||
|
||||
const getTravelTimeTool: AgentTool = {
|
||||
name: "get_travel_time",
|
||||
description: "查询两点之间的驾车预估时间和距离。用于验证活动之间的交通是否合理",
|
||||
parameters: {
|
||||
type: "object",
|
||||
properties: {
|
||||
origin_lat: { type: "number", description: "起点纬度" },
|
||||
origin_lng: { type: "number", description: "起点经度" },
|
||||
dest_lat: { type: "number", description: "终点纬度" },
|
||||
dest_lng: { type: "number", description: "终点经度" },
|
||||
},
|
||||
required: ["origin_lat", "origin_lng", "dest_lat", "dest_lng"],
|
||||
},
|
||||
execute: async (args) => {
|
||||
const oLat = Number(args.origin_lat);
|
||||
const oLng = Number(args.origin_lng);
|
||||
const dLat = Number(args.dest_lat);
|
||||
const dLng = Number(args.dest_lng);
|
||||
const apiKey = requireAmapApiKey();
|
||||
const url = new URL("https://restapi.amap.com/v5/direction/driving");
|
||||
url.searchParams.set("key", apiKey);
|
||||
url.searchParams.set("origin", `${oLng},${oLat}`);
|
||||
url.searchParams.set("destination", `${dLng},${dLat}`);
|
||||
url.searchParams.set("show_fields", "cost");
|
||||
try {
|
||||
const res = await fetch(url.toString());
|
||||
const data = await res.json();
|
||||
if (data.status !== "1" || !data.route?.paths?.length) {
|
||||
return JSON.stringify({ error: "未找到路线" });
|
||||
}
|
||||
const path = data.route.paths[0];
|
||||
return JSON.stringify({
|
||||
distanceKm: Math.round(Number(path.distance) / 100) / 10,
|
||||
durationMin: Math.ceil(Number(path.duration) / 60),
|
||||
});
|
||||
} catch {
|
||||
return JSON.stringify({ error: "路线查询失败" });
|
||||
}
|
||||
},
|
||||
progressBefore: () => "正在查询交通时间...",
|
||||
progressAfter: (_args, result) => {
|
||||
const r = JSON.parse(result);
|
||||
if (r.error) return `路线查询失败`;
|
||||
return `驾车约 ${r.durationMin} 分钟(${r.distanceKm}km)`;
|
||||
},
|
||||
};
|
||||
|
||||
const finalizePlanTool: AgentTool = {
|
||||
name: "finalize_plan",
|
||||
description: "提交最终行程方案。days 数组中每个元素包含 date、items(活动列表)和 summary",
|
||||
@@ -341,7 +390,7 @@ function buildAgentTools(
|
||||
progressAfter: () => "行程规划完成!",
|
||||
};
|
||||
|
||||
return [listIdeasTool, searchPoiTool, finalizePlanTool];
|
||||
return [listIdeasTool, searchPoiTool, getTravelTimeTool, finalizePlanTool];
|
||||
}
|
||||
|
||||
async function runAgentPlanGeneration(
|
||||
|
||||
Reference in New Issue
Block a user