@zhivex-ai/core 0.5.0-next.0 → 0.6.0

package/dist/mcp.js

@@ -0,0 +1,167 @@
+import { z } from "zod";
+import { serializeJsonValue, tool } from "./messages.js";
+import { createToolRegistry } from "./tool-registry.js";
+const isRecord = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
+const toZodLiteral = (value) => {
+    if (value === null || typeof value === "string" || typeof value === "number" || typeof value === "boolean") {
+        return z.literal(value);
+    }
+    return z.unknown();
+};
+const toZodUnion = (schemas) => {
+    if (!schemas?.length) {
+        return z.unknown();
+    }
+    const parsed = schemas.map((schema) => jsonSchemaToZod(schema));
+    if (parsed.length === 1) {
+        return parsed[0];
+    }
+    return z.union(parsed);
+};
+const applyCommonConstraints = (schema, jsonSchema) => {
+    if (schema instanceof z.ZodString) {
+        let next = schema;
+        if (typeof jsonSchema.minLength === "number") {
+            next = next.min(jsonSchema.minLength);
+        }
+        if (typeof jsonSchema.maxLength === "number") {
+            next = next.max(jsonSchema.maxLength);
+        }
+        return next.describe(typeof jsonSchema.description === "string" ? jsonSchema.description : "");
+    }
+    if (schema instanceof z.ZodNumber) {
+        let next = schema;
+        if (typeof jsonSchema.minimum === "number") {
+            next = next.gte(jsonSchema.minimum);
+        }
+        if (typeof jsonSchema.maximum === "number") {
+            next = next.lte(jsonSchema.maximum);
+        }
+        return next.describe(typeof jsonSchema.description === "string" ? jsonSchema.description : "");
+    }
+    if (schema instanceof z.ZodArray) {
+        let next = schema;
+        if (typeof jsonSchema.minItems === "number") {
+            next = next.min(jsonSchema.minItems);
+        }
+        if (typeof jsonSchema.maxItems === "number") {
+            next = next.max(jsonSchema.maxItems);
+        }
+        return next.describe(typeof jsonSchema.description === "string" ? jsonSchema.description : "");
+    }
+    return typeof jsonSchema.description === "string" ? schema.describe(jsonSchema.description) : schema;
+};
+const jsonSchemaToZod = (schema) => {
+    if (!schema) {
+        return z.unknown();
+    }
+    if (!isRecord(schema)) {
+        return z.unknown();
+    }
+    const jsonSchema = schema;
+    if (jsonSchema.const !== undefined) {
+        return toZodLiteral(jsonSchema.const);
+    }
+    if (Array.isArray(jsonSchema.enum) && jsonSchema.enum.length > 0) {
+        return toZodUnion(jsonSchema.enum);
+    }
+    if (Array.isArray(jsonSchema.oneOf)) {
+        return toZodUnion(jsonSchema.oneOf);
+    }
+    if (Array.isArray(jsonSchema.anyOf)) {
+        return toZodUnion(jsonSchema.anyOf);
+    }
+    if (Array.isArray(jsonSchema.type)) {
+        return toZodUnion(jsonSchema.type.map((type) => ({
+            ...jsonSchema,
+            type
+        })));
+    }
+    switch (jsonSchema.type) {
+        case "string":
+            return applyCommonConstraints(z.string(), jsonSchema);
+        case "number":
+            return applyCommonConstraints(z.number(), jsonSchema);
+        case "integer":
+            return applyCommonConstraints(z.number().int(), jsonSchema);
+        case "boolean":
+            return applyCommonConstraints(z.boolean(), jsonSchema);
+        case "null":
+            return z.null();
+        case "array":
+            return applyCommonConstraints(z.array(jsonSchemaToZod(jsonSchema.items)), jsonSchema);
+        case "object": {
+            const properties = isRecord(jsonSchema.properties) ? jsonSchema.properties : {};
+            const required = Array.isArray(jsonSchema.required)
+                ? new Set(jsonSchema.required.filter((value) => typeof value === "string"))
+                : new Set();
+            const shape = Object.fromEntries(Object.entries(properties).map(([key, value]) => {
+                const propertySchema = jsonSchemaToZod(value);
+                return [key, required.has(key) ? propertySchema : propertySchema.optional()];
+            }));
+            let objectSchema = z.object(shape);
+            if (jsonSchema.additionalProperties === true) {
+                objectSchema = objectSchema.passthrough();
+            }
+            else if (jsonSchema.additionalProperties === false) {
+                objectSchema = objectSchema.strict();
+            }
+            else if (isRecord(jsonSchema.additionalProperties)) {
+                objectSchema = objectSchema.catchall(jsonSchemaToZod(jsonSchema.additionalProperties));
+            }
+            else {
+                objectSchema = objectSchema.passthrough();
+            }
+            return applyCommonConstraints(objectSchema, jsonSchema);
+        }
+        default:
+            return z.unknown();
+    }
+};
+const normalizeListedTools = async (client) => {
+    const listed = await client.listTools();
+    return Array.isArray(listed) ? listed : listed.tools;
+};
+const getToolName = (name, prefix) => (prefix ? `${prefix}${name}` : name);
+export const createMcpToolSet = async (client, options = {}) => {
+    const include = options.includeTools ? new Set(options.includeTools) : undefined;
+    const exclude = options.excludeTools ? new Set(options.excludeTools) : undefined;
+    const listedTools = await normalizeListedTools(client);
+    const toolEntries = listedTools.filter((listedTool) => {
+        if (include && !include.has(listedTool.name)) {
+            return false;
+        }
+        if (exclude?.has(listedTool.name)) {
+            return false;
+        }
+        return true;
+    });
+    const seenNames = new Set();
+    return Object.fromEntries(toolEntries.map((listedTool) => {
+        const toolName = getToolName(listedTool.name, options.toolNamePrefix);
+        if (seenNames.has(toolName)) {
+            throw new Error(`Duplicate MCP tool name "${toolName}".`);
+        }
+        seenNames.add(toolName);
+        return [
+            toolName,
+            tool({
+                name: toolName,
+                description: listedTool.description,
+                schema: jsonSchemaToZod(listedTool.inputSchema),
+                metadata: serializeJsonValue({
+                    source: "mcp",
+                    originalName: listedTool.name,
+                    inputSchema: listedTool.inputSchema ?? null,
+                    annotations: listedTool.annotations ?? null
+                }),
+                execute: async (input) => serializeJsonValue(await client.callTool({
+                    name: listedTool.name,
+                    arguments: input
+                }))
+            })
+        ];
+    }));
+};
+export const createMcpToolRegistry = async (client, options = {}) => createToolRegistry(await createMcpToolSet(client, options));
+//# sourceMappingURL=mcp.js.map

package/dist/mcp.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp.js","sourceRoot":"","sources":["../src/mcp.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,kBAAkB,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACzD,OAAO,EAAE,kBAAkB,EAAqB,MAAM,oBAAoB,CAAC;AAiE3E,MAAM,QAAQ,GAAG,CAAC,KAA4B,EAAsC,EAAE,CACpF,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAEvE,MAAM,YAAY,GAAG,CAAC,KAAgB,EAAE,EAAE;IACxC,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,SAAS,EAAE,CAAC;QAC3G,OAAO,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IAED,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;AACrB,CAAC,CAAC;AAEF,MAAM,UAAU,GAAG,CAAC,OAAgC,EAAgB,EAAE;IACpE,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;QACrB,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;IACrB,CAAC;IAED,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC;IAChE,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;IACnB,CAAC;IAED,OAAO,CAAC,CAAC,KAAK,CAAC,MAAyD,CAAC,CAAC;AAC5E,CAAC,CAAC;AAEF,MAAM,sBAAsB,GAAG,CAAC,MAAoB,EAAE,UAA4B,EAAgB,EAAE;IAClG,IAAI,MAAM,YAAY,CAAC,CAAC,SAAS,EAAE,CAAC;QAClC,IAAI,IAAI,GAAG,MAAM,CAAC;QAClB,IAAI,OAAO,UAAU,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;YAC7C,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,OAAO,UAAU,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;YAC7C,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACxC,CAAC;QACD,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,UAAU,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACjG,CAAC;IAED,IAAI,MAAM,YAAY,CAAC,CAAC,SAAS,EAAE,CAAC;QAClC,IAAI,IAAI,GAAG,MAAM,CAAC;QAClB,IAAI,OAAO,UAAU,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC3C,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACtC,CAAC;QACD,IAAI,OAAO,UAAU,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC3C,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACtC,CAAC;QACD,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,UAAU,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACjG,CAAC;IAED,IAAI,MAAM,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC;QACjC,IAAI,IAAI,GAAG,MAAM,CAAC;QAClB,IAAI,OAAO,UAAU,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC5C,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACvC,CAAC;QACD,IAAI,OAAO,UAAU,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC5C,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACvC,CAAC;QACD,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,UAAU,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACjG,CAAC;IAED,OAAO,OAAO,UAAU,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;AACvG,CAAC,CAAC;AAEF,MAAM,eAAe,GAAG,CAAC,MAA6B,EAAgB,EAAE;IACtE,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;IACrB,CAAC;IAED,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QACtB,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;IACrB,CAAC;IAED,MAAM,UAAU,GAAG,MAA0B,CAAC;IAE9C,IAAI,UAAU,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QACnC,OAAO,YAAY,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACxC,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACjE,OAAO,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;QACpC,OAAO,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACtC,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;QACpC,OAAO,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACtC,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QACnC,OAAO,UAAU,CACf,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC7B,GAAG,UAAU;YACb,IAAI;SACL,CAAC,CAAC,CACJ,CAAC;IACJ,CAAC;IAED,QAAQ,UAAU,CAAC,IAAI,EAAE,CAAC;QACxB,KAAK,QAAQ;YACX,OAAO,sBAAsB,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,UAAU,CAAC,CAAC;QACxD,KAAK,QAAQ;YACX,OAAO,sBAAsB,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,UAAU,CAAC,CAAC;QACxD,KAAK,SAAS;YACZ,OAAO,sBAAsB,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,UAAU,CAAC,CAAC;QAC9D,KAAK,SAAS;YACZ,OAAO,sBAAsB,CAAC,CAAC,CAAC,OAAO,EAAE,EAAE,UAAU,CAAC,CAAC;QACzD,KAAK,MAAM;YACT,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,OAAO;YACV,OAAO,sBAAsB,CAAC,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;QACxF,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,MAAM,UAAU,GAAG,QAAQ,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;YAChF,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;gBACjD,CAAC,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,EAAmB,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC;gBAC5F,CAAC,CAAC,IAAI,GAAG,EAAU,CAAC;YAEtB,MAAM,KAAK,GAAG,MAAM,CAAC,WAAW,CAC9B,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;gBAC9C,MAAM,cAAc,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;gBAC9C,OAAO,CAAC,GAAG,EAAE,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,cAAc,CAAC,QAAQ,EAAE,CAAC,CAAC;YAC/E,CAAC,CAAC,CACH,CAAC;YAEF,IAAI,YAAY,GAAG,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACnC,IAAI,UAAU,CAAC,oBAAoB,KAAK,IAAI,EAAE,CAAC;gBAC7C,YAAY,GAAG,YAAY,CAAC,WAAW,EAAE,CAAC;YAC5C,CAAC;iBAAM,IAAI,UAAU,CAAC,oBAAoB,KAAK,KAAK,EAAE,CAAC;gBACrD,YAAY,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC;YACvC,CAAC;iBAAM,IAAI,QAAQ,CAAC,UAAU,CAAC,oBAAoB,CAAC,EAAE,CAAC;gBACrD,YAAY,GAAG,YAAY,CAAC,QAAQ,CAAC,eAAe,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC,CAAC;YACzF,CAAC;iBAAM,CAAC;gBACN,YAAY,GAAG,YAAY,CAAC,WAAW,EAAE,CAAC;YAC5C,CAAC;YAED,OAAO,sBAAsB,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QAC1D,CAAC;QACD;YACE,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;IACvB,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,oBAAoB,GAAG,KAAK,EAAE,MAAiB,EAA4B,EAAE;IACjF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;IACxC,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;AACvD,CAAC,CAAC;AAEF,MAAM,WAAW,GAAG,CAAC,IAAY,EAAE,MAAe,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;AAE5F,MAAM,CAAC,MAAM,gBAAgB,GAAG,KAAK,EAAE,MAAiB,EAAE,UAA6B,EAAE,EAAoB,EAAE;IAC7G,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACjF,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACjF,MAAM,WAAW,GAAG,MAAM,oBAAoB,CAAC,MAAM,CAAC,CAAC;IACvD,MAAM,WAAW,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,EAAE;QACpD,IAAI,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YAC7C,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,OAAO,EAAE,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YAClC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IAEpC,OAAO,MAAM,CAAC,WAAW,CACvB,WAAW,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;QAC7B,MAAM,QAAQ,GAAG,WAAW,CAAC,UAAU,CAAC,IAAI,EAAE,OAAO,CAAC,cAAc,CAAC,CAAC;QACtE,IAAI,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,4BAA4B,QAAQ,IAAI,CAAC,CAAC;QAC5D,CAAC;QAED,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAExB,OAAO;YACL,QAAQ;YACR,IAAI,CAAC;gBACH,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,UAAU,CAAC,WAAW;gBACnC,MAAM,EAAE,eAAe,CAAC,UAAU,CAAC,WAAW,CAAC;gBAC/C,QAAQ,EAAE,kBAAkB,CAAC;oBAC3B,MAAM,EAAE,KAAK;oBACb,YAAY,EAAE,UAAU,CAAC,IAAI;oBAC7B,WAAW,EAAE,UAAU,CAAC,WAAW,IAAI,IAAI;oBAC3C,WAAW,EAAE,UAAU,CAAC,WAAW,IAAI,IAAI;iBAC5C,CAA8B;gBAC/B,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CACvB,kBAAkB,CAChB,MAAM,MAAM,CAAC,QAAQ,CAAC;oBACpB,IAAI,EAAE,UAAU,CAAC,IAAI;oBACrB,SAAS,EAAE,KAAkB;iBAC9B,CAAC,CACH;aACJ,CAAC;SACH,CAAC;IACJ,CAAC,CAAC,CACH,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,qBAAqB,GAAG,KAAK,EACxC,MAAiB,EACjB,UAA6B,EAAE,EACR,EAAE,CAAC,kBAAkB,CAAC,MAAM,gBAAgB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC"}

package/dist/messages.d.ts

@@ -1,7 +1,8 @@
-import type { ContentPart, FinishReason, GenerateResult, JsonValue, LanguageModel, ModelMessage, StreamEvent, ToolCall, ToolDefinition, ToolExecutionResult } from "./types.js";
+import type { AgentCapabilities, AgentSupportTier, ContentPart, FinishReason, GenerateResult, HostedToolDefinition, HostedToolClass, JsonValue, LanguageModel, ModelMessage, StreamEvent, ToolCall, ToolDefinition, ToolExecutionResult } from "./types.js";
 export declare const textPart: (text: string) => ContentPart;
 export declare const toolCallPart: (toolCall: ToolCall) => ContentPart;
 export declare const toolResultPart: (toolResult: ToolExecutionResult) => ContentPart;
+export declare const providerDataPart: (provider: string, data: JsonValue) => ContentPart;
 export declare const createTextMessage: (role: ModelMessage["role"], text: string) => ModelMessage;
 export declare const system: (text: string) => ModelMessage;
 export declare const user: (input: string | ContentPart[]) => ModelMessage;
@@ -10,6 +11,13 @@ export declare const getTextFromParts: (parts: ContentPart[]) => string;
 export declare const getTextFromMessages: (messages: ModelMessage[]) => string;
 export declare const serializeJsonValue: (value: unknown) => JsonValue;
 export declare const tool: <TTool extends ToolDefinition>(definition: TTool) => TTool;
+export declare const hostedTool: <TTool extends HostedToolDefinition>(definition: Omit<TTool, "kind">) => TTool;
+export declare const isHostedToolDefinition: (toolDefinition: ToolDefinition | HostedToolDefinition) => toolDefinition is HostedToolDefinition;
+export declare const isCallableToolDefinition: (toolDefinition: ToolDefinition | HostedToolDefinition) => toolDefinition is ToolDefinition;
+export declare const getHostedToolClass: (toolDefinition: HostedToolDefinition) => HostedToolClass;
+export declare const isHostedToolClass: (toolDefinition: HostedToolDefinition, toolClass: HostedToolClass) => boolean;
+export declare const getAgentCapabilities: (model: LanguageModel) => AgentCapabilities;
+export declare const getAgentSupportTier: (model: LanguageModel) => AgentSupportTier;
 export declare const normalizeFinishReason: (reason: string | undefined | null) => FinishReason | undefined;
 export declare const resultMessages: (result: GenerateResult) => ModelMessage[];
 export declare const validateMessageParts: (model: LanguageModel, messages: ModelMessage[]) => void;

package/dist/messages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"messages.d.ts","sourceRoot":"","sources":["../src/messages.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,WAAW,EACX,YAAY,EACZ,cAAc,EACd,SAAS,EACT,aAAa,EACb,YAAY,EACZ,WAAW,EACX,QAAQ,EACR,cAAc,EACd,mBAAmB,EACpB,MAAM,YAAY,CAAC;AAEpB,eAAO,MAAM,QAAQ,GAAI,MAAM,MAAM,KAAG,WAAuC,CAAC;AAIhF,eAAO,MAAM,YAAY,GAAI,UAAU,QAAQ,KAAG,WAGhD,CAAC;AAEH,eAAO,MAAM,cAAc,GAAI,YAAY,mBAAmB,KAAG,WAG/D,CAAC;AAEH,eAAO,MAAM,iBAAiB,GAAI,MAAM,YAAY,CAAC,MAAM,CAAC,EAAE,MAAM,MAAM,KAAG,YAG3E,CAAC;AAEH,eAAO,MAAM,MAAM,GAAI,MAAM,MAAM,KAAG,YAGpC,CAAC;AAEH,eAAO,MAAM,IAAI,GAAI,OAAO,MAAM,GAAG,WAAW,EAAE,KAAG,YAGnD,CAAC;AAEH,eAAO,MAAM,SAAS,GAAI,OAAO,MAAM,GAAG,WAAW,EAAE,KAAG,YAGxD,CAAC;AAEH,eAAO,MAAM,gBAAgB,GAAI,OAAO,WAAW,EAAE,KAAG,MAI3C,CAAC;AAEd,eAAO,MAAM,mBAAmB,GAAI,UAAU,YAAY,EAAE,KAAG,MAIlD,CAAC;AAEd,eAAO,MAAM,kBAAkB,GAAI,OAAO,OAAO,KAAG,SAA2D,CAAC;AAEhH,eAAO,MAAM,IAAI,GAAI,KAAK,SAAS,cAAc,EAAE,YAAY,KAAK,KAAG,KAAmB,CAAC;AAE3F,eAAO,MAAM,qBAAqB,GAAI,QAAQ,MAAM,GAAG,SAAS,GAAG,IAAI,KAAG,YAAY,GAAG,SAsBxF,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,QAAQ,cAAc,KAAG,YAAY,EAWnE,CAAC;AAEF,eAAO,MAAM,oBAAoB,GAAI,OAAO,aAAa,EAAE,UAAU,YAAY,EAAE,SAwBlF,CAAC;AAEF,eAAO,MAAM,sBAAsB,GAAI,QAAQ,WAAW,EAAE,KAAG,QAAQ,EAGpC,CAAC"}
+{"version":3,"file":"messages.d.ts","sourceRoot":"","sources":["../src/messages.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,iBAAiB,EACjB,gBAAgB,EAChB,WAAW,EACX,YAAY,EACZ,cAAc,EACd,oBAAoB,EACpB,eAAe,EACf,SAAS,EACT,aAAa,EACb,YAAY,EACZ,WAAW,EACX,QAAQ,EACR,cAAc,EACd,mBAAmB,EACpB,MAAM,YAAY,CAAC;AAEpB,eAAO,MAAM,QAAQ,GAAI,MAAM,MAAM,KAAG,WAAuC,CAAC;AAIhF,eAAO,MAAM,YAAY,GAAI,UAAU,QAAQ,KAAG,WAGhD,CAAC;AAEH,eAAO,MAAM,cAAc,GAAI,YAAY,mBAAmB,KAAG,WAG/D,CAAC;AAEH,eAAO,MAAM,gBAAgB,GAAI,UAAU,MAAM,EAAE,MAAM,SAAS,KAAG,WAInE,CAAC;AAEH,eAAO,MAAM,iBAAiB,GAAI,MAAM,YAAY,CAAC,MAAM,CAAC,EAAE,MAAM,MAAM,KAAG,YAG3E,CAAC;AAEH,eAAO,MAAM,MAAM,GAAI,MAAM,MAAM,KAAG,YAGpC,CAAC;AAEH,eAAO,MAAM,IAAI,GAAI,OAAO,MAAM,GAAG,WAAW,EAAE,KAAG,YAGnD,CAAC;AAEH,eAAO,MAAM,SAAS,GAAI,OAAO,MAAM,GAAG,WAAW,EAAE,KAAG,YAGxD,CAAC;AAEH,eAAO,MAAM,gBAAgB,GAAI,OAAO,WAAW,EAAE,KAAG,MAI3C,CAAC;AAEd,eAAO,MAAM,mBAAmB,GAAI,UAAU,YAAY,EAAE,KAAG,MAIlD,CAAC;AAEd,eAAO,MAAM,kBAAkB,GAAI,OAAO,OAAO,KAAG,SAA2D,CAAC;AAEhH,eAAO,MAAM,IAAI,GAAI,KAAK,SAAS,cAAc,EAAE,YAAY,KAAK,KAAG,KAAmB,CAAC;AA0C3F,eAAO,MAAM,UAAU,GAAI,KAAK,SAAS,oBAAoB,EAAE,YAAY,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,KAAG,KAKpF,CAAC;AAEd,eAAO,MAAM,sBAAsB,GACjC,gBAAgB,cAAc,GAAG,oBAAoB,KACpD,cAAc,IAAI,oBAAoF,CAAC;AAE1G,eAAO,MAAM,wBAAwB,GACnC,gBAAgB,cAAc,GAAG,oBAAoB,KACpD,cAAc,IAAI,cAAyD,CAAC;AAE/E,eAAO,MAAM,kBAAkB,GAAI,gBAAgB,oBAAoB,KAAG,eACR,CAAC;AAEnE,eAAO,MAAM,iBAAiB,GAC5B,gBAAgB,oBAAoB,EACpC,WAAW,eAAe,YACyB,CAAC;AActD,eAAO,MAAM,oBAAoB,GAAI,OAAO,aAAa,KAAG,iBAG1D,CAAC;AAEH,eAAO,MAAM,mBAAmB,GAAI,OAAO,aAAa,KAAG,gBAA2D,CAAC;AAEvH,eAAO,MAAM,qBAAqB,GAAI,QAAQ,MAAM,GAAG,SAAS,GAAG,IAAI,KAAG,YAAY,GAAG,SAsBxF,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,QAAQ,cAAc,KAAG,YAAY,EAWnE,CAAC;AAEF,eAAO,MAAM,oBAAoB,GAAI,OAAO,aAAa,EAAE,UAAU,YAAY,EAAE,SAwBlF,CAAC;AAEF,eAAO,MAAM,sBAAsB,GAAI,QAAQ,WAAW,EAAE,KAAG,QAAQ,EAGpC,CAAC"}

package/dist/messages.js

@@ -9,6 +9,11 @@ export const toolResultPart = (toolResult) => ({
     type: "tool-result",
     toolResult
 });
+export const providerDataPart = (provider, data) => ({
+    type: "provider-data",
+    provider,
+    data
+});
 export const createTextMessage = (role, text) => ({
     role,
     parts: [textPart(text)]
@@ -35,6 +40,56 @@ export const getTextFromMessages = (messages) => messages
     .join("");
 export const serializeJsonValue = (value) => JSON.parse(JSON.stringify(value));
 export const tool = (definition) => definition;
+const inferHostedToolClass = (definition) => {
+    const normalizedType = definition.type.toLowerCase();
+    if (normalizedType.includes("web_search") ||
+        normalizedType.includes("web-search") ||
+        normalizedType.includes("googlesearch")) {
+        return "web-search";
+    }
+    if (normalizedType.includes("file_search") ||
+        normalizedType.includes("file-search")) {
+        return "file-search";
+    }
+    if (normalizedType === "mcp" || normalizedType.includes("mcp_toolset") || normalizedType.includes("mcp-toolset")) {
+        return normalizedType.includes("toolset") ? "toolset" : "remote-mcp";
+    }
+    if (normalizedType.includes("computer_use") ||
+        normalizedType.includes("computer-use")) {
+        return "computer-use";
+    }
+    if (normalizedType.includes("codeexecution") ||
+        normalizedType.includes("code_execution") ||
+        normalizedType.includes("code-execution")) {
+        return "code-execution";
+    }
+    return "custom";
+};
+export const hostedTool = (definition) => ({
+    kind: "hosted",
+    toolClass: inferHostedToolClass(definition),
+    ...definition
+});
+export const isHostedToolDefinition = (toolDefinition) => "kind" in toolDefinition && toolDefinition.kind === "hosted";
+export const isCallableToolDefinition = (toolDefinition) => !isHostedToolDefinition(toolDefinition);
+export const getHostedToolClass = (toolDefinition) => toolDefinition.toolClass ?? inferHostedToolClass(toolDefinition);
+export const isHostedToolClass = (toolDefinition, toolClass) => getHostedToolClass(toolDefinition) === toolClass;
+const emptyAgentCapabilities = {
+    supportTier: "tier-c",
+    toolChoiceNone: false,
+    approvalRequests: false,
+    hostedWebSearch: false,
+    hostedFileSearch: false,
+    remoteMcp: false,
+    computerUse: false,
+    codeExecution: false,
+    toolsets: false
+};
+export const getAgentCapabilities = (model) => ({
+    ...emptyAgentCapabilities,
+    ...(model.capabilities.agentCapabilities ?? {})
+});
+export const getAgentSupportTier = (model) => getAgentCapabilities(model).supportTier;
 export const normalizeFinishReason = (reason) => {
     if (!reason) {
         return undefined;

package/dist/messages.js.map

@@ -1 +1 @@
-{"version":3,"file":"messages.js","sourceRoot":"","sources":["../src/messages.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,MAAM,aAAa,CAAC;AActD,MAAM,CAAC,MAAM,QAAQ,GAAG,CAAC,IAAY,EAAe,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;AAEhF,MAAM,qBAAqB,GAAG,CAAC,KAA6B,EAAiB,EAAE,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;AAExI,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,QAAkB,EAAe,EAAE,CAAC,CAAC;IAChE,IAAI,EAAE,WAAW;IACjB,QAAQ;CACT,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,UAA+B,EAAe,EAAE,CAAC,CAAC;IAC/E,IAAI,EAAE,aAAa;IACnB,UAAU;CACX,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,IAA0B,EAAE,IAAY,EAAgB,EAAE,CAAC,CAAC;IAC5F,IAAI;IACJ,KAAK,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;CACxB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,MAAM,GAAG,CAAC,IAAY,EAAgB,EAAE,CAAC,CAAC;IACrD,IAAI,EAAE,QAAQ;IACd,KAAK,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;CACxB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,IAAI,GAAG,CAAC,KAA6B,EAAgB,EAAE,CAAC,CAAC;IACpE,IAAI,EAAE,MAAM;IACZ,KAAK,EAAE,qBAAqB,CAAC,KAAK,CAAC;CACpC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,SAAS,GAAG,CAAC,KAA6B,EAAgB,EAAE,CAAC,CAAC;IACzE,IAAI,EAAE,WAAW;IACjB,KAAK,EAAE,qBAAqB,CAAC,KAAK,CAAC;CACpC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,KAAoB,EAAU,EAAE,CAC/D,KAAK;KACF,MAAM,CAAC,CAAC,IAAI,EAAkD,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;KACtF,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;KACxB,IAAI,CAAC,EAAE,CAAC,CAAC;AAEd,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,QAAwB,EAAU,EAAE,CACtE,QAAQ;KACL,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,WAAW,CAAC;KACjD,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;KACjD,IAAI,CAAC,EAAE,CAAC,CAAC;AAEd,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,KAAc,EAAa,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAc,CAAC;AAEhH,MAAM,CAAC,MAAM,IAAI,GAAG,CAA+B,UAAiB,EAAS,EAAE,CAAC,UAAU,CAAC;AAE3F,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,MAAiC,EAA4B,EAAE;IACnG,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,QAAQ,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC;QAC7B,KAAK,MAAM,CAAC;QACZ,KAAK,UAAU;YACb,OAAO,MAAM,CAAC;QAChB,KAAK,QAAQ,CAAC;QACd,KAAK,YAAY;YACf,OAAO,QAAQ,CAAC;QAClB,KAAK,YAAY,CAAC;QAClB,KAAK,UAAU;YACb,OAAO,YAAY,CAAC;QACtB,KAAK,gBAAgB;YACnB,OAAO,gBAAgB,CAAC;QAC1B,KAAK,OAAO;YACV,OAAO,OAAO,CAAC;QACjB;YACE,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,MAAsB,EAAkB,EAAE;IACvE,IAAI,MAAM,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC;QAC5B,OAAO,MAAM,CAAC,QAAQ,CAAC;IACzB,CAAC;IACD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC1B,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;QAChB,OAAO,CAAC,iBAAiB,CAAC,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;IACvD,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,KAAoB,EAAE,QAAwB,EAAE,EAAE;IACrF,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YACjC,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;gBACxD,MAAM,IAAI,uBAAuB,CAC/B,UAAU,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,kCAAkC,CAC5E,CAAC;YACJ,CAAC;YAED,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;gBACtD,MAAM,IAAI,uBAAuB,CAC/B,UAAU,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,iCAAiC,CAC3E,CAAC;YACJ,CAAC;YAED,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;gBAC7D,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;oBAC9B,MAAM,IAAI,uBAAuB,CAC/B,UAAU,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,kCAAkC,CAC5E,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,MAAqB,EAAc,EAAE,CAC1E,MAAM;KACH,MAAM,CAAC,CAAC,KAAK,EAAwD,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,WAAW,CAAC;KACnG,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC"}
+{"version":3,"file":"messages.js","sourceRoot":"","sources":["../src/messages.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,MAAM,aAAa,CAAC;AAkBtD,MAAM,CAAC,MAAM,QAAQ,GAAG,CAAC,IAAY,EAAe,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;AAEhF,MAAM,qBAAqB,GAAG,CAAC,KAA6B,EAAiB,EAAE,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;AAExI,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,QAAkB,EAAe,EAAE,CAAC,CAAC;IAChE,IAAI,EAAE,WAAW;IACjB,QAAQ;CACT,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,UAA+B,EAAe,EAAE,CAAC,CAAC;IAC/E,IAAI,EAAE,aAAa;IACnB,UAAU;CACX,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,QAAgB,EAAE,IAAe,EAAe,EAAE,CAAC,CAAC;IACnF,IAAI,EAAE,eAAe;IACrB,QAAQ;IACR,IAAI;CACL,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,IAA0B,EAAE,IAAY,EAAgB,EAAE,CAAC,CAAC;IAC5F,IAAI;IACJ,KAAK,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;CACxB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,MAAM,GAAG,CAAC,IAAY,EAAgB,EAAE,CAAC,CAAC;IACrD,IAAI,EAAE,QAAQ;IACd,KAAK,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;CACxB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,IAAI,GAAG,CAAC,KAA6B,EAAgB,EAAE,CAAC,CAAC;IACpE,IAAI,EAAE,MAAM;IACZ,KAAK,EAAE,qBAAqB,CAAC,KAAK,CAAC;CACpC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,SAAS,GAAG,CAAC,KAA6B,EAAgB,EAAE,CAAC,CAAC;IACzE,IAAI,EAAE,WAAW;IACjB,KAAK,EAAE,qBAAqB,CAAC,KAAK,CAAC;CACpC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,KAAoB,EAAU,EAAE,CAC/D,KAAK;KACF,MAAM,CAAC,CAAC,IAAI,EAAkD,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;KACtF,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;KACxB,IAAI,CAAC,EAAE,CAAC,CAAC;AAEd,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,QAAwB,EAAU,EAAE,CACtE,QAAQ;KACL,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,WAAW,CAAC;KACjD,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;KACjD,IAAI,CAAC,EAAE,CAAC,CAAC;AAEd,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,KAAc,EAAa,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAc,CAAC;AAEhH,MAAM,CAAC,MAAM,IAAI,GAAG,CAA+B,UAAiB,EAAS,EAAE,CAAC,UAAU,CAAC;AAE3F,MAAM,oBAAoB,GAAG,CAAC,UAA8C,EAAmB,EAAE;IAC/F,MAAM,cAAc,GAAG,UAAU,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;IAErD,IACE,cAAc,CAAC,QAAQ,CAAC,YAAY,CAAC;QACrC,cAAc,CAAC,QAAQ,CAAC,YAAY,CAAC;QACrC,cAAc,CAAC,QAAQ,CAAC,cAAc,CAAC,EACvC,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,IACE,cAAc,CAAC,QAAQ,CAAC,aAAa,CAAC;QACtC,cAAc,CAAC,QAAQ,CAAC,aAAa,CAAC,EACtC,CAAC;QACD,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,IAAI,cAAc,KAAK,KAAK,IAAI,cAAc,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,cAAc,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;QACjH,OAAO,cAAc,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC;IACvE,CAAC;IAED,IACE,cAAc,CAAC,QAAQ,CAAC,cAAc,CAAC;QACvC,cAAc,CAAC,QAAQ,CAAC,cAAc,CAAC,EACvC,CAAC;QACD,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,IACE,cAAc,CAAC,QAAQ,CAAC,eAAe,CAAC;QACxC,cAAc,CAAC,QAAQ,CAAC,gBAAgB,CAAC;QACzC,cAAc,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EACzC,CAAC;QACD,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,UAAU,GAAG,CAAqC,UAA+B,EAAS,EAAE,CACvG,CAAC;IACC,IAAI,EAAE,QAAQ;IACd,SAAS,EAAE,oBAAoB,CAAC,UAAU,CAAC;IAC3C,GAAG,UAAU;CACd,CAAU,CAAC;AAEd,MAAM,CAAC,MAAM,sBAAsB,GAAG,CACpC,cAAqD,EACb,EAAE,CAAC,MAAM,IAAI,cAAc,IAAI,cAAc,CAAC,IAAI,KAAK,QAAQ,CAAC;AAE1G,MAAM,CAAC,MAAM,wBAAwB,GAAG,CACtC,cAAqD,EACnB,EAAE,CAAC,CAAC,sBAAsB,CAAC,cAAc,CAAC,CAAC;AAE/E,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,cAAoC,EAAmB,EAAE,CAC1F,cAAc,CAAC,SAAS,IAAI,oBAAoB,CAAC,cAAc,CAAC,CAAC;AAEnE,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAC/B,cAAoC,EACpC,SAA0B,EAC1B,EAAE,CAAC,kBAAkB,CAAC,cAAc,CAAC,KAAK,SAAS,CAAC;AAEtD,MAAM,sBAAsB,GAAsB;IAChD,WAAW,EAAE,QAAQ;IACrB,cAAc,EAAE,KAAK;IACrB,gBAAgB,EAAE,KAAK;IACvB,eAAe,EAAE,KAAK;IACtB,gBAAgB,EAAE,KAAK;IACvB,SAAS,EAAE,KAAK;IAChB,WAAW,EAAE,KAAK;IAClB,aAAa,EAAE,KAAK;IACpB,QAAQ,EAAE,KAAK;CAChB,CAAC;AAEF,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,KAAoB,EAAqB,EAAE,CAAC,CAAC;IAChF,GAAG,sBAAsB;IACzB,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,iBAAiB,IAAI,EAAE,CAAC;CAChD,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,KAAoB,EAAoB,EAAE,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC;AAEvH,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,MAAiC,EAA4B,EAAE;IACnG,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,QAAQ,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC;QAC7B,KAAK,MAAM,CAAC;QACZ,KAAK,UAAU;YACb,OAAO,MAAM,CAAC;QAChB,KAAK,QAAQ,CAAC;QACd,KAAK,YAAY;YACf,OAAO,QAAQ,CAAC;QAClB,KAAK,YAAY,CAAC;QAClB,KAAK,UAAU;YACb,OAAO,YAAY,CAAC;QACtB,KAAK,gBAAgB;YACnB,OAAO,gBAAgB,CAAC;QAC1B,KAAK,OAAO;YACV,OAAO,OAAO,CAAC;QACjB;YACE,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,MAAsB,EAAkB,EAAE;IACvE,IAAI,MAAM,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC;QAC5B,OAAO,MAAM,CAAC,QAAQ,CAAC;IACzB,CAAC;IACD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC1B,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;QAChB,OAAO,CAAC,iBAAiB,CAAC,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;IACvD,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,KAAoB,EAAE,QAAwB,EAAE,EAAE;IACrF,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YACjC,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;gBACxD,MAAM,IAAI,uBAAuB,CAC/B,UAAU,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,kCAAkC,CAC5E,CAAC;YACJ,CAAC;YAED,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;gBACtD,MAAM,IAAI,uBAAuB,CAC/B,UAAU,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,iCAAiC,CAC3E,CAAC;YACJ,CAAC;YAED,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;gBAC7D,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;oBAC9B,MAAM,IAAI,uBAAuB,CAC/B,UAAU,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,kCAAkC,CAC5E,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,MAAqB,EAAc,EAAE,CAC1E,MAAM;KACH,MAAM,CAAC,CAAC,KAAK,EAAwD,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,WAAW,CAAC;KACnG,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC"}

package/dist/observability.d.ts

@@ -0,0 +1,51 @@
+import type { AgentTelemetryObserver, LanguageModelMiddleware, ProviderOptions } from "./types.js";
+type PrimitiveSpanAttribute = string | number | boolean | null;
+type SpanAttributeValue = PrimitiveSpanAttribute | PrimitiveSpanAttribute[];
+export interface OTelSpanLike {
+    setAttribute?(key: string, value: SpanAttributeValue): unknown;
+    addEvent?(name: string, attributes?: Record<string, SpanAttributeValue>): unknown;
+    recordException?(error: Error): unknown;
+    setStatus?(status: unknown): unknown;
+    end(): unknown;
+}
+export interface OTelTracerLike {
+    startSpan(name: string, options?: {
+        attributes?: Record<string, SpanAttributeValue>;
+    }): OTelSpanLike;
+}
+export declare class OTelSpanHandle {
+    readonly span: OTelSpanLike;
+    private readonly statusSetter?;
+    constructor(span: OTelSpanLike, statusSetter?: ((span: OTelSpanLike, error: Error) => Promise<void> | void) | undefined);
+    end(options?: {
+        attributes?: Record<string, unknown>;
+        error?: Error;
+    }): Promise<void>;
+}
+export declare class OTelObserver {
+    private readonly tracer;
+    private readonly statusSetter?;
+    constructor(tracer: OTelTracerLike, statusSetter?: ((span: OTelSpanLike, error: Error) => Promise<void> | void) | undefined);
+    startSpan(name: string, attributes?: Record<string, unknown>): OTelSpanHandle;
+}
+export declare const createOtelObserver: (options?: {
+    tracer?: OTelTracerLike;
+    tracerName?: string;
+    version?: string;
+}) => Promise<OTelObserver>;
+export declare const createOtelAgentObserver: (options?: {
+    observer?: OTelObserver;
+    tracer?: OTelTracerLike;
+    tracerName?: string;
+    version?: string;
+    spanNamePrefix?: string;
+}) => Promise<AgentTelemetryObserver>;
+export declare const createOtelTelemetryMiddleware: <TProviderOptions extends ProviderOptions = ProviderOptions>(options?: {
+    observer?: OTelObserver;
+    tracer?: OTelTracerLike;
+    tracerName?: string;
+    version?: string;
+    spanNamePrefix?: string;
+}) => Promise<LanguageModelMiddleware<TProviderOptions>>;
+export {};
+//# sourceMappingURL=observability.d.ts.map

package/dist/observability.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"observability.d.ts","sourceRoot":"","sources":["../src/observability.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAEV,sBAAsB,EACtB,uBAAuB,EAEvB,eAAe,EAChB,MAAM,YAAY,CAAC;AAEpB,KAAK,sBAAsB,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,IAAI,CAAC;AAC/D,KAAK,kBAAkB,GAAG,sBAAsB,GAAG,sBAAsB,EAAE,CAAC;AAE5E,MAAM,WAAW,YAAY;IAC3B,YAAY,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,kBAAkB,GAAG,OAAO,CAAC;IAC/D,QAAQ,CAAC,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,kBAAkB,CAAC,GAAG,OAAO,CAAC;IAClF,eAAe,CAAC,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC;IACxC,SAAS,CAAC,CAAC,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC;IACrC,GAAG,IAAI,OAAO,CAAC;CAChB;AAED,MAAM,WAAW,cAAc;IAC7B,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAA;KAAE,GAAG,YAAY,CAAC;CACtG;AAyDD,qBAAa,cAAc;IAEvB,QAAQ,CAAC,IAAI,EAAE,YAAY;IAC3B,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC;gBADrB,IAAI,EAAE,YAAY,EACV,YAAY,CAAC,GAAE,CAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,aAAA;IAGtF,GAAG,CAAC,OAAO,GAAE;QACjB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACrC,KAAK,CAAC,EAAE,KAAK,CAAC;KACV;CAQP;AAED,qBAAa,YAAY;IAErB,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC;gBADb,MAAM,EAAE,cAAc,EACtB,YAAY,CAAC,GAAE,CAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,aAAA;IAG5F,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,cAAc;CAM9E;AAeD,eAAO,MAAM,kBAAkB,GAAU,UAAS;IAChD,MAAM,CAAC,EAAE,cAAc,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;CACb,KAAG,OAAO,CAAC,YAAY,CAe5B,CAAC;AAEF,eAAO,MAAM,uBAAuB,GAAU,UAAS;IACrD,QAAQ,CAAC,EAAE,YAAY,CAAC;IACxB,MAAM,CAAC,EAAE,cAAc,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,cAAc,CAAC,EAAE,MAAM,CAAC;CACpB,KAAG,OAAO,CAAC,sBAAsB,CA6HtC,CAAC;AAEF,eAAO,MAAM,6BAA6B,GAAU,gBAAgB,SAAS,eAAe,GAAG,eAAe,EAAE,UAAS;IACvH,QAAQ,CAAC,EAAE,YAAY,CAAC;IACxB,MAAM,CAAC,EAAE,cAAc,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,cAAc,CAAC,EAAE,MAAM,CAAC;CACpB,KAAG,OAAO,CAAC,uBAAuB,CAAC,gBAAgB,CAAC,CAuFzD,CAAC"}

package/dist/observability.js

@@ -0,0 +1,282 @@
+import { createTelemetryMiddleware } from "./middleware.js";
+const OTEL_API_MODULE = "@opentelemetry/api";
+// Keep OpenTelemetry optional for consumers that never use the OTEL helpers.
+const loadOptionalOtelApi = async () => {
+    try {
+        return await import(OTEL_API_MODULE);
+    }
+    catch {
+        return undefined;
+    }
+};
+const toSpanAttributeValue = (value) => {
+    if (value === null || typeof value === "string" || typeof value === "number" || typeof value === "boolean") {
+        return value;
+    }
+    if (Array.isArray(value)) {
+        const normalized = value
+            .map((item) => (item === null || typeof item === "string" || typeof item === "number" || typeof item === "boolean" ? item : JSON.stringify(item)))
+            .filter((item) => item !== undefined);
+        return normalized.length ? normalized : undefined;
+    }
+    if (value === undefined) {
+        return undefined;
+    }
+    try {
+        return JSON.stringify(value);
+    }
+    catch {
+        return String(value);
+    }
+};
+const toSpanAttributes = (attributes) => {
+    const entries = Object.entries(attributes ?? {})
+        .map(([key, value]) => [key, toSpanAttributeValue(value)])
+        .filter((entry) => entry[1] !== undefined);
+    return Object.fromEntries(entries);
+};
+const safeSetAttributes = (span, attributes) => {
+    for (const [key, value] of Object.entries(toSpanAttributes(attributes))) {
+        span.setAttribute?.(key, value);
+    }
+};
+const safeAddEvent = (span, name, attributes) => {
+    if (!span?.addEvent) {
+        return;
+    }
+    span.addEvent(name, toSpanAttributes(attributes));
+};
+export class OTelSpanHandle {
+    span;
+    statusSetter;
+    constructor(span, statusSetter) {
+        this.span = span;
+        this.statusSetter = statusSetter;
+    }
+    async end(options = {}) {
+        safeSetAttributes(this.span, options.attributes);
+        if (options.error) {
+            this.span.recordException?.(options.error);
+            await this.statusSetter?.(this.span, options.error);
+        }
+        this.span.end();
+    }
+}
+export class OTelObserver {
+    tracer;
+    statusSetter;
+    constructor(tracer, statusSetter) {
+        this.tracer = tracer;
+        this.statusSetter = statusSetter;
+    }
+    startSpan(name, attributes) {
+        const span = this.tracer.startSpan(name, {
+            attributes: toSpanAttributes(attributes)
+        });
+        return new OTelSpanHandle(span, this.statusSetter);
+    }
+}
+const createDefaultStatusSetter = async () => {
+    const otel = await loadOptionalOtelApi();
+    if (!otel) {
+        return undefined;
+    }
+    return (span, error) => {
+        span.setStatus?.({
+            code: otel.SpanStatusCode.ERROR,
+            message: error.message
+        });
+    };
+};
+export const createOtelObserver = async (options = {}) => {
+    const statusSetter = await createDefaultStatusSetter();
+    if (options.tracer) {
+        return new OTelObserver(options.tracer, statusSetter);
+    }
+    const otel = await loadOptionalOtelApi();
+    if (!otel) {
+        throw new Error('OpenTelemetry is not installed. Install "@opentelemetry/api" to use OTEL observability helpers.', { cause: new Error(`Missing optional dependency: ${OTEL_API_MODULE}`) });
+    }
+    return new OTelObserver(otel.trace.getTracer(options.tracerName ?? "zhivex-ai", options.version), statusSetter);
+};
+export const createOtelAgentObserver = async (options = {}) => {
+    const observer = options.observer ?? (await createOtelObserver(options));
+    const runSpans = new Map();
+    const stepSpans = new Map();
+    const prefix = options.spanNamePrefix ?? "zhivex.agent";
+    return async (event) => {
+        if (event.type === "run-start") {
+            runSpans.set(event.runId, observer.startSpan(`${prefix}.run`, {
+                "zhivex.run_id": event.runId,
+                "zhivex.agent_id": event.agentId,
+                "zhivex.provider": event.provider,
+                "zhivex.model_id": event.modelId,
+                "zhivex.max_steps": event.maxSteps
+            }));
+            return;
+        }
+        if (event.type === "step-start") {
+            stepSpans.set(`${event.runId}:${event.stepIndex}`, observer.startSpan(`${prefix}.step`, {
+                "zhivex.run_id": event.runId,
+                "zhivex.agent_id": event.agentId,
+                "zhivex.step_index": event.stepIndex
+            }));
+            return;
+        }
+        if (event.type === "step-finish") {
+            const key = `${event.runId}:${event.step.index}`;
+            const handle = stepSpans.get(key);
+            stepSpans.delete(key);
+            await handle?.end({
+                attributes: {
+                    "zhivex.step_status": event.step.status,
+                    "zhivex.tool_results": event.step.toolResults.length
+                }
+            });
+            return;
+        }
+        if (event.type === "approval-request") {
+            safeAddEvent(runSpans.get(event.runId)?.span, "approval-request", {
+                "zhivex.approval_id": event.approval.id,
+                "zhivex.approval_provider": event.approval.provider,
+                "zhivex.approval_name": event.approval.name
+            });
+            return;
+        }
+        if (event.type === "approval-resolved") {
+            safeAddEvent(runSpans.get(event.runId)?.span, "approval-resolved", {
+                "zhivex.approval_id": event.approval.approvalRequestId,
+                "zhivex.approval_provider": event.approval.provider,
+                "zhivex.approved": event.approval.approve,
+                "zhivex.reason": event.approval.reason
+            });
+            return;
+        }
+        if (event.type === "tool-approval") {
+            safeAddEvent(runSpans.get(event.runId)?.span, "tool-approval", {
+                "zhivex.tool_name": event.toolCall.name,
+                "zhivex.tool_call_id": event.toolCall.id,
+                "zhivex.approved": event.approved,
+                "zhivex.reason": event.reason
+            });
+            return;
+        }
+        if (event.type === "memory-loaded") {
+            safeAddEvent(runSpans.get(event.runId)?.span, "memory-loaded", {
+                "zhivex.message_count": event.messageCount
+            });
+            return;
+        }
+        if (event.type === "guardrail-triggered") {
+            safeAddEvent(runSpans.get(event.runId)?.span, "guardrail-triggered", {
+                "zhivex.stage": event.stage,
+                "zhivex.reason": event.reason,
+                ...event.metadata
+            });
+            return;
+        }
+        if (event.type === "handoff") {
+            safeAddEvent(runSpans.get(event.runId)?.span, "handoff", {
+                "zhivex.handoff_id": event.handoff.id,
+                "zhivex.from_agent_id": event.handoff.fromAgentId,
+                "zhivex.to_agent_id": event.handoff.toAgentId
+            });
+            return;
+        }
+        if (event.type === "state-saved") {
+            safeAddEvent(runSpans.get(event.runId)?.span, "state-saved", {
+                "zhivex.status": event.status
+            });
+            return;
+        }
+        if (event.type === "run-finish") {
+            const handle = runSpans.get(event.runId);
+            runSpans.delete(event.runId);
+            await handle?.end({
+                attributes: {
+                    "zhivex.status": event.status,
+                    "zhivex.current_step": event.state.currentStep,
+                    "zhivex.pending_approvals": event.state.pendingApprovals.length,
+                    "zhivex.finish_reason": event.state.finishReason,
+                    "zhivex.provider_finish_reason": event.state.providerFinishReason
+                },
+                error: event.status === "failed" && event.state.error?.message
+                    ? new Error(event.state.error.message)
+                    : undefined
+            });
+        }
+    };
+};
+export const createOtelTelemetryMiddleware = async (options = {}) => {
+    const observer = options.observer ?? (await createOtelObserver(options));
+    const spans = new Map();
+    const prefix = options.spanNamePrefix ?? "zhivex.model";
+    const keyFor = (event) => {
+        if (event.type === "generate-start" || event.type === "generate-finish" || event.type === "generate-error") {
+            return `generate:${event.model.provider}:${event.model.modelId}:${event.startedAt}`;
+        }
+        if (event.type === "stream-start" || event.type === "stream-finish" || event.type === "stream-error") {
+            return `stream:${event.model.provider}:${event.model.modelId}:${event.startedAt}`;
+        }
+        return `tool:${event.model.provider}:${event.model.modelId}:${event.step}:${event.toolCall.id}:${event.startedAt}`;
+    };
+    return createTelemetryMiddleware({
+        onEvent: async (event) => {
+            if (event.type === "generate-start" || event.type === "stream-start") {
+                spans.set(keyFor(event), observer.startSpan(`${prefix}.${event.type === "generate-start" ? "generate" : "stream"}`, {
+                    "zhivex.provider": event.model.provider,
+                    "zhivex.model_id": event.model.modelId
+                }));
+                return;
+            }
+            if (event.type === "tool-execution-start") {
+                spans.set(keyFor(event), observer.startSpan(`${prefix}.tool`, {
+                    "zhivex.provider": event.model.provider,
+                    "zhivex.model_id": event.model.modelId,
+                    "zhivex.step": event.step,
+                    "zhivex.tool_name": event.toolCall.name,
+                    "zhivex.tool_call_id": event.toolCall.id
+                }));
+                return;
+            }
+            const handle = spans.get(keyFor(event));
+            spans.delete(keyFor(event));
+            if (event.type === "generate-finish") {
+                await handle?.end({
+                    attributes: {
+                        "zhivex.latency_ms": event.latencyMs
+                    }
+                });
+                return;
+            }
+            if (event.type === "stream-finish") {
+                await handle?.end({
+                    attributes: {
+                        "zhivex.finish_reason": event.finishReason,
+                        "zhivex.provider_finish_reason": event.providerFinishReason,
+                        "zhivex.latency_ms": event.latencyMs
+                    }
+                });
+                return;
+            }
+            if (event.type === "tool-execution-finish") {
+                await handle?.end({
+                    attributes: {
+                        "zhivex.latency_ms": event.latencyMs,
+                        "zhivex.tool_error": event.toolResult.isError
+                    }
+                });
+                return;
+            }
+            if (event.type === "generate-error" || event.type === "stream-error" || event.type === "tool-execution-error") {
+                await handle?.end({
+                    attributes: {
+                        "zhivex.latency_ms": event.latencyMs
+                    },
+                    error: event.error
+                });
+            }
+        }
+    });
+};
+//# sourceMappingURL=observability.js.map