41 lines
1.1 KiB
JavaScript
41 lines
1.1 KiB
JavaScript
import * as z from "zod"
|
||
import { tool } from "langchain"
|
||
|
||
/**
|
||
* 获取城市经纬度工具
|
||
* @param {string} city - 城市名称
|
||
* @returns {Object|null} - 返回经纬度对象 {lat, lng}，如果未找到则返回null
|
||
*/
|
||
export const getLatLngTool = tool(
|
||
async (input) => {
|
||
const { city } = input
|
||
try {
|
||
// 使用open-meteo地理编码API获取经纬度
|
||
console.log('获取城市经纬度:', city)
|
||
const url = `https://geocoding-api.open-meteo.com/v1/search?name=${encodeURIComponent(city)}&language=zh&count=1`
|
||
const res = await fetch(url)
|
||
const data = await res.json()
|
||
|
||
if (data.results?.length) {
|
||
return {
|
||
lat: data.results[0].latitude,
|
||
lng: data.results[0].longitude
|
||
}
|
||
}
|
||
|
||
// 如果没有找到结果，返回null
|
||
return null
|
||
} catch (error) {
|
||
console.error('地理编码API调用失败:', error)
|
||
return null
|
||
}
|
||
},
|
||
{
|
||
name: "get_lat_lng",
|
||
description: `根据城市名称获取经纬度信息`,
|
||
schema: z.object({
|
||
city: z.string().describe("城市名称，例如：北京市海淀区")
|
||
})
|
||
}
|
||
)
|