updates

app/api/chat/route.ts

@@ -3,9 +3,11 @@ import { OpenAIStream, StreamingTextResponse } from "ai";
 import { createUrlSurfer } from "@/app/tools/surfer";
 import { createSearchApi } from "@/app/tools/search";
 import { createOddsApi } from "@/app/tools/odds";
+import { createSportsResultsApi } from "@/app/tools/scores";
 
 const [, urlSurferSchema] = createUrlSurfer();
 const [, serpApiSchema] = createSearchApi({ apiKey: process.env.SERP_API_KEY || '' });
+const [, sportsApiResultsSchema] = createSportsResultsApi({ apiKey: process.env.SERP_API_KEY || '' });
 const [, oddsApiSchema] = createOddsApi({ apiKey: process.env.ODDS_API_KEY || '' });
 
 const config = new Configuration({
@@ -16,7 +18,8 @@ const openai = new OpenAIApi(config);
 const functions: any[] = [
   urlSurferSchema,
   serpApiSchema,
-  oddsApiSchema
+  oddsApiSchema,
+  sportsApiResultsSchema
 ];
 
 export async function POST(req: Request) {

app/tools/odds.ts

@@ -5,7 +5,7 @@ function createOddsApi({ apiKey }: { apiKey: string }) {
   const paramsSchema = z.object({
     input: z.string(),
   });
-  const name = 'oddsApi';
+  const name = 'sports odds';
   const description = 'A realtime Sports Odds API. Useful for when you need to answer questions about sports odds, currently NBA and NFL. Input should be a string including a sport (nba or nfl), optionally including type (h2h, spread or o/u). Outputs a JSON array of results.';
 
   const execute = async ({ input }: z.infer<typeof paramsSchema>) => {
@@ -26,12 +26,12 @@ function createOddsApi({ apiKey }: { apiKey: string }) {
             sportKey = 'upcoming';
         }
 
-        if (input.includes('spread')) {
+        if (input.includes('money line')) {
             market = 'spread';
         } else if (input.includes('o/u')) {
             market = 'totals';
         } else {
-            market = 'h2h';
+            market = 'spread';
         }
         const activeSports = await fetch(`https://api.the-odds-api.com/v4/sports/${sportKey}/odds?apiKey=${apiKey}&oddsFormat=${oddsFormat}&dateFormat=${dateFormat}&market=${market}&regions=${regions}`);
         const oddsResponse = await activeSports.json();

app/tools/scores.ts

@@ -0,0 +1,30 @@
+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 {
+        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: ${error}`);
+    }
+  };
+
+  return new Tool(paramsSchema, name, description, execute).tool;
+}
+
+export { createSportsResultsApi };

app/tools/search.ts

@@ -6,7 +6,7 @@ function createSearchApi({ apiKey }: { apiKey: string }) {
   const paramsSchema = z.object({
     input: z.string(),
   });
-  const name = 'searchApi';
+  const name = 'search';
   const description = 'A custom search engine. Useful for when you need to answer questions about current events. Input should be a search query. Outputs a JSON array of results.';
 
   const execute = async ({ input }: z.infer<typeof paramsSchema>) => {