From 1d960a1b9a6092b9d3086ed9165c7a6dc5855c34 Mon Sep 17 00:00:00 2001
From: Mats Julius Funke <125814808+matsjfunke@users.noreply.github.com>
Date: Sun, 3 Aug 2025 10:34:19 +0200
Subject: [PATCH] fix: added explicit handling for float & int

---
 client/src/utils/schemaUtils.ts | 52 ++++++++++++++++++++++++++++++---
 1 file changed, 48 insertions(+), 4 deletions(-)

diff --git a/client/src/utils/schemaUtils.ts b/client/src/utils/schemaUtils.ts
index 62401bf8..7a175142 100644
--- a/client/src/utils/schemaUtils.ts
+++ b/client/src/utils/schemaUtils.ts
@@ -151,42 +151,86 @@ export function isPropertyRequired(
  * @returns A normalized schema or the original schema
  */
 export function normalizeUnionType(schema: JsonSchemaType): JsonSchemaType {
-  // Handle anyOf with string and null (FastMCP pattern)
+  // Handle anyOf with exactly string and null (FastMCP pattern)
   if (
     schema.anyOf &&
+    schema.anyOf.length === 2 &&
     schema.anyOf.some((t) => (t as JsonSchemaType).type === "string") &&
     schema.anyOf.some((t) => (t as JsonSchemaType).type === "null")
   ) {
     return { ...schema, type: "string", anyOf: undefined };
   }
 
-  // Handle anyOf with boolean and null (FastMCP pattern)
+  // Handle anyOf with exactly boolean and null (FastMCP pattern)
   if (
     schema.anyOf &&
+    schema.anyOf.length === 2 &&
     schema.anyOf.some((t) => (t as JsonSchemaType).type === "boolean") &&
     schema.anyOf.some((t) => (t as JsonSchemaType).type === "null")
   ) {
     return { ...schema, type: "boolean", anyOf: undefined };
   }
 
-  // Handle array type with string and null
+  // Handle anyOf with exactly number and null (FastMCP pattern)
+  if (
+    schema.anyOf &&
+    schema.anyOf.length === 2 &&
+    schema.anyOf.some((t) => (t as JsonSchemaType).type === "number") &&
+    schema.anyOf.some((t) => (t as JsonSchemaType).type === "null")
+  ) {
+    return { ...schema, type: "number", anyOf: undefined };
+  }
+
+  // Handle anyOf with exactly integer and null (FastMCP pattern)
+  if (
+    schema.anyOf &&
+    schema.anyOf.length === 2 &&
+    schema.anyOf.some((t) => (t as JsonSchemaType).type === "integer") &&
+    schema.anyOf.some((t) => (t as JsonSchemaType).type === "null")
+  ) {
+    return { ...schema, type: "integer", anyOf: undefined };
+  }
+
+  // Handle array type with exactly string and null
   if (
     Array.isArray(schema.type) &&
+    schema.type.length === 2 &&
     schema.type.includes("string") &&
     schema.type.includes("null")
   ) {
     return { ...schema, type: "string" };
   }
 
-  // Handle array type with boolean and null
+  // Handle array type with exactly boolean and null
   if (
     Array.isArray(schema.type) &&
+    schema.type.length === 2 &&
     schema.type.includes("boolean") &&
     schema.type.includes("null")
   ) {
     return { ...schema, type: "boolean" };
   }
 
+  // Handle array type with exactly number and null
+  if (
+    Array.isArray(schema.type) &&
+    schema.type.length === 2 &&
+    schema.type.includes("number") &&
+    schema.type.includes("null")
+  ) {
+    return { ...schema, type: "number" };
+  }
+
+  // Handle array type with exactly integer and null
+  if (
+    Array.isArray(schema.type) &&
+    schema.type.length === 2 &&
+    schema.type.includes("integer") &&
+    schema.type.includes("null")
+  ) {
+    return { ...schema, type: "integer" };
+  }
+
   return schema;
 }