Fix build-schemas.ts to handle namespace exports

Co-authored-by: huangyiirene <7665279+huangyiirene@users.noreply.github.com>

Author: Copilot

packages/spec/scripts/build-schemas.ts

@@ -15,25 +15,30 @@ console.log(`Generating JSON Schemas to ${OUT_DIR}...`);
 
 let count = 0;
 
-// Iterate over all exports in the protocol
-for (const [key, value] of Object.entries(Protocol)) {
-  // Check if it looks like a Zod Schema
-  // Zod schemas usually have a parse method and safeParse method, or we can check instanceof
-  if (value instanceof z.ZodType) {
-    const schemaName = key.endsWith('Schema') ? key.replace('Schema', '') : key;
-    
-    // Convert to JSON Schema
-    const jsonSchema = zodToJsonSchema(value, {
-      name: schemaName,
-      $refStrategy: "none" // We want self-contained schemas for now, or use 'relative' if we handle refs
-    });
-
-    const fileName = `${schemaName}.json`;
-    const filePath = path.join(OUT_DIR, fileName);
-
-    fs.writeFileSync(filePath, JSON.stringify(jsonSchema, null, 2));
-    console.log(`✓ ${fileName}`);
-    count++;
+// Protocol now exports namespaces (Data, UI, System, AI, API)
+// We need to iterate through each namespace
+for (const [namespaceName, namespaceExports] of Object.entries(Protocol)) {
+  if (typeof namespaceExports === 'object' && namespaceExports !== null) {
+    // Iterate over all exports in each namespace
+    for (const [key, value] of Object.entries(namespaceExports)) {
+      // Check if it looks like a Zod Schema
+      if (value instanceof z.ZodType) {
+        const schemaName = key.endsWith('Schema') ? key.replace('Schema', '') : key;
+        
+        // Convert to JSON Schema
+        const jsonSchema = zodToJsonSchema(value, {
+          name: schemaName,
+          $refStrategy: "none" // We want self-contained schemas for now
+        });
+
+        const fileName = `${schemaName}.json`;
+        const filePath = path.join(OUT_DIR, fileName);
+
+        fs.writeFileSync(filePath, JSON.stringify(jsonSchema, null, 2));
+        console.log(`✓ ${fileName}`);
+        count++;
+      }
+    }
   }
 }