@cmdoss/memwal-sdk 0.6.0

package/dist/hooks.wip/useMemoryServices.js

@@ -0,0 +1,264 @@
+/**
+ * useMemoryServices - Core hook for client-side memory services
+ *
+ * Manages lifecycle of browser-compatible services:
+ * - EmbeddingService (Gemini API)
+ * - BrowserHnswIndexService (hnswlib-wasm + IndexedDB)
+ * - BrowserKnowledgeGraphManager (IndexedDB)
+ *
+ * Services are singletons per user address, shared across components.
+ */
+import { useEffect, useState, useMemo } from 'react';
+import { EmbeddingService } from '../services/EmbeddingService.js';
+import { GeminiAIService } from '../services/GeminiAIService.js';
+import { BrowserHnswIndexService } from '../vector/BrowserHnswIndexService.js';
+import { BrowserKnowledgeGraphManager } from '../graph/BrowserKnowledgeGraphManager.js';
+import { StorageService } from '../services/StorageService.js';
+import { EncryptionService } from '../services/EncryptionService.js';
+// Singleton store for services (shared across all hooks)
+const servicesStore = new Map();
+/**
+ * Initialize and manage memory services for client-side operations
+ *
+ * @param userAddress - User's blockchain address (used as unique identifier)
+ * @param config - Optional configuration for services
+ * @returns Memory services and loading state
+ *
+ * @example
+ * ```tsx
+ * function MyComponent() {
+ *   const account = useCurrentAccount();
+ *   const { embeddingService, hnswService, isReady } = useMemoryServices(
+ *     account?.address,
+ *     { geminiApiKey: process.env.NEXT_PUBLIC_GEMINI_API_KEY }
+ *   );
+ *
+ *   if (!isReady) return <Loading />;
+ *   // Use services...
+ * }
+ * ```
+ */
+export function useMemoryServices(userAddress, config = {}) {
+    const [isReady, setIsReady] = useState(false);
+    const [isLoading, setIsLoading] = useState(false);
+    const [error, setError] = useState(null);
+    // Memoize config to prevent re-initialization
+    const memoizedConfig = useMemo(() => ({
+        geminiApiKey: config.geminiApiKey || '',
+        embeddingModel: config.embeddingModel || 'text-embedding-004',
+        embeddingDimension: config.embeddingDimension || 3072,
+        hnswMaxElements: config.hnswMaxElements || 10000,
+        hnswM: config.hnswM || 16,
+        hnswEfConstruction: config.hnswEfConstruction || 200,
+        batchSize: config.batchSize || 50,
+        batchDelayMs: config.batchDelayMs || 5000,
+        packageId: config.packageId || '',
+        walrusAggregator: config.walrusAggregator || 'https://aggregator.walrus-testnet.walrus.space',
+        walrusPublisher: config.walrusPublisher || 'https://publisher.walrus-testnet.walrus.space',
+        sealServerObjectIds: config.sealServerObjectIds || [
+            '0x73d05d62c18d9374e3ea529e8e0ed6161da1a141a94d3f76ae3fe4e99356db75',
+            '0xf5d14a81a982144ae441cd7d64b09027f116a468bd36e7eca494f750591623c8'
+        ],
+        suiClient: config.suiClient
+    }), [
+        config.geminiApiKey,
+        config.embeddingModel,
+        config.embeddingDimension,
+        config.hnswMaxElements,
+        config.hnswM,
+        config.hnswEfConstruction,
+        config.batchSize,
+        config.batchDelayMs,
+        config.packageId,
+        config.walrusAggregator,
+        config.walrusPublisher,
+        config.sealServerObjectIds,
+        config.suiClient
+    ]);
+    // Initialize services on mount
+    useEffect(() => {
+        if (!userAddress) {
+            setIsReady(false);
+            return;
+        }
+        // Increment ref count on mount
+        if (servicesStore.has(userAddress)) {
+            const existing = servicesStore.get(userAddress);
+            existing.refCount++;
+            setIsReady(true);
+            console.log(`✅ Using existing services for ${userAddress} (refCount: ${existing.refCount})`);
+        }
+        let mounted = true;
+        const initializeServices = async () => {
+            try {
+                setIsLoading(true);
+                setError(null);
+                // Check if services already exist (already checked above, but keeping for clarity)
+                if (servicesStore.has(userAddress)) {
+                    setIsReady(true);
+                    setIsLoading(false);
+                    return;
+                }
+                // DEBUG: Log the API key status
+                console.log('🔍 useMemoryServices - Initializing with config:', {
+                    hasGeminiApiKey: !!memoizedConfig.geminiApiKey,
+                    apiKeyPreview: memoizedConfig.geminiApiKey ? `${memoizedConfig.geminiApiKey.substring(0, 10)}...` : 'UNDEFINED',
+                    model: memoizedConfig.embeddingModel,
+                    dimensions: memoizedConfig.embeddingDimension
+                });
+                // Initialize embedding service
+                const embeddingService = new EmbeddingService({
+                    apiKey: memoizedConfig.geminiApiKey,
+                    model: memoizedConfig.embeddingModel,
+                    dimensions: memoizedConfig.embeddingDimension
+                });
+                // Initialize Gemini AI service for metadata extraction
+                const geminiAIService = new GeminiAIService({
+                    apiKey: memoizedConfig.geminiApiKey,
+                    model: process.env.AI_CHAT_MODEL || 'google/gemini-2.5-flash',
+                    temperature: 0.1
+                });
+                // Initialize HNSW service
+                const hnswService = new BrowserHnswIndexService({
+                    dimension: memoizedConfig.embeddingDimension,
+                    maxElements: memoizedConfig.hnswMaxElements,
+                    m: memoizedConfig.hnswM,
+                    efConstruction: memoizedConfig.hnswEfConstruction
+                }, {
+                    maxBatchSize: memoizedConfig.batchSize,
+                    batchDelayMs: memoizedConfig.batchDelayMs
+                });
+                // Initialize knowledge graph manager
+                const graphManager = new BrowserKnowledgeGraphManager();
+                // Initialize storage service
+                const storageService = new StorageService({
+                    walrusAggregatorUrl: memoizedConfig.walrusAggregator,
+                    walrusPublisherUrl: memoizedConfig.walrusPublisher,
+                    packageId: memoizedConfig.packageId,
+                    suiClient: memoizedConfig.suiClient,
+                    network: 'testnet',
+                    useUploadRelay: true
+                });
+                // Initialize encryption service
+                const encryptionService = new EncryptionService(memoizedConfig.suiClient, {
+                    packageId: memoizedConfig.packageId,
+                    encryptionConfig: {
+                        enabled: true,
+                        keyServers: memoizedConfig.sealServerObjectIds
+                    }
+                });
+                // Try to load existing index from IndexedDB
+                try {
+                    await hnswService.loadIndexFromDB(userAddress);
+                    console.log('✅ Loaded existing HNSW index from IndexedDB');
+                }
+                catch (err) {
+                    console.log('ℹ️ No existing index found, will create on first add');
+                }
+                // Store services
+                servicesStore.set(userAddress, {
+                    embedding: embeddingService,
+                    geminiAI: geminiAIService,
+                    hnsw: hnswService,
+                    graph: graphManager,
+                    storage: storageService,
+                    encryption: encryptionService,
+                    refCount: 1
+                });
+                if (mounted) {
+                    setIsReady(true);
+                    setIsLoading(false);
+                }
+            }
+            catch (err) {
+                console.error('Failed to initialize memory services:', err);
+                if (mounted) {
+                    setError(err);
+                    setIsLoading(false);
+                }
+            }
+        };
+        initializeServices();
+        // Cleanup on unmount
+        return () => {
+            mounted = false;
+            if (userAddress && servicesStore.has(userAddress)) {
+                const services = servicesStore.get(userAddress);
+                services.refCount--;
+                // Destroy services if no more references
+                if (services.refCount <= 0) {
+                    console.log(`🧹 Cleaning up services for user ${userAddress}`);
+                    services.hnsw.destroy();
+                    services.graph.destroy();
+                    servicesStore.delete(userAddress);
+                }
+            }
+        };
+    }, [userAddress, memoizedConfig]);
+    // Return current services (get from store directly)
+    if (!userAddress) {
+        return {
+            embeddingService: null,
+            geminiAIService: null,
+            hnswService: null,
+            graphManager: null,
+            storageService: null,
+            encryptionService: null,
+            isReady: false,
+            isLoading,
+            error
+        };
+    }
+    // Get services from store
+    const services = servicesStore.get(userAddress);
+    if (!services || !isReady) {
+        return {
+            embeddingService: null,
+            geminiAIService: null,
+            hnswService: null,
+            graphManager: null,
+            storageService: null,
+            encryptionService: null,
+            isReady: false,
+            isLoading,
+            error
+        };
+    }
+    return {
+        embeddingService: services.embedding,
+        geminiAIService: services.geminiAI,
+        hnswService: services.hnsw,
+        graphManager: services.graph,
+        storageService: services.storage,
+        encryptionService: services.encryption,
+        isReady,
+        isLoading,
+        error
+    };
+}
+/**
+ * Clear all services for a specific user (useful for logout)
+ */
+export function clearMemoryServices(userAddress) {
+    if (servicesStore.has(userAddress)) {
+        const services = servicesStore.get(userAddress);
+        services.hnsw.destroy();
+        services.graph.destroy();
+        servicesStore.delete(userAddress);
+        console.log(`✅ Cleared services for user ${userAddress}`);
+    }
+}
+/**
+ * Get current service statistics
+ */
+export function getMemoryServicesStats() {
+    return {
+        activeUsers: servicesStore.size,
+        services: Array.from(servicesStore.entries()).map(([userAddress, services]) => ({
+            userAddress,
+            refCount: services.refCount,
+            hnswStats: services.hnsw.getCacheStats()
+        }))
+    };
+}
+//# sourceMappingURL=useMemoryServices.js.map

package/dist/hooks.wip/useMemoryServices.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useMemoryServices.js","sourceRoot":"","sources":["../../src/hooks.wip/useMemoryServices.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,6BAA6B,CAAC;AAC9D,OAAO,EAAE,uBAAuB,EAAE,MAAM,mCAAmC,CAAC;AAC5E,OAAO,EAAE,4BAA4B,EAAE,MAAM,uCAAuC,CAAC;AACrF,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AA8BlE,yDAAyD;AACzD,MAAM,aAAa,GAAG,IAAI,GAAG,EAQzB,CAAC;AAEL;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAM,UAAU,iBAAiB,CAC/B,WAAoB,EACpB,SAA+B,EAAE;IAEjC,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC9C,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAClD,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAe,IAAI,CAAC,CAAC;IAEvD,8CAA8C;IAC9C,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;QACpC,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,EAAE;QACvC,cAAc,EAAE,MAAM,CAAC,cAAc,IAAI,oBAAoB;QAC7D,kBAAkB,EAAE,MAAM,CAAC,kBAAkB,IAAI,IAAI;QACrD,eAAe,EAAE,MAAM,CAAC,eAAe,IAAI,KAAK;QAChD,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,EAAE;QACzB,kBAAkB,EAAE,MAAM,CAAC,kBAAkB,IAAI,GAAG;QACpD,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,EAAE;QACjC,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,IAAI;QACzC,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,EAAE;QACjC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,IAAI,gDAAgD;QAC7F,eAAe,EAAE,MAAM,CAAC,eAAe,IAAI,+CAA+C;QAC1F,mBAAmB,EAAE,MAAM,CAAC,mBAAmB,IAAI;YACjD,oEAAoE;YACpE,oEAAoE;SACrE;QACD,SAAS,EAAE,MAAM,CAAC,SAAS;KAC5B,CAAC,EAAE;QACF,MAAM,CAAC,YAAY;QACnB,MAAM,CAAC,cAAc;QACrB,MAAM,CAAC,kBAAkB;QACzB,MAAM,CAAC,eAAe;QACtB,MAAM,CAAC,KAAK;QACZ,MAAM,CAAC,kBAAkB;QACzB,MAAM,CAAC,SAAS;QAChB,MAAM,CAAC,YAAY;QACnB,MAAM,CAAC,SAAS;QAChB,MAAM,CAAC,gBAAgB;QACvB,MAAM,CAAC,eAAe;QACtB,MAAM,CAAC,mBAAmB;QAC1B,MAAM,CAAC,SAAS;KACjB,CAAC,CAAC;IAEH,+BAA+B;IAC/B,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,UAAU,CAAC,KAAK,CAAC,CAAC;YAClB,OAAO;QACT,CAAC;QAED,+BAA+B;QAC/B,IAAI,aAAa,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;YACnC,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,WAAW,CAAE,CAAC;YACjD,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACpB,UAAU,CAAC,IAAI,CAAC,CAAC;YACjB,OAAO,CAAC,GAAG,CAAC,iCAAiC,WAAW,eAAe,QAAQ,CAAC,QAAQ,GAAG,CAAC,CAAC;QAC/F,CAAC;QAED,IAAI,OAAO,GAAG,IAAI,CAAC;QAEnB,MAAM,kBAAkB,GAAG,KAAK,IAAI,EAAE;YACpC,IAAI,CAAC;gBACH,YAAY,CAAC,IAAI,CAAC,CAAC;gBACnB,QAAQ,CAAC,IAAI,CAAC,CAAC;gBAEf,mFAAmF;gBACnF,IAAI,aAAa,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;oBACnC,UAAU,CAAC,IAAI,CAAC,CAAC;oBACjB,YAAY,CAAC,KAAK,CAAC,CAAC;oBACpB,OAAO;gBACT,CAAC;gBAED,gCAAgC;gBAChC,OAAO,CAAC,GAAG,CAAC,kDAAkD,EAAE;oBAC9D,eAAe,EAAE,CAAC,CAAC,cAAc,CAAC,YAAY;oBAC9C,aAAa,EAAE,cAAc,CAAC,YAAY,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,WAAW;oBAC/G,KAAK,EAAE,cAAc,CAAC,cAAc;oBACpC,UAAU,EAAE,cAAc,CAAC,kBAAkB;iBAC9C,CAAC,CAAC;gBAEH,+BAA+B;gBAC/B,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAAC;oBAC5C,MAAM,EAAE,cAAc,CAAC,YAAY;oBACnC,KAAK,EAAE,cAAc,CAAC,cAAc;oBACpC,UAAU,EAAE,cAAc,CAAC,kBAAkB;iBAC9C,CAAC,CAAC;gBAEH,uDAAuD;gBACvD,MAAM,eAAe,GAAG,IAAI,eAAe,CAAC;oBAC1C,MAAM,EAAE,cAAc,CAAC,YAAY;oBACnC,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,yBAAyB;oBAC7D,WAAW,EAAE,GAAG;iBACjB,CAAC,CAAC;gBAEH,0BAA0B;gBAC1B,MAAM,WAAW,GAAG,IAAI,uBAAuB,CAC7C;oBACE,SAAS,EAAE,cAAc,CAAC,kBAAkB;oBAC5C,WAAW,EAAE,cAAc,CAAC,eAAe;oBAC3C,CAAC,EAAE,cAAc,CAAC,KAAK;oBACvB,cAAc,EAAE,cAAc,CAAC,kBAAkB;iBAClD,EACD;oBACE,YAAY,EAAE,cAAc,CAAC,SAAS;oBACtC,YAAY,EAAE,cAAc,CAAC,YAAY;iBAC1C,CACF,CAAC;gBAEF,qCAAqC;gBACrC,MAAM,YAAY,GAAG,IAAI,4BAA4B,EAAE,CAAC;gBAExD,6BAA6B;gBAC7B,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC;oBACxC,mBAAmB,EAAE,cAAc,CAAC,gBAAgB;oBACpD,kBAAkB,EAAE,cAAc,CAAC,eAAe;oBAClD,SAAS,EAAE,cAAc,CAAC,SAAS;oBACnC,SAAS,EAAE,cAAc,CAAC,SAAS;oBACnC,OAAO,EAAE,SAAS;oBAClB,cAAc,EAAE,IAAI;iBACrB,CAAC,CAAC;gBAEH,gCAAgC;gBAChC,MAAM,iBAAiB,GAAG,IAAI,iBAAiB,CAC7C,cAAc,CAAC,SAAS,EACxB;oBACE,SAAS,EAAE,cAAc,CAAC,SAAS;oBACnC,gBAAgB,EAAE;wBAChB,OAAO,EAAE,IAAI;wBACb,UAAU,EAAE,cAAc,CAAC,mBAAmB;qBAC/C;iBACF,CACF,CAAC;gBAEF,4CAA4C;gBAC5C,IAAI,CAAC;oBACH,MAAM,WAAW,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;oBAC/C,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;gBAC7D,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,OAAO,CAAC,GAAG,CAAC,sDAAsD,CAAC,CAAC;gBACtE,CAAC;gBAED,iBAAiB;gBACjB,aAAa,CAAC,GAAG,CAAC,WAAW,EAAE;oBAC7B,SAAS,EAAE,gBAAgB;oBAC3B,QAAQ,EAAE,eAAe;oBACzB,IAAI,EAAE,WAAW;oBACjB,KAAK,EAAE,YAAY;oBACnB,OAAO,EAAE,cAAc;oBACvB,UAAU,EAAE,iBAAiB;oBAC7B,QAAQ,EAAE,CAAC;iBACZ,CAAC,CAAC;gBAEH,IAAI,OAAO,EAAE,CAAC;oBACZ,UAAU,CAAC,IAAI,CAAC,CAAC;oBACjB,YAAY,CAAC,KAAK,CAAC,CAAC;gBACtB,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,uCAAuC,EAAE,GAAG,CAAC,CAAC;gBAC5D,IAAI,OAAO,EAAE,CAAC;oBACZ,QAAQ,CAAC,GAAY,CAAC,CAAC;oBACvB,YAAY,CAAC,KAAK,CAAC,CAAC;gBACtB,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QAEF,kBAAkB,EAAE,CAAC;QAErB,qBAAqB;QACrB,OAAO,GAAG,EAAE;YACV,OAAO,GAAG,KAAK,CAAC;YAEhB,IAAI,WAAW,IAAI,aAAa,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;gBAClD,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,WAAW,CAAE,CAAC;gBACjD,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBAEpB,yCAAyC;gBACzC,IAAI,QAAQ,CAAC,QAAQ,IAAI,CAAC,EAAE,CAAC;oBAC3B,OAAO,CAAC,GAAG,CAAC,oCAAoC,WAAW,EAAE,CAAC,CAAC;oBAC/D,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;oBACxB,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;oBACzB,aAAa,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;gBACpC,CAAC;YACH,CAAC;QACH,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC,CAAC;IAElC,oDAAoD;IACpD,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO;YACL,gBAAgB,EAAE,IAAI;YACtB,eAAe,EAAE,IAAI;YACrB,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,IAAI;YAClB,cAAc,EAAE,IAAI;YACpB,iBAAiB,EAAE,IAAI;YACvB,OAAO,EAAE,KAAK;YACd,SAAS;YACT,KAAK;SACN,CAAC;IACJ,CAAC;IAED,0BAA0B;IAC1B,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAEhD,IAAI,CAAC,QAAQ,IAAI,CAAC,OAAO,EAAE,CAAC;QAC1B,OAAO;YACL,gBAAgB,EAAE,IAAI;YACtB,eAAe,EAAE,IAAI;YACrB,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,IAAI;YAClB,cAAc,EAAE,IAAI;YACpB,iBAAiB,EAAE,IAAI;YACvB,OAAO,EAAE,KAAK;YACd,SAAS;YACT,KAAK;SACN,CAAC;IACJ,CAAC;IAED,OAAO;QACL,gBAAgB,EAAE,QAAQ,CAAC,SAAS;QACpC,eAAe,EAAE,QAAQ,CAAC,QAAQ;QAClC,WAAW,EAAE,QAAQ,CAAC,IAAI;QAC1B,YAAY,EAAE,QAAQ,CAAC,KAAK;QAC5B,cAAc,EAAE,QAAQ,CAAC,OAAO;QAChC,iBAAiB,EAAE,QAAQ,CAAC,UAAU;QACtC,OAAO;QACP,SAAS;QACT,KAAK;KACN,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,WAAmB;IACrD,IAAI,aAAa,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;QACnC,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,WAAW,CAAE,CAAC;QACjD,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QACxB,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;QACzB,aAAa,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAClC,OAAO,CAAC,GAAG,CAAC,+BAA+B,WAAW,EAAE,CAAC,CAAC;IAC5D,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,sBAAsB;IACpC,OAAO;QACL,WAAW,EAAE,aAAa,CAAC,IAAI;QAC/B,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,WAAW,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC;YAC9E,WAAW;YACX,QAAQ,EAAE,QAAQ,CAAC,QAAQ;YAC3B,SAAS,EAAE,QAAQ,CAAC,IAAI,CAAC,aAAa,EAAE;SACzC,CAAC,CAAC;KACJ,CAAC;AACJ,CAAC"}

package/dist/hooks.wip/usePDWRAG.d.ts

@@ -0,0 +1,112 @@
+/**
+ * usePDWRAG - React Hook for RAG with PDWVectorStore
+ *
+ * Convenience hook for building RAG applications in React.
+ * Combines usePDWVectorStore with RAG chain creation.
+ *
+ * @example
+ * ```typescript
+ * import { usePDWRAG } from 'personal-data-wallet-sdk/hooks';
+ * import { useCurrentAccount } from '@mysten/dapp-kit';
+ * import { ChatGoogleGenerativeAI } from '@langchain/google-genai';
+ *
+ * function MyRAGApp() {
+ *   const account = useCurrentAccount();
+ *   const llm = new ChatGoogleGenerativeAI({ apiKey: geminiApiKey });
+ *
+ *   const { query, isProcessing, answer, sources, error } = usePDWRAG({
+ *     userAddress: account?.address,
+ *     packageId: '0x...',
+ *     geminiApiKey: process.env.NEXT_PUBLIC_GEMINI_API_KEY!,
+ *     llm,
+ *   });
+ *
+ *   const handleAsk = () => {
+ *     query('What did I do last week?');
+ *   };
+ *
+ *   return (
+ *     <div>
+ *       <button onClick={handleAsk} disabled={isProcessing}>
+ *         Ask AI
+ *       </button>
+ *       {answer && <p>{answer}</p>}
+ *     </div>
+ *   );
+ * }
+ * ```
+ */
+import type { Runnable } from '@langchain/core/runnables';
+import type { BaseMessage } from '@langchain/core/messages';
+import { type UsePDWVectorStoreOptions } from './usePDWVectorStore';
+export interface UsePDWRAGOptions extends UsePDWVectorStoreOptions {
+    /**
+     * LangChain LLM for generation (any chat model like ChatOpenAI, ChatAnthropic, ChatGoogleGenerativeAI, etc.)
+     */
+    llm: Runnable<any, BaseMessage>;
+    /**
+     * System prompt for the RAG chain
+     */
+    systemPrompt?: string;
+    /**
+     * Number of documents to retrieve
+     * @default 5
+     */
+    k?: number;
+    /**
+     * Minimum similarity threshold
+     * @default 0.5
+     */
+    minSimilarity?: number;
+    /**
+     * Whether to return source documents
+     * @default false
+     */
+    returnSourceDocuments?: boolean;
+    /**
+     * Metadata filters for retrieval
+     */
+    filter?: Record<string, any>;
+}
+export interface UsePDWRAGReturn {
+    /**
+     * Query function - ask a question
+     */
+    query: (question: string) => Promise<void>;
+    /**
+     * Whether a query is currently processing
+     */
+    isProcessing: boolean;
+    /**
+     * Latest answer from the RAG chain
+     */
+    answer: string | null;
+    /**
+     * Source documents (if returnSourceDocuments is true)
+     */
+    sources: Array<{
+        content: string;
+        metadata: any;
+        similarity?: number;
+    }> | null;
+    /**
+     * Error if query failed
+     */
+    error: Error | null;
+    /**
+     * Whether the RAG system is ready
+     */
+    isReady: boolean;
+    /**
+     * Clear current answer and sources
+     */
+    clear: () => void;
+}
+/**
+ * React hook for RAG with PDWVectorStore
+ *
+ * Combines vector store initialization with RAG chain creation
+ * and provides a simple query interface.
+ */
+export declare function usePDWRAG(options: UsePDWRAGOptions): UsePDWRAGReturn;
+//# sourceMappingURL=usePDWRAG.d.ts.map

package/dist/hooks.wip/usePDWRAG.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"usePDWRAG.d.ts","sourceRoot":"","sources":["../../src/hooks.wip/usePDWRAG.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAqCG;AAGH,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAqB,KAAK,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAIvF,MAAM,WAAW,gBAAiB,SAAQ,wBAAwB;IAChE;;OAEG;IACH,GAAG,EAAE,QAAQ,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;IAEhC;;OAEG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IAEtB;;;OAGG;IACH,CAAC,CAAC,EAAE,MAAM,CAAC;IAEX;;;OAGG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;IAEvB;;;OAGG;IACH,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAEhC;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC9B;AAED,MAAM,WAAW,eAAe;IAC9B;;OAEG;IACH,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAE3C;;OAEG;IACH,YAAY,EAAE,OAAO,CAAC;IAEtB;;OAEG;IACH,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IAEtB;;OAEG;IACH,OAAO,EAAE,KAAK,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,EAAE,GAAG,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,CAAC,GAAG,IAAI,CAAC;IAEV;;OAEG;IACH,KAAK,EAAE,KAAK,GAAG,IAAI,CAAC;IAEpB;;OAEG;IACH,OAAO,EAAE,OAAO,CAAC;IAEjB;;OAEG;IACH,KAAK,EAAE,MAAM,IAAI,CAAC;CACnB;AAED;;;;;GAKG;AACH,wBAAgB,SAAS,CAAC,OAAO,EAAE,gBAAgB,GAAG,eAAe,CAyFpE"}

package/dist/hooks.wip/usePDWRAG.js

@@ -0,0 +1,121 @@
+/**
+ * usePDWRAG - React Hook for RAG with PDWVectorStore
+ *
+ * Convenience hook for building RAG applications in React.
+ * Combines usePDWVectorStore with RAG chain creation.
+ *
+ * @example
+ * ```typescript
+ * import { usePDWRAG } from 'personal-data-wallet-sdk/hooks';
+ * import { useCurrentAccount } from '@mysten/dapp-kit';
+ * import { ChatGoogleGenerativeAI } from '@langchain/google-genai';
+ *
+ * function MyRAGApp() {
+ *   const account = useCurrentAccount();
+ *   const llm = new ChatGoogleGenerativeAI({ apiKey: geminiApiKey });
+ *
+ *   const { query, isProcessing, answer, sources, error } = usePDWRAG({
+ *     userAddress: account?.address,
+ *     packageId: '0x...',
+ *     geminiApiKey: process.env.NEXT_PUBLIC_GEMINI_API_KEY!,
+ *     llm,
+ *   });
+ *
+ *   const handleAsk = () => {
+ *     query('What did I do last week?');
+ *   };
+ *
+ *   return (
+ *     <div>
+ *       <button onClick={handleAsk} disabled={isProcessing}>
+ *         Ask AI
+ *       </button>
+ *       {answer && <p>{answer}</p>}
+ *     </div>
+ *   );
+ * }
+ * ```
+ */
+import { useState, useCallback } from 'react';
+import { usePDWVectorStore } from './usePDWVectorStore.js';
+import { createPDWRAG, createPDWRAGWithSources } from '../langchain/createPDWRAG.js';
+/**
+ * React hook for RAG with PDWVectorStore
+ *
+ * Combines vector store initialization with RAG chain creation
+ * and provides a simple query interface.
+ */
+export function usePDWRAG(options) {
+    const { llm, systemPrompt, k = 5, minSimilarity = 0.5, returnSourceDocuments = false, filter, ...vectorStoreOptions } = options;
+    // Initialize vector store
+    const { vectorStore, isReady: vectorStoreReady, error: vectorStoreError } = usePDWVectorStore(vectorStoreOptions);
+    // Query state
+    const [isProcessing, setIsProcessing] = useState(false);
+    const [answer, setAnswer] = useState(null);
+    const [sources, setSources] = useState(null);
+    const [error, setError] = useState(vectorStoreError);
+    // Query function
+    const query = useCallback(async (question) => {
+        if (!vectorStore || !vectorStoreReady) {
+            setError(new Error('Vector store not ready'));
+            return;
+        }
+        setIsProcessing(true);
+        setError(null);
+        try {
+            const ragConfig = {
+                vectorStore,
+                llm,
+                systemPrompt,
+                k,
+                minSimilarity,
+                filter,
+            };
+            if (returnSourceDocuments) {
+                // Use RAG with sources
+                const chain = await createPDWRAGWithSources(ragConfig);
+                const result = await chain.invoke({ question });
+                if (typeof result === 'object' && 'answer' in result) {
+                    setAnswer(result.answer);
+                    setSources(result.sourceDocuments || null);
+                }
+                else {
+                    setAnswer(String(result));
+                    setSources(null);
+                }
+            }
+            else {
+                // Use regular RAG
+                const chain = await createPDWRAG(ragConfig);
+                const result = await chain.invoke({ question });
+                setAnswer(String(result));
+                setSources(null);
+            }
+        }
+        catch (err) {
+            const errorObj = err instanceof Error ? err : new Error(String(err));
+            setError(errorObj);
+            setAnswer(null);
+            setSources(null);
+        }
+        finally {
+            setIsProcessing(false);
+        }
+    }, [vectorStore, vectorStoreReady, llm, systemPrompt, k, minSimilarity, returnSourceDocuments, filter]);
+    // Clear function
+    const clear = useCallback(() => {
+        setAnswer(null);
+        setSources(null);
+        setError(null);
+    }, []);
+    return {
+        query,
+        isProcessing,
+        answer,
+        sources,
+        error: error || vectorStoreError,
+        isReady: vectorStoreReady && !isProcessing,
+        clear,
+    };
+}
+//# sourceMappingURL=usePDWRAG.js.map

package/dist/hooks.wip/usePDWRAG.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usePDWRAG.js","sourceRoot":"","sources":["../../src/hooks.wip/usePDWRAG.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAqCG;AAEH,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AAG9C,OAAO,EAAE,iBAAiB,EAAiC,MAAM,qBAAqB,CAAC;AACvF,OAAO,EAAE,YAAY,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AA+ElF;;;;;GAKG;AACH,MAAM,UAAU,SAAS,CAAC,OAAyB;IACjD,MAAM,EACJ,GAAG,EACH,YAAY,EACZ,CAAC,GAAG,CAAC,EACL,aAAa,GAAG,GAAG,EACnB,qBAAqB,GAAG,KAAK,EAC7B,MAAM,EACN,GAAG,kBAAkB,EACtB,GAAG,OAAO,CAAC;IAEZ,0BAA0B;IAC1B,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,gBAAgB,EAAE,KAAK,EAAE,gBAAgB,EAAE,GACvE,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;IAExC,cAAc;IACd,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IACxD,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,QAAQ,CAAgB,IAAI,CAAC,CAAC;IAC1D,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,GAAG,QAAQ,CAAe,IAAI,CAAC,CAAC;IAC3D,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAe,gBAAgB,CAAC,CAAC;IAEnE,iBAAiB;IACjB,MAAM,KAAK,GAAG,WAAW,CACvB,KAAK,EAAE,QAAgB,EAAE,EAAE;QACzB,IAAI,CAAC,WAAW,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACtC,QAAQ,CAAC,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC,CAAC;YAC9C,OAAO;QACT,CAAC;QAED,eAAe,CAAC,IAAI,CAAC,CAAC;QACtB,QAAQ,CAAC,IAAI,CAAC,CAAC;QAEf,IAAI,CAAC;YACH,MAAM,SAAS,GAAiB;gBAC9B,WAAW;gBACX,GAAG;gBACH,YAAY;gBACZ,CAAC;gBACD,aAAa;gBACb,MAAM;aACP,CAAC;YAEF,IAAI,qBAAqB,EAAE,CAAC;gBAC1B,uBAAuB;gBACvB,MAAM,KAAK,GAAG,MAAM,uBAAuB,CAAC,SAAS,CAAC,CAAC;gBACvD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;gBAEhD,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,QAAQ,IAAI,MAAM,EAAE,CAAC;oBACrD,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBACzB,UAAU,CAAC,MAAM,CAAC,eAAe,IAAI,IAAI,CAAC,CAAC;gBAC7C,CAAC;qBAAM,CAAC;oBACN,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;oBAC1B,UAAU,CAAC,IAAI,CAAC,CAAC;gBACnB,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,kBAAkB;gBAClB,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,SAAS,CAAC,CAAC;gBAC5C,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;gBAChD,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;gBAC1B,UAAU,CAAC,IAAI,CAAC,CAAC;YACnB,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YACrE,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACnB,SAAS,CAAC,IAAI,CAAC,CAAC;YAChB,UAAU,CAAC,IAAI,CAAC,CAAC;QACnB,CAAC;gBAAS,CAAC;YACT,eAAe,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;IACH,CAAC,EACD,CAAC,WAAW,EAAE,gBAAgB,EAAE,GAAG,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,qBAAqB,EAAE,MAAM,CAAC,CACpG,CAAC;IAEF,iBAAiB;IACjB,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;QAC7B,SAAS,CAAC,IAAI,CAAC,CAAC;QAChB,UAAU,CAAC,IAAI,CAAC,CAAC;QACjB,QAAQ,CAAC,IAAI,CAAC,CAAC;IACjB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO;QACL,KAAK;QACL,YAAY;QACZ,MAAM;QACN,OAAO;QACP,KAAK,EAAE,KAAK,IAAI,gBAAgB;QAChC,OAAO,EAAE,gBAAgB,IAAI,CAAC,YAAY;QAC1C,KAAK;KACN,CAAC;AACJ,CAAC"}

package/dist/hooks.wip/usePDWVectorStore.d.ts

@@ -0,0 +1,68 @@
+/**
+ * usePDWVectorStore - React Hook for PDWVectorStore
+ *
+ * Convenience hook for initializing and managing PDWVectorStore in React applications.
+ * Handles initialization, lifecycle, and provides easy access to the vector store.
+ *
+ * @example
+ * ```typescript
+ * import { usePDWVectorStore } from 'personal-data-wallet-sdk/hooks';
+ * import { useCurrentAccount } from '@mysten/dapp-kit';
+ *
+ * function MyComponent() {
+ *   const account = useCurrentAccount();
+ *
+ *   const { vectorStore, isReady, error } = usePDWVectorStore({
+ *     userAddress: account?.address,
+ *     packageId: '0x...',
+ *     walrusAggregator: 'https://...',
+ *     geminiApiKey: process.env.NEXT_PUBLIC_GEMINI_API_KEY!
+ *   });
+ *
+ *   if (!isReady) return <div>Initializing...</div>;
+ *   if (error) return <div>Error: {error.message}</div>;
+ *
+ *   // Use vectorStore...
+ * }
+ * ```
+ */
+import { PDWEmbeddings } from '../langchain/PDWEmbeddings';
+import { PDWVectorStore } from '../langchain/PDWVectorStore';
+import type { PDWVectorStoreConfig } from '../langchain/PDWVectorStore';
+export interface UsePDWVectorStoreOptions extends PDWVectorStoreConfig {
+    /**
+     * Whether to initialize immediately
+     * @default true
+     */
+    enabled?: boolean;
+}
+export interface UsePDWVectorStoreReturn {
+    /**
+     * Initialized PDWVectorStore instance
+     */
+    vectorStore: PDWVectorStore | null;
+    /**
+     * PDWEmbeddings instance
+     */
+    embeddings: PDWEmbeddings | null;
+    /**
+     * Whether the vector store is ready to use
+     */
+    isReady: boolean;
+    /**
+     * Initialization error if any
+     */
+    error: Error | null;
+    /**
+     * Manually reinitialize the vector store
+     */
+    reinitialize: () => void;
+}
+/**
+ * React hook for managing PDWVectorStore
+ *
+ * Automatically initializes PDWEmbeddings and PDWVectorStore,
+ * handles lifecycle, and provides ready state.
+ */
+export declare function usePDWVectorStore(options: UsePDWVectorStoreOptions): UsePDWVectorStoreReturn;
+//# sourceMappingURL=usePDWVectorStore.d.ts.map

package/dist/hooks.wip/usePDWVectorStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"usePDWVectorStore.d.ts","sourceRoot":"","sources":["../../src/hooks.wip/usePDWVectorStore.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AAGH,OAAO,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAC3D,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAC7D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AAExE,MAAM,WAAW,wBAAyB,SAAQ,oBAAoB;IACpE;;;OAGG;IACH,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,uBAAuB;IACtC;;OAEG;IACH,WAAW,EAAE,cAAc,GAAG,IAAI,CAAC;IAEnC;;OAEG;IACH,UAAU,EAAE,aAAa,GAAG,IAAI,CAAC;IAEjC;;OAEG;IACH,OAAO,EAAE,OAAO,CAAC;IAEjB;;OAEG;IACH,KAAK,EAAE,KAAK,GAAG,IAAI,CAAC;IAEpB;;OAEG;IACH,YAAY,EAAE,MAAM,IAAI,CAAC;CAC1B;AAED;;;;;GAKG;AACH,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,wBAAwB,GAChC,uBAAuB,CA+EzB"}