feat(tools): add type coercion to the schema

Signed-off-by: Tomas Dvorak <toomas2d@gmail.com>
i-am-bee · Oct 7, 2024 · f0995ea · f0995ea
1 parent 9b60f8a
commit f0995ea
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/tools/search/duckDuckGoSearch.ts b/src/tools/search/duckDuckGoSearch.ts
@@ -80,7 +80,7 @@ export class DuckDuckGoSearchTool extends Tool<
   inputSchema() {
     return z.object({
       query: z.string({ description: `Search query` }).min(1).max(128),
-      page: z
+      page: z.coerce
         .number()
         .int()
         .min(1)

diff --git a/src/tools/weather/openMeteo.ts b/src/tools/weather/openMeteo.ts
@@ -79,8 +79,8 @@ export class OpenMeteoTool extends Tool<
             .strip(),
           z
             .object({
-              latitude: z.number(),
-              longitude: z.number(),
+              latitude: z.coerce.number(),
+              longitude: z.coerce.number(),
             })
             .strip(),
         ]),
@@ -91,7 +91,7 @@ export class OpenMeteoTool extends Tool<
           .union([z.string().date(), z.string().datetime()])
           .describe("End date for the weather forecast in the format YYYY-MM-DD (UTC)")
           .optional(),
-        elevation: z.number().nullish(),
+        elevation: z.coerce.number().nullish(),
         temperature_unit: z.enum(["celsius", "fahrenheit"]).default("celsius"),
       })
       .strip();