@rws-framework/ai-tools 3.2.6 → 3.3.1

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@rws-framework/ai-tools",
   "private": false,
-  "version": "3.2.6",
+  "version": "3.3.1",
   "description": "",
   "main": "src/index.ts",
   "scripts": {},

package/src/models/prompts/inc/input-output-manager.ts

@@ -36,8 +36,7 @@ export class InputOutputManager {
     readInput(): CompoundInput[] {
         const enchantedInput: CompoundInput[] = this.enhancedInput.map(enchantment => ({ 
             role: enchantment.input.role || 'user', 
-            type: enchantment.input.type, 
-            text: enchantment.input.text 
+            content: enchantment.input.content
         }));
 
         return [...enchantedInput, ...this.input];
@@ -87,7 +86,13 @@ export class InputOutputManager {
         if (callback) {
             callback(messages, prompt);
         } else {
-            this.input = [{ type: 'text', text: prompt }, ...this.input];
+            this.input = [{ 
+                role: 'user', 
+                content: { 
+                    type: 'text', 
+                    text: prompt 
+                } 
+            }, ...this.input];
         }
     }
 

package/src/services/LangChainRAGService.ts

@@ -141,13 +141,13 @@ export class LangChainRAGService {
      */
     async searchKnowledge(request: IRAGSearchRequest): Promise<IRAGResponse<{ results: ISearchResult[] }>> {
         this.log('log', `[SEARCH] Starting knowledge search for query: "${request.query}"`);
-        this.log('debug', `[SEARCH] Search parameters: maxResults=${request.maxResults || 5}, threshold=${request.threshold || 0.3}`);
+        this.log('debug', `[SEARCH] Search parameters: maxResults=${request.maxResults || 5}, threshold=${request.threshold || 0.3}, temporaryDocumentSearch=${request.temporaryDocumentSearch}`);
 
         try {
             await this.ensureInitialized();
 
             const knowledgeIds = request.filter?.knowledgeIds || [];
-            console.log('knowledgeIds', knowledgeIds);
+            console.log('knowledgeIds', knowledgeIds, 'temporaryDocumentSearch:', request.temporaryDocumentSearch);
             
             if (knowledgeIds.length === 0) {
                 this.log('warn', '[SEARCH] No knowledge IDs provided for search, returning empty results');
@@ -157,16 +157,29 @@ export class LangChainRAGService {
                 };
             }
 
-            // Load all knowledge vectors in parallel
+            // Load all knowledge vectors in parallel (including temporary documents)
             const knowledgeVectorPromises = knowledgeIds.map(async (knowledgeId) => {
-                const vectorData = await this.loadKnowledgeVectorWithEmbeddings(knowledgeId);
-                return {
-                    knowledgeId,                    
-                    chunks: vectorData.chunks
-                };
+                try {
+                    const vectorData = await this.loadKnowledgeVectorWithEmbeddings(knowledgeId);
+                    return {
+                        knowledgeId,                    
+                        chunks: vectorData.chunks
+                    };
+                } catch (loadError) {
+                    this.log('warn', `[SEARCH] Failed to load knowledge ${knowledgeId}:`, loadError);
+                    return null;
+                }
             });
 
-            const knowledgeVectors = await Promise.all(knowledgeVectorPromises);
+            const knowledgeVectors = (await Promise.all(knowledgeVectorPromises)).filter(v => v !== null);
+            
+            if (knowledgeVectors.length === 0) {
+                this.log('warn', '[SEARCH] No knowledge vectors could be loaded for search');
+                return {
+                    success: true,
+                    data: { results: [] }
+                };
+            }
 
             // Use optimized vector search service
             const searchResponse = await this.vectorSearchService.searchSimilar({
@@ -205,25 +218,25 @@ export class LangChainRAGService {
     /**
      * Remove knowledge from index
      */
-    async removeKnowledge(knowledgeId: string | number): Promise<boolean> {
-        this.log('log', `[REMOVE] Starting removal of knowledge: ${knowledgeId}`);
+    async removeKnowledge(fileId: string | number): Promise<boolean> {
+        this.log('log', `[REMOVE] Starting removal of knowledge: ${fileId}`);
 
         try {
             await this.ensureInitialized();
 
             // Remove the individual knowledge vector file
-            const vectorFilePath = this.getKnowledgeVectorPath(knowledgeId);
+            const vectorFilePath = this.getKnowledgeVectorPath(fileId);
             if (fs.existsSync(vectorFilePath)) {
                 fs.unlinkSync(vectorFilePath);
-                this.log('log', `[REMOVE] Successfully removed vector file for knowledge ${knowledgeId}`);
+                this.log('log', `[REMOVE] Successfully removed vector file for knowledge ${fileId}`);
                 return true;
             } else {
-                this.log('warn', `[REMOVE] Vector file not found for knowledge ${knowledgeId}`);
+                this.log('warn', `[REMOVE] Vector file not found for knowledge ${fileId}`);
                 return true; // Consider it successful if file doesn't exist
             }
 
         } catch (error: any) {
-            this.log('error', `[REMOVE] Failed to remove knowledge ${knowledgeId}:`, error);
+            this.log('error', `[REMOVE] Failed to remove knowledge ${fileId}:`, error);
             return false;
         }
     }

package/src/types/rag.types.ts

@@ -24,6 +24,7 @@ export interface IRAGSearchRequest {
     query: string;
     maxResults?: number;
     threshold?: number;
+    temporaryDocumentSearch?: boolean; // Flag for searching temporary documents (web search)
     filter?: {
         knowledgeIds?: (string | number)[];
         documentIds?: (string | number)[];