@mastra/core 0.1.27-alpha.0 → 0.1.27-alpha.2

package/README.md

@@ -1,40 +1,8 @@
-# Future
+# @mastra/core
 
 This is a guide to developing the core package.
 
-### Ensure Docker is Installed
-
-If you haven't already, install Docker: (Docker Installation)[https://www.docker.com/]
-
-### Start Docker Containers
-
-1. Navigate to the .dev directory:`cd .dev`
-2. Start the Docker containers:
-   `docker-compose up -d`
-
-This command will spin up the necessary services, including the database and the Inngest server.
-
-### Setup Environment Variables
-
-1. Copy the sample environment variable files:
-   packages/admin/.env.local.sample
-   packages/core/.env.local.sample
-2. Duplicate and rename each .env.local.sample file to .env:
-   cp packages/admin/.env.local.sample packages/admin/.env`
-cp packages/core/.env.local.sample packages/core/.env`
-
-Note: We won’t need to do this again after the first setup if the CLI persists credentials.
-
 ### Install Dependencies for Core Package
 
 1. Navigate to the core directory:`cd packages/core`
-2. Install the dependencies:`pnpm install`
-
-### Build and Initialize CLI
-
-1. Navigate to the cli directory:`cd packages/cli`
-2. Build the project and run the initialization script:`pnpm run build && node dist/index.js init`
-3. When prompted, supply the necessary credentials:
-
-Database URL: postgresql://postgres:postgres@localhost:5432/postgres
-Inngest Server URL: http://localhost:8288
+2. Install the dependencies:`pnpm install`

package/dist/{next-src/agent → agent}/index.d.ts

@@ -1,3 +1,4 @@
+import { UserContent } from 'ai';
 import { Integration } from '../integration';
 import { Logger } from '../logger';
 import { AllTools, ToolApi } from '../tools/types';
@@ -48,12 +49,12 @@ export declare class Agent<TIntegrations extends Integration[] | undefined = und
     private getAgentParams;
     private createOpenAICompatibleModel;
     text({ messages, onStepFinish, maxSteps, }: {
-        messages: string[];
+        messages: UserContent[];
         onStepFinish?: (step: string) => void;
         maxSteps?: number;
     }): Promise<import("ai").GenerateTextResult<{}>>;
     stream({ messages, onStepFinish, maxSteps, }: {
-        messages: string[];
+        messages: UserContent[];
         onStepFinish?: (step: string) => void;
         maxSteps?: number;
     }): Promise<import("ai").StreamTextResult<{}>>;

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/agent/index.ts"],"names":[],"mappings":"AAGA,OAAO,EAOL,WAAW,EACZ,MAAM,IAAI,CAAC;AACZ,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAgB,MAAM,EAAE,MAAM,WAAW,CAAC;AACjD,OAAO,EAAE,QAAQ,EAAY,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAE7D,KAAK,sBAAsB,GAAG,OAAO,GAAG,aAAa,GAAG,eAAe,CAAC;AAExE,KAAK,kBAAkB,GAAG;IACxB,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,IAAI,EAAE,sBAAsB,CAAC;IAC7B,UAAU,EAAE,MAAM,GAAG,UAAU,CAAC;CACjC,CAAC;AAEF,KAAK,yBAAyB,GAC1B,wBAAwB,GACxB,0BAA0B,GAC1B,yBAAyB,GACzB,4BAA4B,CAAC;AAEjC,KAAK,qBAAqB,GAAG;IAC3B,QAAQ,EAAE,kBAAkB,CAAC;IAC7B,IAAI,EAAE,yBAAyB,CAAC;IAChC,UAAU,EAAE,MAAM,GAAG,UAAU,CAAC;CACjC,CAAC;AAEF,KAAK,oBAAoB,GACrB,uCAAuC,GACvC,sCAAsC,GACtC,cAAc,GACd,aAAa,CAAC;AAElB,KAAK,gBAAgB,GAAG;IACtB,QAAQ,EAAE,aAAa,CAAC;IACxB,IAAI,EAAE,oBAAoB,CAAC;IAC3B,UAAU,EAAE,MAAM,GAAG,UAAU,CAAC;CACjC,CAAC;AAEF,KAAK,WAAW,GACZ,kBAAkB,GAClB,qBAAqB,GACrB,gBAAgB,CAAC;AAIrB,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,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;IAEvD,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;IAYD;;;OAGG;IACH,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC;IAKxC;;;OAGG;IACH,WAAW,CAAC,MAAM,EAAE,MAAM;IAK1B,OAAO,CAAC,YAAY;IAepB,OAAO,CAAC,YAAY;IAuBpB,OAAO,CAAC,cAAc;IAwFtB,OAAO,CAAC,2BAA2B;IAgB7B,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;IAmDK,MAAM,CAAC,EACX,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;CAkDF"}

package/dist/core.cjs.development.js

@@ -2,12 +2,12 @@
 
 Object.defineProperty(exports, '__esModule', { value: true });
 
-var redis = require('@upstash/redis');
-var fs = require('fs');
-var path = require('path');
 var anthropic = require('@ai-sdk/anthropic');
 var openai = require('@ai-sdk/openai');
 var ai = require('ai');
+var redis = require('@upstash/redis');
+var fs = require('fs');
+var path = require('path');
 var zod = require('zod');
 
 function asyncGeneratorStep(n, t, e, r, o, a, c) {
@@ -1218,10 +1218,148 @@ var GmailIntegration = /*#__PURE__*/function (_Integration2) {
   return GmailIntegration;
 }(Integration);
 
+exports.PropertyType = void 0;
+(function (PropertyType) {
+  PropertyType["LONG_TEXT"] = "LONG_TEXT";
+  PropertyType["SINGLE_LINE_TEXT"] = "SINGLE_LINE_TEXT";
+  PropertyType["SINGLE_SELECT"] = "SINGLE_SELECT";
+  PropertyType["MULTI_SELECT"] = "MULTI_SELECT";
+  PropertyType["CHECKBOX"] = "CHECKBOX";
+  PropertyType["DATE"] = "DATE";
+  PropertyType["USER"] = "USER";
+  PropertyType["BADGE_LIST"] = "BADGE_LIST";
+  PropertyType["CURRENCY"] = "CURRENCY";
+  PropertyType["URL"] = "URL";
+  PropertyType["PHONE"] = "PHONE";
+  PropertyType["CONTACT"] = "CONTACT";
+  PropertyType["COMPANY"] = "COMPANY";
+  PropertyType["PERSON"] = "PERSON";
+  PropertyType["ENRICHMENT"] = "ENRICHMENT";
+  PropertyType["COMPOSITE"] = "COMPOSITE";
+  PropertyType["BOOLEAN"] = "BOOLEAN";
+  PropertyType["NUMBER"] = "NUMBER";
+  PropertyType["FLOAT"] = "FLOAT";
+  PropertyType["JSON_OBJECT"] = "JSON_OBJECT";
+  PropertyType["JSON_ARRAY"] = "JSON_ARRAY";
+})(exports.PropertyType || (exports.PropertyType = {}));
+exports.FilterOperators = void 0;
+(function (FilterOperators) {
+  FilterOperators["IS"] = "is";
+  FilterOperators["EQUAL"] = "eq";
+  FilterOperators["NOT_EQUAL"] = "not_eq";
+  FilterOperators["CONTAINS"] = "contains";
+  FilterOperators["IN"] = "in";
+  FilterOperators["NOT_IN"] = "not_in";
+  FilterOperators["GREATER_THAN"] = "gt";
+  FilterOperators["LESS_THAN"] = "lt";
+  FilterOperators["NOT_CONTAINS"] = "not_contains";
+  FilterOperators["GREATER_THAN_OR_EQUAL"] = "gte";
+  FilterOperators["LESS_THAN_OR_EQUAL"] = "lte";
+  FilterOperators["OP"] = "op";
+  FilterOperators["SET"] = "set";
+  FilterOperators["NOT_SET"] = "not_set";
+})(exports.FilterOperators || (exports.FilterOperators = {}));
+
+var _FieldTypePrimitiveMa;
+var FieldTypePrimitiveMap = (_FieldTypePrimitiveMa = {}, _FieldTypePrimitiveMa[exports.PropertyType.BADGE_LIST] = 'string', _FieldTypePrimitiveMa[exports.PropertyType.CHECKBOX] = 'boolean', _FieldTypePrimitiveMa[exports.PropertyType.BOOLEAN] = 'boolean', _FieldTypePrimitiveMa[exports.PropertyType.NUMBER] = 'number', _FieldTypePrimitiveMa[exports.PropertyType.FLOAT] = 'number', _FieldTypePrimitiveMa[exports.PropertyType.JSON_ARRAY] = 'string', _FieldTypePrimitiveMa[exports.PropertyType.JSON_OBJECT] = 'string', _FieldTypePrimitiveMa[exports.PropertyType.CURRENCY] = 'number', _FieldTypePrimitiveMa[exports.PropertyType.DATE] = 'date', _FieldTypePrimitiveMa[exports.PropertyType.LONG_TEXT] = 'string', _FieldTypePrimitiveMa[exports.PropertyType.MULTI_SELECT] = 'string', _FieldTypePrimitiveMa[exports.PropertyType.PHONE] = 'string', _FieldTypePrimitiveMa[exports.PropertyType.SINGLE_LINE_TEXT] = 'string', _FieldTypePrimitiveMa[exports.PropertyType.SINGLE_SELECT] = 'string', _FieldTypePrimitiveMa[exports.PropertyType.URL] = 'string', _FieldTypePrimitiveMa[exports.PropertyType.USER] = 'string', _FieldTypePrimitiveMa[exports.PropertyType.CONTACT] = 'string', _FieldTypePrimitiveMa[exports.PropertyType.COMPANY] = 'string', _FieldTypePrimitiveMa[exports.PropertyType.COMPOSITE] = 'string', _FieldTypePrimitiveMa[exports.PropertyType.PERSON] = 'string', _FieldTypePrimitiveMa[exports.PropertyType.ENRICHMENT] = 'string', _FieldTypePrimitiveMa);
+/**
+ * Split a string on commas and strip quotes from each part
+ * @param str input string
+ * @returns array of strings
+ */
+var splitAndStripQuotes = function splitAndStripQuotes(str) {
+  // Raw regex pattern for splitting on non-quoted commas
+  var pattern = /,\s*(?=(?:[^"]*"[^"]*")*[^"]*$)/;
+  // Split the string
+  var parts = str.split(pattern);
+  // Strip quotes from each part
+  var cleanedParts = parts.map(function (part) {
+    return part.trim().replace(/^"|"$/g, '');
+  });
+  return cleanedParts;
+};
+var transformFilterValueArray = function transformFilterValueArray(value) {
+  if (typeof value === 'string') return value.split(',');
+  return value;
+};
+var transformFilterValueBoolean = function transformFilterValueBoolean(value) {
+  if (typeof value === 'string') return value === 'true';
+  return value;
+};
+/**
+ * This returns the SQL field for a JSON field with appropriate casting.
+ * @param column - JSON field to cast.
+ * @param type - Type of the JSON field.
+ */
+var getJSONField = function getJSONField(column, type) {
+  var primitiveType = FieldTypePrimitiveMap[type];
+  switch (primitiveType) {
+    case 'number':
+      {
+        // cast to number if field is numeric
+        return "(" + column + ")::bigint";
+      }
+    case 'boolean':
+      {
+        // cast to boolean if field is boolean
+        return "(" + column + ")::boolean";
+      }
+    case 'date':
+      {
+        return "(" + column + ")::timestamp";
+      }
+  }
+  return "(" + column + ")::text";
+};
+/**
+ * @description
+ * Builds a query string from filter and sort query strings
+ * @param filterQueryString Filter query string
+ * @param sortQueryString Sort query string
+ * @returns
+ */
+function buildQueryString(_ref) {
+  var filterQueryString = _ref.filterQueryString,
+    sortQueryString = _ref.sortQueryString;
+  if (!filterQueryString && !sortQueryString) return '';
+  if (!filterQueryString) return "" + sortQueryString;
+  if (!sortQueryString) return "" + filterQueryString;
+  return filterQueryString + "&" + sortQueryString;
+}
+
+var filterQuerySchema = /*#__PURE__*/zod.z.object({
+  is: /*#__PURE__*/zod.z.array(zod.z.string()).or(zod.z.string()).optional(),
+  eq: /*#__PURE__*/zod.z.array(zod.z.string()).or(zod.z.string()).optional(),
+  not_eq: /*#__PURE__*/zod.z.array(zod.z.string()).or(zod.z.string()).optional(),
+  contains: /*#__PURE__*/zod.z.array(zod.z.string()).or(zod.z.string()).optional(),
+  not_contains: /*#__PURE__*/zod.z.array(zod.z.string()).or(zod.z.string()).optional(),
+  gt: /*#__PURE__*/zod.z.array(zod.z.string()).or(zod.z.string()).optional(),
+  gte: /*#__PURE__*/zod.z.array(zod.z.string()).or(zod.z.string()).optional(),
+  lt: /*#__PURE__*/zod.z.array(zod.z.string()).or(zod.z.string()).optional(),
+  lte: /*#__PURE__*/zod.z.array(zod.z.string()).or(zod.z.string()).optional(),
+  "in": /*#__PURE__*/zod.z.array(zod.z.string()).or(zod.z.string()).optional(),
+  not_in: /*#__PURE__*/zod.z.array(zod.z.string()).or(zod.z.string()).optional(),
+  op: /*#__PURE__*/zod.z["enum"](['or', 'and']).optional()["default"]('or'),
+  set: /*#__PURE__*/zod.z.string().optional().transform(transformFilterValueBoolean),
+  not_set: /*#__PURE__*/zod.z.string().optional().transform(transformFilterValueBoolean)
+});
+var sortQuerySchema = /*#__PURE__*/zod.z.string().transform(transformFilterValueArray);
+
+var MastraEngine =
+/**
+ * Initializes the database connection
+ * @param config Configuration object for database connection
+ */
+function MastraEngine(config) {};
+
+var MastraVector = function MastraVector() {};
+
 var Mastra = /*#__PURE__*/function () {
   function Mastra(config) {
     var _this = this,
       _config$integrations;
+    this.engine = void 0;
+    this.vectors = void 0;
     this.tools = void 0;
     this.agents = void 0;
     this.integrations = void 0;
@@ -1283,33 +1421,47 @@ var Mastra = /*#__PURE__*/function () {
         agent.__setLogger(agentLogger);
       }
     });
-    this.syncs = new Map();
-    Object.entries(config.syncs).forEach(function (_ref2) {
-      var key = _ref2[0],
-        sync = _ref2[1];
-      _this.syncs.set(key, sync);
-    });
+    if (config.syncs && !config.engine) {
+      throw new Error('Engine is required to run syncs');
+    }
+    this.syncs = config.syncs || {};
+    if (config.engine) {
+      this.engine = config.engine;
+    }
+    if (config.vectors) {
+      this.vectors = config.vectors;
+    }
   }
   var _proto = Mastra.prototype;
   _proto.sync = /*#__PURE__*/function () {
     var _sync = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(key, params) {
-      var sync;
+      var _this$syncs;
+      var syncFn;
       return _regeneratorRuntime().wrap(function _callee$(_context) {
         while (1) switch (_context.prev = _context.next) {
           case 0:
-            sync = this.syncs.get(key);
-            if (sync) {
-              _context.next = 3;
+            if (this.engine) {
+              _context.next = 2;
               break;
             }
             throw new Error("Sync function " + key + " not found");
-          case 3:
-            _context.next = 5;
-            return sync({
+          case 2:
+            syncFn = (_this$syncs = this.syncs) == null ? void 0 : _this$syncs[key];
+            if (syncFn) {
+              _context.next = 5;
+              break;
+            }
+            throw new Error("Sync function " + key + " not found");
+          case 5:
+            _context.next = 7;
+            return syncFn({
               tools: this.tools,
-              params: params
+              params: params,
+              engine: this.engine,
+              agents: this.agents,
+              vectors: this.vectors
             });
-          case 5:
+          case 7:
           case "end":
             return _context.stop();
         }
@@ -1335,9 +1487,9 @@ var Mastra = /*#__PURE__*/function () {
     return integration;
   };
   _proto.availableIntegrations = function availableIntegrations() {
-    return Array.from(this.integrations.entries()).map(function (_ref3) {
-      var name = _ref3[0],
-        integration = _ref3[1];
+    return Array.from(this.integrations.entries()).map(function (_ref2) {
+      var name = _ref2[0],
+        integration = _ref2[1];
       return {
         name: name,
         integration: integration
@@ -1347,9 +1499,9 @@ var Mastra = /*#__PURE__*/function () {
   _proto.getTools = function getTools() {
     return this.tools;
   };
-  _proto.setLogger = function setLogger(_ref4) {
-    var key = _ref4.key,
-      logger = _ref4.logger;
+  _proto.setLogger = function setLogger(_ref3) {
+    var key = _ref3.key,
+      logger = _ref3.logger;
     this.logger.set(key, logger);
   };
   _proto.getLogger = function getLogger(key) {
@@ -1360,13 +1512,23 @@ var Mastra = /*#__PURE__*/function () {
 
 exports.Agent = Agent;
 exports.BaseLogger = BaseLogger;
+exports.FieldTypePrimitiveMap = FieldTypePrimitiveMap;
 exports.GmailIntegration = GmailIntegration;
 exports.Integration = Integration;
 exports.LogLevel = LogLevel;
 exports.Mastra = Mastra;
+exports.MastraEngine = MastraEngine;
+exports.MastraVector = MastraVector;
 exports.MultiLogger = MultiLogger;
 exports.RegisteredLogger = RegisteredLogger;
+exports.buildQueryString = buildQueryString;
 exports.createLogger = createLogger;
 exports.createMultiLogger = createMultiLogger;
 exports.createTool = createTool;
+exports.filterQuerySchema = filterQuerySchema;
+exports.getJSONField = getJSONField;
+exports.sortQuerySchema = sortQuerySchema;
+exports.splitAndStripQuotes = splitAndStripQuotes;
+exports.transformFilterValueArray = transformFilterValueArray;
+exports.transformFilterValueBoolean = transformFilterValueBoolean;
 //# sourceMappingURL=core.cjs.development.js.map