@agentium/transport 2.3.2 → 2.4.0

package/dist/index.cjs

@@ -1117,6 +1117,27 @@ var SCHEMAS = {
       sessionId: { type: "string" },
       userId: { type: "string" }
     }
+  },
+  CorrectionRequest: {
+    type: "object",
+    required: ["originalValue", "correctedValue"],
+    properties: {
+      originalValue: { type: "string", description: "What the agent produced (the wrong value)" },
+      correctedValue: { type: "string", description: "What it should have been" },
+      field: { type: "string", description: "The specific field/aspect corrected" },
+      reason: { type: "string", description: "Why the correction applies" },
+      entityKey: { type: "string", description: "Real-world entity this applies to, e.g. a vendor ID" },
+      runId: { type: "string", description: "Run that produced the corrected output" },
+      sessionId: { type: "string" },
+      userId: { type: "string" },
+      tenantId: { type: "string" },
+      scope: {
+        type: "string",
+        enum: ["user", "agent", "tenant", "global"],
+        description: "Visibility scope (default: agent)"
+      },
+      tags: { type: "array", items: { type: "string" } }
+    }
   }
 };
 function buildAgentDescription(agent) {
@@ -1328,6 +1349,43 @@ ${agentDesc}`,
           }
         }
       };
+      spec.paths[`${prefix}/agents/${name}/corrections`] = {
+        post: {
+          tags: ["Agents"],
+          summary: `Record a correction for agent: ${name}`,
+          description: `Record a human correction of agent **${name}**'s output. The correction is stored as vectorized knowledge and retrieved on future relevant runs so the mistake is not repeated. Requires the agent to have \`memory.corrections\` configured.`,
+          operationId: `recordCorrection_${name}`,
+          requestBody: {
+            required: true,
+            content: {
+              "application/json": {
+                schema: { $ref: "#/components/schemas/CorrectionRequest" }
+              }
+            }
+          },
+          responses: {
+            "201": {
+              description: "Correction recorded"
+            },
+            "400": {
+              description: "Bad request",
+              content: {
+                "application/json": {
+                  schema: { $ref: "#/components/schemas/Error" }
+                }
+              }
+            },
+            "404": {
+              description: "Corrections not enabled for this agent",
+              content: {
+                "application/json": {
+                  schema: { $ref: "#/components/schemas/Error" }
+                }
+              }
+            }
+          }
+        }
+      };
     }
   }
   if (routerOpts.teams && Object.keys(routerOpts.teams).length > 0) {
@@ -1651,6 +1709,45 @@ function createAgentRouter(opts) {
           }
         }
       });
+      router.post(`/agents/${name}/corrections`, async (req, res) => {
+        try {
+          const memory = agent.memory;
+          if (!memory || !memory.getCorrectionStore?.()) {
+            return res.status(404).json({
+              error: `Corrections are not enabled for agent "${name}". Configure memory.corrections with a vectorStore.`
+            });
+          }
+          const validated = validateBody(req.body, {
+            originalValue: "string",
+            correctedValue: "string",
+            field: "string?",
+            reason: "string?",
+            entityKey: "string?",
+            runId: "string?",
+            sessionId: "string?",
+            userId: "string?",
+            tenantId: "string?",
+            scope: "string?"
+          });
+          const correction = await memory.recordCorrection({
+            agentName: name,
+            runId: validated.runId,
+            sessionId: validated.sessionId,
+            field: validated.field,
+            originalValue: validated.originalValue,
+            correctedValue: validated.correctedValue,
+            reason: validated.reason,
+            entityKey: validated.entityKey,
+            tags: Array.isArray(req.body?.tags) ? req.body.tags : void 0,
+            scope: validated.scope,
+            userId: validated.userId,
+            tenantId: validated.tenantId
+          });
+          res.status(201).json(correction);
+        } catch (error) {
+          res.status(400).json({ error: error.message });
+        }
+      });
     }
   }
   if (opts.teams) {
@@ -1906,8 +2003,7 @@ function createAgentRouter(opts) {
     router.get("/agents/:name/checkpoints", async (req, res) => {
       const agent = reg.getAgent(req.params.name);
       if (!agent) return res.status(404).json({ error: `Agent "${req.params.name}" not found` });
-      const config = agent.config;
-      const checkpointMgr = config?._checkpointManager;
+      const checkpointMgr = agent.checkpointManager ?? agent.config?._checkpointManager;
       if (!checkpointMgr) return res.json([]);
       const runId = req.query.runId;
       if (!runId) return res.status(400).json({ error: "runId query param required" });
@@ -1921,8 +2017,7 @@ function createAgentRouter(opts) {
     router.post("/agents/:name/rollback/:checkpointId", async (req, res) => {
       const agent = reg.getAgent(req.params.name);
       if (!agent) return res.status(404).json({ error: `Agent "${req.params.name}" not found` });
-      const config = agent.config;
-      const checkpointMgr = config?._checkpointManager;
+      const checkpointMgr = agent.checkpointManager ?? agent.config?._checkpointManager;
       if (!checkpointMgr) return res.status(400).json({ error: "Checkpointing not enabled for this agent" });
       try {
         const checkpoint = await checkpointMgr.rollback(req.params.checkpointId);
@@ -1944,11 +2039,11 @@ function createAgentRouter(opts) {
 `);
       };
       for (const agent of reg.agents.values()) {
-        agent.eventBus.on("tool.approval.request", listener);
+        agent.eventBus?.on("tool.approval.request", listener);
       }
       req.on("close", () => {
         for (const agent of reg.agents.values()) {
-          agent.eventBus.off("tool.approval.request", listener);
+          agent.eventBus?.off("tool.approval.request", listener);
         }
       });
     });

package/dist/index.d.cts

@@ -1,4 +1,4 @@
-import { Agent, A2AAgentCard, MCPToolProviderConfig, ToolDef, MCPToolProvider, Registry, Servable, Team, Workflow, Toolkit, EventBus, VisionAgent, VoiceAgent } from '@agentium/core';
+import { Agent, A2AAgentCard, MCPToolProviderConfig, ToolDef, MCPToolProvider, Registry, Servable, ServableAgent, Team, Workflow, Toolkit, EventBus, VisionAgent, VoiceAgent } from '@agentium/core';
 
 interface A2AServerOptions {
     agents: Record<string, Agent>;
@@ -184,7 +184,7 @@ interface RouterOptions {
      * Each item is classified by its `.kind` and keyed by `.name`.
      */
     serve?: Servable[];
-    agents?: Record<string, Agent>;
+    agents?: Record<string, Agent | ServableAgent>;
     teams?: Record<string, Team>;
     workflows?: Record<string, Workflow<any>>;
     middleware?: any[];
@@ -435,7 +435,7 @@ interface GatewayOptions {
      * Each item is classified by its `.kind` and keyed by `.name`.
      */
     serve?: Servable[];
-    agents?: Record<string, Agent>;
+    agents?: Record<string, Agent | ServableAgent>;
     teams?: Record<string, Team>;
     io: any;
     namespace?: string;

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { Agent, A2AAgentCard, MCPToolProviderConfig, ToolDef, MCPToolProvider, Registry, Servable, Team, Workflow, Toolkit, EventBus, VisionAgent, VoiceAgent } from '@agentium/core';
+import { Agent, A2AAgentCard, MCPToolProviderConfig, ToolDef, MCPToolProvider, Registry, Servable, ServableAgent, Team, Workflow, Toolkit, EventBus, VisionAgent, VoiceAgent } from '@agentium/core';
 
 interface A2AServerOptions {
     agents: Record<string, Agent>;
@@ -184,7 +184,7 @@ interface RouterOptions {
      * Each item is classified by its `.kind` and keyed by `.name`.
      */
     serve?: Servable[];
-    agents?: Record<string, Agent>;
+    agents?: Record<string, Agent | ServableAgent>;
     teams?: Record<string, Team>;
     workflows?: Record<string, Workflow<any>>;
     middleware?: any[];
@@ -435,7 +435,7 @@ interface GatewayOptions {
      * Each item is classified by its `.kind` and keyed by `.name`.
      */
     serve?: Servable[];
-    agents?: Record<string, Agent>;
+    agents?: Record<string, Agent | ServableAgent>;
     teams?: Record<string, Team>;
     io: any;
     namespace?: string;

package/dist/index.js

@@ -1069,6 +1069,27 @@ var SCHEMAS = {
       sessionId: { type: "string" },
       userId: { type: "string" }
     }
+  },
+  CorrectionRequest: {
+    type: "object",
+    required: ["originalValue", "correctedValue"],
+    properties: {
+      originalValue: { type: "string", description: "What the agent produced (the wrong value)" },
+      correctedValue: { type: "string", description: "What it should have been" },
+      field: { type: "string", description: "The specific field/aspect corrected" },
+      reason: { type: "string", description: "Why the correction applies" },
+      entityKey: { type: "string", description: "Real-world entity this applies to, e.g. a vendor ID" },
+      runId: { type: "string", description: "Run that produced the corrected output" },
+      sessionId: { type: "string" },
+      userId: { type: "string" },
+      tenantId: { type: "string" },
+      scope: {
+        type: "string",
+        enum: ["user", "agent", "tenant", "global"],
+        description: "Visibility scope (default: agent)"
+      },
+      tags: { type: "array", items: { type: "string" } }
+    }
   }
 };
 function buildAgentDescription(agent) {
@@ -1280,6 +1301,43 @@ ${agentDesc}`,
           }
         }
       };
+      spec.paths[`${prefix}/agents/${name}/corrections`] = {
+        post: {
+          tags: ["Agents"],
+          summary: `Record a correction for agent: ${name}`,
+          description: `Record a human correction of agent **${name}**'s output. The correction is stored as vectorized knowledge and retrieved on future relevant runs so the mistake is not repeated. Requires the agent to have \`memory.corrections\` configured.`,
+          operationId: `recordCorrection_${name}`,
+          requestBody: {
+            required: true,
+            content: {
+              "application/json": {
+                schema: { $ref: "#/components/schemas/CorrectionRequest" }
+              }
+            }
+          },
+          responses: {
+            "201": {
+              description: "Correction recorded"
+            },
+            "400": {
+              description: "Bad request",
+              content: {
+                "application/json": {
+                  schema: { $ref: "#/components/schemas/Error" }
+                }
+              }
+            },
+            "404": {
+              description: "Corrections not enabled for this agent",
+              content: {
+                "application/json": {
+                  schema: { $ref: "#/components/schemas/Error" }
+                }
+              }
+            }
+          }
+        }
+      };
     }
   }
   if (routerOpts.teams && Object.keys(routerOpts.teams).length > 0) {
@@ -1603,6 +1661,45 @@ function createAgentRouter(opts) {
           }
         }
       });
+      router.post(`/agents/${name}/corrections`, async (req, res) => {
+        try {
+          const memory = agent.memory;
+          if (!memory || !memory.getCorrectionStore?.()) {
+            return res.status(404).json({
+              error: `Corrections are not enabled for agent "${name}". Configure memory.corrections with a vectorStore.`
+            });
+          }
+          const validated = validateBody(req.body, {
+            originalValue: "string",
+            correctedValue: "string",
+            field: "string?",
+            reason: "string?",
+            entityKey: "string?",
+            runId: "string?",
+            sessionId: "string?",
+            userId: "string?",
+            tenantId: "string?",
+            scope: "string?"
+          });
+          const correction = await memory.recordCorrection({
+            agentName: name,
+            runId: validated.runId,
+            sessionId: validated.sessionId,
+            field: validated.field,
+            originalValue: validated.originalValue,
+            correctedValue: validated.correctedValue,
+            reason: validated.reason,
+            entityKey: validated.entityKey,
+            tags: Array.isArray(req.body?.tags) ? req.body.tags : void 0,
+            scope: validated.scope,
+            userId: validated.userId,
+            tenantId: validated.tenantId
+          });
+          res.status(201).json(correction);
+        } catch (error) {
+          res.status(400).json({ error: error.message });
+        }
+      });
     }
   }
   if (opts.teams) {
@@ -1858,8 +1955,7 @@ function createAgentRouter(opts) {
     router.get("/agents/:name/checkpoints", async (req, res) => {
       const agent = reg.getAgent(req.params.name);
       if (!agent) return res.status(404).json({ error: `Agent "${req.params.name}" not found` });
-      const config = agent.config;
-      const checkpointMgr = config?._checkpointManager;
+      const checkpointMgr = agent.checkpointManager ?? agent.config?._checkpointManager;
       if (!checkpointMgr) return res.json([]);
       const runId = req.query.runId;
       if (!runId) return res.status(400).json({ error: "runId query param required" });
@@ -1873,8 +1969,7 @@ function createAgentRouter(opts) {
     router.post("/agents/:name/rollback/:checkpointId", async (req, res) => {
       const agent = reg.getAgent(req.params.name);
       if (!agent) return res.status(404).json({ error: `Agent "${req.params.name}" not found` });
-      const config = agent.config;
-      const checkpointMgr = config?._checkpointManager;
+      const checkpointMgr = agent.checkpointManager ?? agent.config?._checkpointManager;
       if (!checkpointMgr) return res.status(400).json({ error: "Checkpointing not enabled for this agent" });
       try {
         const checkpoint = await checkpointMgr.rollback(req.params.checkpointId);
@@ -1896,11 +1991,11 @@ function createAgentRouter(opts) {
 `);
       };
       for (const agent of reg.agents.values()) {
-        agent.eventBus.on("tool.approval.request", listener);
+        agent.eventBus?.on("tool.approval.request", listener);
       }
       req.on("close", () => {
         for (const agent of reg.agents.values()) {
-          agent.eventBus.off("tool.approval.request", listener);
+          agent.eventBus?.off("tool.approval.request", listener);
         }
       });
     });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agentium/transport",
-  "version": "2.3.2",
+  "version": "2.4.0",
   "description": "HTTP and WebSocket transport layer for Agentium agents",
   "license": "MIT",
   "repository": {
@@ -42,7 +42,7 @@
     "typescript": "^5.6.0"
   },
   "peerDependencies": {
-    "@agentium/core": "^2.3.2",
+    "@agentium/core": "^2.4.0",
     "@types/express": "^4.0.0 || ^5.0.0",
     "express": "^4.0.0 || ^5.0.0",
     "multer": ">=2.0.0",