@mastra/core 0.1.27-alpha.5 → 0.1.27-alpha.6

package/dist/agent/index.d.ts

@@ -4,9 +4,9 @@ import { Logger } from '../logger';
 import { AllTools, ToolApi } from '../tools/types';
 import { LLM } from '../llm';
 import { ModelConfig } from '../llm/types';
-export declare class Agent<TIntegrations extends Integration[] | undefined = undefined, TTools extends Record<string, ToolApi<TIntegrations>> | undefined = undefined, TKeys extends keyof AllTools<TIntegrations, TTools> = keyof AllTools<TIntegrations, TTools>> {
+export declare class Agent<TTools, TIntegrations extends Integration[] | undefined = undefined, TKeys extends keyof AllTools<TTools, TIntegrations> = keyof AllTools<TTools, TIntegrations>> {
     name: string;
-    readonly llm: LLM<TIntegrations, TTools, TKeys>;
+    readonly llm: LLM<TTools, TIntegrations, TKeys>;
     readonly instructions: string;
     readonly model: ModelConfig;
     readonly enabledTools: Partial<Record<TKeys, boolean>>;

package/dist/agent/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/agent/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAe,WAAW,EAAE,MAAM,IAAI,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAgB,MAAM,EAAE,MAAM,WAAW,CAAC;AACjD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,GAAG,EAAE,MAAM,QAAQ,CAAC;AAC7B,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAE3C,qBAAa,KAAK,CAChB,aAAa,SAAS,WAAW,EAAE,GAAG,SAAS,GAAG,SAAS,EAC3D,MAAM,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC,GAAG,SAAS,GAAG,SAAS,EAC7E,KAAK,SAAS,MAAM,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC,GAAG,MAAM,QAAQ,CAClE,aAAa,EACb,MAAM,CACP;IAEM,IAAI,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,aAAa,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;IAChD,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;IACvD,MAAM,EAAE,MAAM,CAAC;gBAEH,MAAM,EAAE;QAClB,IAAI,EAAE,MAAM,CAAC;QACb,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,WAAW,CAAC;QACnB,YAAY,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;KAChD;IAcD;;;OAGG;IACH,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC;IAKxC;;;OAGG;IACH,WAAW,CAAC,MAAM,EAAE,MAAM;IAKpB,IAAI,CAAC,EACT,QAAQ,EACR,YAAY,EACZ,QAAY,GACb,EAAE;QACD,QAAQ,EAAE,WAAW,EAAE,CAAC;QACxB,YAAY,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;QACtC,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB;IAsBK,MAAM,CAAC,EACX,QAAQ,EACR,YAAY,EACZ,QAAQ,EACR,QAAY,GACb,EAAE;QACD,QAAQ,EAAE,WAAW,EAAE,CAAC;QACxB,YAAY,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;QACtC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;QACpD,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB;CAsBF"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/agent/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAe,WAAW,EAAE,MAAM,IAAI,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAgB,MAAM,EAAE,MAAM,WAAW,CAAC;AACjD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,GAAG,EAAE,MAAM,QAAQ,CAAC;AAC7B,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAE3C,qBAAa,KAAK,CAChB,MAAM,EACN,aAAa,SAAS,WAAW,EAAE,GAAG,SAAS,GAAG,SAAS,EAC3D,KAAK,SAAS,MAAM,QAAQ,CAAC,MAAM,EAAE,aAAa,CAAC,GAAG,MAAM,QAAQ,CAClE,MAAM,EACN,aAAa,CACd;IAEM,IAAI,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,MAAM,EAAE,aAAa,EAAE,KAAK,CAAC,CAAC;IAChD,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;IACvD,MAAM,EAAE,MAAM,CAAC;gBAEH,MAAM,EAAE;QAClB,IAAI,EAAE,MAAM,CAAC;QACb,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,WAAW,CAAC;QACnB,YAAY,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;KAChD;IAcD;;;OAGG;IACH,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC;IAKxC;;;OAGG;IACH,WAAW,CAAC,MAAM,EAAE,MAAM;IAKpB,IAAI,CAAC,EACT,QAAQ,EACR,YAAY,EACZ,QAAY,GACb,EAAE;QACD,QAAQ,EAAE,WAAW,EAAE,CAAC;QACxB,YAAY,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;QACtC,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB;IAsBK,MAAM,CAAC,EACX,QAAQ,EACR,YAAY,EACZ,QAAQ,EACR,QAAY,GACb,EAAE;QACD,QAAQ,EAAE,WAAW,EAAE,CAAC;QACxB,YAAY,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;QACtC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;QACpD,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB;CAsBF"}

package/dist/core.cjs.development.js

@@ -1191,16 +1191,7 @@ function createTool(opts) {
 var _excluded = ["client"];
 var Integration = /*#__PURE__*/function () {
   function Integration() {
-    var _this = this;
     this.authType = 'API_KEY';
-    this.executeTool = function (tool, params) {
-      var toolExecutor = _this.tools[tool];
-      return toolExecutor.executor(_extends({}, params, {
-        getIntegration: function getIntegration() {
-          return _this;
-        }
-      }));
-    };
   }
   var _proto = Integration.prototype;
   _proto.getApiClient = /*#__PURE__*/function () {
@@ -1221,7 +1212,7 @@ var Integration = /*#__PURE__*/function () {
     return getApiClient;
   }();
   _proto._generateIntegrationTools = function _generateIntegrationTools() {
-    var _this2 = this;
+    var _this = this;
     var _this$baseClient = this.baseClient,
       clientMethods = _objectWithoutPropertiesLoose(_this$baseClient, _excluded);
     var schemas = this.toolSchemas;
@@ -1244,7 +1235,7 @@ var Integration = /*#__PURE__*/function () {
                 case 0:
                   data = _ref.data;
                   _context2.next = 3;
-                  return _this2.getApiClient();
+                  return _this.getApiClient();
                 case 3:
                   client = _context2.sent;
                   value = client[key];
@@ -1282,43 +1273,6 @@ var Integration = /*#__PURE__*/function () {
     }
   }]);
 }();
-var GmailIntegration = /*#__PURE__*/function (_Integration2) {
-  function GmailIntegration(config) {
-    var _this3;
-    _this3 = _Integration2.call(this) || this;
-    _this3.name = 'GMAIL';
-    _this3.logoUrl = '';
-    _this3.tools = {
-      gmailGetProfile: createTool({
-        label: 'Get Gmail Profile',
-        schema: zod.z.object({}),
-        description: 'Get the profile of the authenticated user',
-        executor: function () {
-          var _executor2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3() {
-            return _regeneratorRuntime().wrap(function _callee3$(_context3) {
-              while (1) switch (_context3.prev = _context3.next) {
-                case 0:
-                  return _context3.abrupt("return", {
-                    email: ''
-                  });
-                case 1:
-                case "end":
-                  return _context3.stop();
-              }
-            }, _callee3);
-          }));
-          function executor() {
-            return _executor2.apply(this, arguments);
-          }
-          return executor;
-        }()
-      })
-    };
-    return _this3;
-  }
-  _inheritsLoose(GmailIntegration, _Integration2);
-  return GmailIntegration;
-}(Integration);
 
 exports.PropertyType = void 0;
 (function (PropertyType) {
@@ -1501,7 +1455,7 @@ var Mastra = /*#__PURE__*/function () {
     // Merge custom tools with integration tools
     var allTools = _extends({}, configuredTools, integrationTools);
     // Hydrate tools with integration tools
-    var hydratedTools = Object.entries(allTools).reduce(function (memo, _ref) {
+    var hydratedTools = Object.entries(allTools != null ? allTools : {}).reduce(function (memo, _ref) {
       var key = _ref[0],
         val = _ref[1];
       memo[key] = _extends({}, val, {
@@ -1599,12 +1553,47 @@ var Mastra = /*#__PURE__*/function () {
     return agent;
   };
   _proto.getIntegration = function getIntegration(name) {
-    var integration = this.integrations.get(name.toUpperCase());
+    var stringifiedName = String(name);
+    var integration = this.integrations.get(stringifiedName.toUpperCase());
     if (!integration) {
-      throw new Error("Integration with name " + name + " not found");
+      throw new Error("Integration with name " + stringifiedName + " not found");
     }
     return integration;
   };
+  _proto.getTool = function getTool(name) {
+    var _this2 = this;
+    var tool = this.tools[name];
+    if (!tool) {
+      throw new Error("Tool with name " + String(name) + " not found");
+    }
+    return _extends({}, tool, {
+      execute: function () {
+        var _execute = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(params) {
+          return _regeneratorRuntime().wrap(function _callee2$(_context2) {
+            while (1) switch (_context2.prev = _context2.next) {
+              case 0:
+                return _context2.abrupt("return", tool.executor({
+                  data: params,
+                  getIntegration: function getIntegration(name) {
+                    return _this2.getIntegration(name);
+                  },
+                  agents: _this2.agents,
+                  llm: _this2.llm,
+                  engine: _this2.engine
+                }));
+              case 1:
+              case "end":
+                return _context2.stop();
+            }
+          }, _callee2);
+        }));
+        function execute(_x3) {
+          return _execute.apply(this, arguments);
+        }
+        return execute;
+      }()
+    });
+  };
   _proto.availableIntegrations = function availableIntegrations() {
     return Array.from(this.integrations.entries()).map(function (_ref2) {
       var name = _ref2[0],
@@ -2389,7 +2378,6 @@ function _detectUnreachableSteps2() {
 exports.Agent = Agent;
 exports.BaseLogger = BaseLogger;
 exports.FieldTypePrimitiveMap = FieldTypePrimitiveMap;
-exports.GmailIntegration = GmailIntegration;
 exports.Integration = Integration;
 exports.LogLevel = LogLevel;
 exports.Mastra = Mastra;