| import { Tool } from 'openai-function-calling-tools'; | |
| import { z } from 'zod'; | |
| import { getJson } from 'serpapi'; | |
| function createSportsResultsApi({ apiKey }: { apiKey: string }) { | |
| const paramsSchema = z.object({ | |
| input: z.string(), | |
| }); | |
| const name = 'sports_results'; | |
| const description = 'A custom search engine for sports scores and results. Useful for when you need to answer questions about current sports scores (like nba or nfl). Input should be a search query. Outputs a JSON array of results.'; | |
| const execute = async ({ input }: z.infer<typeof paramsSchema>) => { | |
| try { | |
| if (input.toLowerCase().includes('nba')) { | |
| const results = await fetch("https://data.nba.com/data/10s/v2015/json/mobile_teams/nba/2023/scores/00_todays_scores.json"); | |
| const games = await results.json(); | |
| return JSON.stringify(games.gs.g); | |
| } else { | |
| const response = await getJson({ | |
| engine: "google", | |
| api_key: apiKey, | |
| q: input, | |
| location: "Seattle, Washington", | |
| }); | |
| return JSON.stringify(response['sports_results']); | |
| } | |
| } catch (error) { | |
| throw new Error(`Error in serpApi or NBA API: ${error}`); | |
| } | |
| }; | |
| return new Tool(paramsSchema, name, description, execute).tool; | |
| } | |
| export { createSportsResultsApi }; | |