@mastra/server 0.13.2-alpha.0 → 0.13.2-alpha.2

package/dist/{chunk-VOY6MGKA.js → chunk-43YZEULM.js}

@@ -14,7 +14,8 @@ __export(agents_exports, {
   getEvalsByAgentIdHandler: () => getEvalsByAgentIdHandler,
   getLiveEvalsByAgentIdHandler: () => getLiveEvalsByAgentIdHandler,
   streamGenerateHandler: () => streamGenerateHandler,
-  streamVNextGenerateHandler: () => streamVNextGenerateHandler
+  streamVNextGenerateHandler: () => streamVNextGenerateHandler,
+  updateAgentModelHandler: () => updateAgentModelHandler
 });
 async function getAgentsHandler({ mastra, runtimeContext }) {
   try {
@@ -295,7 +296,24 @@ function streamVNextGenerateHandler({
     return handleError(error, "error streaming agent response");
   }
 }
+function updateAgentModelHandler({
+  mastra,
+  agentId,
+  body
+}) {
+  try {
+    const agent = mastra.getAgent(agentId);
+    if (!agent) {
+      throw new HTTPException(404, { message: "Agent not found" });
+    }
+    const { model } = body;
+    agent.__updateModel({ model });
+    return { message: "Agent model updated" };
+  } catch (error) {
+    return handleError(error, "error updating agent model");
+  }
+}
 
-export { agents_exports, generateHandler, getAgentByIdHandler, getAgentsHandler, getEvalsByAgentIdHandler, getLiveEvalsByAgentIdHandler, streamGenerateHandler, streamVNextGenerateHandler };
-//# sourceMappingURL=chunk-VOY6MGKA.js.map
-//# sourceMappingURL=chunk-VOY6MGKA.js.map
+export { agents_exports, generateHandler, getAgentByIdHandler, getAgentsHandler, getEvalsByAgentIdHandler, getLiveEvalsByAgentIdHandler, streamGenerateHandler, streamVNextGenerateHandler, updateAgentModelHandler };
+//# sourceMappingURL=chunk-43YZEULM.js.map
+//# sourceMappingURL=chunk-43YZEULM.js.map

package/dist/chunk-43YZEULM.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/server/handlers/agents.ts"],"names":["acc","key"],"mappings":";;;;;;;;AAAA,IAAA,cAAA,GAAA;AAAA,QAAA,CAAA,cAAA,EAAA;AAAA,EAAA,eAAA,EAAA,MAAA,eAAA;AAAA,EAAA,mBAAA,EAAA,MAAA,mBAAA;AAAA,EAAA,gBAAA,EAAA,MAAA,gBAAA;AAAA,EAAA,wBAAA,EAAA,MAAA,wBAAA;AAAA,EAAA,4BAAA,EAAA,MAAA,4BAAA;AAAA,EAAA,qBAAA,EAAA,MAAA,qBAAA;AAAA,EAAA,0BAAA,EAAA,MAAA,0BAAA;AAAA,EAAA,uBAAA,EAAA,MAAA;AAAA,CAAA,CAAA;AAiBA,eAAsB,gBAAA,CAAiB,EAAE,MAAA,EAAQ,cAAA,EAAe,EAAiD;AAC/G,EAAA,IAAI;AACF,IAAA,MAAM,MAAA,GAAS,OAAO,SAAA,EAAU;AAEhC,IAAA,MAAM,mBAAA,GAAsB,MAAM,OAAA,CAAQ,GAAA;AAAA,MACxC,MAAA,CAAO,QAAQ,MAAM,CAAA,CAAE,IAAI,OAAO,CAAC,EAAA,EAAI,KAAK,CAAA,KAAM;AAChD,QAAA,MAAM,eAAe,MAAM,KAAA,CAAM,eAAA,CAAgB,EAAE,gBAAgB,CAAA;AACnE,QAAA,MAAM,QAAQ,MAAM,KAAA,CAAM,QAAA,CAAS,EAAE,gBAAgB,CAAA;AACrD,QAAA,MAAM,MAAM,MAAM,KAAA,CAAM,MAAA,CAAO,EAAE,gBAAgB,CAAA;AACjD,QAAA,MAAM,yBAAyB,MAAM,KAAA,CAAM,yBAAA,CAA0B,EAAE,gBAAgB,CAAA;AACvF,QAAA,MAAM,uBAAuB,MAAM,KAAA,CAAM,uBAAA,CAAwB,EAAE,gBAAgB,CAAA;AAEnF,QAAA,MAAM,oBAAA,GAAuB,MAAA,CAAO,OAAA,CAAQ,KAAA,IAAS,EAAE,CAAA,CAAE,MAAA,CAAY,CAAC,GAAA,EAAK,CAAC,GAAA,EAAK,IAAI,CAAA,KAAM;AACzF,UAAA,MAAM,KAAA,GAAQ,IAAA;AACd,UAAA,GAAA,CAAI,GAAG,CAAA,GAAI;AAAA,YACT,GAAG,KAAA;AAAA,YACH,WAAA,EAAa,MAAM,WAAA,GAAc,SAAA,CAAU,YAAgB,KAAA,CAAM,WAAW,CAAC,CAAA,GAAI,MAAA;AAAA,YACjF,YAAA,EAAc,MAAM,YAAA,GAAe,SAAA,CAAU,YAAgB,KAAA,CAAM,YAAY,CAAC,CAAA,GAAI;AAAA,WACtF;AACA,UAAA,OAAO,GAAA;AAAA,QACT,CAAA,EAAG,EAAE,CAAA;AAEL,QAAA,IAAI,2BAA2B,EAAC;AAEhC,QAAA,IAAI,kBAAkB,KAAA,EAAO;AAC3B,UAAA,MAAM,MAAA,GAAS,OAAO,SAAA,EAAU;AAChC,UAAA,IAAI;AACF,YAAA,MAAM,YAAY,MAAM,KAAA,CAAM,YAAA,CAAa,EAAE,gBAAgB,CAAA;AAC7D,YAAA,wBAAA,GAA2B,MAAA,CAAO,OAAA,CAAQ,SAAA,IAAa,EAAE,CAAA,CAAE,MAAA,CAAY,CAAC,GAAA,EAAK,CAAC,GAAA,EAAK,QAAQ,CAAA,KAAM;AAC/F,cAAA,OAAO;AAAA,gBACL,GAAG,GAAA;AAAA,gBACH,CAAC,GAAG,GAAG;AAAA,kBACL,MAAM,QAAA,CAAS;AAAA;AACjB,eACF;AAAA,YACF,CAAA,EAAG,EAAE,CAAA;AAAA,UACP,SAAS,KAAA,EAAO;AACd,YAAA,MAAA,CAAO,MAAM,mCAAA,EAAqC,EAAE,WAAW,KAAA,CAAM,IAAA,EAAM,OAAO,CAAA;AAAA,UACpF;AAAA,QACF;AAEA,QAAA,OAAO;AAAA,UACL,EAAA;AAAA,UACA,MAAM,KAAA,CAAM,IAAA;AAAA,UACZ,YAAA;AAAA,UACA,KAAA,EAAO,oBAAA;AAAA,UACP,SAAA,EAAW,wBAAA;AAAA,UACX,QAAA,EAAU,KAAK,WAAA,EAAY;AAAA,UAC3B,OAAA,EAAS,KAAK,UAAA,EAAW;AAAA,UACzB,sBAAA;AAAA,UACA;AAAA,SACF;AAAA,MACF,CAAC;AAAA,KACH;AAEA,IAAA,MAAM,gBAAA,GAAmB,oBAAoB,MAAA,CAE3C,CAAC,KAAK,EAAE,EAAA,EAAI,GAAG,IAAA,EAAK,KAAM;AAC1B,MAAA,GAAA,CAAI,EAAE,CAAA,GAAI,IAAA;AACV,MAAA,OAAO,GAAA;AAAA,IACT,CAAA,EAAG,EAAE,CAAA;AAEL,IAAA,OAAO,gBAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,sBAAsB,CAAA;AAAA,EAClD;AACF;AAEA,eAAsB,mBAAA,CAAoB;AAAA,EACxC,MAAA;AAAA,EACA,cAAA;AAAA,EACA,OAAA;AAAA,EACA,YAAA,GAAe;AACjB,CAAA,EAA0F;AACxF,EAAA,IAAI;AACF,IAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA;AAErC,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,mBAAmB,CAAA;AAAA,IAC7D;AAEA,IAAA,MAAM,QAAQ,MAAM,KAAA,CAAM,QAAA,CAAS,EAAE,gBAAgB,CAAA;AAErD,IAAA,MAAM,oBAAA,GAAuB,MAAA,CAAO,OAAA,CAAQ,KAAA,IAAS,EAAE,CAAA,CAAE,MAAA,CAAY,CAAC,GAAA,EAAK,CAAC,GAAA,EAAK,IAAI,CAAA,KAAM;AACzF,MAAA,MAAM,KAAA,GAAQ,IAAA;AACd,MAAA,GAAA,CAAI,GAAG,CAAA,GAAI;AAAA,QACT,GAAG,KAAA;AAAA,QACH,WAAA,EAAa,MAAM,WAAA,GAAc,SAAA,CAAU,YAAgB,KAAA,CAAM,WAAW,CAAC,CAAA,GAAI,MAAA;AAAA,QACjF,YAAA,EAAc,MAAM,YAAA,GAAe,SAAA,CAAU,YAAgB,KAAA,CAAM,YAAY,CAAC,CAAA,GAAI;AAAA,OACtF;AACA,MAAA,OAAO,GAAA;AAAA,IACT,CAAA,EAAG,EAAE,CAAA;AAEL,IAAA,IAAI,2BAA2B,EAAC;AAEhC,IAAA,IAAI,kBAAkB,KAAA,EAAO;AAC3B,MAAA,MAAM,MAAA,GAAS,OAAO,SAAA,EAAU;AAChC,MAAA,IAAI;AACF,QAAA,MAAM,YAAY,MAAM,KAAA,CAAM,YAAA,CAAa,EAAE,gBAAgB,CAAA;AAE7D,QAAA,wBAAA,GAA2B,MAAA,CAAO,OAAA,CAAQ,SAAA,IAAa,EAAE,CAAA,CAAE,MAAA,CAAY,CAAC,GAAA,EAAK,CAAC,GAAA,EAAK,QAAQ,CAAA,KAAM;AAC/F,UAAA,OAAO;AAAA,YACL,GAAG,GAAA;AAAA,YACH,CAAC,GAAG,GAAG;AAAA,cACL,MAAM,QAAA,CAAS,IAAA;AAAA,cACf,KAAA,EAAO,MAAA,CAAO,OAAA,CAAQ,QAAA,CAAS,KAAK,CAAA,CAAE,MAAA,CAAY,CAACA,IAAAA,EAAK,CAACC,IAAAA,EAAK,IAAI,CAAA,KAAM;AACtE,gBAAA,OAAO;AAAA,kBACL,GAAGD,IAAAA;AAAA,kBACH,CAACC,IAAG,GAAG;AAAA,oBACL,IAAI,IAAA,CAAK,EAAA;AAAA,oBACT,aAAa,IAAA,CAAK;AAAA;AACpB,iBACF;AAAA,cACF,CAAA,EAAG,EAAE;AAAA;AACP,WACF;AAAA,QACF,CAAA,EAAG,EAAE,CAAA;AAAA,MACP,SAAS,KAAA,EAAO;AACd,QAAA,MAAA,CAAO,MAAM,mCAAA,EAAqC,EAAE,WAAW,KAAA,CAAM,IAAA,EAAM,OAAO,CAAA;AAAA,MACpF;AAAA,IACF;AAEA,IAAA,IAAI,mBAAA,GAAsB,cAAA;AAC1B,IAAA,IAAI,YAAA,EAAc;AAChB,MAAA,mBAAA,GAAsB,IAAI,MAAM,cAAA,EAAgB;AAAA,QAC9C,GAAA,CAAI,QAAQ,IAAA,EAAM;AAChB,UAAA,IAAI,SAAS,KAAA,EAAO;AAClB,YAAA,OAAO,SAAU,GAAA,EAAa;AAC5B,cAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,GAAA,CAAI,GAAG,CAAA;AAC5B,cAAA,OAAO,KAAA,IAAS,IAAI,GAAG,CAAA,CAAA,CAAA;AAAA,YACzB,CAAA;AAAA,UACF;AACA,UAAA,OAAO,OAAA,CAAQ,GAAA,CAAI,MAAA,EAAQ,IAAI,CAAA;AAAA,QACjC;AAAA,OACD,CAAA;AAAA,IACH;AAEA,IAAA,MAAM,eAAe,MAAM,KAAA,CAAM,gBAAgB,EAAE,cAAA,EAAgB,qBAAqB,CAAA;AACxF,IAAA,MAAM,MAAM,MAAM,KAAA,CAAM,MAAA,CAAO,EAAE,gBAAgB,CAAA;AACjD,IAAA,MAAM,yBAAyB,MAAM,KAAA,CAAM,0BAA0B,EAAE,cAAA,EAAgB,qBAAqB,CAAA;AAC5G,IAAA,MAAM,uBAAuB,MAAM,KAAA,CAAM,wBAAwB,EAAE,cAAA,EAAgB,qBAAqB,CAAA;AAExG,IAAA,OAAO;AAAA,MACL,MAAM,KAAA,CAAM,IAAA;AAAA,MACZ,YAAA;AAAA,MACA,KAAA,EAAO,oBAAA;AAAA,MACP,SAAA,EAAW,wBAAA;AAAA,MACX,QAAA,EAAU,KAAK,WAAA,EAAY;AAAA,MAC3B,OAAA,EAAS,KAAK,UAAA,EAAW;AAAA,MACzB,sBAAA;AAAA,MACA;AAAA,KACF;AAAA,EACF,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,qBAAqB,CAAA;AAAA,EACjD;AACF;AAEA,eAAsB,wBAAA,CAAyB;AAAA,EAC7C,MAAA;AAAA,EACA,cAAA;AAAA,EACA;AACF,CAAA,EAAkE;AAChE,EAAA,IAAI;AACF,IAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA;AACrC,IAAA,MAAM,KAAA,GAAS,MAAM,MAAA,CAAO,UAAA,EAAW,EAAG,sBAAsB,KAAA,CAAM,IAAA,EAAM,MAAM,CAAA,IAAM,EAAC;AACzF,IAAA,MAAM,eAAe,MAAM,KAAA,CAAM,eAAA,CAAgB,EAAE,gBAAgB,CAAA;AACnE,IAAA,OAAO;AAAA,MACL,EAAA,EAAI,OAAA;AAAA,MACJ,MAAM,KAAA,CAAM,IAAA;AAAA,MACZ,YAAA;AAAA,MACA;AAAA,KACF;AAAA,EACF,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,0BAA0B,CAAA;AAAA,EACtD;AACF;AAEA,eAAsB,4BAAA,CAA6B;AAAA,EACjD,MAAA;AAAA,EACA,cAAA;AAAA,EACA;AACF,CAAA,EAAkE;AAChE,EAAA,IAAI;AACF,IAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA;AACrC,IAAA,MAAM,KAAA,GAAS,MAAM,MAAA,CAAO,UAAA,EAAW,EAAG,sBAAsB,KAAA,CAAM,IAAA,EAAM,MAAM,CAAA,IAAM,EAAC;AACzF,IAAA,MAAM,eAAe,MAAM,KAAA,CAAM,eAAA,CAAgB,EAAE,gBAAgB,CAAA;AAEnE,IAAA,OAAO;AAAA,MACL,EAAA,EAAI,OAAA;AAAA,MACJ,MAAM,KAAA,CAAM,IAAA;AAAA,MACZ,YAAA;AAAA,MACA;AAAA,KACF;AAAA,EACF,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,0BAA0B,CAAA;AAAA,EACtD;AACF;AAEA,eAAsB,eAAA,CAAgB;AAAA,EACpC,MAAA;AAAA,EACA,cAAA;AAAA,EACA,OAAA;AAAA,EACA,IAAA;AAAA,EACA;AACF,CAAA,EASG;AACD,EAAA,IAAI;AACF,IAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA;AAErC,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,mBAAmB,CAAA;AAAA,IAC7D;AAEA,IAAA,MAAM,EAAE,UAAU,UAAA,EAAY,UAAA,EAAY,gBAAgB,mBAAA,EAAqB,GAAG,MAAK,GAAI,IAAA;AAE3F,IAAA,MAAM,kBAAkB,UAAA,IAAc,UAAA;AAEtC,IAAA,MAAM,mBAAA,GAAsB,IAAI,cAAA,CAAwC;AAAA,MACtE,GAAG,KAAA,CAAM,IAAA,CAAK,cAAA,CAAe,SAAS,CAAA;AAAA,MACtC,GAAG,MAAM,IAAA,CAAK,MAAA,CAAO,QAAQ,mBAAA,IAAuB,EAAE,CAAC;AAAA,KACxD,CAAA;AAED,IAAA,YAAA,CAAa,EAAE,UAAU,CAAA;AAEzB,IAAA,MAAM,MAAA,GAAS,MAAM,KAAA,CAAM,QAAA,CAAS,QAAA,EAAU;AAAA,MAC5C,GAAG,IAAA;AAAA;AAAA,MAEH,UAAA,EAAY,eAAA;AAAA,MACZ,cAAA,EAAgB,mBAAA;AAAA,MAChB,MAAA,EAAQ;AAAA,KACT,CAAA;AAED,IAAA,OAAO,MAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,6BAA6B,CAAA;AAAA,EACzD;AACF;AAEA,eAAsB,qBAAA,CAAsB;AAAA,EAC1C,MAAA;AAAA,EACA,cAAA;AAAA,EACA,OAAA;AAAA,EACA,IAAA;AAAA,EACA;AACF,CAAA,EASkC;AAChC,EAAA,IAAI;AACF,IAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA;AAErC,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,mBAAmB,CAAA;AAAA,IAC7D;AAEA,IAAA,MAAM,EAAE,UAAU,UAAA,EAAY,UAAA,EAAY,gBAAgB,mBAAA,EAAqB,GAAG,MAAK,GAAI,IAAA;AAE3F,IAAA,MAAM,kBAAkB,UAAA,IAAc,UAAA;AAEtC,IAAA,MAAM,mBAAA,GAAsB,IAAI,cAAA,CAAwC;AAAA,MACtE,GAAG,KAAA,CAAM,IAAA,CAAK,cAAA,CAAe,SAAS,CAAA;AAAA,MACtC,GAAG,MAAM,IAAA,CAAK,MAAA,CAAO,QAAQ,mBAAA,IAAuB,EAAE,CAAC;AAAA,KACxD,CAAA;AAED,IAAA,YAAA,CAAa,EAAE,UAAU,CAAA;AAEzB,IAAA,MAAM,YAAA,GAAe,MAAM,KAAA,CAAM,MAAA,CAAO,QAAA,EAAU;AAAA,MAChD,GAAG,IAAA;AAAA;AAAA,MAEH,UAAA,EAAY,eAAA;AAAA,MACZ,cAAA,EAAgB,mBAAA;AAAA,MAChB,MAAA,EAAQ;AAAA,KACT,CAAA;AAED,IAAA,MAAM,cAAA,GAAiB,IAAA,CAAK,MAAA,GACxB,YAAA,CAAa,oBAAA,CAAqB;AAAA,MAChC,OAAA,EAAS;AAAA,QACP,mBAAA,EAAqB;AAAA;AACvB,KACD,CAAA,GACD,YAAA,CAAa,oBAAA,CAAqB;AAAA,MAChC,SAAA,EAAW,IAAA;AAAA,MACX,aAAA,EAAe,IAAA;AAAA,MACf,eAAA,EAAiB,CAAC,KAAA,KAAe;AAC/B,QAAA,OAAO,CAAA,iDAAA,EAAoD,iBAAiB,KAAA,GAAQ,KAAA,CAAM,UAAU,IAAA,CAAK,SAAA,CAAU,KAAK,CAAC,CAAA,CAAA;AAAA,MAC3H,CAAA;AAAA,MACA,OAAA,EAAS;AAAA,QACP,mBAAA,EAAqB;AAAA;AACvB,KACD,CAAA;AAEL,IAAA,OAAO,cAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,gCAAgC,CAAA;AAAA,EAC5D;AACF;AAEO,SAAS,0BAAA,CAA2B;AAAA,EACzC,MAAA;AAAA,EACA,cAAA;AAAA,EACA,OAAA;AAAA,EACA,IAAA;AAAA,EACA;AACF,CAAA,EAOqC;AACnC,EAAA,IAAI;AACF,IAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA;AAErC,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,mBAAmB,CAAA;AAAA,IAC7D;AAEA,IAAA,MAAM,EAAE,QAAA,EAAU,cAAA,EAAgB,mBAAA,EAAqB,GAAG,MAAK,GAAI,IAAA;AACnE,IAAA,MAAM,mBAAA,GAAsB,IAAI,cAAA,CAAwC;AAAA,MACtE,GAAG,KAAA,CAAM,IAAA,CAAK,cAAA,CAAe,SAAS,CAAA;AAAA,MACtC,GAAG,MAAM,IAAA,CAAK,MAAA,CAAO,QAAQ,mBAAA,IAAuB,EAAE,CAAC;AAAA,KACxD,CAAA;AAED,IAAA,YAAA,CAAa,EAAE,UAAU,CAAA;AAEzB,IAAA,MAAM,YAAA,GAAe,KAAA,CAAM,WAAA,CAAY,QAAA,EAAU;AAAA,MAC/C,GAAG,IAAA;AAAA,MACH,cAAA,EAAgB,mBAAA;AAAA,MAChB;AAAA,KACD,CAAA;AAED,IAAA,OAAO,YAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,gCAAgC,CAAA;AAAA,EAC5D;AACF;AAEO,SAAS,uBAAA,CAAwB;AAAA,EACtC,MAAA;AAAA,EACA,OAAA;AAAA,EACA;AACF,CAAA,EAKwB;AACtB,EAAA,IAAI;AACF,IAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA;AAErC,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,mBAAmB,CAAA;AAAA,IAC7D;AAEA,IAAA,MAAM,EAAE,OAAM,GAAI,IAAA;AAElB,IAAA,KAAA,CAAM,aAAA,CAAc,EAAE,KAAA,EAAO,CAAA;AAE7B,IAAA,OAAO,EAAE,SAAS,qBAAA,EAAsB;AAAA,EAC1C,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,4BAA4B,CAAA;AAAA,EACxD;AACF","file":"chunk-43YZEULM.js","sourcesContent":["import type { Agent, MastraLanguageModel } from '@mastra/core/agent';\nimport { RuntimeContext } from '@mastra/core/runtime-context';\nimport { stringify } from 'superjson';\nimport zodToJsonSchema from 'zod-to-json-schema';\nimport { HTTPException } from '../http-exception';\nimport type { Context } from '../types';\n\nimport { handleError } from './error';\nimport { validateBody } from './utils';\n\ntype GetBody<\n  T extends keyof Agent & { [K in keyof Agent]: Agent[K] extends (...args: any) => any ? K : never }[keyof Agent],\n> = {\n  messages: Parameters<Agent[T]>[0];\n} & Parameters<Agent[T]>[1];\n\n// Agent handlers\nexport async function getAgentsHandler({ mastra, runtimeContext }: Context & { runtimeContext: RuntimeContext }) {\n  try {\n    const agents = mastra.getAgents();\n\n    const serializedAgentsMap = await Promise.all(\n      Object.entries(agents).map(async ([id, agent]) => {\n        const instructions = await agent.getInstructions({ runtimeContext });\n        const tools = await agent.getTools({ runtimeContext });\n        const llm = await agent.getLLM({ runtimeContext });\n        const defaultGenerateOptions = await agent.getDefaultGenerateOptions({ runtimeContext });\n        const defaultStreamOptions = await agent.getDefaultStreamOptions({ runtimeContext });\n\n        const serializedAgentTools = Object.entries(tools || {}).reduce<any>((acc, [key, tool]) => {\n          const _tool = tool as any;\n          acc[key] = {\n            ..._tool,\n            inputSchema: _tool.inputSchema ? stringify(zodToJsonSchema(_tool.inputSchema)) : undefined,\n            outputSchema: _tool.outputSchema ? stringify(zodToJsonSchema(_tool.outputSchema)) : undefined,\n          };\n          return acc;\n        }, {});\n\n        let serializedAgentWorkflows = {};\n\n        if ('getWorkflows' in agent) {\n          const logger = mastra.getLogger();\n          try {\n            const workflows = await agent.getWorkflows({ runtimeContext });\n            serializedAgentWorkflows = Object.entries(workflows || {}).reduce<any>((acc, [key, workflow]) => {\n              return {\n                ...acc,\n                [key]: {\n                  name: workflow.name,\n                },\n              };\n            }, {});\n          } catch (error) {\n            logger.error('Error getting workflows for agent', { agentName: agent.name, error });\n          }\n        }\n\n        return {\n          id,\n          name: agent.name,\n          instructions,\n          tools: serializedAgentTools,\n          workflows: serializedAgentWorkflows,\n          provider: llm?.getProvider(),\n          modelId: llm?.getModelId(),\n          defaultGenerateOptions: defaultGenerateOptions as any,\n          defaultStreamOptions: defaultStreamOptions as any,\n        };\n      }),\n    );\n\n    const serializedAgents = serializedAgentsMap.reduce<\n      Record<string, Omit<(typeof serializedAgentsMap)[number], 'id'>>\n    >((acc, { id, ...rest }) => {\n      acc[id] = rest;\n      return acc;\n    }, {});\n\n    return serializedAgents;\n  } catch (error) {\n    return handleError(error, 'Error getting agents');\n  }\n}\n\nexport async function getAgentByIdHandler({\n  mastra,\n  runtimeContext,\n  agentId,\n  isPlayground = false,\n}: Context & { isPlayground?: boolean; runtimeContext: RuntimeContext; agentId: string }) {\n  try {\n    const agent = mastra.getAgent(agentId);\n\n    if (!agent) {\n      throw new HTTPException(404, { message: 'Agent not found' });\n    }\n\n    const tools = await agent.getTools({ runtimeContext });\n\n    const serializedAgentTools = Object.entries(tools || {}).reduce<any>((acc, [key, tool]) => {\n      const _tool = tool as any;\n      acc[key] = {\n        ..._tool,\n        inputSchema: _tool.inputSchema ? stringify(zodToJsonSchema(_tool.inputSchema)) : undefined,\n        outputSchema: _tool.outputSchema ? stringify(zodToJsonSchema(_tool.outputSchema)) : undefined,\n      };\n      return acc;\n    }, {});\n\n    let serializedAgentWorkflows = {};\n\n    if ('getWorkflows' in agent) {\n      const logger = mastra.getLogger();\n      try {\n        const workflows = await agent.getWorkflows({ runtimeContext });\n\n        serializedAgentWorkflows = Object.entries(workflows || {}).reduce<any>((acc, [key, workflow]) => {\n          return {\n            ...acc,\n            [key]: {\n              name: workflow.name,\n              steps: Object.entries(workflow.steps).reduce<any>((acc, [key, step]) => {\n                return {\n                  ...acc,\n                  [key]: {\n                    id: step.id,\n                    description: step.description,\n                  },\n                };\n              }, {}),\n            },\n          };\n        }, {});\n      } catch (error) {\n        logger.error('Error getting workflows for agent', { agentName: agent.name, error });\n      }\n    }\n\n    let proxyRuntimeContext = runtimeContext;\n    if (isPlayground) {\n      proxyRuntimeContext = new Proxy(runtimeContext, {\n        get(target, prop) {\n          if (prop === 'get') {\n            return function (key: string) {\n              const value = target.get(key);\n              return value ?? `<${key}>`;\n            };\n          }\n          return Reflect.get(target, prop);\n        },\n      });\n    }\n\n    const instructions = await agent.getInstructions({ runtimeContext: proxyRuntimeContext });\n    const llm = await agent.getLLM({ runtimeContext });\n    const defaultGenerateOptions = await agent.getDefaultGenerateOptions({ runtimeContext: proxyRuntimeContext });\n    const defaultStreamOptions = await agent.getDefaultStreamOptions({ runtimeContext: proxyRuntimeContext });\n\n    return {\n      name: agent.name,\n      instructions,\n      tools: serializedAgentTools,\n      workflows: serializedAgentWorkflows,\n      provider: llm?.getProvider(),\n      modelId: llm?.getModelId(),\n      defaultGenerateOptions: defaultGenerateOptions as any,\n      defaultStreamOptions: defaultStreamOptions as any,\n    };\n  } catch (error) {\n    return handleError(error, 'Error getting agent');\n  }\n}\n\nexport async function getEvalsByAgentIdHandler({\n  mastra,\n  runtimeContext,\n  agentId,\n}: Context & { runtimeContext: RuntimeContext; agentId: string }) {\n  try {\n    const agent = mastra.getAgent(agentId);\n    const evals = (await mastra.getStorage()?.getEvalsByAgentName?.(agent.name, 'test')) || [];\n    const instructions = await agent.getInstructions({ runtimeContext });\n    return {\n      id: agentId,\n      name: agent.name,\n      instructions,\n      evals,\n    };\n  } catch (error) {\n    return handleError(error, 'Error getting test evals');\n  }\n}\n\nexport async function getLiveEvalsByAgentIdHandler({\n  mastra,\n  runtimeContext,\n  agentId,\n}: Context & { runtimeContext: RuntimeContext; agentId: string }) {\n  try {\n    const agent = mastra.getAgent(agentId);\n    const evals = (await mastra.getStorage()?.getEvalsByAgentName?.(agent.name, 'live')) || [];\n    const instructions = await agent.getInstructions({ runtimeContext });\n\n    return {\n      id: agentId,\n      name: agent.name,\n      instructions,\n      evals,\n    };\n  } catch (error) {\n    return handleError(error, 'Error getting live evals');\n  }\n}\n\nexport async function generateHandler({\n  mastra,\n  runtimeContext,\n  agentId,\n  body,\n  abortSignal,\n}: Context & {\n  runtimeContext: RuntimeContext;\n  agentId: string;\n  body: GetBody<'generate'> & {\n    // @deprecated use resourceId\n    resourceid?: string;\n    runtimeContext?: Record<string, unknown>;\n  };\n  abortSignal?: AbortSignal;\n}) {\n  try {\n    const agent = mastra.getAgent(agentId);\n\n    if (!agent) {\n      throw new HTTPException(404, { message: 'Agent not found' });\n    }\n\n    const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;\n    // Use resourceId if provided, fall back to resourceid (deprecated)\n    const finalResourceId = resourceId ?? resourceid;\n\n    const finalRuntimeContext = new RuntimeContext<Record<string, unknown>>([\n      ...Array.from(runtimeContext.entries()),\n      ...Array.from(Object.entries(agentRuntimeContext ?? {})),\n    ]);\n\n    validateBody({ messages });\n\n    const result = await agent.generate(messages, {\n      ...rest,\n      // @ts-expect-error TODO fix types\n      resourceId: finalResourceId,\n      runtimeContext: finalRuntimeContext,\n      signal: abortSignal,\n    });\n\n    return result;\n  } catch (error) {\n    return handleError(error, 'Error generating from agent');\n  }\n}\n\nexport async function streamGenerateHandler({\n  mastra,\n  runtimeContext,\n  agentId,\n  body,\n  abortSignal,\n}: Context & {\n  runtimeContext: RuntimeContext;\n  agentId: string;\n  body: GetBody<'stream'> & {\n    // @deprecated use resourceId\n    resourceid?: string;\n    runtimeContext?: string;\n  };\n  abortSignal?: AbortSignal;\n}): Promise<Response | undefined> {\n  try {\n    const agent = mastra.getAgent(agentId);\n\n    if (!agent) {\n      throw new HTTPException(404, { message: 'Agent not found' });\n    }\n\n    const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;\n    // Use resourceId if provided, fall back to resourceid (deprecated)\n    const finalResourceId = resourceId ?? resourceid;\n\n    const finalRuntimeContext = new RuntimeContext<Record<string, unknown>>([\n      ...Array.from(runtimeContext.entries()),\n      ...Array.from(Object.entries(agentRuntimeContext ?? {})),\n    ]);\n\n    validateBody({ messages });\n\n    const streamResult = await agent.stream(messages, {\n      ...rest,\n      // @ts-expect-error TODO fix types\n      resourceId: finalResourceId,\n      runtimeContext: finalRuntimeContext,\n      signal: abortSignal,\n    });\n\n    const streamResponse = rest.output\n      ? streamResult.toTextStreamResponse({\n          headers: {\n            'Transfer-Encoding': 'chunked',\n          },\n        })\n      : streamResult.toDataStreamResponse({\n          sendUsage: true,\n          sendReasoning: true,\n          getErrorMessage: (error: any) => {\n            return `An error occurred while processing your request. ${error instanceof Error ? error.message : JSON.stringify(error)}`;\n          },\n          headers: {\n            'Transfer-Encoding': 'chunked',\n          },\n        });\n\n    return streamResponse;\n  } catch (error) {\n    return handleError(error, 'error streaming agent response');\n  }\n}\n\nexport function streamVNextGenerateHandler({\n  mastra,\n  runtimeContext,\n  agentId,\n  body,\n  abortSignal,\n}: Context & {\n  runtimeContext: RuntimeContext;\n  agentId: string;\n  body: GetBody<'streamVNext'> & {\n    runtimeContext?: string;\n  };\n  abortSignal?: AbortSignal;\n}): ReturnType<Agent['streamVNext']> {\n  try {\n    const agent = mastra.getAgent(agentId);\n\n    if (!agent) {\n      throw new HTTPException(404, { message: 'Agent not found' });\n    }\n\n    const { messages, runtimeContext: agentRuntimeContext, ...rest } = body;\n    const finalRuntimeContext = new RuntimeContext<Record<string, unknown>>([\n      ...Array.from(runtimeContext.entries()),\n      ...Array.from(Object.entries(agentRuntimeContext ?? {})),\n    ]);\n\n    validateBody({ messages });\n\n    const streamResult = agent.streamVNext(messages, {\n      ...rest,\n      runtimeContext: finalRuntimeContext,\n      abortSignal,\n    });\n\n    return streamResult;\n  } catch (error) {\n    return handleError(error, 'error streaming agent response');\n  }\n}\n\nexport function updateAgentModelHandler({\n  mastra,\n  agentId,\n  body,\n}: Context & {\n  agentId: string;\n  body: {\n    model: MastraLanguageModel;\n  };\n}): { message: string } {\n  try {\n    const agent = mastra.getAgent(agentId);\n\n    if (!agent) {\n      throw new HTTPException(404, { message: 'Agent not found' });\n    }\n\n    const { model } = body;\n\n    agent.__updateModel({ model });\n\n    return { message: 'Agent model updated' };\n  } catch (error) {\n    return handleError(error, 'error updating agent model');\n  }\n}\n"]}

package/dist/{chunk-TRWPUPGD.js → chunk-JNWFG7JP.js}

@@ -1,5 +1,5 @@
 import { __export } from './chunk-PZ5AY32C.js';
-import { A2AError } from '@mastra/core/a2a';
+import { MastraA2AError } from '@mastra/core/a2a';
 import { z } from 'zod';
 
 // src/server/handlers/a2a.ts
@@ -7,19 +7,19 @@ var a2a_exports = {};
 __export(a2a_exports, {
   getAgentCardByIdHandler: () => getAgentCardByIdHandler,
   getAgentExecutionHandler: () => getAgentExecutionHandler,
+  handleMessageSend: () => handleMessageSend,
+  handleMessageStream: () => handleMessageStream,
   handleTaskCancel: () => handleTaskCancel,
-  handleTaskGet: () => handleTaskGet,
-  handleTaskSend: () => handleTaskSend,
-  handleTaskSendSubscribe: () => handleTaskSendSubscribe
+  handleTaskGet: () => handleTaskGet
 });
 function normalizeError(error, reqId, taskId, logger) {
   let a2aError;
-  if (error instanceof A2AError) {
+  if (error instanceof MastraA2AError) {
     a2aError = error;
   } else if (error instanceof Error) {
-    a2aError = A2AError.internalError(error.message, { stack: error.stack });
+    a2aError = MastraA2AError.internalError(error.message, { stack: error.stack });
   } else {
-    a2aError = A2AError.internalError("An unknown error occurred.", error);
+    a2aError = MastraA2AError.internalError("An unknown error occurred.", error);
   }
   if (taskId && !a2aError.taskId) {
     a2aError.taskId = taskId;
@@ -37,7 +37,7 @@ function createErrorResponse(id, error) {
 }
 function createSuccessResponse(id, result) {
   if (!id) {
-    throw A2AError.internalError("Cannot create success response for null ID.");
+    throw MastraA2AError.internalError("Cannot create success response for null ID.");
   }
   return {
     jsonrpc: "2.0",
@@ -52,7 +52,7 @@ function convertToCoreMessage(message) {
   };
 }
 function convertToCoreMessagePart(part) {
-  switch (part.type) {
+  switch (part.kind) {
     case "text":
       return {
         type: "text",
@@ -61,7 +61,7 @@ function convertToCoreMessagePart(part) {
     case "file":
       return {
         type: "file",
-        data: new URL(part.file.uri),
+        data: "uri" in part.file ? new URL(part.file.uri) : part.file.bytes,
         mimeType: part.file.mimeType
       };
     case "data":
@@ -69,39 +69,15 @@ function convertToCoreMessagePart(part) {
   }
 }
 
-// src/server/a2a/store.ts
-var InMemoryTaskStore = class {
-  store = /* @__PURE__ */ new Map();
-  activeCancellations = /* @__PURE__ */ new Set();
-  async load({ agentId, taskId }) {
-    const entry = this.store.get(`${agentId}-${taskId}`);
-    if (!entry) {
-      return null;
-    }
-    return { task: { ...entry.task }, history: [...entry.history] };
-  }
-  async save({ agentId, data }) {
-    const key = `${agentId}-${data.task.id}`;
-    if (!data.task.id) {
-      throw new Error("Task ID is required");
-    }
-    this.store.set(key, {
-      task: { ...data.task },
-      history: [...data.history]
-    });
-  }
-};
-
 // src/server/a2a/tasks.ts
 function isTaskStatusUpdate(update) {
   return "state" in update && !("parts" in update);
 }
 function isArtifactUpdate(update) {
-  return "parts" in update;
+  return "kind" in update && update.kind === "artifact-update";
 }
-function applyUpdateToTaskAndHistory(current, update) {
-  let newTask = structuredClone(current.task);
-  let newHistory = structuredClone(current.history);
+function applyUpdateToTask(current, update) {
+  let newTask = structuredClone(current);
   if (isTaskStatusUpdate(update)) {
     newTask.status = {
       ...newTask.status,
@@ -110,58 +86,42 @@ function applyUpdateToTaskAndHistory(current, update) {
       // Apply updates
       timestamp: (/* @__PURE__ */ new Date()).toISOString()
     };
-    if (update.message?.role === "agent") {
-      newHistory.push(update.message);
-    }
   } else if (isArtifactUpdate(update)) {
     if (!newTask.artifacts) {
       newTask.artifacts = [];
     } else {
       newTask.artifacts = [...newTask.artifacts];
     }
-    const existingIndex = update.index ?? -1;
-    let replaced = false;
-    if (existingIndex >= 0 && existingIndex < newTask.artifacts.length) {
-      const existingArtifact = newTask.artifacts[existingIndex];
+    const artifact = update.artifact;
+    const existingIndex = newTask.artifacts.findIndex((a) => a.name === artifact.name);
+    const existingArtifact = newTask.artifacts[existingIndex];
+    if (existingArtifact) {
       if (update.append) {
         const appendedArtifact = JSON.parse(JSON.stringify(existingArtifact));
-        appendedArtifact.parts.push(...update.parts);
-        if (update.metadata) {
+        appendedArtifact.parts.push(...artifact.parts);
+        if (artifact.metadata) {
           appendedArtifact.metadata = {
             ...appendedArtifact.metadata || {},
-            ...update.metadata
+            ...artifact.metadata
           };
         }
-        if (update.lastChunk !== void 0) appendedArtifact.lastChunk = update.lastChunk;
-        if (update.description) appendedArtifact.description = update.description;
+        if (artifact.description) appendedArtifact.description = artifact.description;
         newTask.artifacts[existingIndex] = appendedArtifact;
-        replaced = true;
       } else {
-        newTask.artifacts[existingIndex] = { ...update };
-        replaced = true;
-      }
-    } else if (update.name) {
-      const namedIndex = newTask.artifacts.findIndex((a) => a.name === update.name);
-      if (namedIndex >= 0) {
-        newTask.artifacts[namedIndex] = { ...update };
-        replaced = true;
-      }
-    }
-    if (!replaced) {
-      newTask.artifacts.push({ ...update });
-      if (newTask.artifacts.some((a) => a.index !== void 0)) {
-        newTask.artifacts.sort((a, b) => (a.index ?? 0) - (b.index ?? 0));
+        newTask.artifacts[existingIndex] = { ...artifact };
       }
+    } else {
+      newTask.artifacts.push({ ...artifact });
     }
   }
-  return { task: newTask, history: newHistory };
+  return newTask;
 }
-async function loadOrCreateTaskAndHistory({
+async function loadOrCreateTask({
   agentId,
   taskId,
   taskStore,
   message,
-  sessionId,
+  contextId,
   metadata,
   logger
 }) {
@@ -169,47 +129,40 @@ async function loadOrCreateTaskAndHistory({
   if (!data) {
     const initialTask = {
       id: taskId,
-      sessionId,
+      contextId: contextId || crypto.randomUUID(),
       status: {
         state: "submitted",
         timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-        message: null
+        message: void 0
       },
       artifacts: [],
-      metadata
-    };
-    const initialData = {
-      task: initialTask,
-      history: [message]
+      history: [message],
+      metadata,
+      kind: "task"
     };
-    logger?.info(`[Task ${taskId}] Created new task and history.`);
-    await taskStore.save({ agentId, data: initialData });
-    return initialData;
+    logger?.info(`[Task ${taskId}] Created new task.`);
+    await taskStore.save({ agentId, data: initialTask });
+    return initialTask;
   }
-  logger?.info(`[Task ${taskId}] Loaded existing task and history.`);
-  let updatedData = {
-    task: data.task,
-    history: [...data.history, message]
-  };
-  const { status } = data.task;
+  logger?.info(`[Task ${taskId}] Loaded existing task.`);
+  let updatedData = data;
+  updatedData.history = [...data.history || [], message];
+  const { status } = data;
   const finalStates = ["completed", "failed", "canceled"];
   if (finalStates.includes(status.state)) {
     logger?.warn(`[Task ${taskId}] Received message for task in final state ${status.state}. Restarting.`);
-    updatedData = applyUpdateToTaskAndHistory(updatedData, {
+    updatedData = applyUpdateToTask(updatedData, {
       state: "submitted",
-      message: null
+      message: void 0
     });
   } else if (status.state === "input-required") {
     logger?.info(`[Task ${taskId}] Changing state from 'input-required' to 'working'.`);
-    updatedData = applyUpdateToTaskAndHistory(updatedData, { state: "working" });
+    updatedData = applyUpdateToTask(updatedData, { state: "working" });
   } else if (status.state === "working") {
     logger?.warn(`[Task ${taskId}] Received message while already 'working'. Proceeding.`);
   }
   await taskStore.save({ agentId, data: updatedData });
-  return {
-    task: { ...updatedData.task },
-    history: [...updatedData.history]
-  };
+  return updatedData;
 }
 function createTaskContext({
   task,
@@ -226,15 +179,22 @@ function createTaskContext({
 }
 
 // src/server/handlers/a2a.ts
-var taskSendParamsSchema = z.object({
-  id: z.string().min(1, "Invalid or missing task ID (params.id)."),
+var messageSendParamsSchema = z.object({
   message: z.object({
+    role: z.enum(["user", "agent"]),
     parts: z.array(
       z.object({
-        type: z.enum(["text"]),
+        kind: z.enum(["text"]),
         text: z.string()
       })
-    )
+    ),
+    kind: z.literal("message"),
+    messageId: z.string(),
+    contextId: z.string().optional(),
+    taskId: z.string().optional(),
+    referenceTaskIds: z.array(z.string()).optional(),
+    extensions: z.array(z.string()).optional(),
+    metadata: z.record(z.any()).optional()
   })
 });
 async function getAgentCardByIdHandler({
@@ -281,17 +241,17 @@ async function getAgentCardByIdHandler({
   };
   return agentCard;
 }
-function validateTaskSendParams(params) {
+function validateMessageSendParams(params) {
   try {
-    taskSendParamsSchema.parse(params);
+    messageSendParamsSchema.parse(params);
   } catch (error) {
     if (error instanceof z.ZodError) {
-      throw A2AError.invalidParams(error.errors[0].message);
+      throw MastraA2AError.invalidParams(error.errors[0].message);
     }
     throw error;
   }
 }
-async function handleTaskSend({
+async function handleMessageSend({
   requestId,
   params,
   taskStore,
@@ -300,20 +260,22 @@ async function handleTaskSend({
   logger,
   runtimeContext
 }) {
-  validateTaskSendParams(params);
-  const { id: taskId, message, sessionId, metadata } = params;
-  let currentData = await loadOrCreateTaskAndHistory({
+  validateMessageSendParams(params);
+  const { message, metadata } = params;
+  const { contextId } = message;
+  const taskId = message.taskId || crypto.randomUUID();
+  let currentData = await loadOrCreateTask({
     taskId,
     taskStore,
     agentId,
     message,
-    sessionId,
+    contextId,
     metadata
   });
   createTaskContext({
-    task: currentData.task,
+    task: currentData,
     userMessage: message,
-    history: currentData.history,
+    history: currentData.history || [],
     activeCancellations: taskStore.activeCancellations
   });
   try {
@@ -321,16 +283,18 @@ async function handleTaskSend({
       runId: taskId,
       runtimeContext
     });
-    currentData = applyUpdateToTaskAndHistory(currentData, {
+    currentData = applyUpdateToTask(currentData, {
       state: "completed",
       message: {
+        messageId: crypto.randomUUID(),
         role: "agent",
         parts: [
           {
-            type: "text",
+            kind: "text",
             text
           }
-        ]
+        ],
+        kind: "message"
       }
     });
     await taskStore.save({ agentId, data: currentData });
@@ -338,24 +302,26 @@ async function handleTaskSend({
     const failureStatusUpdate = {
       state: "failed",
       message: {
+        messageId: crypto.randomUUID(),
         role: "agent",
         parts: [
           {
-            type: "text",
+            kind: "text",
             text: `Handler failed: ${handlerError instanceof Error ? handlerError.message : String(handlerError)}`
           }
-        ]
+        ],
+        kind: "message"
       }
     };
-    currentData = applyUpdateToTaskAndHistory(currentData, failureStatusUpdate);
+    currentData = applyUpdateToTask(currentData, failureStatusUpdate);
     try {
       await taskStore.save({ agentId, data: currentData });
     } catch (saveError) {
-      logger?.error(`Failed to save task ${taskId} after handler error:`, saveError?.message);
+      logger?.error(`Failed to save task ${currentData.id} after handler error:`, saveError?.message);
     }
-    return normalizeError(handlerError, requestId, taskId, logger);
+    return normalizeError(handlerError, requestId, currentData.id, logger);
   }
-  return createSuccessResponse(requestId, currentData.task);
+  return createSuccessResponse(requestId, currentData);
 }
 async function handleTaskGet({
   requestId,
@@ -365,11 +331,11 @@ async function handleTaskGet({
 }) {
   const task = await taskStore.load({ agentId, taskId });
   if (!task) {
-    throw A2AError.taskNotFound(taskId);
+    throw MastraA2AError.taskNotFound(taskId);
   }
   return createSuccessResponse(requestId, task);
 }
-async function* handleTaskSendSubscribe({
+async function* handleMessageStream({
   requestId,
   params,
   taskStore,
@@ -381,13 +347,15 @@ async function* handleTaskSendSubscribe({
   yield createSuccessResponse(requestId, {
     state: "working",
     message: {
+      messageId: crypto.randomUUID(),
+      kind: "message",
       role: "agent",
-      parts: [{ type: "text", text: "Generating response..." }]
+      parts: [{ kind: "text", text: "Generating response..." }]
     }
   });
   let result;
   try {
-    result = await handleTaskSend({
+    result = await handleMessageSend({
       requestId,
       params,
       taskStore,
@@ -397,7 +365,7 @@ async function* handleTaskSendSubscribe({
       logger
     });
   } catch (err) {
-    if (!(err instanceof A2AError)) {
+    if (!(err instanceof MastraA2AError)) {
       throw err;
     }
     result = createErrorResponse(requestId, err.toJSONRPCError());
@@ -416,25 +384,27 @@ async function handleTaskCancel({
     taskId
   });
   if (!data) {
-    throw A2AError.taskNotFound(taskId);
+    throw MastraA2AError.taskNotFound(taskId);
   }
   const finalStates = ["completed", "failed", "canceled"];
-  if (finalStates.includes(data.task.status.state)) {
-    logger?.info(`Task ${taskId} already in final state ${data.task.status.state}, cannot cancel.`);
-    return createSuccessResponse(requestId, data.task);
+  if (finalStates.includes(data.status.state)) {
+    logger?.info(`Task ${taskId} already in final state ${data.status.state}, cannot cancel.`);
+    return createSuccessResponse(requestId, data);
   }
   taskStore.activeCancellations.add(taskId);
   const cancelUpdate = {
     state: "canceled",
     message: {
       role: "agent",
-      parts: [{ type: "text", text: "Task cancelled by request." }]
+      parts: [{ kind: "text", text: "Task cancelled by request." }],
+      kind: "message",
+      messageId: crypto.randomUUID()
     }
   };
-  data = applyUpdateToTaskAndHistory(data, cancelUpdate);
+  data = applyUpdateToTask(data, cancelUpdate);
   await taskStore.save({ agentId, data });
   taskStore.activeCancellations.delete(taskId);
-  return createSuccessResponse(requestId, data.task);
+  return createSuccessResponse(requestId, data);
 }
 async function getAgentExecutionHandler({
   requestId,
@@ -443,16 +413,16 @@ async function getAgentExecutionHandler({
   runtimeContext,
   method,
   params,
-  taskStore = new InMemoryTaskStore(),
+  taskStore,
   logger
 }) {
   const agent = mastra.getAgent(agentId);
   let taskId;
   try {
-    taskId = params.id;
+    taskId = "id" in params ? params.id : params.message?.taskId || "No task ID provided";
     switch (method) {
-      case "tasks/send": {
-        const result2 = await handleTaskSend({
+      case "message/send": {
+        const result2 = await handleMessageSend({
           requestId,
           params,
           taskStore,
@@ -462,8 +432,8 @@ async function getAgentExecutionHandler({
         });
         return result2;
       }
-      case "tasks/sendSubscribe":
-        const result = await handleTaskSendSubscribe({
+      case "message/stream":
+        const result = await handleMessageStream({
           requestId,
           taskStore,
           params,
@@ -491,16 +461,16 @@ async function getAgentExecutionHandler({
         return result2;
       }
       default:
-        throw A2AError.methodNotFound(method);
+        throw MastraA2AError.methodNotFound(method);
     }
   } catch (error) {
-    if (error instanceof A2AError && taskId && !error.taskId) {
+    if (error instanceof MastraA2AError && taskId && !error.taskId) {
       error.taskId = taskId;
     }
     return normalizeError(error, requestId, taskId, logger);
   }
 }
 
-export { a2a_exports, getAgentCardByIdHandler, getAgentExecutionHandler, handleTaskCancel, handleTaskGet, handleTaskSend, handleTaskSendSubscribe };
-//# sourceMappingURL=chunk-TRWPUPGD.js.map
-//# sourceMappingURL=chunk-TRWPUPGD.js.map
+export { a2a_exports, getAgentCardByIdHandler, getAgentExecutionHandler, handleMessageSend, handleMessageStream, handleTaskCancel, handleTaskGet };
+//# sourceMappingURL=chunk-JNWFG7JP.js.map
+//# sourceMappingURL=chunk-JNWFG7JP.js.map

package/dist/chunk-JNWFG7JP.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/server/handlers/a2a.ts","../src/server/a2a/protocol.ts","../src/server/a2a/tasks.ts"],"names":["MastraA2AError","result"],"mappings":";;;;;AAAA,IAAA,WAAA,GAAA;AAAA,QAAA,CAAA,WAAA,EAAA;AAAA,EAAA,uBAAA,EAAA,MAAA,uBAAA;AAAA,EAAA,wBAAA,EAAA,MAAA,wBAAA;AAAA,EAAA,iBAAA,EAAA,MAAA,iBAAA;AAAA,EAAA,mBAAA,EAAA,MAAA,mBAAA;AAAA,EAAA,gBAAA,EAAA,MAAA,gBAAA;AAAA,EAAA,aAAA,EAAA,MAAA;AAAA,CAAA,CAAA;ACMO,SAAS,cAAA,CACd,KAAA,EACA,KAAA,EACA,MAAA,EACA,MAAA,EACgC;AAChC,EAAA,IAAI,QAAA;AACJ,EAAA,IAAI,iBAAiB,cAAA,EAAgB;AACnC,IAAA,QAAA,GAAW,KAAA;AAAA,EACb,CAAA,MAAA,IAAW,iBAAiB,KAAA,EAAO;AAEjC,IAAA,QAAA,GAAW,cAAA,CAAe,cAAc,KAAA,CAAM,OAAA,EAAS,EAAE,KAAA,EAAO,KAAA,CAAM,OAAO,CAAA;AAAA,EAC/E,CAAA,MAAO;AAEL,IAAA,QAAA,GAAW,cAAA,CAAe,aAAA,CAAc,4BAAA,EAA8B,KAAK,CAAA;AAAA,EAC7E;AAGA,EAAA,IAAI,MAAA,IAAU,CAAC,QAAA,CAAS,MAAA,EAAQ;AAC9B,IAAA,QAAA,CAAS,MAAA,GAAS,MAAA;AAAA,EACpB;AAEA,EAAA,MAAA,EAAQ,KAAA,CAAM,mCAAmC,QAAA,CAAS,MAAA,IAAU,KAAK,CAAA,SAAA,EAAY,KAAA,IAAS,KAAK,CAAA,EAAA,CAAA,EAAM,QAAQ,CAAA;AAEjH,EAAA,OAAO,mBAAA,CAAoB,KAAA,EAAO,QAAA,CAAS,cAAA,EAAgB,CAAA;AAC7D;AAEO,SAAS,mBAAA,CACd,IACA,KAAA,EACgC;AAEhC,EAAA,OAAO;AAAA,IACL,OAAA,EAAS,KAAA;AAAA,IACT,EAAA;AAAA;AAAA,IACA;AAAA,GACF;AACF;AAEO,SAAS,qBAAA,CAAyB,IAA4B,MAAA,EAA+B;AAClG,EAAA,IAAI,CAAC,EAAA,EAAI;AAEP,IAAA,MAAM,cAAA,CAAe,cAAc,6CAA6C,CAAA;AAAA,EAClF;AAEA,EAAA,OAAO;AAAA,IACL,OAAA,EAAS,KAAA;AAAA,IACT,EAAA;AAAA,IACA;AAAA,GACF;AACF;AAEO,SAAS,qBAAqB,OAAA,EAA+B;AAClE,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,OAAA,CAAQ,IAAA,KAAS,MAAA,GAAS,MAAA,GAAS,WAAA;AAAA,IACzC,SAAS,OAAA,CAAQ,KAAA,CAAM,IAAI,CAAA,GAAA,KAAO,wBAAA,CAAyB,GAAG,CAAC;AAAA,GACjE;AACF;AAEA,SAAS,yBAAyB,IAAA,EAAY;AAC5C,EAAA,QAAQ,KAAK,IAAA;AAAM,IACjB,KAAK,MAAA;AACH,MAAA,OAAO;AAAA,QACL,IAAA,EAAM,MAAA;AAAA,QACN,MAAM,IAAA,CAAK;AAAA,OACb;AAAA,IACF,KAAK,MAAA;AACH,MAAA,OAAO;AAAA,QACL,IAAA,EAAM,MAAA;AAAA,QACN,IAAA,EAAM,KAAA,IAAS,IAAA,CAAK,IAAA,GAAO,IAAI,GAAA,CAAI,IAAA,CAAK,IAAA,CAAK,GAAG,CAAA,GAAI,IAAA,CAAK,IAAA,CAAK,KAAA;AAAA,QAC9D,QAAA,EAAU,KAAK,IAAA,CAAK;AAAA,OACtB;AAAA,IACF,KAAK,MAAA;AACH,MAAA,MAAM,IAAI,MAAM,+CAA+C,CAAA;AAAA;AAErE;;;ACrEA,SAAS,mBAAmB,MAAA,EAAuF;AACjH,EAAA,OAAO,OAAA,IAAW,MAAA,IAAU,EAAE,OAAA,IAAW,MAAA,CAAA;AAC3C;AAEA,SAAS,iBAAiB,MAAA,EAAiF;AACzG,EAAA,OAAO,MAAA,IAAU,MAAA,IAAU,MAAA,CAAO,IAAA,KAAS,iBAAA;AAC7C;AAEO,SAAS,iBAAA,CACd,SACA,MAAA,EACM;AACN,EAAA,IAAI,OAAA,GAAU,gBAAgB,OAAO,CAAA;AAErC,EAAA,IAAI,kBAAA,CAAmB,MAAM,CAAA,EAAG;AAE9B,IAAA,OAAA,CAAQ,MAAA,GAAS;AAAA,MACf,GAAG,OAAA,CAAQ,MAAA;AAAA;AAAA,MACX,GAAG,MAAA;AAAA;AAAA,MACH,SAAA,EAAA,iBAAW,IAAI,IAAA,EAAK,EAAE,WAAA;AAAY,KACpC;AAAA,EACF,CAAA,MAAA,IAAW,gBAAA,CAAiB,MAAM,CAAA,EAAG;AAEnC,IAAA,IAAI,CAAC,QAAQ,SAAA,EAAW;AACtB,MAAA,OAAA,CAAQ,YAAY,EAAC;AAAA,IACvB,CAAA,MAAO;AAEL,MAAA,OAAA,CAAQ,SAAA,GAAY,CAAC,GAAG,OAAA,CAAQ,SAAS,CAAA;AAAA,IAC3C;AAEA,IAAA,MAAM,WAAW,MAAA,CAAO,QAAA;AACxB,IAAA,MAAM,aAAA,GAAgB,QAAQ,SAAA,CAAU,SAAA,CAAU,OAAK,CAAA,CAAE,IAAA,KAAS,SAAS,IAAI,CAAA;AAC/E,IAAA,MAAM,gBAAA,GAAmB,OAAA,CAAQ,SAAA,CAAU,aAAa,CAAA;AAExD,IAAA,IAAI,gBAAA,EAAkB;AACpB,MAAA,IAAI,OAAO,MAAA,EAAQ;AAEjB,QAAA,MAAM,mBAAmB,IAAA,CAAK,KAAA,CAAM,IAAA,CAAK,SAAA,CAAU,gBAAgB,CAAC,CAAA;AACpE,QAAA,gBAAA,CAAiB,KAAA,CAAM,IAAA,CAAK,GAAG,QAAA,CAAS,KAAK,CAAA;AAC7C,QAAA,IAAI,SAAS,QAAA,EAAU;AACrB,UAAA,gBAAA,CAAiB,QAAA,GAAW;AAAA,YAC1B,GAAI,gBAAA,CAAiB,QAAA,IAAY,EAAC;AAAA,YAClC,GAAG,QAAA,CAAS;AAAA,WACd;AAAA,QACF;AACA,QAAA,IAAI,QAAA,CAAS,WAAA,EAAa,gBAAA,CAAiB,WAAA,GAAc,QAAA,CAAS,WAAA;AAClE,QAAA,OAAA,CAAQ,SAAA,CAAU,aAAa,CAAA,GAAI,gBAAA;AAAA,MACrC,CAAA,MAAO;AAEL,QAAA,OAAA,CAAQ,SAAA,CAAU,aAAa,CAAA,GAAI,EAAE,GAAG,QAAA,EAAS;AAAA,MACnD;AAAA,IACF,CAAA,MAAO;AACL,MAAA,OAAA,CAAQ,SAAA,CAAU,IAAA,CAAK,EAAE,GAAG,UAAU,CAAA;AAAA,IACxC;AAAA,EACF;AAEA,EAAA,OAAO,OAAA;AACT;AAEA,eAAsB,gBAAA,CAAiB;AAAA,EACrC,OAAA;AAAA,EACA,MAAA;AAAA,EACA,SAAA;AAAA,EACA,OAAA;AAAA,EACA,SAAA;AAAA,EACA,QAAA;AAAA,EACA;AACF,CAAA,EAQkB;AAChB,EAAA,MAAM,OAAO,MAAM,SAAA,CAAU,KAAK,EAAE,OAAA,EAAS,QAAQ,CAAA;AAGrD,EAAA,IAAI,CAAC,IAAA,EAAM;AACT,IAAA,MAAM,WAAA,GAAoB;AAAA,MACxB,EAAA,EAAI,MAAA;AAAA,MACJ,SAAA,EAAW,SAAA,IAAa,MAAA,CAAO,UAAA,EAAW;AAAA,MAC1C,MAAA,EAAQ;AAAA,QACN,KAAA,EAAO,WAAA;AAAA,QACP,SAAA,EAAA,iBAAW,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY;AAAA,QAClC,OAAA,EAAS;AAAA,OACX;AAAA,MACA,WAAW,EAAC;AAAA,MACZ,OAAA,EAAS,CAAC,OAAO,CAAA;AAAA,MACjB,QAAA;AAAA,MACA,IAAA,EAAM;AAAA,KACR;AAEA,IAAA,MAAA,EAAQ,IAAA,CAAK,CAAA,MAAA,EAAS,MAAM,CAAA,mBAAA,CAAqB,CAAA;AACjD,IAAA,MAAM,UAAU,IAAA,CAAK,EAAE,OAAA,EAAS,IAAA,EAAM,aAAa,CAAA;AAEnD,IAAA,OAAO,WAAA;AAAA,EACT;AAGA,EAAA,MAAA,EAAQ,IAAA,CAAK,CAAA,MAAA,EAAS,MAAM,CAAA,uBAAA,CAAyB,CAAA;AAGrD,EAAA,IAAI,WAAA,GAAc,IAAA;AAClB,EAAA,WAAA,CAAY,UAAU,CAAC,GAAI,KAAK,OAAA,IAAW,IAAK,OAAO,CAAA;AAGvD,EAAA,MAAM,EAAE,QAAO,GAAI,IAAA;AACnB,EAAA,MAAM,WAAA,GAA2B,CAAC,WAAA,EAAa,QAAA,EAAU,UAAU,CAAA;AAEnE,EAAA,IAAI,WAAA,CAAY,QAAA,CAAS,MAAA,CAAO,KAAK,CAAA,EAAG;AACtC,IAAA,MAAA,EAAQ,KAAK,CAAA,MAAA,EAAS,MAAM,CAAA,2CAAA,EAA8C,MAAA,CAAO,KAAK,CAAA,aAAA,CAAe,CAAA;AACrG,IAAA,WAAA,GAAc,kBAAkB,WAAA,EAAa;AAAA,MAC3C,KAAA,EAAO,WAAA;AAAA,MACP,OAAA,EAAS;AAAA,KACV,CAAA;AAAA,EACH,CAAA,MAAA,IAAW,MAAA,CAAO,KAAA,KAAU,gBAAA,EAAkB;AAC5C,IAAA,MAAA,EAAQ,IAAA,CAAK,CAAA,MAAA,EAAS,MAAM,CAAA,oDAAA,CAAsD,CAAA;AAClF,IAAA,WAAA,GAAc,iBAAA,CAAkB,WAAA,EAAa,EAAE,KAAA,EAAO,WAAW,CAAA;AAAA,EACnE,CAAA,MAAA,IAAW,MAAA,CAAO,KAAA,KAAU,SAAA,EAAW;AACrC,IAAA,MAAA,EAAQ,IAAA,CAAK,CAAA,MAAA,EAAS,MAAM,CAAA,uDAAA,CAAyD,CAAA;AAAA,EACvF;AAEA,EAAA,MAAM,UAAU,IAAA,CAAK,EAAE,OAAA,EAAS,IAAA,EAAM,aAAa,CAAA;AAEnD,EAAA,OAAO,WAAA;AACT;AAEO,SAAS,iBAAA,CAAkB;AAAA,EAChC,IAAA;AAAA,EACA,WAAA;AAAA,EACA,OAAA;AAAA,EACA;AACF,CAAA,EAKgB;AACd,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,gBAAgB,IAAI,CAAA;AAAA,IAC1B,WAAA;AAAA,IACA,OAAA,EAAS,gBAAgB,OAAO,CAAA;AAAA,IAChC,WAAA,EAAa,MAAM,mBAAA,CAAoB,GAAA,CAAI,KAAK,EAAE;AAAA,GACpD;AACF;;;AF5IA,IAAM,uBAAA,GAA0B,EAAE,MAAA,CAAO;AAAA,EACvC,OAAA,EAAS,EAAE,MAAA,CAAO;AAAA,IAChB,MAAM,CAAA,CAAE,IAAA,CAAK,CAAC,MAAA,EAAQ,OAAO,CAAC,CAAA;AAAA,IAC9B,OAAO,CAAA,CAAE,KAAA;AAAA,MACP,EAAE,MAAA,CAAO;AAAA,QACP,IAAA,EAAM,CAAA,CAAE,IAAA,CAAK,CAAC,MAAM,CAAC,CAAA;AAAA,QACrB,IAAA,EAAM,EAAE,MAAA;AAAO,OAChB;AAAA,KACH;AAAA,IACA,IAAA,EAAM,CAAA,CAAE,OAAA,CAAQ,SAAS,CAAA;AAAA,IACzB,SAAA,EAAW,EAAE,MAAA,EAAO;AAAA,IACpB,SAAA,EAAW,CAAA,CAAE,MAAA,EAAO,CAAE,QAAA,EAAS;AAAA,IAC/B,MAAA,EAAQ,CAAA,CAAE,MAAA,EAAO,CAAE,QAAA,EAAS;AAAA,IAC5B,kBAAkB,CAAA,CAAE,KAAA,CAAM,EAAE,MAAA,EAAQ,EAAE,QAAA,EAAS;AAAA,IAC/C,YAAY,CAAA,CAAE,KAAA,CAAM,EAAE,MAAA,EAAQ,EAAE,QAAA,EAAS;AAAA,IACzC,UAAU,CAAA,CAAE,MAAA,CAAO,EAAE,GAAA,EAAK,EAAE,QAAA;AAAS,GACtC;AACH,CAAC,CAAA;AAED,eAAsB,uBAAA,CAAwB;AAAA,EAC5C,MAAA;AAAA,EACA,OAAA;AAAA,EACA,YAAA,GAAe,QAAQ,OAAO,CAAA,CAAA;AAAA,EAC9B,QAAA,GAAW;AAAA,IACT,YAAA,EAAc,QAAA;AAAA,IACd,GAAA,EAAK;AAAA,GACP;AAAA,EACA,OAAA,GAAU,KAAA;AAAA,EACV;AACF,CAAA,EASuB;AACrB,EAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA;AAErC,EAAA,IAAI,CAAC,KAAA,EAAO;AACV,IAAA,MAAM,IAAI,KAAA,CAAM,CAAA,cAAA,EAAiB,OAAO,CAAA,UAAA,CAAY,CAAA;AAAA,EACtD;AAEA,EAAA,MAAM,CAAC,YAAA,EAAc,KAAK,CAAA,GAAI,MAAM,QAAQ,GAAA,CAAI;AAAA,IAC9C,KAAA,CAAM,eAAA,CAAgB,EAAE,cAAA,EAAgB,CAAA;AAAA,IACxC,KAAA,CAAM,QAAA,CAAS,EAAE,cAAA,EAAgB;AAAA,GAClC,CAAA;AAGD,EAAA,MAAM,SAAA,GAAuB;AAAA,IAC3B,IAAA,EAAM,MAAM,EAAA,IAAM,OAAA;AAAA,IAClB,WAAA,EAAa,YAAA;AAAA,IACb,GAAA,EAAK,YAAA;AAAA,IACL,QAAA;AAAA,IACA,OAAA;AAAA,IACA,YAAA,EAAc;AAAA,MACZ,SAAA,EAAW,IAAA;AAAA;AAAA,MACX,iBAAA,EAAmB,KAAA;AAAA,MACnB,sBAAA,EAAwB;AAAA,KAC1B;AAAA,IACA,iBAAA,EAAmB,CAAC,MAAM,CAAA;AAAA,IAC1B,kBAAA,EAAoB,CAAC,MAAM,CAAA;AAAA;AAAA,IAE3B,MAAA,EAAQ,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA,CAAE,IAAI,CAAC,CAAC,MAAA,EAAQ,IAAI,CAAA,MAAO;AAAA,MACrD,EAAA,EAAI,MAAA;AAAA,MACJ,IAAA,EAAM,MAAA;AAAA,MACN,WAAA,EAAa,IAAA,CAAK,WAAA,IAAe,CAAA,MAAA,EAAS,MAAM,CAAA,CAAA;AAAA;AAAA,MAEhD,IAAA,EAAM,CAAC,MAAM;AAAA,KACf,CAAE;AAAA,GACJ;AAEA,EAAA,OAAO,SAAA;AACT;AAEA,SAAS,0BAA0B,MAAA,EAA2B;AAC5D,EAAA,IAAI;AACF,IAAA,uBAAA,CAAwB,MAAM,MAAM,CAAA;AAAA,EACtC,SAAS,KAAA,EAAO;AACd,IAAA,IAAI,KAAA,YAAiB,EAAE,QAAA,EAAU;AAC/B,MAAA,MAAMA,eAAe,aAAA,CAAe,KAAA,CAAqB,MAAA,CAAO,CAAC,EAAG,OAAO,CAAA;AAAA,IAC7E;AAEA,IAAA,MAAM,KAAA;AAAA,EACR;AACF;AAEA,eAAsB,iBAAA,CAAkB;AAAA,EACtC,SAAA;AAAA,EACA,MAAA;AAAA,EACA,SAAA;AAAA,EACA,KAAA;AAAA,EACA,OAAA;AAAA,EACA,MAAA;AAAA,EACA;AACF,CAAA,EAQG;AACD,EAAA,yBAAA,CAA0B,MAAM,CAAA;AAEhC,EAAA,MAAM,EAAE,OAAA,EAAS,QAAA,EAAS,GAAI,MAAA;AAC9B,EAAA,MAAM,EAAE,WAAU,GAAI,OAAA;AACtB,EAAA,MAAM,MAAA,GAAS,OAAA,CAAQ,MAAA,IAAU,MAAA,CAAO,UAAA,EAAW;AAGnD,EAAA,IAAI,WAAA,GAAc,MAAM,gBAAA,CAAiB;AAAA,IACvC,MAAA;AAAA,IACA,SAAA;AAAA,IACA,OAAA;AAAA,IACA,OAAA;AAAA,IACA,SAAA;AAAA,IACA;AAAA,GACD,CAAA;AAGD,EAAgB,iBAAA,CAAkB;AAAA,IAChC,IAAA,EAAM,WAAA;AAAA,IACN,WAAA,EAAa,OAAA;AAAA,IACb,OAAA,EAAS,WAAA,CAAY,OAAA,IAAW,EAAC;AAAA,IACjC,qBAAqB,SAAA,CAAU;AAAA,GAChC;AAED,EAAA,IAAI;AACF,IAAA,MAAM,EAAE,IAAA,EAAK,GAAI,MAAM,KAAA,CAAM,SAAS,CAAC,oBAAA,CAAqB,OAAO,CAAC,CAAA,EAAG;AAAA,MACrE,KAAA,EAAO,MAAA;AAAA,MACP;AAAA,KACD,CAAA;AAED,IAAA,WAAA,GAAc,kBAAkB,WAAA,EAAa;AAAA,MAC3C,KAAA,EAAO,WAAA;AAAA,MACP,OAAA,EAAS;AAAA,QACP,SAAA,EAAW,OAAO,UAAA,EAAW;AAAA,QAC7B,IAAA,EAAM,OAAA;AAAA,QACN,KAAA,EAAO;AAAA,UACL;AAAA,YACE,IAAA,EAAM,MAAA;AAAA,YACN;AAAA;AACF,SACF;AAAA,QACA,IAAA,EAAM;AAAA;AACR,KACD,CAAA;AAED,IAAA,MAAM,UAAU,IAAA,CAAK,EAAE,OAAA,EAAS,IAAA,EAAM,aAAa,CAAA;AACpC,EACjB,SAAS,YAAA,EAAc;AAErB,IAAA,MAAM,mBAAA,GAAqD;AAAA,MACzD,KAAA,EAAO,QAAA;AAAA,MACP,OAAA,EAAS;AAAA,QACP,SAAA,EAAW,OAAO,UAAA,EAAW;AAAA,QAC7B,IAAA,EAAM,OAAA;AAAA,QACN,KAAA,EAAO;AAAA,UACL;AAAA,YACE,IAAA,EAAM,MAAA;AAAA,YACN,IAAA,EAAM,mBAAmB,YAAA,YAAwB,KAAA,GAAQ,aAAa,OAAA,GAAU,MAAA,CAAO,YAAY,CAAC,CAAA;AAAA;AACtG,SACF;AAAA,QACA,IAAA,EAAM;AAAA;AACR,KACF;AACA,IAAA,WAAA,GAAc,iBAAA,CAAkB,aAAa,mBAAmB,CAAA;AAEhE,IAAA,IAAI;AACF,MAAA,MAAM,UAAU,IAAA,CAAK,EAAE,OAAA,EAAS,IAAA,EAAM,aAAa,CAAA;AAAA,IACrD,SAAS,SAAA,EAAW;AAElB,MAAA,MAAA,EAAQ,MAAM,CAAA,oBAAA,EAAuB,WAAA,CAAY,EAAE,CAAA,qBAAA,CAAA,EAAyB,WAAW,OAAO,CAAA;AAAA,IAChG;AAEA,IAAA,OAAO,cAAA,CAAe,YAAA,EAAc,SAAA,EAAW,WAAA,CAAY,IAAI,MAAM,CAAA;AAAA,EACvE;AAGA,EAAA,OAAO,qBAAA,CAAsB,WAAW,WAAW,CAAA;AACrD;AAEA,eAAsB,aAAA,CAAc;AAAA,EAClC,SAAA;AAAA,EACA,SAAA;AAAA,EACA,OAAA;AAAA,EACA;AACF,CAAA,EAKG;AACD,EAAA,MAAM,OAAO,MAAM,SAAA,CAAU,KAAK,EAAE,OAAA,EAAS,QAAQ,CAAA;AAErD,EAAA,IAAI,CAAC,IAAA,EAAM;AACT,IAAA,MAAMA,cAAAA,CAAe,aAAa,MAAM,CAAA;AAAA,EAC1C;AAEA,EAAA,OAAO,qBAAA,CAAsB,WAAW,IAAI,CAAA;AAC9C;AAEA,gBAAuB,mBAAA,CAAoB;AAAA,EACzC,SAAA;AAAA,EACA,MAAA;AAAA,EACA,SAAA;AAAA,EACA,KAAA;AAAA,EACA,OAAA;AAAA,EACA,MAAA;AAAA,EACA;AACF,CAAA,EAQG;AACD,EAAA,MAAM,sBAAsB,SAAA,EAAW;AAAA,IACrC,KAAA,EAAO,SAAA;AAAA,IACP,OAAA,EAAS;AAAA,MACP,SAAA,EAAW,OAAO,UAAA,EAAW;AAAA,MAC7B,IAAA,EAAM,SAAA;AAAA,MACN,IAAA,EAAM,OAAA;AAAA,MACN,OAAO,CAAC,EAAE,MAAM,MAAA,EAAQ,IAAA,EAAM,0BAA0B;AAAA;AAC1D,GACD,CAAA;AAED,EAAA,IAAI,MAAA;AACJ,EAAA,IAAI;AACF,IAAA,MAAA,GAAS,MAAM,iBAAA,CAAkB;AAAA,MAC/B,SAAA;AAAA,MACA,MAAA;AAAA,MACA,SAAA;AAAA,MACA,KAAA;AAAA,MACA,OAAA;AAAA,MACA,cAAA;AAAA,MACA;AAAA,KACD,CAAA;AAAA,EACH,SAAS,GAAA,EAAK;AACZ,IAAA,IAAI,EAAE,eAAeA,cAAAA,CAAAA,EAAiB;AACpC,MAAA,MAAM,GAAA;AAAA,IACR;AAEA,IAAA,MAAA,GAAS,mBAAA,CAAoB,SAAA,EAAW,GAAA,CAAI,cAAA,EAAgB,CAAA;AAAA,EAC9D;AAEA,EAAA,MAAM,MAAA;AACR;AAEA,eAAsB,gBAAA,CAAiB;AAAA,EACrC,SAAA;AAAA,EACA,SAAA;AAAA,EACA,OAAA;AAAA,EACA,MAAA;AAAA,EACA;AACF,CAAA,EAMG;AAED,EAAA,IAAI,IAAA,GAAO,MAAM,SAAA,CAAU,IAAA,CAAK;AAAA,IAC9B,OAAA;AAAA,IACA;AAAA,GACD,CAAA;AAED,EAAA,IAAI,CAAC,IAAA,EAAM;AACT,IAAA,MAAMA,cAAAA,CAAe,aAAa,MAAM,CAAA;AAAA,EAC1C;AAGA,EAAA,MAAM,WAAA,GAA2B,CAAC,WAAA,EAAa,QAAA,EAAU,UAAU,CAAA;AAEnE,EAAA,IAAI,WAAA,CAAY,QAAA,CAAS,IAAA,CAAK,MAAA,CAAO,KAAK,CAAA,EAAG;AAC3C,IAAA,MAAA,EAAQ,KAAK,CAAA,KAAA,EAAQ,MAAM,2BAA2B,IAAA,CAAK,MAAA,CAAO,KAAK,CAAA,gBAAA,CAAkB,CAAA;AACzF,IAAA,OAAO,qBAAA,CAAsB,WAAW,IAAI,CAAA;AAAA,EAC9C;AAGA,EAAA,SAAA,CAAU,mBAAA,CAAoB,IAAI,MAAM,CAAA;AAGxC,EAAA,MAAM,YAAA,GAA8C;AAAA,IAClD,KAAA,EAAO,UAAA;AAAA,IACP,OAAA,EAAS;AAAA,MACP,IAAA,EAAM,OAAA;AAAA,MACN,OAAO,CAAC,EAAE,MAAM,MAAA,EAAQ,IAAA,EAAM,8BAA8B,CAAA;AAAA,MAC5D,IAAA,EAAM,SAAA;AAAA,MACN,SAAA,EAAW,OAAO,UAAA;AAAW;AAC/B,GACF;AAEA,EAAA,IAAA,GAAO,iBAAA,CAAkB,MAAM,YAAY,CAAA;AAG3C,EAAA,MAAM,SAAA,CAAU,IAAA,CAAK,EAAE,OAAA,EAAS,MAAM,CAAA;AAGtC,EAAA,SAAA,CAAU,mBAAA,CAAoB,OAAO,MAAM,CAAA;AAG3C,EAAA,OAAO,qBAAA,CAAsB,WAAW,IAAI,CAAA;AAC9C;AAEA,eAAsB,wBAAA,CAAyB;AAAA,EAC7C,SAAA;AAAA,EACA,MAAA;AAAA,EACA,OAAA;AAAA,EACA,cAAA;AAAA,EACA,MAAA;AAAA,EACA,MAAA;AAAA,EACA,SAAA;AAAA,EACA;AACF,CAAA,EAQiB;AACf,EAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA;AAErC,EAAA,IAAI,MAAA;AAEJ,EAAA,IAAI;AAGF,IAAA,MAAA,GAAS,QAAQ,MAAA,GAAS,MAAA,CAAO,EAAA,GAAK,MAAA,CAAO,SAAS,MAAA,IAAU,qBAAA;AAGhE,IAAA,QAAQ,MAAA;AAAQ,MACd,KAAK,cAAA,EAAgB;AACnB,QAAA,MAAMC,OAAAA,GAAS,MAAM,iBAAA,CAAkB;AAAA,UACrC,SAAA;AAAA,UACA,MAAA;AAAA,UACA,SAAA;AAAA,UACA,KAAA;AAAA,UACA,OAAA;AAAA,UACA;AAAA,SACD,CAAA;AACD,QAAA,OAAOA,OAAAA;AAAA,MACT;AAAA,MACA,KAAK,gBAAA;AACH,QAAA,MAAM,MAAA,GAAS,MAAM,mBAAA,CAAoB;AAAA,UACvC,SAAA;AAAA,UACA,SAAA;AAAA,UACA,MAAA;AAAA,UACA,KAAA;AAAA,UACA,OAAA;AAAA,UACA;AAAA,SACD,CAAA;AACD,QAAA,OAAO,MAAA;AAAA,MAET,KAAK,WAAA,EAAa;AAChB,QAAA,MAAMA,OAAAA,GAAS,MAAM,aAAA,CAAc;AAAA,UACjC,SAAA;AAAA,UACA,SAAA;AAAA,UACA,OAAA;AAAA,UACA;AAAA,SACD,CAAA;AAED,QAAA,OAAOA,OAAAA;AAAA,MACT;AAAA,MACA,KAAK,cAAA,EAAgB;AACnB,QAAA,MAAMA,OAAAA,GAAS,MAAM,gBAAA,CAAiB;AAAA,UACpC,SAAA;AAAA,UACA,SAAA;AAAA,UACA,OAAA;AAAA,UACA;AAAA,SACD,CAAA;AAED,QAAA,OAAOA,OAAAA;AAAA,MACT;AAAA,MACA;AACE,QAAA,MAAMD,cAAAA,CAAe,eAAe,MAAM,CAAA;AAAA;AAC9C,EACF,SAAS,KAAA,EAAO;AACd,IAAA,IAAI,KAAA,YAAiBA,cAAAA,IAAkB,MAAA,IAAU,CAAC,MAAM,MAAA,EAAQ;AAC9D,MAAA,KAAA,CAAM,MAAA,GAAS,MAAA;AAAA,IACjB;AAEA,IAAA,OAAO,cAAA,CAAe,KAAA,EAAO,SAAA,EAAW,MAAA,EAAQ,MAAM,CAAA;AAAA,EACxD;AACF","file":"chunk-JNWFG7JP.js","sourcesContent":["import { MastraA2AError } from '@mastra/core/a2a';\nimport type {\n  MessageSendParams,\n  TaskQueryParams,\n  TaskIdParams,\n  AgentCard,\n  TaskStatus,\n  TaskState,\n} from '@mastra/core/a2a';\nimport type { Agent } from '@mastra/core/agent';\nimport type { IMastraLogger } from '@mastra/core/logger';\nimport type { RuntimeContext } from '@mastra/core/runtime-context';\nimport { z } from 'zod';\nimport { convertToCoreMessage, normalizeError, createSuccessResponse, createErrorResponse } from '../a2a/protocol';\nimport type { InMemoryTaskStore } from '../a2a/store';\nimport { applyUpdateToTask, createTaskContext, loadOrCreateTask } from '../a2a/tasks';\nimport type { Context } from '../types';\n\nconst messageSendParamsSchema = z.object({\n  message: z.object({\n    role: z.enum(['user', 'agent']),\n    parts: z.array(\n      z.object({\n        kind: z.enum(['text']),\n        text: z.string(),\n      }),\n    ),\n    kind: z.literal('message'),\n    messageId: z.string(),\n    contextId: z.string().optional(),\n    taskId: z.string().optional(),\n    referenceTaskIds: z.array(z.string()).optional(),\n    extensions: z.array(z.string()).optional(),\n    metadata: z.record(z.any()).optional(),\n  }),\n});\n\nexport async function getAgentCardByIdHandler({\n  mastra,\n  agentId,\n  executionUrl = `/a2a/${agentId}`,\n  provider = {\n    organization: 'Mastra',\n    url: 'https://mastra.ai',\n  },\n  version = '1.0',\n  runtimeContext,\n}: Context & {\n  runtimeContext: RuntimeContext;\n  agentId: keyof ReturnType<typeof mastra.getAgents>;\n  executionUrl?: string;\n  version?: string;\n  provider?: {\n    organization: string;\n    url: string;\n  };\n}): Promise<AgentCard> {\n  const agent = mastra.getAgent(agentId);\n\n  if (!agent) {\n    throw new Error(`Agent with ID ${agentId} not found`);\n  }\n\n  const [instructions, tools] = await Promise.all([\n    agent.getInstructions({ runtimeContext }),\n    agent.getTools({ runtimeContext }),\n  ]);\n\n  // Extract agent information to create the AgentCard\n  const agentCard: AgentCard = {\n    name: agent.id || agentId,\n    description: instructions,\n    url: executionUrl,\n    provider,\n    version,\n    capabilities: {\n      streaming: true, // All agents support streaming\n      pushNotifications: false,\n      stateTransitionHistory: false,\n    },\n    defaultInputModes: ['text'],\n    defaultOutputModes: ['text'],\n    // Convert agent tools to skills format for A2A protocol\n    skills: Object.entries(tools).map(([toolId, tool]) => ({\n      id: toolId,\n      name: toolId,\n      description: tool.description || `Tool: ${toolId}`,\n      // Optional fields\n      tags: ['tool'],\n    })),\n  };\n\n  return agentCard;\n}\n\nfunction validateMessageSendParams(params: MessageSendParams) {\n  try {\n    messageSendParamsSchema.parse(params);\n  } catch (error) {\n    if (error instanceof z.ZodError) {\n      throw MastraA2AError.invalidParams((error as z.ZodError).errors[0]!.message);\n    }\n\n    throw error;\n  }\n}\n\nexport async function handleMessageSend({\n  requestId,\n  params,\n  taskStore,\n  agent,\n  agentId,\n  logger,\n  runtimeContext,\n}: {\n  requestId: string;\n  params: MessageSendParams;\n  taskStore: InMemoryTaskStore;\n  agent: Agent;\n  agentId: string;\n  logger?: IMastraLogger;\n  runtimeContext: RuntimeContext;\n}) {\n  validateMessageSendParams(params);\n\n  const { message, metadata } = params;\n  const { contextId } = message;\n  const taskId = message.taskId || crypto.randomUUID();\n\n  // Load or create task\n  let currentData = await loadOrCreateTask({\n    taskId,\n    taskStore,\n    agentId,\n    message,\n    contextId,\n    metadata,\n  });\n\n  // Use the new TaskContext definition, passing history\n  const context = createTaskContext({\n    task: currentData,\n    userMessage: message,\n    history: currentData.history || [],\n    activeCancellations: taskStore.activeCancellations,\n  });\n\n  try {\n    const { text } = await agent.generate([convertToCoreMessage(message)], {\n      runId: taskId,\n      runtimeContext,\n    });\n\n    currentData = applyUpdateToTask(currentData, {\n      state: 'completed',\n      message: {\n        messageId: crypto.randomUUID(),\n        role: 'agent',\n        parts: [\n          {\n            kind: 'text',\n            text: text,\n          },\n        ],\n        kind: 'message',\n      },\n    });\n\n    await taskStore.save({ agentId, data: currentData });\n    context.task = currentData;\n  } catch (handlerError) {\n    // If handler throws, apply 'failed' status, save, and rethrow\n    const failureStatusUpdate: Omit<TaskStatus, 'timestamp'> = {\n      state: 'failed',\n      message: {\n        messageId: crypto.randomUUID(),\n        role: 'agent',\n        parts: [\n          {\n            kind: 'text',\n            text: `Handler failed: ${handlerError instanceof Error ? handlerError.message : String(handlerError)}`,\n          },\n        ],\n        kind: 'message',\n      },\n    };\n    currentData = applyUpdateToTask(currentData, failureStatusUpdate);\n\n    try {\n      await taskStore.save({ agentId, data: currentData });\n    } catch (saveError) {\n      // @ts-expect-error saveError is an unknown error\n      logger?.error(`Failed to save task ${currentData.id} after handler error:`, saveError?.message);\n    }\n\n    return normalizeError(handlerError, requestId, currentData.id, logger); // Rethrow original error\n  }\n\n  // The loop finished, send the final task state\n  return createSuccessResponse(requestId, currentData);\n}\n\nexport async function handleTaskGet({\n  requestId,\n  taskStore,\n  agentId,\n  taskId,\n}: {\n  requestId: string;\n  taskStore: InMemoryTaskStore;\n  agentId: string;\n  taskId: string;\n}) {\n  const task = await taskStore.load({ agentId, taskId });\n\n  if (!task) {\n    throw MastraA2AError.taskNotFound(taskId);\n  }\n\n  return createSuccessResponse(requestId, task);\n}\n\nexport async function* handleMessageStream({\n  requestId,\n  params,\n  taskStore,\n  agent,\n  agentId,\n  logger,\n  runtimeContext,\n}: {\n  requestId: string;\n  params: MessageSendParams;\n  taskStore: InMemoryTaskStore;\n  agent: Agent;\n  agentId: string;\n  logger?: IMastraLogger;\n  runtimeContext: RuntimeContext;\n}) {\n  yield createSuccessResponse(requestId, {\n    state: 'working',\n    message: {\n      messageId: crypto.randomUUID(),\n      kind: 'message',\n      role: 'agent',\n      parts: [{ kind: 'text', text: 'Generating response...' }],\n    },\n  });\n\n  let result;\n  try {\n    result = await handleMessageSend({\n      requestId,\n      params,\n      taskStore,\n      agent,\n      agentId,\n      runtimeContext,\n      logger,\n    });\n  } catch (err) {\n    if (!(err instanceof MastraA2AError)) {\n      throw err;\n    }\n\n    result = createErrorResponse(requestId, err.toJSONRPCError());\n  }\n\n  yield result;\n}\n\nexport async function handleTaskCancel({\n  requestId,\n  taskStore,\n  agentId,\n  taskId,\n  logger,\n}: {\n  requestId: string;\n  taskStore: InMemoryTaskStore;\n  agentId: string;\n  taskId: string;\n  logger?: IMastraLogger;\n}) {\n  // Load task and history\n  let data = await taskStore.load({\n    agentId,\n    taskId,\n  });\n\n  if (!data) {\n    throw MastraA2AError.taskNotFound(taskId);\n  }\n\n  // Check if cancelable (not already in a final state)\n  const finalStates: TaskState[] = ['completed', 'failed', 'canceled'];\n\n  if (finalStates.includes(data.status.state)) {\n    logger?.info(`Task ${taskId} already in final state ${data.status.state}, cannot cancel.`);\n    return createSuccessResponse(requestId, data);\n  }\n\n  // Signal cancellation\n  taskStore.activeCancellations.add(taskId);\n\n  // Apply 'canceled' state update\n  const cancelUpdate: Omit<TaskStatus, 'timestamp'> = {\n    state: 'canceled',\n    message: {\n      role: 'agent',\n      parts: [{ kind: 'text', text: 'Task cancelled by request.' }],\n      kind: 'message',\n      messageId: crypto.randomUUID(),\n    },\n  };\n\n  data = applyUpdateToTask(data, cancelUpdate);\n\n  // Save the updated state\n  await taskStore.save({ agentId, data });\n\n  // Remove from active cancellations *after* saving\n  taskStore.activeCancellations.delete(taskId);\n\n  // Return the updated task object\n  return createSuccessResponse(requestId, data);\n}\n\nexport async function getAgentExecutionHandler({\n  requestId,\n  mastra,\n  agentId,\n  runtimeContext,\n  method,\n  params,\n  taskStore,\n  logger,\n}: Context & {\n  requestId: string;\n  runtimeContext: RuntimeContext;\n  agentId: string;\n  method: 'message/send' | 'message/stream' | 'tasks/get' | 'tasks/cancel';\n  params: MessageSendParams | TaskQueryParams | TaskIdParams;\n  taskStore: InMemoryTaskStore;\n  logger?: IMastraLogger;\n}): Promise<any> {\n  const agent = mastra.getAgent(agentId);\n\n  let taskId: string | undefined; // For error context\n\n  try {\n    // Attempt to get task ID early for error context. Cast params to any to access id.\n    // Proper validation happens within specific handlers.\n    taskId = 'id' in params ? params.id : params.message?.taskId || 'No task ID provided';\n\n    // 2. Route based on method\n    switch (method) {\n      case 'message/send': {\n        const result = await handleMessageSend({\n          requestId,\n          params: params as MessageSendParams,\n          taskStore,\n          agent,\n          agentId,\n          runtimeContext,\n        });\n        return result;\n      }\n      case 'message/stream':\n        const result = await handleMessageStream({\n          requestId,\n          taskStore,\n          params: params as MessageSendParams,\n          agent,\n          agentId,\n          runtimeContext,\n        });\n        return result;\n\n      case 'tasks/get': {\n        const result = await handleTaskGet({\n          requestId,\n          taskStore,\n          agentId,\n          taskId,\n        });\n\n        return result;\n      }\n      case 'tasks/cancel': {\n        const result = await handleTaskCancel({\n          requestId,\n          taskStore,\n          agentId,\n          taskId,\n        });\n\n        return result;\n      }\n      default:\n        throw MastraA2AError.methodNotFound(method);\n    }\n  } catch (error) {\n    if (error instanceof MastraA2AError && taskId && !error.taskId) {\n      error.taskId = taskId; // Add task ID context if missing\n    }\n\n    return normalizeError(error, requestId, taskId, logger);\n  }\n}\n","import { MastraA2AError } from '@mastra/core/a2a';\n\nimport type { JSONRPCError, JSONRPCResponse, Message, Part } from '@mastra/core/a2a';\nimport type { CoreMessage } from '@mastra/core/llm';\nimport type { IMastraLogger } from '@mastra/core/logger';\n\nexport function normalizeError(\n  error: any,\n  reqId: number | string | null,\n  taskId?: string,\n  logger?: IMastraLogger,\n): JSONRPCResponse<null, unknown> {\n  let a2aError: MastraA2AError;\n  if (error instanceof MastraA2AError) {\n    a2aError = error;\n  } else if (error instanceof Error) {\n    // Generic JS error\n    a2aError = MastraA2AError.internalError(error.message, { stack: error.stack });\n  } else {\n    // Unknown error type\n    a2aError = MastraA2AError.internalError('An unknown error occurred.', error);\n  }\n\n  // Ensure Task ID context is present if possible\n  if (taskId && !a2aError.taskId) {\n    a2aError.taskId = taskId;\n  }\n\n  logger?.error(`Error processing request (Task: ${a2aError.taskId ?? 'N/A'}, ReqID: ${reqId ?? 'N/A'}):`, a2aError);\n\n  return createErrorResponse(reqId, a2aError.toJSONRPCError());\n}\n\nexport function createErrorResponse(\n  id: number | string | null,\n  error: JSONRPCError<unknown>,\n): JSONRPCResponse<null, unknown> {\n  // For errors, ID should be the same as request ID, or null if that couldn't be determined\n  return {\n    jsonrpc: '2.0',\n    id: id, // Can be null if request ID was invalid/missing\n    error: error,\n  };\n}\n\nexport function createSuccessResponse<T>(id: number | string | null, result: T): JSONRPCResponse<T> {\n  if (!id) {\n    // This shouldn't happen for methods that expect a response, but safeguard\n    throw MastraA2AError.internalError('Cannot create success response for null ID.');\n  }\n\n  return {\n    jsonrpc: '2.0',\n    id: id,\n    result: result,\n  };\n}\n\nexport function convertToCoreMessage(message: Message): CoreMessage {\n  return {\n    role: message.role === 'user' ? 'user' : 'assistant',\n    content: message.parts.map(msg => convertToCoreMessagePart(msg)),\n  };\n}\n\nfunction convertToCoreMessagePart(part: Part) {\n  switch (part.kind) {\n    case 'text':\n      return {\n        type: 'text',\n        text: part.text,\n      } as const;\n    case 'file':\n      return {\n        type: 'file',\n        data: 'uri' in part.file ? new URL(part.file.uri) : part.file.bytes,\n        mimeType: part.file.mimeType!,\n      } as const;\n    case 'data':\n      throw new Error('Data parts are not supported in core messages');\n  }\n}\n","import type {\n  Message,\n  Task,\n  TaskState,\n  TaskStatus,\n  TaskContext,\n  TaskArtifactUpdateEvent,\n  Artifact,\n} from '@mastra/core/a2a';\nimport type { IMastraLogger } from '@mastra/core/logger';\nimport type { InMemoryTaskStore } from './store';\n\nfunction isTaskStatusUpdate(update: TaskStatus | TaskArtifactUpdateEvent): update is Omit<TaskStatus, 'timestamp'> {\n  return 'state' in update && !('parts' in update);\n}\n\nfunction isArtifactUpdate(update: TaskStatus | TaskArtifactUpdateEvent): update is TaskArtifactUpdateEvent {\n  return 'kind' in update && update.kind === 'artifact-update';\n}\n\nexport function applyUpdateToTask(\n  current: Task,\n  update: Omit<TaskStatus, 'timestamp'> | TaskArtifactUpdateEvent,\n): Task {\n  let newTask = structuredClone(current);\n\n  if (isTaskStatusUpdate(update)) {\n    // Merge status update\n    newTask.status = {\n      ...newTask.status, // Keep existing properties if not overwritten\n      ...update, // Apply updates\n      timestamp: new Date().toISOString(),\n    };\n  } else if (isArtifactUpdate(update)) {\n    // Handle artifact update\n    if (!newTask.artifacts) {\n      newTask.artifacts = [];\n    } else {\n      // Ensure we're working with a copy of the artifacts array\n      newTask.artifacts = [...newTask.artifacts];\n    }\n\n    const artifact = update.artifact;\n    const existingIndex = newTask.artifacts.findIndex(a => a.name === artifact.name);\n    const existingArtifact = newTask.artifacts[existingIndex];\n\n    if (existingArtifact) {\n      if (update.append) {\n        // Create a deep copy for modification to avoid mutating original\n        const appendedArtifact = JSON.parse(JSON.stringify(existingArtifact)) as Artifact;\n        appendedArtifact.parts.push(...artifact.parts);\n        if (artifact.metadata) {\n          appendedArtifact.metadata = {\n            ...(appendedArtifact.metadata || {}),\n            ...artifact.metadata,\n          };\n        }\n        if (artifact.description) appendedArtifact.description = artifact.description;\n        newTask.artifacts[existingIndex] = appendedArtifact; // Replace with appended version\n      } else {\n        // Overwrite artifact at index (with a copy of the update)\n        newTask.artifacts[existingIndex] = { ...artifact };\n      }\n    } else {\n      newTask.artifacts.push({ ...artifact });\n    }\n  }\n\n  return newTask;\n}\n\nexport async function loadOrCreateTask({\n  agentId,\n  taskId,\n  taskStore,\n  message,\n  contextId,\n  metadata,\n  logger,\n}: {\n  agentId: string;\n  taskId: string;\n  taskStore: InMemoryTaskStore;\n  message: Message;\n  contextId?: string;\n  metadata?: Record<string, unknown>;\n  logger?: IMastraLogger;\n}): Promise<Task> {\n  const data = await taskStore.load({ agentId, taskId });\n\n  // Create new task if none exists\n  if (!data) {\n    const initialTask: Task = {\n      id: taskId,\n      contextId: contextId || crypto.randomUUID(),\n      status: {\n        state: 'submitted',\n        timestamp: new Date().toISOString(),\n        message: undefined,\n      },\n      artifacts: [],\n      history: [message],\n      metadata: metadata,\n      kind: 'task',\n    };\n\n    logger?.info(`[Task ${taskId}] Created new task.`);\n    await taskStore.save({ agentId, data: initialTask });\n\n    return initialTask;\n  }\n\n  // Handle existing task\n  logger?.info(`[Task ${taskId}] Loaded existing task.`);\n\n  // Add message to history and prepare updated data\n  let updatedData = data;\n  updatedData.history = [...(data.history || []), message];\n\n  // Handle state transitions\n  const { status } = data;\n  const finalStates: TaskState[] = ['completed', 'failed', 'canceled'];\n\n  if (finalStates.includes(status.state)) {\n    logger?.warn(`[Task ${taskId}] Received message for task in final state ${status.state}. Restarting.`);\n    updatedData = applyUpdateToTask(updatedData, {\n      state: 'submitted',\n      message: undefined,\n    });\n  } else if (status.state === 'input-required') {\n    logger?.info(`[Task ${taskId}] Changing state from 'input-required' to 'working'.`);\n    updatedData = applyUpdateToTask(updatedData, { state: 'working' });\n  } else if (status.state === 'working') {\n    logger?.warn(`[Task ${taskId}] Received message while already 'working'. Proceeding.`);\n  }\n\n  await taskStore.save({ agentId, data: updatedData });\n\n  return updatedData;\n}\n\nexport function createTaskContext({\n  task,\n  userMessage,\n  history,\n  activeCancellations,\n}: {\n  task: Task;\n  userMessage: Message;\n  history: Message[];\n  activeCancellations: Set<string>;\n}): TaskContext {\n  return {\n    task: structuredClone(task),\n    userMessage,\n    history: structuredClone(history),\n    isCancelled: () => activeCancellations.has(task.id),\n  };\n}\n"]}