pomljs/dist/util/schema.cjs.map

Prompt Orchestration Markup Language

{"version":3,"file":"schema.cjs","sources":["../../.build/util/schema.js"],"sourcesContent":["import { z } from 'zod';\nimport { jsonSchemaToZod } from 'json-schema-to-zod';\n/**\n * A unified schema representation that can work with both Zod and OpenAPI schemas.\n * Provides conversion methods between different schema formats.\n */\nexport class Schema {\n    zodSchema;\n    openApiSchema;\n    constructor(zodSchema, openApiSchema) {\n        this.zodSchema = zodSchema;\n        this.openApiSchema = openApiSchema;\n        if (!zodSchema && !openApiSchema) {\n            throw new Error(\"At least one schema must be provided\");\n        }\n        this.zodSchema = zodSchema;\n        this.openApiSchema = openApiSchema;\n    }\n    /**\n     * Creates a Schema instance from an OpenAPI schema object.\n     * @param openApiSchema The OpenAPI schema object\n     * @returns A new Schema instance\n     */\n    static fromOpenAPI(openApiSchema) {\n        const schema = new Schema(undefined, openApiSchema);\n        // TODO: openapi schema full validation should be added here\n        if (typeof openApiSchema !== 'object' || openApiSchema === null) {\n            throw new Error(\"Invalid OpenAPI schema provided\");\n        }\n        return schema;\n    }\n    /**\n     * Creates a Schema instance from a Zod schema.\n     * @param zodSchema The Zod schema\n     * @returns A new Schema instance\n     */\n    static fromZod(zodSchema) {\n        return new Schema(zodSchema);\n    }\n    /**\n     * Converts the schema to a Zod schema.\n     * @returns The Zod schema representation\n     * @throws Error if no schema is available\n     */\n    toZod() {\n        if (this.zodSchema) {\n            return this.zodSchema;\n        }\n        else if (this.openApiSchema) {\n            // It's not safe and even prohibited to use eval in browser environments.\n            // We need to make z available in the eval context\n            const zodSchemaString = jsonSchemaToZod(this.openApiSchema);\n            // Create a function that has z in scope and evaluate the schema string\n            const evalWithZ = new Function('z', `return ${zodSchemaString}`);\n            return evalWithZ(z);\n        }\n        else {\n            throw new Error(\"No Zod schema available\");\n        }\n    }\n    /**\n     * Converts the schema to an OpenAPI/JSON schema format.\n     * @returns The OpenAPI schema representation\n     * @throws Error if no schema is available\n     */\n    toOpenAPI() {\n        if (this.openApiSchema) {\n            return this.openApiSchema;\n        }\n        else if (this.zodSchema) {\n            const schema = z.toJSONSchema(this.zodSchema);\n            // pop the $schema property if it exists\n            if (schema.$schema) {\n                delete schema.$schema;\n            }\n            return schema;\n        }\n        else {\n            throw new Error(\"No schema available\");\n        }\n    }\n}\n/**\n * Manages a collection of tool schemas and provides conversion methods\n * for different AI provider formats (OpenAI, Vercel, etc.).\n */\nexport class ToolsSchema {\n    tools;\n    constructor() {\n        this.tools = new Map();\n    }\n    /**\n     * Adds a tool with a Zod schema to the collection.\n     * @param name The name of the tool\n     * @param description A description of what the tool does\n     * @param zodSchema The Zod schema for the tool's input parameters\n     */\n    addZodTool(name, description, zodSchema) {\n        const schema = Schema.fromZod(zodSchema);\n        if (this.tools.has(name)) {\n            throw new Error(`Tool with name \"${name}\" already exists`);\n        }\n        this.tools.set(name, {\n            name,\n            description,\n            inputSchema: schema\n        });\n    }\n    /**\n     * Adds a tool with an OpenAPI schema to the collection.\n     * @param name The name of the tool\n     * @param description A description of what the tool does\n     * @param openApiSchema The OpenAPI schema for the tool's input parameters\n     */\n    addOpenAPITool(name, description, openApiSchema) {\n        const schema = Schema.fromOpenAPI(openApiSchema);\n        if (this.tools.has(name)) {\n            throw new Error(`Tool with name \"${name}\" already exists`);\n        }\n        this.tools.set(name, {\n            name,\n            description,\n            inputSchema: schema\n        });\n    }\n    /**\n     * Add a tool with pre-parsed schema.\n     * @param name The name of the tool\n     * @param description A description of what the tool does\n     * @param schema The pre-parsed schema for the tool's input parameters\n     */\n    addTool(name, description, schema) {\n        if (this.tools.has(name)) {\n            throw new Error(`Tool with name \"${name}\" already exists`);\n        }\n        this.tools.set(name, {\n            name,\n            description,\n            inputSchema: schema\n        });\n    }\n    /**\n     * Converts the tools collection to Vercel AI SDK format.\n     * @returns An object mapping tool names to their Vercel AI SDK representations\n     */\n    toVercel() {\n        const vercelTools = {};\n        for (const [name, tool] of this.tools) {\n            vercelTools[name] = {\n                description: tool.description,\n                parameters: tool.inputSchema.toZod()\n            };\n        }\n        return vercelTools;\n    }\n    /**\n     * Converts the tools collection to OpenAI function calling format.\n     * @returns An array of OpenAI function definitions\n     * @example\n     * [\n     *   {\n     *     \"type\": \"function\",\n     *     \"name\": \"get_horoscope\",\n     *     \"description\": \"Get today's horoscope for an astrological sign.\",\n     *     \"parameters\": {\n     *       \"type\": \"object\",\n     *       \"properties\": {\n     *         \"sign\": {\n     *           \"type\": \"string\",\n     *           \"description\": \"An astrological sign like Taurus or Aquarius\"\n     *         }\n     *       },\n     *       \"required\": [\"sign\"]\n     *     }\n     *   }\n     * ]\n     */\n    toOpenAI() {\n        const openAITools = [];\n        // FIXME: Handle error gracefully here.\n        for (const [_, tool] of this.tools) {\n            openAITools.push({\n                type: 'function',\n                name: tool.name,\n                description: tool.description,\n                parameters: tool.inputSchema.toOpenAPI()\n            });\n        }\n        return openAITools;\n    }\n    /**\n     * Gets a tool by name.\n     * @param name The name of the tool to retrieve\n     * @returns The tool schema if found, undefined otherwise\n     */\n    getTool(name) {\n        return this.tools.get(name);\n    }\n    /**\n     * Gets all tools in the collection.\n     * @returns An array of all tool schemas\n     */\n    getTools() {\n        return Array.from(this.tools.values());\n    }\n    /**\n     * Removes a tool from the collection.\n     * @param name The name of the tool to remove\n     * @returns true if the tool was removed, false if it didn't exist\n     */\n    removeTool(name) {\n        return this.tools.delete(name);\n    }\n    /**\n     * Gets the number of tools in the collection.\n     * @returns The number of tools\n     */\n    size() {\n        return this.tools.size;\n    }\n    /**\n     * Clears all tools from the collection.\n     */\n    clear() {\n        this.tools.clear();\n    }\n}\n//# sourceMappingURL=schema.js.map"],"names":["jsonSchemaToZod","z"],"mappings":";;;;;AAEA;AACA;AACA;AACA;AACO,MAAM,MAAM,CAAC;AACpB,IAAI,SAAS;AACb,IAAI,aAAa;AACjB,IAAI,WAAW,CAAC,SAAS,EAAE,aAAa,EAAE;AAC1C,QAAQ,IAAI,CAAC,SAAS,GAAG,SAAS;AAClC,QAAQ,IAAI,CAAC,aAAa,GAAG,aAAa;AAC1C,QAAQ,IAAI,CAAC,SAAS,IAAI,CAAC,aAAa,EAAE;AAC1C,YAAY,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC;AACnE;AACA,QAAQ,IAAI,CAAC,SAAS,GAAG,SAAS;AAClC,QAAQ,IAAI,CAAC,aAAa,GAAG,aAAa;AAC1C;AACA;AACA;AACA;AACA;AACA;AACA,IAAI,OAAO,WAAW,CAAC,aAAa,EAAE;AACtC,QAAQ,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,SAAS,EAAE,aAAa,CAAC;AAC3D;AACA,QAAQ,IAAI,OAAO,aAAa,KAAK,QAAQ,IAAI,aAAa,KAAK,IAAI,EAAE;AACzE,YAAY,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC;AAC9D;AACA,QAAQ,OAAO,MAAM;AACrB;AACA;AACA;AACA;AACA;AACA;AACA,IAAI,OAAO,OAAO,CAAC,SAAS,EAAE;AAC9B,QAAQ,OAAO,IAAI,MAAM,CAAC,SAAS,CAAC;AACpC;AACA;AACA;AACA;AACA;AACA;AACA,IAAI,KAAK,GAAG;AACZ,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE;AAC5B,YAAY,OAAO,IAAI,CAAC,SAAS;AACjC;AACA,aAAa,IAAI,IAAI,CAAC,aAAa,EAAE;AACrC;AACA;AACA,YAAY,MAAM,eAAe,GAAGA,+BAAe,CAAC,IAAI,CAAC,aAAa,CAAC;AACvE;AACA,YAAY,MAAM,SAAS,GAAG,IAAI,QAAQ,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC,CAAC;AAC5E,YAAY,OAAO,SAAS,CAACC,KAAC,CAAC;AAC/B;AACA,aAAa;AACb,YAAY,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC;AACtD;AACA;AACA;AACA;AACA;AACA;AACA;AACA,IAAI,SAAS,GAAG;AAChB,QAAQ,IAAI,IAAI,CAAC,aAAa,EAAE;AAChC,YAAY,OAAO,IAAI,CAAC,aAAa;AACrC;AACA,aAAa,IAAI,IAAI,CAAC,SAAS,EAAE;AACjC,YAAY,MAAM,MAAM,GAAGA,KAAC,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC;AACzD;AACA,YAAY,IAAI,MAAM,CAAC,OAAO,EAAE;AAChC,gBAAgB,OAAO,MAAM,CAAC,OAAO;AACrC;AACA,YAAY,OAAO,MAAM;AACzB;AACA,aAAa;AACb,YAAY,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC;AAClD;AACA;AACA;AACA;AACA;AACA;AACA;AACO,MAAM,WAAW,CAAC;AACzB,IAAI,KAAK;AACT,IAAI,WAAW,GAAG;AAClB,QAAQ,IAAI,CAAC,KAAK,GAAG,IAAI,GAAG,EAAE;AAC9B;AACA;AACA;AACA;AACA;AACA;AACA;AACA,IAAI,UAAU,CAAC,IAAI,EAAE,WAAW,EAAE,SAAS,EAAE;AAC7C,QAAQ,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC;AAChD,QAAQ,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;AAClC,YAAY,MAAM,IAAI,KAAK,CAAC,CAAC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;AACtE;AACA,QAAQ,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE;AAC7B,YAAY,IAAI;AAChB,YAAY,WAAW;AACvB,YAAY,WAAW,EAAE;AACzB,SAAS,CAAC;AACV;AACA;AACA;AACA;AACA;AACA;AACA;AACA,IAAI,cAAc,CAAC,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE;AACrD,QAAQ,MAAM,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC;AACxD,QAAQ,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;AAClC,YAAY,MAAM,IAAI,KAAK,CAAC,CAAC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;AACtE;AACA,QAAQ,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE;AAC7B,YAAY,IAAI;AAChB,YAAY,WAAW;AACvB,YAAY,WAAW,EAAE;AACzB,SAAS,CAAC;AACV;AACA;AACA;AACA;AACA;AACA;AACA;AACA,IAAI,OAAO,CAAC,IAAI,EAAE,WAAW,EAAE,MAAM,EAAE;AACvC,QAAQ,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;AAClC,YAAY,MAAM,IAAI,KAAK,CAAC,CAAC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;AACtE;AACA,QAAQ,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE;AAC7B,YAAY,IAAI;AAChB,YAAY,WAAW;AACvB,YAAY,WAAW,EAAE;AACzB,SAAS,CAAC;AACV;AACA;AACA;AACA;AACA;AACA,IAAI,QAAQ,GAAG;AACf,QAAQ,MAAM,WAAW,GAAG,EAAE;AAC9B,QAAQ,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE;AAC/C,YAAY,WAAW,CAAC,IAAI,CAAC,GAAG;AAChC,gBAAgB,WAAW,EAAE,IAAI,CAAC,WAAW;AAC7C,gBAAgB,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,KAAK;AAClD,aAAa;AACb;AACA,QAAQ,OAAO,WAAW;AAC1B;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,IAAI,QAAQ,GAAG;AACf,QAAQ,MAAM,WAAW,GAAG,EAAE;AAC9B;AACA,QAAQ,KAAK,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE;AAC5C,YAAY,WAAW,CAAC,IAAI,CAAC;AAC7B,gBAAgB,IAAI,EAAE,UAAU;AAChC,gBAAgB,IAAI,EAAE,IAAI,CAAC,IAAI;AAC/B,gBAAgB,WAAW,EAAE,IAAI,CAAC,WAAW;AAC7C,gBAAgB,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,SAAS;AACtD,aAAa,CAAC;AACd;AACA,QAAQ,OAAO,WAAW;AAC1B;AACA;AACA;AACA;AACA;AACA;AACA,IAAI,OAAO,CAAC,IAAI,EAAE;AAClB,QAAQ,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC;AACnC;AACA;AACA;AACA;AACA;AACA,IAAI,QAAQ,GAAG;AACf,QAAQ,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;AAC9C;AACA;AACA;AACA;AACA;AACA;AACA,IAAI,UAAU,CAAC,IAAI,EAAE;AACrB,QAAQ,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC;AACtC;AACA;AACA;AACA;AACA;AACA,IAAI,IAAI,GAAG;AACX,QAAQ,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI;AAC9B;AACA;AACA;AACA;AACA,IAAI,KAAK,GAAG;AACZ,QAAQ,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE;AAC1B;AACA;;;;;"}