@mastra/core 0.1.27-alpha.28 → 0.1.27-alpha.29

package/dist/agent/index.d.ts

@@ -2,7 +2,7 @@ import { CoreAssistantMessage, CoreMessage, CoreToolMessage, CoreUserMessage, Us
 import { ZodSchema } from 'zod';
 import { Integration } from '../integration';
 import { LLM } from '../llm';
-import { ModelConfig, StructuredOutput } from '../llm/types';
+import { GenerateReturn, ModelConfig, StructuredOutput } from '../llm/types';
 import { Logger } from '../logger';
 import { MastraMemory } from '../memory';
 import { Run } from '../run/types';
@@ -89,6 +89,17 @@ export declare class Agent<TTools, TIntegrations extends Integration[] | undefin
             threadId: string;
         }) => Promise<void>;
     };
+    generate<S extends boolean = false, Z extends ZodSchema | undefined = undefined>(messages: string | string[] | CoreMessage[], { schema, stream, context, threadId: threadIdInFn, resourceid, maxSteps, onFinish, onStepFinish, runId, }?: {
+        resourceid?: string;
+        context?: CoreMessage[];
+        threadId?: string;
+        runId?: string;
+        stream?: S;
+        schema?: Z;
+        onFinish?: (result: string) => Promise<void> | void;
+        onStepFinish?: (step: string) => void;
+        maxSteps?: number;
+    }): Promise<GenerateReturn<S, Z>>;
     text({ messages, context, onStepFinish, maxSteps, threadId: threadIdInFn, resourceid, runId, }: {
         resourceid?: string;
         threadId?: string;

package/dist/core.cjs.development.js

@@ -1538,7 +1538,15 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
           case 0:
             _ref8 = _temp === void 0 ? {} : _temp, schema = _ref8.schema, stream = _ref8.stream, _ref8$maxSteps = _ref8.maxSteps, maxSteps = _ref8$maxSteps === void 0 ? 5 : _ref8$maxSteps, onFinish = _ref8.onFinish, onStepFinish = _ref8.onStepFinish, enabledTools = _ref8.enabledTools, convertedTools = _ref8.convertedTools, runId = _ref8.runId;
             if (Array.isArray(messages)) {
-              msgs = messages;
+              msgs = messages == null ? void 0 : messages.map(function (m) {
+                if (typeof m === 'string') {
+                  return {
+                    role: 'user',
+                    content: m
+                  };
+                }
+                return m;
+              });
             } else {
               msgs = [{
                 role: 'user',
@@ -2886,28 +2894,206 @@ exports.Agent = (_tools = /*#__PURE__*/_classPrivateFieldLooseKey("tools"), _log
       }()
     };
   };
-  _proto.text = /*#__PURE__*/function () {
-    var _text = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(_ref8) {
-      var messages, context, onStepFinish, _ref8$maxSteps, maxSteps, threadIdInFn, resourceid, runId, _this$__primitive, before, after, _yield$before, threadId, messageObjects, convertedTools, result;
-      return _regeneratorRuntime().wrap(function _callee9$(_context9) {
-        while (1) switch (_context9.prev = _context9.next) {
+  _proto.generate = /*#__PURE__*/function () {
+    var _generate = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee11(messages, _temp) {
+      var _ref8, schema, stream, context, threadIdInFn, resourceid, _ref8$maxSteps, maxSteps, _onFinish, onStepFinish, runId, messagesToUse, _this$__primitive, before, after, _yield$before, threadId, messageObjects, convertedTools, _result, result;
+      return _regeneratorRuntime().wrap(function _callee11$(_context11) {
+        while (1) switch (_context11.prev = _context11.next) {
           case 0:
-            messages = _ref8.messages, context = _ref8.context, onStepFinish = _ref8.onStepFinish, _ref8$maxSteps = _ref8.maxSteps, maxSteps = _ref8$maxSteps === void 0 ? 5 : _ref8$maxSteps, threadIdInFn = _ref8.threadId, resourceid = _ref8.resourceid, runId = _ref8.runId;
+            _ref8 = _temp === void 0 ? {} : _temp, schema = _ref8.schema, stream = _ref8.stream, context = _ref8.context, threadIdInFn = _ref8.threadId, resourceid = _ref8.resourceid, _ref8$maxSteps = _ref8.maxSteps, maxSteps = _ref8$maxSteps === void 0 ? 5 : _ref8$maxSteps, _onFinish = _ref8.onFinish, onStepFinish = _ref8.onStepFinish, runId = _ref8.runId;
+            messagesToUse = [];
+            if (Array.isArray(messages)) {
+              messagesToUse = messages.map(function (content) {
+                if (typeof content === 'string') {
+                  return content;
+                }
+                return content.content;
+              });
+            } else {
+              messagesToUse = [messages];
+            }
             _this$__primitive = this.__primitive({
-              messages: messages,
+              messages: messagesToUse,
               context: context,
               threadId: threadIdInFn,
               resourceid: resourceid,
               runId: runId
             }), before = _this$__primitive.before, after = _this$__primitive.after;
-            _context9.next = 4;
+            _context11.next = 6;
             return before();
-          case 4:
-            _yield$before = _context9.sent;
+          case 6:
+            _yield$before = _context11.sent;
             threadId = _yield$before.threadId;
             messageObjects = _yield$before.messageObjects;
             convertedTools = _yield$before.convertedTools;
-            _context9.next = 10;
+            if (!(stream && schema)) {
+              _context11.next = 12;
+              break;
+            }
+            return _context11.abrupt("return", this.llm.__streamObject({
+              messages: messageObjects,
+              structuredOutput: schema,
+              enabledTools: this.enabledTools,
+              convertedTools: convertedTools,
+              onStepFinish: onStepFinish,
+              onFinish: function () {
+                var _onFinish2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(result) {
+                  var res;
+                  return _regeneratorRuntime().wrap(function _callee9$(_context9) {
+                    while (1) switch (_context9.prev = _context9.next) {
+                      case 0:
+                        _context9.prev = 0;
+                        res = JSON.parse(result) || {};
+                        _context9.next = 4;
+                        return after({
+                          result: res,
+                          threadId: threadId
+                        });
+                      case 4:
+                        _context9.next = 9;
+                        break;
+                      case 6:
+                        _context9.prev = 6;
+                        _context9.t0 = _context9["catch"](0);
+                        console.error('Error saving memory on finish', _context9.t0);
+                      case 9:
+                        _onFinish == null || _onFinish(result);
+                      case 10:
+                      case "end":
+                        return _context9.stop();
+                    }
+                  }, _callee9, null, [[0, 6]]);
+                }));
+                function onFinish(_x10) {
+                  return _onFinish2.apply(this, arguments);
+                }
+                return onFinish;
+              }(),
+              maxSteps: maxSteps,
+              runId: runId
+            }));
+          case 12:
+            if (!stream) {
+              _context11.next = 14;
+              break;
+            }
+            return _context11.abrupt("return", this.llm.__stream({
+              messages: messageObjects,
+              enabledTools: this.enabledTools,
+              convertedTools: convertedTools,
+              onStepFinish: onStepFinish,
+              onFinish: function () {
+                var _onFinish3 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee10(result) {
+                  var res;
+                  return _regeneratorRuntime().wrap(function _callee10$(_context10) {
+                    while (1) switch (_context10.prev = _context10.next) {
+                      case 0:
+                        _context10.prev = 0;
+                        res = JSON.parse(result) || {};
+                        _context10.next = 4;
+                        return after({
+                          result: res,
+                          threadId: threadId
+                        });
+                      case 4:
+                        _context10.next = 9;
+                        break;
+                      case 6:
+                        _context10.prev = 6;
+                        _context10.t0 = _context10["catch"](0);
+                        console.error('Error saving memory on finish', _context10.t0);
+                      case 9:
+                        _onFinish == null || _onFinish(result);
+                      case 10:
+                      case "end":
+                        return _context10.stop();
+                    }
+                  }, _callee10, null, [[0, 6]]);
+                }));
+                function onFinish(_x11) {
+                  return _onFinish3.apply(this, arguments);
+                }
+                return onFinish;
+              }(),
+              maxSteps: maxSteps,
+              runId: runId
+            }));
+          case 14:
+            if (!schema) {
+              _context11.next = 21;
+              break;
+            }
+            _context11.next = 17;
+            return this.llm.__textObject({
+              messages: messageObjects,
+              structuredOutput: schema,
+              enabledTools: this.enabledTools,
+              convertedTools: convertedTools,
+              onStepFinish: onStepFinish,
+              maxSteps: maxSteps,
+              runId: runId
+            });
+          case 17:
+            _result = _context11.sent;
+            _context11.next = 20;
+            return after({
+              result: _result,
+              threadId: threadId
+            });
+          case 20:
+            return _context11.abrupt("return", _result);
+          case 21:
+            _context11.next = 23;
+            return this.llm.__text({
+              messages: messageObjects,
+              enabledTools: this.enabledTools,
+              convertedTools: convertedTools,
+              onStepFinish: onStepFinish,
+              maxSteps: maxSteps,
+              runId: runId
+            });
+          case 23:
+            result = _context11.sent;
+            _context11.next = 26;
+            return after({
+              result: result,
+              threadId: threadId
+            });
+          case 26:
+            return _context11.abrupt("return", result);
+          case 27:
+          case "end":
+            return _context11.stop();
+        }
+      }, _callee11, this);
+    }));
+    function generate(_x8, _x9) {
+      return _generate.apply(this, arguments);
+    }
+    return generate;
+  }();
+  _proto.text = /*#__PURE__*/function () {
+    var _text = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee12(_ref9) {
+      var messages, context, onStepFinish, _ref9$maxSteps, maxSteps, threadIdInFn, resourceid, runId, _this$__primitive2, before, after, _yield$before2, threadId, messageObjects, convertedTools, result;
+      return _regeneratorRuntime().wrap(function _callee12$(_context12) {
+        while (1) switch (_context12.prev = _context12.next) {
+          case 0:
+            messages = _ref9.messages, context = _ref9.context, onStepFinish = _ref9.onStepFinish, _ref9$maxSteps = _ref9.maxSteps, maxSteps = _ref9$maxSteps === void 0 ? 5 : _ref9$maxSteps, threadIdInFn = _ref9.threadId, resourceid = _ref9.resourceid, runId = _ref9.runId;
+            _this$__primitive2 = this.__primitive({
+              messages: messages,
+              context: context,
+              threadId: threadIdInFn,
+              resourceid: resourceid,
+              runId: runId
+            }), before = _this$__primitive2.before, after = _this$__primitive2.after;
+            _context12.next = 4;
+            return before();
+          case 4:
+            _yield$before2 = _context12.sent;
+            threadId = _yield$before2.threadId;
+            messageObjects = _yield$before2.messageObjects;
+            convertedTools = _yield$before2.convertedTools;
+            _context12.next = 10;
             return this.llm.__text({
               messages: messageObjects,
               enabledTools: this.enabledTools,
@@ -2917,47 +3103,47 @@ exports.Agent = (_tools = /*#__PURE__*/_classPrivateFieldLooseKey("tools"), _log
               runId: runId
             });
           case 10:
-            result = _context9.sent;
-            _context9.next = 13;
+            result = _context12.sent;
+            _context12.next = 13;
             return after({
               result: result,
               threadId: threadId
             });
           case 13:
-            return _context9.abrupt("return", result);
+            return _context12.abrupt("return", result);
           case 14:
           case "end":
-            return _context9.stop();
+            return _context12.stop();
         }
-      }, _callee9, this);
+      }, _callee12, this);
     }));
-    function text(_x8) {
+    function text(_x12) {
       return _text.apply(this, arguments);
     }
     return text;
   }();
   _proto.textObject = /*#__PURE__*/function () {
-    var _textObject = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee10(_ref9) {
-      var messages, context, structuredOutput, onStepFinish, _ref9$maxSteps, maxSteps, threadIdInFn, resourceid, runId, _this$__primitive2, before, after, _yield$before2, threadId, messageObjects, convertedTools, result;
-      return _regeneratorRuntime().wrap(function _callee10$(_context10) {
-        while (1) switch (_context10.prev = _context10.next) {
+    var _textObject = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee13(_ref10) {
+      var messages, context, structuredOutput, onStepFinish, _ref10$maxSteps, maxSteps, threadIdInFn, resourceid, runId, _this$__primitive3, before, after, _yield$before3, threadId, messageObjects, convertedTools, result;
+      return _regeneratorRuntime().wrap(function _callee13$(_context13) {
+        while (1) switch (_context13.prev = _context13.next) {
           case 0:
-            messages = _ref9.messages, context = _ref9.context, structuredOutput = _ref9.structuredOutput, onStepFinish = _ref9.onStepFinish, _ref9$maxSteps = _ref9.maxSteps, maxSteps = _ref9$maxSteps === void 0 ? 5 : _ref9$maxSteps, threadIdInFn = _ref9.threadId, resourceid = _ref9.resourceid, runId = _ref9.runId;
-            _this$__primitive2 = this.__primitive({
+            messages = _ref10.messages, context = _ref10.context, structuredOutput = _ref10.structuredOutput, onStepFinish = _ref10.onStepFinish, _ref10$maxSteps = _ref10.maxSteps, maxSteps = _ref10$maxSteps === void 0 ? 5 : _ref10$maxSteps, threadIdInFn = _ref10.threadId, resourceid = _ref10.resourceid, runId = _ref10.runId;
+            _this$__primitive3 = this.__primitive({
               messages: messages,
               context: context,
               threadId: threadIdInFn,
               resourceid: resourceid,
               runId: runId
-            }), before = _this$__primitive2.before, after = _this$__primitive2.after;
-            _context10.next = 4;
+            }), before = _this$__primitive3.before, after = _this$__primitive3.after;
+            _context13.next = 4;
             return before();
           case 4:
-            _yield$before2 = _context10.sent;
-            threadId = _yield$before2.threadId;
-            messageObjects = _yield$before2.messageObjects;
-            convertedTools = _yield$before2.convertedTools;
-            _context10.next = 10;
+            _yield$before3 = _context13.sent;
+            threadId = _yield$before3.threadId;
+            messageObjects = _yield$before3.messageObjects;
+            convertedTools = _yield$before3.convertedTools;
+            _context13.next = 10;
             return this.llm.__textObject({
               messages: messageObjects,
               structuredOutput: structuredOutput,
@@ -2968,81 +3154,81 @@ exports.Agent = (_tools = /*#__PURE__*/_classPrivateFieldLooseKey("tools"), _log
               runId: runId
             });
           case 10:
-            result = _context10.sent;
-            _context10.next = 13;
+            result = _context13.sent;
+            _context13.next = 13;
             return after({
               result: result,
               threadId: threadId
             });
           case 13:
-            return _context10.abrupt("return", result);
+            return _context13.abrupt("return", result);
           case 14:
           case "end":
-            return _context10.stop();
+            return _context13.stop();
         }
-      }, _callee10, this);
+      }, _callee13, this);
     }));
-    function textObject(_x9) {
+    function textObject(_x13) {
       return _textObject.apply(this, arguments);
     }
     return textObject;
   }();
   _proto.stream = /*#__PURE__*/function () {
-    var _stream = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee12(_ref10) {
-      var messages, context, onStepFinish, _onFinish, _ref10$maxSteps, maxSteps, threadIdInFn, resourceid, runId, _this$__primitive3, before, after, _yield$before3, threadId, messageObjects, convertedTools;
-      return _regeneratorRuntime().wrap(function _callee12$(_context12) {
-        while (1) switch (_context12.prev = _context12.next) {
+    var _stream = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee15(_ref11) {
+      var messages, context, onStepFinish, _onFinish4, _ref11$maxSteps, maxSteps, threadIdInFn, resourceid, runId, _this$__primitive4, before, after, _yield$before4, threadId, messageObjects, convertedTools;
+      return _regeneratorRuntime().wrap(function _callee15$(_context15) {
+        while (1) switch (_context15.prev = _context15.next) {
           case 0:
-            messages = _ref10.messages, context = _ref10.context, onStepFinish = _ref10.onStepFinish, _onFinish = _ref10.onFinish, _ref10$maxSteps = _ref10.maxSteps, maxSteps = _ref10$maxSteps === void 0 ? 5 : _ref10$maxSteps, threadIdInFn = _ref10.threadId, resourceid = _ref10.resourceid, runId = _ref10.runId;
-            _this$__primitive3 = this.__primitive({
+            messages = _ref11.messages, context = _ref11.context, onStepFinish = _ref11.onStepFinish, _onFinish4 = _ref11.onFinish, _ref11$maxSteps = _ref11.maxSteps, maxSteps = _ref11$maxSteps === void 0 ? 5 : _ref11$maxSteps, threadIdInFn = _ref11.threadId, resourceid = _ref11.resourceid, runId = _ref11.runId;
+            _this$__primitive4 = this.__primitive({
               messages: messages,
               context: context,
               threadId: threadIdInFn,
               resourceid: resourceid,
               runId: runId
-            }), before = _this$__primitive3.before, after = _this$__primitive3.after;
-            _context12.next = 4;
+            }), before = _this$__primitive4.before, after = _this$__primitive4.after;
+            _context15.next = 4;
             return before();
           case 4:
-            _yield$before3 = _context12.sent;
-            threadId = _yield$before3.threadId;
-            messageObjects = _yield$before3.messageObjects;
-            convertedTools = _yield$before3.convertedTools;
-            return _context12.abrupt("return", this.llm.__stream({
+            _yield$before4 = _context15.sent;
+            threadId = _yield$before4.threadId;
+            messageObjects = _yield$before4.messageObjects;
+            convertedTools = _yield$before4.convertedTools;
+            return _context15.abrupt("return", this.llm.__stream({
               messages: messageObjects,
               enabledTools: this.enabledTools,
               convertedTools: convertedTools,
               onStepFinish: onStepFinish,
               onFinish: function () {
-                var _onFinish2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee11(result) {
+                var _onFinish5 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee14(result) {
                   var res;
-                  return _regeneratorRuntime().wrap(function _callee11$(_context11) {
-                    while (1) switch (_context11.prev = _context11.next) {
+                  return _regeneratorRuntime().wrap(function _callee14$(_context14) {
+                    while (1) switch (_context14.prev = _context14.next) {
                       case 0:
-                        _context11.prev = 0;
+                        _context14.prev = 0;
                         res = JSON.parse(result) || {};
-                        _context11.next = 4;
+                        _context14.next = 4;
                         return after({
                           result: res,
                           threadId: threadId
                         });
                       case 4:
-                        _context11.next = 9;
+                        _context14.next = 9;
                         break;
                       case 6:
-                        _context11.prev = 6;
-                        _context11.t0 = _context11["catch"](0);
-                        console.error('Error saving memory on finish', _context11.t0);
+                        _context14.prev = 6;
+                        _context14.t0 = _context14["catch"](0);
+                        console.error('Error saving memory on finish', _context14.t0);
                       case 9:
-                        _onFinish == null || _onFinish(result);
+                        _onFinish4 == null || _onFinish4(result);
                       case 10:
                       case "end":
-                        return _context11.stop();
+                        return _context14.stop();
                     }
-                  }, _callee11, null, [[0, 6]]);
+                  }, _callee14, null, [[0, 6]]);
                 }));
-                function onFinish(_x11) {
-                  return _onFinish2.apply(this, arguments);
+                function onFinish(_x15) {
+                  return _onFinish5.apply(this, arguments);
                 }
                 return onFinish;
               }(),
@@ -3051,72 +3237,72 @@ exports.Agent = (_tools = /*#__PURE__*/_classPrivateFieldLooseKey("tools"), _log
             }));
           case 9:
           case "end":
-            return _context12.stop();
+            return _context15.stop();
         }
-      }, _callee12, this);
+      }, _callee15, this);
     }));
-    function stream(_x10) {
+    function stream(_x14) {
       return _stream.apply(this, arguments);
     }
     return stream;
   }();
   _proto.streamObject = /*#__PURE__*/function () {
-    var _streamObject = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee14(_ref11) {
-      var messages, context, structuredOutput, onStepFinish, _onFinish3, _ref11$maxSteps, maxSteps, threadIdInFn, resourceid, runId, _this$__primitive4, before, after, _yield$before4, threadId, messageObjects, convertedTools;
-      return _regeneratorRuntime().wrap(function _callee14$(_context14) {
-        while (1) switch (_context14.prev = _context14.next) {
+    var _streamObject = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee17(_ref12) {
+      var messages, context, structuredOutput, onStepFinish, _onFinish6, _ref12$maxSteps, maxSteps, threadIdInFn, resourceid, runId, _this$__primitive5, before, after, _yield$before5, threadId, messageObjects, convertedTools;
+      return _regeneratorRuntime().wrap(function _callee17$(_context17) {
+        while (1) switch (_context17.prev = _context17.next) {
           case 0:
-            messages = _ref11.messages, context = _ref11.context, structuredOutput = _ref11.structuredOutput, onStepFinish = _ref11.onStepFinish, _onFinish3 = _ref11.onFinish, _ref11$maxSteps = _ref11.maxSteps, maxSteps = _ref11$maxSteps === void 0 ? 5 : _ref11$maxSteps, threadIdInFn = _ref11.threadId, resourceid = _ref11.resourceid, runId = _ref11.runId;
-            _this$__primitive4 = this.__primitive({
+            messages = _ref12.messages, context = _ref12.context, structuredOutput = _ref12.structuredOutput, onStepFinish = _ref12.onStepFinish, _onFinish6 = _ref12.onFinish, _ref12$maxSteps = _ref12.maxSteps, maxSteps = _ref12$maxSteps === void 0 ? 5 : _ref12$maxSteps, threadIdInFn = _ref12.threadId, resourceid = _ref12.resourceid, runId = _ref12.runId;
+            _this$__primitive5 = this.__primitive({
               messages: messages,
               context: context,
               threadId: threadIdInFn,
               resourceid: resourceid,
               runId: runId
-            }), before = _this$__primitive4.before, after = _this$__primitive4.after;
-            _context14.next = 4;
+            }), before = _this$__primitive5.before, after = _this$__primitive5.after;
+            _context17.next = 4;
             return before();
           case 4:
-            _yield$before4 = _context14.sent;
-            threadId = _yield$before4.threadId;
-            messageObjects = _yield$before4.messageObjects;
-            convertedTools = _yield$before4.convertedTools;
-            return _context14.abrupt("return", this.llm.__streamObject({
+            _yield$before5 = _context17.sent;
+            threadId = _yield$before5.threadId;
+            messageObjects = _yield$before5.messageObjects;
+            convertedTools = _yield$before5.convertedTools;
+            return _context17.abrupt("return", this.llm.__streamObject({
               messages: messageObjects,
               structuredOutput: structuredOutput,
               enabledTools: this.enabledTools,
               convertedTools: convertedTools,
               onStepFinish: onStepFinish,
               onFinish: function () {
-                var _onFinish4 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee13(result) {
+                var _onFinish7 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee16(result) {
                   var res;
-                  return _regeneratorRuntime().wrap(function _callee13$(_context13) {
-                    while (1) switch (_context13.prev = _context13.next) {
+                  return _regeneratorRuntime().wrap(function _callee16$(_context16) {
+                    while (1) switch (_context16.prev = _context16.next) {
                       case 0:
-                        _context13.prev = 0;
+                        _context16.prev = 0;
                         res = JSON.parse(result) || {};
-                        _context13.next = 4;
+                        _context16.next = 4;
                         return after({
                           result: res,
                           threadId: threadId
                         });
                       case 4:
-                        _context13.next = 9;
+                        _context16.next = 9;
                         break;
                       case 6:
-                        _context13.prev = 6;
-                        _context13.t0 = _context13["catch"](0);
-                        console.error('Error saving memory on finish', _context13.t0);
+                        _context16.prev = 6;
+                        _context16.t0 = _context16["catch"](0);
+                        console.error('Error saving memory on finish', _context16.t0);
                       case 9:
-                        _onFinish3 == null || _onFinish3(result);
+                        _onFinish6 == null || _onFinish6(result);
                       case 10:
                       case "end":
-                        return _context13.stop();
+                        return _context16.stop();
                     }
-                  }, _callee13, null, [[0, 6]]);
+                  }, _callee16, null, [[0, 6]]);
                 }));
-                function onFinish(_x13) {
-                  return _onFinish4.apply(this, arguments);
+                function onFinish(_x17) {
+                  return _onFinish7.apply(this, arguments);
                 }
                 return onFinish;
               }(),
@@ -3125,11 +3311,11 @@ exports.Agent = (_tools = /*#__PURE__*/_classPrivateFieldLooseKey("tools"), _log
             }));
           case 9:
           case "end":
-            return _context14.stop();
+            return _context17.stop();
         }
-      }, _callee14, this);
+      }, _callee17, this);
     }));
-    function streamObject(_x12) {
+    function streamObject(_x16) {
       return _streamObject.apply(this, arguments);
     }
     return streamObject;