@mastra/core 0.1.27-alpha.1 → 0.1.27-alpha.3

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

@@ -2,12 +2,18 @@ import { UserContent } from 'ai';
 import { Integration } from '../integration';
 import { Logger } from '../logger';
 import { AllTools, ToolApi } from '../tools/types';
-type OpenAIVercelModelNames = 'gpt-4' | 'gpt-4-turbo' | 'gpt-3.5-turbo';
+type OpenAIVercelModelNames = 'gpt-4' | 'gpt-4-turbo' | 'gpt-3.5-turbo' | 'gpt-4o' | 'gpt-4o-mini';
 type OpenAIVercelConfig = {
     provider: 'OPEN_AI_VERCEL';
     name: OpenAIVercelModelNames;
     toolChoice: 'auto' | 'required';
 };
+type GoogleVercelModelNames = 'gemini-1.5-pro-latest' | 'gemini-1.5-pro' | 'gemini-1.5-flash-latest' | 'gemini-1.5-flash';
+type GoogleVercelConfig = {
+    provider: 'GOOGLE_VERCEL';
+    name: GoogleVercelModelNames;
+    toolChoice: 'auto' | 'required';
+};
 type AnthropicVercelModelNames = 'claude-3-opus-20240229' | 'claude-3-sonnet-20240229' | 'claude-3-haiku-20240307' | 'claude-3-5-sonnet-20240620';
 type AnthropicVercelConfig = {
     provider: 'ANTHROPIC_VERCEL';
@@ -20,7 +26,7 @@ type GroqVercelConfig = {
     name: GroqVercelModelNames;
     toolChoice: 'auto' | 'required';
 };
-type ModelConfig = OpenAIVercelConfig | AnthropicVercelConfig | GroqVercelConfig;
+type ModelConfig = OpenAIVercelConfig | AnthropicVercelConfig | GroqVercelConfig | GoogleVercelConfig;
 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>> {
     #private;
     name: string;
@@ -53,9 +59,10 @@ export declare class Agent<TIntegrations extends Integration[] | undefined = und
         onStepFinish?: (step: string) => void;
         maxSteps?: number;
     }): Promise<import("ai").GenerateTextResult<{}>>;
-    stream({ messages, onStepFinish, maxSteps, }: {
+    stream({ messages, onStepFinish, onFinish, maxSteps, }: {
         messages: UserContent[];
         onStepFinish?: (step: string) => void;
+        onFinish?: (result: string) => Promise<void> | 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,GACvB,OAAO,GACP,aAAa,GACb,eAAe,GACf,QAAQ,GACR,aAAa,CAAC;AAElB,KAAK,kBAAkB,GAAG;IACxB,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,IAAI,EAAE,sBAAsB,CAAC;IAC7B,UAAU,EAAE,MAAM,GAAG,UAAU,CAAC;CACjC,CAAC;AAEF,KAAK,sBAAsB,GACvB,uBAAuB,GACvB,gBAAgB,GAChB,yBAAyB,GACzB,kBAAkB,CAAC;AAEvB,KAAK,kBAAkB,GAAG;IACxB,QAAQ,EAAE,eAAe,CAAC;IAC1B,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,GAChB,kBAAkB,CAAC;AAIvB,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;IAyBpB,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,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;CAqDF"}

package/dist/core.cjs.development.js

@@ -2,13 +2,13 @@
 
 Object.defineProperty(exports, '__esModule', { value: true });
 
-var redis = require('@upstash/redis');
-var fs = require('fs');
-var path = require('path');
+var zod = require('zod');
 var anthropic = require('@ai-sdk/anthropic');
 var openai = require('@ai-sdk/openai');
 var ai = require('ai');
-var zod = require('zod');
+var redis = require('@upstash/redis');
+var fs = require('fs');
+var path = require('path');
 
 function asyncGeneratorStep(n, t, e, r, o, a, c) {
   try {
@@ -854,7 +854,9 @@ var Agent = /*#__PURE__*/function () {
       if (enabled && tool) {
         memo[k] = {
           description: tool.description,
-          parameters: tool.schema,
+          parameters: zod.z.object({
+            data: tool.schema
+          }),
           execute: tool.executor
         };
       }
@@ -1003,13 +1005,13 @@ var Agent = /*#__PURE__*/function () {
     return text;
   }();
   _proto.stream = /*#__PURE__*/function () {
-    var _stream = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(_ref4) {
+    var _stream = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(_ref4) {
       var _this3 = this;
-      var messages, _onStepFinish3, _ref4$maxSteps, maxSteps, params, argsForExecute, messageObjects;
-      return _regeneratorRuntime().wrap(function _callee4$(_context4) {
-        while (1) switch (_context4.prev = _context4.next) {
+      var messages, _onStepFinish3, _onFinish, _ref4$maxSteps, maxSteps, params, argsForExecute, messageObjects;
+      return _regeneratorRuntime().wrap(function _callee5$(_context5) {
+        while (1) switch (_context5.prev = _context5.next) {
           case 0:
-            messages = _ref4.messages, _onStepFinish3 = _ref4.onStepFinish, _ref4$maxSteps = _ref4.maxSteps, maxSteps = _ref4$maxSteps === void 0 ? 5 : _ref4$maxSteps;
+            messages = _ref4.messages, _onStepFinish3 = _ref4.onStepFinish, _onFinish = _ref4.onFinish, _ref4$maxSteps = _ref4.maxSteps, maxSteps = _ref4$maxSteps === void 0 ? 5 : _ref4$maxSteps;
             this.logger.info("Starting stream generation for agent " + this.name);
             params = this.getAgentParams({
               tools: this.convertTools(),
@@ -1048,6 +1050,23 @@ var Agent = /*#__PURE__*/function () {
                   return _onStepFinish4.apply(this, arguments);
                 }
                 return onStepFinish;
+              }(),
+              onFinish: function () {
+                var _onFinish2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(props) {
+                  return _regeneratorRuntime().wrap(function _callee4$(_context4) {
+                    while (1) switch (_context4.prev = _context4.next) {
+                      case 0:
+                        _onFinish == null || _onFinish(JSON.stringify(props, null, 2));
+                      case 1:
+                      case "end":
+                        return _context4.stop();
+                    }
+                  }, _callee4);
+                }));
+                function onFinish(_x5) {
+                  return _onFinish2.apply(this, arguments);
+                }
+                return onFinish;
               }()
             };
             messageObjects = messages.map(function (content) {
@@ -1061,17 +1080,17 @@ var Agent = /*#__PURE__*/function () {
               content: this.instructions
             });
             this.logger.debug("Streaming text with " + messageObjects.length + " messages");
-            _context4.next = 9;
+            _context5.next = 9;
             return ai.streamText(_extends({
               messages: messageObjects
             }, argsForExecute));
           case 9:
-            return _context4.abrupt("return", _context4.sent);
+            return _context5.abrupt("return", _context5.sent);
           case 10:
           case "end":
-            return _context4.stop();
+            return _context5.stop();
         }
-      }, _callee4, this);
+      }, _callee5, this);
     }));
     function stream(_x3) {
       return _stream.apply(this, arguments);