@mastra/rag 2.4.2 → 2.5.0

package/dist/tools/bedrock-knowledge-base.d.ts

@@ -0,0 +1,29 @@
+export interface BedrockKBToolOptions {
+    /** The ID of the Bedrock Knowledge Base. */
+    knowledgeBaseId: string;
+    /** AWS region. Defaults to AWS_REGION env var or us-east-1. */
+    region?: string;
+    /** Maximum number of results. Defaults to 5. */
+    numberOfResults?: number;
+    /** Use AgenticRetrieveStream for complex queries with query decomposition and managed reranking. Falls back to plain Retrieve on failure. Defaults to true. */
+    useAgenticRetrieval?: boolean;
+    /** Default AWS user ID for access-controlled retrieval. A userId in the tool request context takes precedence. */
+    userId?: string;
+}
+export interface BedrockKBResult {
+    content: string;
+    source?: string;
+    score?: number;
+    metadata: Record<string, unknown>;
+}
+export declare function createBedrockKBTool(options: BedrockKBToolOptions): import("@mastra/core/tools").Tool<{
+    queryText: string;
+}, {
+    results: {
+        content: string;
+        metadata: Record<string, unknown>;
+        source?: string | undefined;
+        score?: number | undefined;
+    }[];
+}, unknown, unknown, import("@mastra/core/tools").ToolExecutionContext<unknown, unknown, unknown>, `bedrock_knowledge_base_${string}`, unknown>;
+//# sourceMappingURL=bedrock-knowledge-base.d.ts.map

package/dist/tools/bedrock-knowledge-base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock-knowledge-base.d.ts","sourceRoot":"","sources":["../../src/tools/bedrock-knowledge-base.ts"],"names":[],"mappings":"AASA,MAAM,WAAW,oBAAoB;IACnC,4CAA4C;IAC5C,eAAe,EAAE,MAAM,CAAC;IACxB,+DAA+D;IAC/D,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,gDAAgD;IAChD,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,+JAA+J;IAC/J,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,kHAAkH;IAClH,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAmCD,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,oBAAoB;;;;;;;;;gJA2FhE"}

package/dist/tools/index.d.ts

@@ -2,4 +2,6 @@ export * from './document-chunker.js';
 export * from './graph-rag.js';
 export * from './vector-query.js';
 export type { VectorStoreResolver, VectorStoreResolverContext, VectorQueryToolOptions, GraphRagToolOptions, DatabaseConfig, PineconeConfig, PgVectorConfig, ChromaConfig, TurbopufferConfig, } from './types.js';
+export { createBedrockKBTool } from './bedrock-knowledge-base.js';
+export type { BedrockKBToolOptions, BedrockKBResult } from './bedrock-knowledge-base.js';
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,cAAc,oBAAoB,CAAC;AACnC,cAAc,aAAa,CAAC;AAC5B,cAAc,gBAAgB,CAAC;AAC/B,YAAY,EACV,mBAAmB,EACnB,0BAA0B,EAC1B,sBAAsB,EACtB,mBAAmB,EACnB,cAAc,EACd,cAAc,EACd,cAAc,EACd,YAAY,EACZ,iBAAiB,GAClB,MAAM,SAAS,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,cAAc,oBAAoB,CAAC;AACnC,cAAc,aAAa,CAAC;AAC5B,cAAc,gBAAgB,CAAC;AAC/B,YAAY,EACV,mBAAmB,EACnB,0BAA0B,EAC1B,sBAAsB,EACtB,mBAAmB,EACnB,cAAc,EACd,cAAc,EACd,cAAc,EACd,YAAY,EACZ,iBAAiB,GAClB,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAC/D,YAAY,EAAE,oBAAoB,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/rag",
-  "version": "2.4.2",
+  "version": "2.5.0",
   "description": "",
   "type": "module",
   "main": "dist/index.js",
@@ -25,6 +25,7 @@
   "author": "",
   "license": "Apache-2.0",
   "dependencies": {
+    "@aws-sdk/client-bedrock-agent-runtime": "^3.1076.0",
     "big.js": "^7.0.1",
     "js-tiktoken": "^1.0.21",
     "node-html-better-parser": "^1.5.8",
@@ -43,16 +44,17 @@
     "@vitest/ui": "4.1.10",
     "ai": "^4.3.19",
     "dotenv": "^17.4.2",
+    "esbuild": "^0.28.0",
     "eslint": "^10.7.0",
-    "tsup": "^8.5.1",
+    "tsdown": "0.22.9",
     "tsx": "^4.23.1",
     "typescript": "^6.0.3",
     "vitest": "4.1.10",
     "zod": "^4.4.3",
-    "@internal/ai-sdk-v4": "0.0.62",
-    "@internal/lint": "0.0.115",
-    "@mastra/core": "1.52.0",
-    "@internal/types-builder": "0.0.90"
+    "@internal/ai-sdk-v4": "0.0.69",
+    "@internal/types-builder": "0.0.97",
+    "@internal/lint": "0.0.122",
+    "@mastra/core": "1.58.0"
   },
   "keywords": [
     "rag",
@@ -83,10 +85,11 @@
     "node": ">=22.13.0"
   },
   "scripts": {
-    "build:lib": "tsup --silent --config tsup.config.ts",
+    "build:lib": "tsdown --silent --config tsdown.config.ts",
     "build:watch": "pnpm build:lib --watch",
     "vitest": "vitest",
     "test": "vitest run",
-    "lint": "eslint ."
+    "lint": "oxlint . && eslint .",
+    "lint:fix": "oxlint --fix . && eslint --fix ."
   }
 }