@igniter-js/agents 0.1.11 → 0.1.13

package/dist/index.mjs

@@ -3,8 +3,6 @@ import { experimental_createMCPClient } from '@ai-sdk/mcp';
 import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
 import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
 import { ToolLoopAgent, tool } from 'ai';
-import { IgniterTelemetryEvents } from '@igniter-js/telemetry';
-import { z } from 'zod';
 import { readFile, writeFile, mkdir } from 'fs/promises';
 import { join } from 'path';
 
@@ -481,153 +479,6 @@ var IgniterAgentManagerCore = class _IgniterAgentManagerCore {
     );
   }
 };
-var BaseAgentAttributesSchema = z.object({
-  /**
-   * The agent name.
-   */
-  "ctx.agent.name": z.string(),
-  /**
-   * The current scope (if scoped).
-   */
-  "ctx.agent.scope": z.string().optional(),
-  /**
-   * The scope identifier value.
-   */
-  "ctx.agent.scopeId": z.string().optional()
-});
-var LifecycleAttributesSchema = BaseAgentAttributesSchema.extend({
-  /**
-   * Number of toolsets registered.
-   */
-  "ctx.lifecycle.toolsetCount": z.number().optional(),
-  /**
-   * Number of MCP connections configured.
-   */
-  "ctx.lifecycle.mcpCount": z.number().optional(),
-  /**
-   * Whether memory adapter is configured.
-   */
-  "ctx.lifecycle.hasMemory": z.boolean().optional()
-});
-var GenerationAttributesSchema = BaseAgentAttributesSchema.extend({
-  /**
-   * The model used for generation.
-   */
-  "ctx.generation.model": z.string().optional(),
-  /**
-   * Number of input messages.
-   */
-  "ctx.generation.inputMessages": z.number().optional(),
-  /**
-   * Number of input tokens (if available).
-   */
-  "ctx.generation.inputTokens": z.number().optional(),
-  /**
-   * Number of output tokens (if available).
-   */
-  "ctx.generation.outputTokens": z.number().optional(),
-  /**
-   * Total tokens used (if available).
-   */
-  "ctx.generation.totalTokens": z.number().optional(),
-  /**
-   * Generation duration in milliseconds.
-   */
-  "ctx.generation.durationMs": z.number().optional(),
-  /**
-   * Number of tool calls made during generation.
-   */
-  "ctx.generation.toolCalls": z.number().optional(),
-  /**
-   * Whether the response was streamed.
-   */
-  "ctx.generation.streamed": z.boolean().optional()
-});
-var ToolAttributesSchema = BaseAgentAttributesSchema.extend({
-  /**
-   * The toolset name.
-   */
-  "ctx.tool.toolset": z.string(),
-  /**
-   * The tool name (within the toolset).
-   */
-  "ctx.tool.name": z.string(),
-  /**
-   * The full tool identifier (toolset_name).
-   */
-  "ctx.tool.fullName": z.string(),
-  /**
-   * Tool execution duration in milliseconds.
-   */
-  "ctx.tool.durationMs": z.number().optional()
-});
-var MCPAttributesSchema = BaseAgentAttributesSchema.extend({
-  /**
-   * The MCP configuration name.
-   */
-  "ctx.mcp.name": z.string(),
-  /**
-   * The MCP transport type (stdio, http).
-   */
-  "ctx.mcp.type": z.enum(["stdio", "http"]),
-  /**
-   * Number of tools provided by this MCP.
-   */
-  "ctx.mcp.toolCount": z.number().optional(),
-  /**
-   * Connection duration in milliseconds.
-   */
-  "ctx.mcp.durationMs": z.number().optional()
-});
-var MemoryAttributesSchema = BaseAgentAttributesSchema.extend({
-  /**
-   * The memory operation type.
-   */
-  "ctx.memory.operation": z.enum([
-    "getWorkingMemory",
-    "updateWorkingMemory",
-    "getMessages",
-    "saveMessage",
-    "getChats",
-    "saveChat",
-    "getChat",
-    "updateChatTitle",
-    "deleteChat"
-  ]),
-  /**
-   * The memory scope (user, chat, global).
-   */
-  "ctx.memory.scope": z.string().optional(),
-  /**
-   * Number of items affected.
-   */
-  "ctx.memory.count": z.number().optional(),
-  /**
-   * Operation duration in milliseconds.
-   */
-  "ctx.memory.durationMs": z.number().optional()
-});
-var ErrorAttributesSchema = BaseAgentAttributesSchema.extend({
-  /**
-   * The error code.
-   */
-  "ctx.error.code": z.string(),
-  /**
-   * The sanitized error message (no sensitive data).
-   */
-  "ctx.error.message": z.string().optional(),
-  /**
-   * The operation that failed.
-   */
-  "ctx.error.operation": z.string().optional(),
-  /**
-   * The component that threw the error.
-   */
-  "ctx.error.component": z.string().optional()
-});
-var IgniterAgentTelemetryEvents = IgniterTelemetryEvents.namespace(
-  "igniter.agent"
-).event("lifecycle.start.started", LifecycleAttributesSchema).event("lifecycle.start.success", LifecycleAttributesSchema).event("lifecycle.start.error", ErrorAttributesSchema).event("lifecycle.stop.started", LifecycleAttributesSchema).event("lifecycle.stop.success", LifecycleAttributesSchema).event("lifecycle.stop.error", ErrorAttributesSchema).event("generation.generate.started", GenerationAttributesSchema).event("generation.generate.success", GenerationAttributesSchema).event("generation.generate.error", ErrorAttributesSchema).event("generation.stream.started", GenerationAttributesSchema).event("generation.stream.chunk", GenerationAttributesSchema).event("generation.stream.success", GenerationAttributesSchema).event("generation.stream.error", ErrorAttributesSchema).event("tool.execute.started", ToolAttributesSchema).event("tool.execute.success", ToolAttributesSchema).event("tool.execute.error", ErrorAttributesSchema).event("mcp.connect.started", MCPAttributesSchema).event("mcp.connect.success", MCPAttributesSchema).event("mcp.connect.error", ErrorAttributesSchema).event("mcp.disconnect.started", MCPAttributesSchema).event("mcp.disconnect.success", MCPAttributesSchema).event("mcp.disconnect.error", ErrorAttributesSchema).event("memory.operation.started", MemoryAttributesSchema).event("memory.operation.success", MemoryAttributesSchema).event("memory.operation.error", ErrorAttributesSchema).build();
 
 // src/core/memory.ts
 var MEMORY_ERROR_CODE = "IGNITER_AGENT_MEMORY_PROVIDER_ERROR" /* MEMORY_PROVIDER_ERROR */;
@@ -649,7 +500,7 @@ var IgniterAgentMemoryCore = class {
     return attributes;
   }
   emitStart(operation, scope) {
-    this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("memory.operation.started"), {
+    this.telemetry?.emit("igniter.agent.memory.operation.started", {
       level: "debug",
       attributes: this.getBaseAttributes(operation, scope)
     });
@@ -662,7 +513,7 @@ var IgniterAgentMemoryCore = class {
     if (count !== void 0) {
       attributes["ctx.memory.count"] = count;
     }
-    this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("memory.operation.success"), {
+    this.telemetry?.emit("igniter.agent.memory.operation.success", {
       level: "debug",
       attributes
     });
@@ -675,7 +526,7 @@ var IgniterAgentMemoryCore = class {
       "ctx.error.operation": operation,
       "ctx.error.component": "memory"
     };
-    this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("memory.operation.error"), {
+    this.telemetry?.emit("igniter.agent.memory.operation.error", {
       level: "error",
       attributes
     });
@@ -898,7 +749,7 @@ var IgniterAgentCore = class {
       "ctx.lifecycle.hasMemory": Boolean(this.memory)
     };
     this.logger?.debug("IgniterAgent.start started", lifecycleAttributes);
-    this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("lifecycle.start.started"), {
+    this.telemetry?.emit("igniter.agent.lifecycle.start.started", {
       level: "debug",
       attributes: lifecycleAttributes
     });
@@ -906,7 +757,7 @@ var IgniterAgentCore = class {
       for (const mcpConfig of mcpConfigs) {
         const mcpStart = Date.now();
         this.hooks.onMCPStart?.(this.getName(), mcpConfig.name);
-        this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("mcp.connect.started"), {
+        this.telemetry?.emit("igniter.agent.mcp.connect.started", {
           level: "debug",
           attributes: {
             "ctx.agent.name": this.getName(),
@@ -917,7 +768,7 @@ var IgniterAgentCore = class {
         try {
           const mcpToolset = await this.initializeMCPClient(mcpConfig);
           toolsets[mcpConfig.name] = mcpToolset;
-          this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("mcp.connect.success"), {
+          this.telemetry?.emit("igniter.agent.mcp.connect.success", {
             level: "debug",
             attributes: {
               "ctx.agent.name": this.getName(),
@@ -929,7 +780,7 @@ var IgniterAgentCore = class {
           });
         } catch (error) {
           const err = error instanceof Error ? error : new Error(String(error));
-          this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("mcp.connect.error"), {
+          this.telemetry?.emit("igniter.agent.mcp.connect.error", {
             level: "error",
             attributes: {
               "ctx.agent.name": this.getName(),
@@ -944,7 +795,7 @@ var IgniterAgentCore = class {
         }
       }
       const durationMs = Date.now() - startTime;
-      this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("lifecycle.start.success"), {
+      this.telemetry?.emit("igniter.agent.lifecycle.start.success", {
         level: "debug",
         attributes: {
           ...lifecycleAttributes,
@@ -959,7 +810,7 @@ var IgniterAgentCore = class {
       this.hooks.onAgentStart?.(this.getName());
     } catch (error) {
       const err = error instanceof Error ? error : new Error(String(error));
-      this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("lifecycle.start.error"), {
+      this.telemetry?.emit("igniter.agent.lifecycle.start.error", {
         level: "error",
         attributes: {
           ...lifecycleAttributes,
@@ -984,14 +835,14 @@ var IgniterAgentCore = class {
       "ctx.lifecycle.hasMemory": Boolean(this.memory)
     };
     this.logger?.debug("IgniterAgent.stop started", lifecycleAttributes);
-    this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("lifecycle.stop.started"), {
+    this.telemetry?.emit("igniter.agent.lifecycle.stop.started", {
       level: "debug",
       attributes: lifecycleAttributes
     });
     try {
       for (const mcpConfig of mcpConfigs) {
         const mcpStart = Date.now();
-        this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("mcp.disconnect.started"), {
+        this.telemetry?.emit("igniter.agent.mcp.disconnect.started", {
           level: "debug",
           attributes: {
             "ctx.agent.name": this.getName(),
@@ -1011,7 +862,7 @@ var IgniterAgentCore = class {
               tools: {}
             };
           }
-          this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("mcp.disconnect.success"), {
+          this.telemetry?.emit("igniter.agent.mcp.disconnect.success", {
             level: "debug",
             attributes: {
               "ctx.agent.name": this.getName(),
@@ -1022,7 +873,7 @@ var IgniterAgentCore = class {
           });
         } catch (error) {
           const err = error instanceof Error ? error : new Error(String(error));
-          this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("mcp.disconnect.error"), {
+          this.telemetry?.emit("igniter.agent.mcp.disconnect.error", {
             level: "error",
             attributes: {
               "ctx.agent.name": this.getName(),
@@ -1037,7 +888,7 @@ var IgniterAgentCore = class {
         }
       }
       const durationMs = Date.now() - startTime;
-      this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("lifecycle.stop.success"), {
+      this.telemetry?.emit("igniter.agent.lifecycle.stop.success", {
         level: "debug",
         attributes: {
           ...lifecycleAttributes,
@@ -1051,7 +902,7 @@ var IgniterAgentCore = class {
       });
     } catch (error) {
       const err = error instanceof Error ? error : new Error(String(error));
-      this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("lifecycle.stop.error"), {
+      this.telemetry?.emit("igniter.agent.lifecycle.stop.error", {
         level: "error",
         attributes: {
           ...lifecycleAttributes,
@@ -1073,7 +924,7 @@ var IgniterAgentCore = class {
       "ctx.generation.streamed": false
     };
     this.logger?.debug("IgniterAgent.generate started", attributes);
-    this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("generation.generate.started"), {
+    this.telemetry?.emit("igniter.agent.generation.generate.started", {
       level: "debug",
       attributes
     });
@@ -1083,7 +934,7 @@ var IgniterAgentCore = class {
       );
       const result = await agent.generate(input);
       const durationMs = Date.now() - startTime;
-      this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("generation.generate.success"), {
+      this.telemetry?.emit("igniter.agent.generation.generate.success", {
         level: "debug",
         attributes: {
           ...attributes,
@@ -1097,7 +948,7 @@ var IgniterAgentCore = class {
       return result;
     } catch (error) {
       const err = error instanceof Error ? error : new Error(String(error));
-      this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("generation.generate.error"), {
+      this.telemetry?.emit("igniter.agent.generation.generate.error", {
         level: "error",
         attributes: {
           ...attributes,
@@ -1119,7 +970,7 @@ var IgniterAgentCore = class {
       "ctx.generation.streamed": true
     };
     this.logger?.debug("IgniterAgent.stream started", attributes);
-    this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("generation.stream.started"), {
+    this.telemetry?.emit("igniter.agent.generation.stream.started", {
       level: "debug",
       attributes
     });
@@ -1130,13 +981,13 @@ var IgniterAgentCore = class {
       const result = await agent.stream(input);
       const durationMs = Date.now() - startTime;
       const emitChunk = () => {
-        this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("generation.stream.chunk"), {
+        this.telemetry?.emit("igniter.agent.generation.stream.chunk", {
           level: "debug",
           attributes
         });
       };
       const wrapped = this.wrapStreamResult(result, emitChunk);
-      this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("generation.stream.success"), {
+      this.telemetry?.emit("igniter.agent.generation.stream.success", {
         level: "debug",
         attributes: {
           ...attributes,
@@ -1150,7 +1001,7 @@ var IgniterAgentCore = class {
       return wrapped;
     } catch (error) {
       const err = error instanceof Error ? error : new Error(String(error));
-      this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("generation.stream.error"), {
+      this.telemetry?.emit("igniter.agent.generation.stream.error", {
         level: "error",
         attributes: {
           ...attributes,
@@ -1211,7 +1062,7 @@ var IgniterAgentCore = class {
       execute: async (input, options) => {
         const startTime = Date.now();
         this.hooks.onToolCallStart?.(agentName, fullName, input);
-        this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("tool.execute.started"), {
+        this.telemetry?.emit("igniter.agent.tool.execute.started", {
           level: "debug",
           attributes: {
             "ctx.agent.name": agentName,
@@ -1228,7 +1079,7 @@ var IgniterAgentCore = class {
           const result = await execute(input, options);
           const durationMs = Date.now() - startTime;
           this.hooks.onToolCallEnd?.(agentName, fullName, result);
-          this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("tool.execute.success"), {
+          this.telemetry?.emit("igniter.agent.tool.execute.success", {
             level: "debug",
             attributes: {
               "ctx.agent.name": agentName,
@@ -1247,7 +1098,7 @@ var IgniterAgentCore = class {
         } catch (error) {
           const err = error instanceof Error ? error : new Error(String(error));
           this.hooks.onToolCallError?.(agentName, fullName, err);
-          this.telemetry?.emit(IgniterAgentTelemetryEvents.get.key("tool.execute.error"), {
+          this.telemetry?.emit("igniter.agent.tool.execute.error", {
             level: "error",
             attributes: {
               "ctx.agent.name": agentName,
@@ -1319,10 +1170,20 @@ var IgniterAgentCore = class {
         });
       }
     }
+    const serializableTools = {};
+    for (const [toolName, tool2] of Object.entries(tools)) {
+      if (!tool2) continue;
+      const { execute, ...toolWithoutExecute } = tool2;
+      serializableTools[toolName] = {
+        ...toolWithoutExecute,
+        execute: tool2.execute
+        // Add back the function reference, not serialized
+      };
+    }
     return new ToolLoopAgent({
       model: this._agent.model,
       instructions: this._agent.instructions ? this._agent.instructions.build(context) : "",
-      tools,
+      tools: serializableTools,
       callOptionsSchema: this._agent.schema
     });
   }