@retrivora-ai/rag-engine 1.1.9 → 1.2.1

package/src/core/Pipeline.ts

@@ -423,4 +423,59 @@ Optimized Search Query:`;
 
     return rewrite.trim() || question;
   }
+
+  /**
+   * Generate 3-5 short, relevant questions based on the vector database content.
+   */
+  async getSuggestions(query: string, namespace?: string): Promise<string[]> {
+    if (!query || query.trim().length < 3) {
+      return [];
+    }
+
+    await this.initialize();
+    const ns = namespace ?? this.config.projectId;
+
+    try {
+      // 1. Retrieve relevant context (top 3 matches)
+      const { sources } = await this.retrieve(query, { namespace: ns, topK: 3 });
+
+      if (sources.length === 0) {
+        return [];
+      }
+
+      // 2. Generate suggestions using LLM
+      const context = sources.map((s) => s.content).join('\n\n---\n\n');
+      const prompt = `
+Based on the following snippets from a document, what are 3 short, relevant questions a user might ask? 
+Focus on questions that can be answered by the context.
+Keep each question under 10 words and make them very specific to the content.
+Return ONLY a JSON array of strings like ["Question 1", "Question 2", "Question 3"].
+
+Context:
+${context}
+
+Suggestions:`;
+
+      const response = await this.llmProvider.chat(
+        [
+          { role: 'system', content: 'You are a helpful assistant that generates search suggestions.' },
+          { role: 'user', content: prompt }
+        ],
+        ''
+      );
+
+      // Simple parsing of JSON array from LLM response
+      const match = response.match(/\[[\s\S]*\]/);
+      if (match) {
+        const suggestions = JSON.parse(match[0]);
+        if (Array.isArray(suggestions)) {
+          return suggestions.map(s => String(s)).slice(0, 3);
+        }
+      }
+    } catch (error) {
+      console.error('[Pipeline] Failed to generate suggestions:', error);
+    }
+
+    return [];
+  }
 }

package/src/core/VectorPlugin.ts

@@ -81,4 +81,11 @@ export class VectorPlugin {
     return this.pipeline.ingest(documents, namespace);
   }
 
+  /**
+   * Get auto-suggestions based on a query prefix.
+   */
+  async getSuggestions(query: string, namespace?: string): Promise<string[]> {
+    await this.validationPromise;
+    return this.pipeline.getSuggestions(query, namespace);
+  }
 }

package/src/handlers/index.ts

@@ -261,5 +261,35 @@ export function createUploadHandler(configOrPlugin?: Partial<RagConfig> | Vector
   };
 }
 
+/**
+ * createSuggestionsHandler — factory for the auto-suggestions endpoint.
+ */
+export function createSuggestionsHandler(configOrPlugin?: Partial<RagConfig> | VectorPlugin) {
+  const plugin =
+    configOrPlugin instanceof VectorPlugin
+      ? configOrPlugin
+      : new VectorPlugin(configOrPlugin);
+
+  return async function POST(req: NextRequest) {
+    try {
+      const body = await req.json();
+      const { query, namespace } = body as {
+        query: string;
+        namespace?: string;
+      };
+
+      if (typeof query !== 'string') {
+        return NextResponse.json({ error: 'query is required' }, { status: 400 });
+      }
+
+      const suggestions = await plugin.getSuggestions(query, namespace);
+      return NextResponse.json({ suggestions });
+    } catch (err) {
+      const message = err instanceof Error ? err.message : 'Internal server error';
+      return NextResponse.json({ error: message }, { status: 500 });
+    }
+  };
+}
+
 // Re-export SSE helper so host apps can use it in custom handlers
 export { sseFrame, sseTextFrame, sseMetaFrame, sseErrorFrame };

package/src/types/index.ts

@@ -50,3 +50,17 @@ export interface RetrievalResult {
 export interface IRetriever {
   retrieve(query: string, options?: Record<string, unknown>): Promise<RetrievalResult>;
 }
+
+export interface SuggestionsResponse {
+  suggestions: string[];
+}
+
+export interface Product {
+  id: string | number;
+  name: string;
+  brand?: string;
+  price?: string | number;
+  image?: string;
+  link?: string;
+  description?: string;
+}