@prefactor/core 0.3.1 → 0.3.2

package/dist/tool-schema.js

@@ -0,0 +1,152 @@
+import { getLogger } from './utils/logging.js';
+const logger = getLogger('tool-schema');
+export function normalizeAgentToolSchemas(agentSchema, { defaultAgentSchema, providerName, }) {
+    const toolSchemas = extractToolSchemas(agentSchema, providerName);
+    return {
+        agentSchema: mergeWithDefaultAgentSchema(stripToolSchemas(agentSchema), defaultAgentSchema),
+        toolSchemas,
+        toolSpanTypes: buildToolSpanTypes(toolSchemas),
+    };
+}
+export function resolveMappedSpanType(toolName, toolSpanTypes, defaultSpanType) {
+    return toolSpanTypes?.[toolName] ?? defaultSpanType;
+}
+function extractToolSchemas(agentSchema, providerName) {
+    const rawToolSchemas = getRawToolSchemas(agentSchema);
+    if (!rawToolSchemas) {
+        return undefined;
+    }
+    const toolSchemas = {};
+    const toolBySpanType = new Map();
+    for (const [toolName, rawConfig] of Object.entries(rawToolSchemas)) {
+        const parsedToolSchema = parseToolSchemaConfig(toolName, rawConfig, providerName, toolBySpanType);
+        if (!parsedToolSchema) {
+            continue;
+        }
+        toolSchemas[toolName] = parsedToolSchema;
+    }
+    return Object.keys(toolSchemas).length > 0 ? toolSchemas : undefined;
+}
+function getRawToolSchemas(agentSchema) {
+    if (!agentSchema || typeof agentSchema !== 'object' || Array.isArray(agentSchema)) {
+        return undefined;
+    }
+    const rawToolSchemas = agentSchema.toolSchemas;
+    if (rawToolSchemas === undefined) {
+        return undefined;
+    }
+    if (!rawToolSchemas || typeof rawToolSchemas !== 'object' || Array.isArray(rawToolSchemas)) {
+        logger.warn('Ignoring invalid agentSchema.toolSchemas: expected an object keyed by tool name.');
+        return undefined;
+    }
+    return rawToolSchemas;
+}
+function parseToolSchemaConfig(toolName, rawConfig, providerName, toolBySpanType) {
+    if (!rawConfig || typeof rawConfig !== 'object' || Array.isArray(rawConfig)) {
+        logger.warn(`Invalid agentSchema.toolSchemas.${toolName}: expected an object with spanType and inputSchema.`);
+        return undefined;
+    }
+    const config = rawConfig;
+    if (typeof config.spanType !== 'string') {
+        logger.warn(`Invalid agentSchema.toolSchemas.${toolName}.spanType: expected a non-empty string.`);
+        return undefined;
+    }
+    const inputSchema = assertValidInputSchema(toolName, config.inputSchema);
+    if (!inputSchema) {
+        return undefined;
+    }
+    const normalizedSpanType = normalizeUniqueToolSpanType(toolName, config.spanType, providerName, toolBySpanType);
+    if (!normalizedSpanType) {
+        return undefined;
+    }
+    return {
+        spanType: normalizedSpanType,
+        inputSchema,
+    };
+}
+function assertValidInputSchema(toolName, inputSchema) {
+    if (!inputSchema || typeof inputSchema !== 'object' || Array.isArray(inputSchema)) {
+        logger.warn(`Invalid agentSchema.toolSchemas.${toolName}.inputSchema: expected an object.`);
+        return undefined;
+    }
+    return inputSchema;
+}
+function normalizeUniqueToolSpanType(toolName, spanType, providerName, toolBySpanType) {
+    const normalizedSpanType = normalizeToolSpanType(spanType, toolName, providerName);
+    if (!normalizedSpanType) {
+        return undefined;
+    }
+    const conflictingTool = toolBySpanType.get(normalizedSpanType);
+    if (conflictingTool && conflictingTool !== toolName) {
+        logger.warn(`Invalid agentSchema.toolSchemas.${toolName}.spanType: normalized span type "${normalizedSpanType}" conflicts with "${conflictingTool}".`);
+        return undefined;
+    }
+    toolBySpanType.set(normalizedSpanType, toolName);
+    return normalizedSpanType;
+}
+function buildToolSpanTypes(toolSchemas) {
+    if (!toolSchemas) {
+        return undefined;
+    }
+    return Object.fromEntries(Object.entries(toolSchemas).map(([toolName, config]) => [toolName, config.spanType]));
+}
+function cloneRecord(value) {
+    if (!value || typeof value !== 'object' || Array.isArray(value)) {
+        return {};
+    }
+    return { ...value };
+}
+function stripToolSchemas(baseSchema) {
+    if (!baseSchema || typeof baseSchema !== 'object' || Array.isArray(baseSchema)) {
+        return baseSchema;
+    }
+    const { toolSchemas: _, ...rest } = baseSchema;
+    return rest;
+}
+function mergeWithDefaultAgentSchema(baseSchema, defaultAgentSchema) {
+    if (!baseSchema) {
+        return defaultAgentSchema;
+    }
+    return {
+        ...defaultAgentSchema,
+        ...baseSchema,
+        span_schemas: {
+            ...cloneRecord(defaultAgentSchema.span_schemas),
+            ...cloneRecord(baseSchema.span_schemas),
+        },
+        span_result_schemas: {
+            ...cloneRecord(defaultAgentSchema.span_result_schemas),
+            ...cloneRecord(baseSchema.span_result_schemas),
+        },
+    };
+}
+function normalizeToolSpanType(spanType, toolName, providerName) {
+    const trimmedSpanType = spanType.trim();
+    if (trimmedSpanType.length === 0) {
+        logger.warn(`Invalid agentSchema.toolSchemas.${toolName}.spanType: expected a non-empty string.`);
+        return undefined;
+    }
+    const providerToolPrefix = `${providerName}:tool:`;
+    if (trimmedSpanType.startsWith(providerToolPrefix)) {
+        const suffix = trimmedSpanType.slice(providerToolPrefix.length).replace(/^:+/, '');
+        if (suffix.length === 0) {
+            logger.warn(`Invalid agentSchema.toolSchemas.${toolName}.spanType: expected a non-empty suffix after normalization.`);
+            return undefined;
+        }
+        return `${providerName}:tool:${suffix}`;
+    }
+    let suffix = trimmedSpanType;
+    if (suffix.startsWith(`${providerName}:`)) {
+        suffix = suffix.slice(`${providerName}:`.length);
+    }
+    if (suffix.startsWith('tool:')) {
+        suffix = suffix.slice('tool:'.length);
+    }
+    suffix = suffix.replace(/^:+/, '');
+    if (suffix.length === 0) {
+        logger.warn(`Invalid agentSchema.toolSchemas.${toolName}.spanType: expected a non-empty suffix after normalization.`);
+        return undefined;
+    }
+    return `${providerName}:tool:${suffix}`;
+}
+//# sourceMappingURL=tool-schema.js.map

package/dist/tool-schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-schema.js","sourceRoot":"","sources":["../src/tool-schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAe/C,MAAM,MAAM,GAAG,SAAS,CAAC,aAAa,CAAC,CAAC;AAExC,MAAM,UAAU,yBAAyB,CACvC,WAAgD,EAChD,EACE,kBAAkB,EAClB,YAAY,GAIb;IAED,MAAM,WAAW,GAAG,kBAAkB,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;IAClE,OAAO;QACL,WAAW,EAAE,2BAA2B,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,kBAAkB,CAAC;QAC3F,WAAW;QACX,aAAa,EAAE,kBAAkB,CAAC,WAAW,CAAC;KAC/C,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,QAAgB,EAChB,aAAiD,EACjD,eAAuB;IAEvB,OAAO,aAAa,EAAE,CAAC,QAAQ,CAAC,IAAI,eAAe,CAAC;AACtD,CAAC;AAED,SAAS,kBAAkB,CACzB,WAAgD,EAChD,YAAoB;IAEpB,MAAM,cAAc,GAAG,iBAAiB,CAAC,WAAW,CAAC,CAAC;IACtD,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,WAAW,GAAqC,EAAE,CAAC;IACzD,MAAM,cAAc,GAAG,IAAI,GAAG,EAAkB,CAAC;IACjD,KAAK,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;QACnE,MAAM,gBAAgB,GAAG,qBAAqB,CAC5C,QAAQ,EACR,SAAS,EACT,YAAY,EACZ,cAAc,CACf,CAAC;QACF,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACtB,SAAS;QACX,CAAC;QAED,WAAW,CAAC,QAAQ,CAAC,GAAG,gBAAgB,CAAC;IAC3C,CAAC;IAED,OAAO,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC;AACvE,CAAC;AAED,SAAS,iBAAiB,CACxB,WAAgD;IAEhD,IAAI,CAAC,WAAW,IAAI,OAAO,WAAW,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;QAClF,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,cAAc,GAAI,WAAyC,CAAC,WAAW,CAAC;IAC9E,IAAI,cAAc,KAAK,SAAS,EAAE,CAAC;QACjC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,CAAC,cAAc,IAAI,OAAO,cAAc,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;QAC3F,MAAM,CAAC,IAAI,CAAC,kFAAkF,CAAC,CAAC;QAChG,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,cAAyC,CAAC;AACnD,CAAC;AAED,SAAS,qBAAqB,CAC5B,QAAgB,EAChB,SAAkB,EAClB,YAAoB,EACpB,cAAmC;IAEnC,IAAI,CAAC,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;QAC5E,MAAM,CAAC,IAAI,CACT,mCAAmC,QAAQ,qDAAqD,CACjG,CAAC;QACF,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,MAAM,GAAG,SAGd,CAAC;IAEF,IAAI,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACxC,MAAM,CAAC,IAAI,CACT,mCAAmC,QAAQ,yCAAyC,CACrF,CAAC;QACF,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,WAAW,GAAG,sBAAsB,CAAC,QAAQ,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;IACzE,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,kBAAkB,GAAG,2BAA2B,CACpD,QAAQ,EACR,MAAM,CAAC,QAAQ,EACf,YAAY,EACZ,cAAc,CACf,CAAC;IACF,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO;QACL,QAAQ,EAAE,kBAAkB;QAC5B,WAAW;KACZ,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB,CAAC,QAAgB,EAAE,WAAoB;IACpE,IAAI,CAAC,WAAW,IAAI,OAAO,WAAW,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;QAClF,MAAM,CAAC,IAAI,CAAC,mCAAmC,QAAQ,mCAAmC,CAAC,CAAC;QAC5F,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,WAAyB,CAAC;AACnC,CAAC;AAED,SAAS,2BAA2B,CAClC,QAAgB,EAChB,QAAgB,EAChB,YAAoB,EACpB,cAAmC;IAEnC,MAAM,kBAAkB,GAAG,qBAAqB,CAAC,QAAQ,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;IACnF,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,eAAe,GAAG,cAAc,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;IAC/D,IAAI,eAAe,IAAI,eAAe,KAAK,QAAQ,EAAE,CAAC;QACpD,MAAM,CAAC,IAAI,CACT,mCAAmC,QAAQ,oCAAoC,kBAAkB,qBAAqB,eAAe,IAAI,CAC1I,CAAC;QACF,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,cAAc,CAAC,GAAG,CAAC,kBAAkB,EAAE,QAAQ,CAAC,CAAC;IACjD,OAAO,kBAAkB,CAAC;AAC5B,CAAC;AAED,SAAS,kBAAkB,CACzB,WAAyD;IAEzD,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CACrF,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,KAAc;IACjC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAChE,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,EAAE,GAAI,KAAiC,EAAE,CAAC;AACnD,CAAC;AAED,SAAS,gBAAgB,CACvB,UAA+C;IAE/C,IAAI,CAAC,UAAU,IAAI,OAAO,UAAU,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QAC/E,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,MAAM,EAAE,WAAW,EAAE,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,UAEnC,CAAC;IAEF,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,2BAA2B,CAClC,UAA+C,EAC/C,kBAA2C;IAE3C,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,kBAAkB,CAAC;IAC5B,CAAC;IAED,OAAO;QACL,GAAG,kBAAkB;QACrB,GAAG,UAAU;QACb,YAAY,EAAE;YACZ,GAAG,WAAW,CAAC,kBAAkB,CAAC,YAAY,CAAC;YAC/C,GAAG,WAAW,CAAC,UAAU,CAAC,YAAY,CAAC;SACxC;QACD,mBAAmB,EAAE;YACnB,GAAG,WAAW,CAAC,kBAAkB,CAAC,mBAAmB,CAAC;YACtD,GAAG,WAAW,CAAC,UAAU,CAAC,mBAAmB,CAAC;SAC/C;KACF,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAC5B,QAAgB,EAChB,QAAgB,EAChB,YAAoB;IAEpB,MAAM,eAAe,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;IACxC,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACjC,MAAM,CAAC,IAAI,CACT,mCAAmC,QAAQ,yCAAyC,CACrF,CAAC;QACF,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,kBAAkB,GAAG,GAAG,YAAY,QAAQ,CAAC;IACnD,IAAI,eAAe,CAAC,UAAU,CAAC,kBAAkB,CAAC,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,eAAe,CAAC,KAAK,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACnF,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,CAAC,IAAI,CACT,mCAAmC,QAAQ,6DAA6D,CACzG,CAAC;YACF,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,GAAG,YAAY,SAAS,MAAM,EAAE,CAAC;IAC1C,CAAC;IAED,IAAI,MAAM,GAAG,eAAe,CAAC;IAC7B,IAAI,MAAM,CAAC,UAAU,CAAC,GAAG,YAAY,GAAG,CAAC,EAAE,CAAC;QAC1C,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,YAAY,GAAG,CAAC,MAAM,CAAC,CAAC;IACnD,CAAC;IACD,IAAI,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QAC/B,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACxC,CAAC;IAED,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACnC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,MAAM,CAAC,IAAI,CACT,mCAAmC,QAAQ,6DAA6D,CACzG,CAAC;QACF,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,GAAG,YAAY,SAAS,MAAM,EAAE,CAAC;AAC1C,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@prefactor/core",
-  "version": "0.3.1",
+  "version": "0.3.2",
   "description": "Framework-agnostic observability primitives for Prefactor",
   "type": "module",
   "main": "dist/index.cjs",