@byted-las/contextlake-openclaw 1.0.0 → 1.0.3

package/src/lib/actions/retrieve.ts

@@ -1,14 +1,15 @@
 import { createStorageProvider } from '../../service/storage/factory';
 import { createMetadataProvider } from '../../service/metadata/factory';
+import { ContextLakeConfig } from '../../utils/config';
 
 interface RetrieveParams {
   query: string;
   top_k?: number;
-  filter?: string; // SQL-like filter
+  filter?: string;
   include_binary?: boolean;
 }
 
-export async function retrieveAssets(params: RetrieveParams, config: any, logger?: any) {
+export async function retrieveAssets(params: RetrieveParams, config: ContextLakeConfig, logger?: any) {
   if (logger) {
       logger.info(`[ContextLake-Action] Calling retrieveAssets with params: ${JSON.stringify(params)}`);
   } else {
@@ -17,46 +18,37 @@ export async function retrieveAssets(params: RetrieveParams, config: any, logger
   }
 
   const metaConfig = config.metadata_storage || { type: 'local', lancedb_uri: './data/contextlake' };
-  const metadataProvider = createMetadataProvider(metaConfig);
+  const metadataProvider = createMetadataProvider(metaConfig as any);
   
   let storageProvider: any;
   if (config.file_storage && config.file_storage.type) {
-      storageProvider = createStorageProvider(config.file_storage);
+      storageProvider = createStorageProvider(config.file_storage as any);
   }
   
   await metadataProvider.connect();
 
   const results = await metadataProvider.search(
-    params.query, 
-    params.top_k || 5, 
+    params.query,
+    params.top_k || 5,
     params.filter
   );
-
-  // Post-process to handle binary data retrieval if requested
   
   if (params.include_binary) {
       const enrichedResults = await Promise.all(results.map(async (doc: any) => {
-          // Remove raw binary_data buffer to avoid clone errors in postMessage
-          const { binary_data, ...rest } = doc;
+          const { binary_data, vector, ...rest } = doc;
           
-          if (rest.vector) {
-              rest.vector = Array.from(rest.vector);
-          }
-
           try {
             if (rest.storage_type === 'inline') {
-                // Return inline binary data (Base64)
-                return { 
-                    ...rest, 
-                    binary_content: binary_data ? binary_data.toString('base64') : null 
+                return {
+                    ...rest,
+                    binary_content: binary_data ? binary_data.toString('base64') : null
                 };
             } else if (rest.storage_type === 'local' || rest.storage_type === 'tos') {
                 if (storageProvider) {
-                    // Download from storage provider
                     const buffer = await storageProvider.downloadFile(rest.url);
-                    return { 
-                        ...rest, 
-                        binary_content: buffer.toString('base64') 
+                    return {
+                        ...rest,
+                        binary_content: buffer.toString('base64')
                     };
                 } else {
                     return { ...rest, binary_error: 'Storage provider not configured' };
@@ -70,21 +62,11 @@ export async function retrieveAssets(params: RetrieveParams, config: any, logger
       return enrichedResults;
   }
 
-  // Remove binary_data from default response to keep it light and serializable
   return results.map((doc: any) => {
-      const { binary_data, ...rest } = doc;
+      const { binary_data, vector, ...rest } = doc;
       
-      // Ensure all fields are standard serializable types (remove complex objects/buffers)
-      if (rest.vector) {
-          // Keep vector as standard array of numbers if it exists, but some clients might choke on large arrays.
-          // Usually vector is fine, but LanceDB might return custom objects like Float32Array which fail structuredClone in postMessage
-          rest.vector = Array.from(rest.vector);
-      }
-      
-      // LanceDB records might be Proxy objects or contain non-clonable getters.
-      // We stringify and parse to get a clean, plain Javascript Object.
       const safeDoc = JSON.parse(JSON.stringify(rest));
       
       return safeDoc;
   });
-}
+}