matt HOFFNER
nba api works
3db52d6
raw
history blame
No virus
1.39 kB
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 };