@output.ai/core 0.0.14 → 0.0.15

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@output.ai/core",
-  "version": "0.0.14",
+  "version": "0.0.15",
   "description": "The core module of the output framework",
   "type": "module",
   "main": "src/index.js",

package/src/worker/catalog_workflow/index.js

@@ -1,6 +1,26 @@
+import { z } from 'zod';
 import { METADATA_ACCESS_SYMBOL } from '#consts';
 import { Catalog, CatalogActivity, CatalogWorkflow } from './catalog.js';
 
+/**
+ * Converts a Zod schema to JSON Schema format.
+ *
+ * @param {any} schema - A zod schema
+ * @returns {object|null} JSON Schema object, or null if schema is invalid
+ */
+const convertToJsonSchema = schema => {
+  if ( !schema ) {
+    return null;
+  }
+
+  try {
+    return z.toJSONSchema( schema );
+  } catch ( error ) {
+    console.warn( 'Invalid schema provided (expected Zod schema):', schema, error );
+    return null;
+  }
+};
+
 /**
  * Converts the list of workflows and the activities into the catalog information.
  *
@@ -14,8 +34,17 @@ export const createCatalog = ( { workflows, activities } ) =>
   workflows.reduce( ( catalog, workflow ) =>
     catalog.addWorkflow( new CatalogWorkflow( {
       ...workflow,
+      inputSchema: convertToJsonSchema( workflow.inputSchema ),
+      outputSchema: convertToJsonSchema( workflow.outputSchema ),
       activities: Object.entries( activities )
         .filter( ( [ k ] ) => k.startsWith( `${workflow.path}#` ) )
-        .map( ( [ _, v ] ) => new CatalogActivity( v[METADATA_ACCESS_SYMBOL] ) )
+        .map( ( [ _, v ] ) => {
+          const metadata = v[METADATA_ACCESS_SYMBOL];
+          return new CatalogActivity( {
+            ...metadata,
+            inputSchema: convertToJsonSchema( metadata.inputSchema ),
+            outputSchema: convertToJsonSchema( metadata.outputSchema )
+          } );
+        } )
     } ) )
   , new Catalog() );

package/src/worker/catalog_workflow/index.spec.js

@@ -1,4 +1,5 @@
 import { describe, it, expect, vi } from 'vitest';
+import { z } from 'zod';
 
 // Provide the same symbol to the module under test and to the test
 const METADATA_ACCESS_SYMBOL = Symbol( '__metadata' );
@@ -19,16 +20,16 @@ describe( 'createCatalog', () => {
         path: '/flows/flow1',
         pathname: '/flows/flow1/workflow.js',
         description: 'desc-flow1',
-        inputSchema: { in: 'f1' },
-        outputSchema: { out: 'f1' }
+        inputSchema: z.object( { in: z.literal( 'f1' ) } ),
+        outputSchema: z.object( { out: z.literal( 'f1' ) } )
       },
       {
         name: 'flow2',
         path: '/flows/flow2',
         pathname: '/flows/flow2/workflow.js',
         description: 'desc-flow2',
-        inputSchema: { in: 'f2' },
-        outputSchema: { out: 'f2' }
+        inputSchema: z.object( { in: z.literal( 'f2' ) } ),
+        outputSchema: z.object( { out: z.literal( 'f2' ) } )
       }
     ];
 
@@ -37,8 +38,8 @@ describe( 'createCatalog', () => {
       name: 'A1',
       path: '/flows/flow1#A1',
       description: 'desc-a1',
-      inputSchema: { in: 'a1' },
-      outputSchema: { out: 'a1' }
+      inputSchema: z.object( { in: z.literal( 'a1' ) } ),
+      outputSchema: z.object( { out: z.literal( 'a1' ) } )
     } );
 
     const activity2 = () => {};
@@ -46,8 +47,8 @@ describe( 'createCatalog', () => {
       name: 'A2',
       path: '/flows/flow1#A2',
       description: 'desc-a2',
-      inputSchema: { in: 'a2' },
-      outputSchema: { out: 'a2' }
+      inputSchema: z.object( { in: z.literal( 'a2' ) } ),
+      outputSchema: z.object( { out: z.literal( 'a2' ) } )
     } );
 
     const activity3 = () => {};
@@ -55,8 +56,8 @@ describe( 'createCatalog', () => {
       name: 'B1',
       path: '/flows/flow2#B1',
       description: 'desc-b1',
-      inputSchema: { in: 'b1' },
-      outputSchema: { out: 'b1' }
+      inputSchema: z.object( { in: z.literal( 'b1' ) } ),
+      outputSchema: z.object( { out: z.literal( 'b1' ) } )
     } );
 
     const activity4 = () => {};
@@ -64,8 +65,8 @@ describe( 'createCatalog', () => {
       name: 'X',
       path: '/other#X',
       description: 'desc-x',
-      inputSchema: { in: 'x' },
-      outputSchema: { out: 'x' }
+      inputSchema: z.object( { in: z.literal( 'x' ) } ),
+      outputSchema: z.object( { out: z.literal( 'x' ) } )
     } );
 
     const activities = {
@@ -96,20 +97,56 @@ describe( 'createCatalog', () => {
         name: 'flow1',
         path: '/flows/flow1',
         description: 'desc-flow1',
-        inputSchema: { in: 'f1' },
-        outputSchema: { out: 'f1' },
+        inputSchema: {
+          $schema: 'https://json-schema.org/draft/2020-12/schema',
+          type: 'object',
+          properties: { in: { type: 'string', const: 'f1' } },
+          required: [ 'in' ],
+          additionalProperties: false
+        },
+        outputSchema: {
+          $schema: 'https://json-schema.org/draft/2020-12/schema',
+          type: 'object',
+          properties: { out: { type: 'string', const: 'f1' } },
+          required: [ 'out' ],
+          additionalProperties: false
+        },
         activities: [
           {
             name: 'A1',
             description: 'desc-a1',
-            inputSchema: { in: 'a1' },
-            outputSchema: { out: 'a1' }
+            inputSchema: {
+              $schema: 'https://json-schema.org/draft/2020-12/schema',
+              type: 'object',
+              properties: { in: { type: 'string', const: 'a1' } },
+              required: [ 'in' ],
+              additionalProperties: false
+            },
+            outputSchema: {
+              $schema: 'https://json-schema.org/draft/2020-12/schema',
+              type: 'object',
+              properties: { out: { type: 'string', const: 'a1' } },
+              required: [ 'out' ],
+              additionalProperties: false
+            }
           },
           {
             name: 'A2',
             description: 'desc-a2',
-            inputSchema: { in: 'a2' },
-            outputSchema: { out: 'a2' }
+            inputSchema: {
+              $schema: 'https://json-schema.org/draft/2020-12/schema',
+              type: 'object',
+              properties: { in: { type: 'string', const: 'a2' } },
+              required: [ 'in' ],
+              additionalProperties: false
+            },
+            outputSchema: {
+              $schema: 'https://json-schema.org/draft/2020-12/schema',
+              type: 'object',
+              properties: { out: { type: 'string', const: 'a2' } },
+              required: [ 'out' ],
+              additionalProperties: false
+            }
           }
         ]
       },
@@ -117,14 +154,38 @@ describe( 'createCatalog', () => {
         name: 'flow2',
         path: '/flows/flow2',
         description: 'desc-flow2',
-        inputSchema: { in: 'f2' },
-        outputSchema: { out: 'f2' },
+        inputSchema: {
+          $schema: 'https://json-schema.org/draft/2020-12/schema',
+          type: 'object',
+          properties: { in: { type: 'string', const: 'f2' } },
+          required: [ 'in' ],
+          additionalProperties: false
+        },
+        outputSchema: {
+          $schema: 'https://json-schema.org/draft/2020-12/schema',
+          type: 'object',
+          properties: { out: { type: 'string', const: 'f2' } },
+          required: [ 'out' ],
+          additionalProperties: false
+        },
         activities: [
           {
             name: 'B1',
             description: 'desc-b1',
-            inputSchema: { in: 'b1' },
-            outputSchema: { out: 'b1' }
+            inputSchema: {
+              $schema: 'https://json-schema.org/draft/2020-12/schema',
+              type: 'object',
+              properties: { in: { type: 'string', const: 'b1' } },
+              required: [ 'in' ],
+              additionalProperties: false
+            },
+            outputSchema: {
+              $schema: 'https://json-schema.org/draft/2020-12/schema',
+              type: 'object',
+              properties: { out: { type: 'string', const: 'b1' } },
+              required: [ 'out' ],
+              additionalProperties: false
+            }
           }
         ]
       }