@mastra/rag 0.1.0 → 0.1.1

package/.turbo/turbo-build.log

@@ -1,18 +1,18 @@
 
-> @mastra/rag@0.1.0-alpha.89 build C:\Users\Ward\projects\mastra\mastra\packages\rag
+> @mastra/rag@0.1.1-alpha.0 build C:\Users\Ward\projects\mastra\mastra\packages\rag
 > tsup src/index.ts --format esm --experimental-dts --clean --treeshake
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.3.6
 TSC Build start
-TSC ⚡️ Build success in 4639ms
+TSC ⚡️ Build success in 15607ms
 DTS Build start
 CLI Target: es2022
-Analysis will use the bundled TypeScript version 5.7.3
-Writing package typings: C:\Users\Ward\projects\mastra\mastra\packages\rag\dist\_tsup-dts-rollup.d.ts
-DTS ⚡️ Build success in 3914ms
 CLI Cleaning output folder
 ESM Build start
-ESM dist\index.js 83.27 KB
-ESM ⚡️ Build success in 164ms
+Analysis will use the bundled TypeScript version 5.7.3
+Writing package typings: C:\Users\Ward\projects\mastra\mastra\packages\rag\dist\_tsup-dts-rollup.d.ts
+DTS ⚡️ Build success in 9490ms
+ESM dist\index.js 83.24 KB
+ESM ⚡️ Build success in 9832ms

package/.turbo/turbo-lint.log

@@ -0,0 +1,16 @@
+
+> @mastra/rag@0.1.0 lint C:\Users\Ward\projects\mastra\mastra\packages\rag
+> eslint .
+
+
+C:\Users\Ward\projects\mastra\mastra\packages\rag\src\document\transformers\token.ts
+   62:14  warning  'error' is defined but never used  @typescript-eslint/no-unused-vars
+  118:14  warning  'error' is defined but never used  @typescript-eslint/no-unused-vars
+
+C:\Users\Ward\projects\mastra\mastra\packages\rag\src\graph-rag\index.test.ts
+  1:36  warning  'beforeAll' is defined but never used. Allowed unused vars must match /^ignored/u  @typescript-eslint/no-unused-vars
+  1:47  warning  'afterAll' is defined but never used. Allowed unused vars must match /^ignored/u   @typescript-eslint/no-unused-vars
+  1:69  warning  'afterEach' is defined but never used. Allowed unused vars must match /^ignored/u  @typescript-eslint/no-unused-vars
+
+✖ 5 problems (0 errors, 5 warnings)
+

package/CHANGELOG.md

@@ -1,5 +1,31 @@
 # @mastra/rag
 
+## 0.1.1
+
+### Patch Changes
+
+- 91ef439: Add eslint and ran autofix
+- Updated dependencies [d59f1a8]
+- Updated dependencies [91ef439]
+- Updated dependencies [4a25be4]
+- Updated dependencies [bf2e88f]
+- Updated dependencies [2f0d707]
+- Updated dependencies [aac1667]
+  - @mastra/core@0.2.1
+
+## 0.1.1-alpha.0
+
+### Patch Changes
+
+- 91ef439: Add eslint and ran autofix
+- Updated dependencies [d59f1a8]
+- Updated dependencies [91ef439]
+- Updated dependencies [4a25be4]
+- Updated dependencies [bf2e88f]
+- Updated dependencies [2f0d707]
+- Updated dependencies [aac1667]
+  - @mastra/core@0.2.1-alpha.0
+
 ## 0.1.0
 
 ### Minor Changes

package/dist/_tsup-dts-rollup.d.ts

@@ -1,17 +1,17 @@
 import { createTool } from '@mastra/core/tools';
 import { Document as Document_2 } from 'llamaindex';
-import { EmbeddingModel } from 'ai';
-import { KeywordExtractPrompt } from 'llamaindex';
-import { LanguageModelV1 } from 'ai';
-import { LLM } from 'llamaindex';
-import { MastraVector } from '@mastra/core/vector';
-import { QueryResult } from '@mastra/core/vector';
-import { QuestionExtractPrompt } from 'llamaindex';
-import { SummaryPrompt } from 'llamaindex';
-import { TiktokenEncoding } from 'js-tiktoken';
-import { TiktokenModel } from 'js-tiktoken';
-import { TitleCombinePrompt } from 'llamaindex';
-import { TitleExtractorPrompt } from 'llamaindex';
+import type { EmbeddingModel } from 'ai';
+import type { KeywordExtractPrompt } from 'llamaindex';
+import type { LanguageModelV1 } from 'ai';
+import type { LLM } from 'llamaindex';
+import type { MastraVector } from '@mastra/core/vector';
+import type { QueryResult } from '@mastra/core/vector';
+import type { QuestionExtractPrompt } from 'llamaindex';
+import type { SummaryPrompt } from 'llamaindex';
+import type { TiktokenEncoding } from 'js-tiktoken';
+import type { TiktokenModel } from 'js-tiktoken';
+import type { TitleCombinePrompt } from 'llamaindex';
+import type { TitleExtractorPrompt } from 'llamaindex';
 
 /**
  * Vector store specific prompts that detail supported operators and examples.

package/dist/index.js

@@ -1,13 +1,14 @@
 import { Document, SummaryExtractor, QuestionsAnsweredExtractor, KeywordExtractor, TitleExtractor, IngestionPipeline } from 'llamaindex';
-import { encodingForModel, getEncoding } from 'js-tiktoken';
 import { parse } from 'node-html-better-parser';
+import { encodingForModel, getEncoding } from 'js-tiktoken';
 import { CohereRelevanceScorer, MastraAgentRelevanceScorer } from '@mastra/core/relevance';
-import '@mastra/core/vector';
-import { embed } from 'ai';
 import { createTool } from '@mastra/core/tools';
 import { z } from 'zod';
+import { embed } from 'ai';
 
 // src/document/document.ts
+
+// src/document/types.ts
 var Language = /* @__PURE__ */ ((Language2) => {
   Language2["CPP"] = "cpp";
   Language2["GO"] = "go";
@@ -37,8 +38,6 @@ var Language = /* @__PURE__ */ ((Language2) => {
   Language2["POWERSHELL"] = "powershell";
   return Language2;
 })(Language || {});
-
-// src/document/transformers/text.ts
 var TextTransformer = class {
   size;
   overlap;
@@ -849,7 +848,7 @@ var MarkdownHeaderTransformer = class {
         const aggChunk = aggregatedChunks[aggregatedChunks.length - 1];
         aggChunk.content += "  \n" + line.content;
       } else if (aggregatedChunks.length > 0 && JSON.stringify(aggregatedChunks?.[aggregatedChunks.length - 1].metadata) !== JSON.stringify(line.metadata) && Object.keys(aggregatedChunks?.[aggregatedChunks.length - 1].metadata).length < Object.keys(line.metadata).length && aggregatedChunks?.[aggregatedChunks.length - 1]?.content?.split("\n")?.slice(-1)[0][0] === "#" && !this.stripHeaders) {
-        if (aggregatedChunks?.[aggregatedChunks.length - 1]) {
+        if (aggregatedChunks && aggregatedChunks?.[aggregatedChunks.length - 1]) {
           const aggChunk = aggregatedChunks[aggregatedChunks.length - 1];
           if (aggChunk) {
             aggChunk.content += "  \n" + line.content;
@@ -1424,7 +1423,7 @@ var GraphRAG = class {
     return this.edges.filter((edge) => edge.source === nodeId && (!edgeType || edge.type === edgeType)).map((edge) => ({
       id: edge.target,
       weight: edge.weight
-    })).filter((node) => node !== undefined);
+    })).filter((node) => node !== void 0);
   }
   // Calculate cosine similarity between two vectors
   cosineSimilarity(vec1, vec2) {
@@ -1686,7 +1685,7 @@ var createGraphRAGTool = ({
           vectorStore,
           queryText,
           model,
-          queryFilter: Object.keys(queryFilter || {}).length > 0 ? queryFilter : undefined,
+          queryFilter: Object.keys(queryFilter || {}).length > 0 ? queryFilter : void 0,
           topK,
           includeVectors: true
         });
@@ -1761,7 +1760,7 @@ var createVectorQueryTool = ({
           vectorStore,
           queryText,
           model,
-          queryFilter: Object.keys(queryFilter || {}).length > 0 ? queryFilter : undefined,
+          queryFilter: Object.keys(queryFilter || {}).length > 0 ? queryFilter : void 0,
           topK
         });
         if (reranker) {

package/eslint.config.js

@@ -0,0 +1,6 @@
+import { createConfig } from '@internal/lint/eslint';
+
+const config = await createConfig();
+
+/** @type {import("eslint").Linter.Config[]} */
+export default [...config];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/rag",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "",
   "type": "module",
   "main": "dist/index.js",
@@ -25,7 +25,7 @@
     "node-html-better-parser": "^1.4.7",
     "pathe": "^2.0.2",
     "zod": "^3.24.1",
-    "@mastra/core": "^0.2.0"
+    "@mastra/core": "^0.2.1"
   },
   "peerDependencies": {
     "ai": "^4.0.0"
@@ -39,7 +39,9 @@
     "dotenv": "^16.4.7",
     "tsup": "^8.0.1",
     "typescript": "^5.7.3",
-    "vitest": "^3.0.4"
+    "vitest": "^3.0.4",
+    "eslint": "^9.20.1",
+    "@internal/lint": "0.0.0"
   },
   "keywords": [
     "rag",
@@ -60,6 +62,7 @@
   "scripts": {
     "build": "tsup src/index.ts --format esm --experimental-dts --clean --treeshake",
     "buld:watch": "pnpm build --watch",
-    "test": "vitest run"
+    "test": "vitest run",
+    "lint": "eslint ."
   }
 }

package/src/document/document.ts

@@ -13,7 +13,7 @@ import { RecursiveJsonTransformer } from './transformers/json';
 import { LatexTransformer } from './transformers/latex';
 import { MarkdownHeaderTransformer, MarkdownTransformer } from './transformers/markdown';
 import { TokenTransformer } from './transformers/token';
-import { type ChunkOptions, type ChunkParams, type ChunkStrategy, type ExtractParams } from './types';
+import type { ChunkOptions, ChunkParams, ChunkStrategy, ExtractParams } from './types';
 
 export class MDocument {
   private chunks: Chunk[];

package/src/document/transformers/character.ts

@@ -1,4 +1,5 @@
-import { type ChunkOptions, Language } from '../types';
+import { Language } from '../types';
+import type { ChunkOptions } from '../types';
 
 import { TextTransformer } from './text';
 

package/src/document/transformers/text.ts

@@ -1,8 +1,8 @@
 import { Document } from 'llamaindex';
 
-import { type ChunkOptions } from '../types';
+import type { ChunkOptions } from '../types';
 
-import { type Transformer } from './transformer';
+import type { Transformer } from './transformer';
 
 export abstract class TextTransformer implements Transformer {
   protected size: number;

package/src/document/transformers/token.ts

@@ -1,4 +1,5 @@
-import { encodingForModel, getEncoding, Tiktoken, type TiktokenModel, type TiktokenEncoding } from 'js-tiktoken';
+import type { TiktokenModel, TiktokenEncoding, Tiktoken } from 'js-tiktoken';
+import { encodingForModel, getEncoding } from 'js-tiktoken';
 
 import { TextTransformer } from './text';
 

package/src/document/transformers/transformer.ts

@@ -1,4 +1,4 @@
-import { Document } from 'llamaindex';
+import type { Document } from 'llamaindex';
 
 export interface Transformer {
   transformDocuments(documents: Document[]): Document[];

package/src/document/types.ts

@@ -1,11 +1,11 @@
-import { type TiktokenEncoding, type TiktokenModel } from 'js-tiktoken';
-import {
-  type LLM,
-  type TitleCombinePrompt,
-  type TitleExtractorPrompt,
-  type SummaryPrompt,
-  type QuestionExtractPrompt,
-  type KeywordExtractPrompt,
+import type { TiktokenEncoding, TiktokenModel } from 'js-tiktoken';
+import type {
+  LLM,
+  TitleCombinePrompt,
+  TitleExtractorPrompt,
+  SummaryPrompt,
+  QuestionExtractPrompt,
+  KeywordExtractPrompt,
 } from 'llamaindex';
 
 export enum Language {

package/src/graph-rag/index.test.ts

@@ -1,6 +1,7 @@
 import { describe, it, expect, vi, beforeAll, afterAll, beforeEach, afterEach } from 'vitest';
 
-import { GraphChunk, GraphEdge, GraphEmbedding, GraphNode, GraphRAG } from './';
+import type { GraphChunk, GraphEdge, GraphEmbedding, GraphNode } from './';
+import { GraphRAG } from './';
 
 describe('GraphRAG', () => {
   beforeEach(() => {

package/src/rerank/index.ts

@@ -1,6 +1,7 @@
-import { MastraAgentRelevanceScorer, CohereRelevanceScorer, type RelevanceScoreProvider } from '@mastra/core/relevance';
-import { type QueryResult } from '@mastra/core/vector';
-import { type LanguageModelV1 } from 'ai';
+import { MastraAgentRelevanceScorer, CohereRelevanceScorer } from '@mastra/core/relevance';
+import type { RelevanceScoreProvider } from '@mastra/core/relevance';
+import type { QueryResult } from '@mastra/core/vector';
+import type { LanguageModelV1 } from 'ai';
 
 // Default weights for different scoring components (must add up to 1)
 const DEFAULT_WEIGHTS = {

package/src/tools/document-chunker.ts

@@ -1,7 +1,7 @@
 import { createTool } from '@mastra/core/tools';
 import { z } from 'zod';
 
-import { type ChunkParams, MDocument } from '../document';
+import type { MDocument, ChunkParams } from '../document';
 
 export const createDocumentChunkerTool = ({
   doc,

package/src/tools/graph-rag.ts

@@ -1,5 +1,5 @@
 import { createTool } from '@mastra/core/tools';
-import { type EmbeddingModel } from 'ai';
+import type { EmbeddingModel } from 'ai';
 import { z } from 'zod';
 
 import { GraphRAG } from '../graph-rag';

package/src/tools/vector-query.ts

@@ -1,8 +1,9 @@
 import { createTool } from '@mastra/core/tools';
-import { type EmbeddingModel } from 'ai';
+import type { EmbeddingModel } from 'ai';
 import { z } from 'zod';
 
-import { rerank, type RerankConfig } from '../rerank';
+import { rerank } from '../rerank';
+import type { RerankConfig } from '../rerank';
 import { vectorQuerySearch, defaultVectorQueryDescription } from '../utils';
 
 export const createVectorQueryTool = ({

package/src/utils/vector-search.ts

@@ -1,5 +1,6 @@
-import { type MastraVector, type QueryResult } from '@mastra/core/vector';
-import { embed, type EmbeddingModel } from 'ai';
+import type { MastraVector, QueryResult } from '@mastra/core/vector';
+import { embed } from 'ai';
+import type { EmbeddingModel } from 'ai';
 
 interface VectorQuerySearchParams {
   indexName: string;