@mastra/rag 1.2.3-alpha.0 → 1.2.3-alpha.2

package/src/utils/default-settings.ts

@@ -1,38 +0,0 @@
-export const defaultVectorQueryDescription = () =>
-  `Access the knowledge base to find information needed to answer user questions.`;
-
-export const defaultGraphRagDescription = () =>
-  `Access and analyze relationships between information in the knowledge base to answer complex questions about connections and patterns.`;
-
-export const queryTextDescription = `The text query to search for in the vector database.
-- ALWAYS provide a non-empty query string
-- Must contain the user's question or search terms
-- Example: "market data" or "financial reports"
-- If the user's query is about a specific topic, use that topic as the queryText
-- Cannot be an empty string
-- Do not include quotes, just the text itself
-- Required for all searches`;
-
-export const topKDescription = `Controls how many matching documents to return.
-- ALWAYS provide a value
-- If no value is provided, use the default (10)
-- Must be a valid and positive number
-- Cannot be NaN
-- Uses provided value if specified
-- Default: 10 results (use this if unsure)
-- Higher values (like 20) provide more context
-- Lower values (like 3) focus on best matches
-- Based on query requirements`;
-
-export const filterDescription = `JSON-formatted criteria to refine search results.
-- ALWAYS provide a filter value
-- If no filter is provided, use the default ("{}")
-- MUST be a valid, complete JSON object with proper quotes and brackets
-- Uses provided filter if specified
-- Default: "{}" (no filtering)
-- Example for no filtering: "filter": "{}"
-- Example: '{"category": "health"}'
-- Based on query intent
-- Do NOT use single quotes or unquoted properties
-- IMPORTANT: Always ensure JSON is properly closed with matching brackets
-- Multiple filters can be combined`;

package/src/utils/index.ts

@@ -1,3 +0,0 @@
-export * from './vector-search';
-export * from './default-settings';
-export * from './tool-schemas';

package/src/utils/tool-schemas.ts

@@ -1,38 +0,0 @@
-import type { Tool } from '@mastra/core/tools';
-import { z } from 'zod';
-import { queryTextDescription, topKDescription, filterDescription } from './default-settings';
-
-export const baseSchema = {
-  queryText: z.string().describe(queryTextDescription),
-  topK: z.coerce.number().describe(topKDescription),
-};
-
-// Output schema includes `sources`, which exposes the full set of retrieved chunks (QueryResult objects)
-// Each source contains all information needed to reference
-// the original document, chunk, and similarity score.
-export const outputSchema = z.object({
-  // Array of metadata or content for compatibility with prior usage
-  relevantContext: z.any(),
-  // Array of full retrieval result objects
-  sources: z.array(
-    z.object({
-      id: z.string(), // Unique chunk/document identifier
-      metadata: z.any(), // All metadata fields (document ID, etc.)
-      vector: z.array(z.number()), // Embedding vector (if available)
-      score: z.number(), // Similarity score for this retrieval
-      document: z.string(), // Full chunk/document text (if available)
-    }),
-  ),
-});
-
-export const filterSchema = z.object({
-  ...baseSchema,
-  filter: z.coerce.string().describe(filterDescription),
-});
-
-export type RagTool<
-  TInput extends z.ZodType<any, z.ZodTypeDef, any> | undefined,
-  TOutput extends z.ZodType<any, z.ZodTypeDef, any> | undefined,
-> = Tool<TInput, TOutput> & {
-  execute: NonNullable<Tool<TInput, TOutput>['execute']>;
-};