@cmdoss/memwal-sdk 0.6.0

package/dist/langchain/PDWVectorStore.d.ts

@@ -0,0 +1,173 @@
+/**
+ * PDWVectorStore - LangChain VectorStore implementation for Personal Data Wallet
+ *
+ * Integrates PDW's decentralized storage (Walrus) and vector search (HNSW) with LangChain.
+ * Enables RAG workflows with memories stored on Sui blockchain and Walrus.
+ *
+ * Features:
+ * - Decentralized vector storage on Walrus
+ * - HNSW-powered similarity search (browser-compatible WebAssembly)
+ * - Blockchain-backed ownership via Sui
+ * - SEAL encryption for privacy (optional)
+ * - Memory metadata and filtering
+ *
+ * @example
+ * ```typescript
+ * import { PDWVectorStore, PDWEmbeddings } from 'personal-data-wallet-sdk/langchain';
+ * import { useCurrentAccount, useSuiClient } from '@mysten/dapp-kit';
+ *
+ * const embeddings = new PDWEmbeddings({ geminiApiKey });
+ * const vectorStore = new PDWVectorStore(embeddings, {
+ *   userAddress: account.address,
+ *   packageId: '0x...',
+ *   walrusAggregator: 'https://aggregator.walrus-testnet.walrus.space',
+ * });
+ *
+ * // Add documents (requires Sui wallet signing)
+ * await vectorStore.addDocuments([
+ *   { pageContent: 'Hello world', metadata: { category: 'greeting' } }
+ * ], { account, signAndExecute, client });
+ *
+ * // Search
+ * const results = await vectorStore.similaritySearch('hi', 5);
+ * ```
+ */
+import { VectorStore } from '@langchain/core/vectorstores';
+import { Document } from '@langchain/core/documents';
+import type { EmbeddingsInterface } from '@langchain/core/embeddings';
+import type { Callbacks } from '@langchain/core/callbacks/manager';
+import type { MaxMarginalRelevanceSearchOptions } from '@langchain/core/vectorstores';
+import type { Transaction } from '@mysten/sui/transactions';
+import type { SuiClient } from '@mysten/sui/client';
+export interface PDWVectorStoreConfig {
+    userAddress: string;
+    packageId: string;
+    accessRegistryId?: string;
+    network?: 'mainnet' | 'testnet' | 'devnet';
+    walrusPublisher?: string;
+    walrusAggregator?: string;
+    walrusNetwork?: 'testnet' | 'mainnet';
+    geminiApiKey: string;
+    embeddingModel?: string;
+    embeddingDimensions?: number;
+    maxElements?: number;
+    efConstruction?: number;
+    m?: number;
+    defaultCategory?: string;
+    defaultImportance?: number;
+    encryptionEnabled?: boolean;
+    sealServerObjectIds?: string[];
+}
+export interface PDWAddDocumentOptions {
+    account: {
+        address: string;
+    };
+    signAndExecute: (params: {
+        transaction: Transaction;
+    }, callbacks: {
+        onSuccess: (result: any) => void;
+        onError: (error: Error) => void;
+    }) => void;
+    client: SuiClient;
+    category?: string;
+    importance?: number;
+    tags?: string[];
+    encrypt?: boolean;
+    onProgress?: (status: string) => void;
+}
+/**
+ * LangChain VectorStore implementation using Personal Data Wallet
+ *
+ * This class wraps PDW's ClientMemoryManager and MemoryIndexService to provide
+ * a standard LangChain VectorStore interface while preserving all PDW features:
+ * - Decentralized storage on Walrus
+ * - Blockchain ownership via Sui
+ * - SEAL encryption
+ * - HNSW vector search
+ */
+export declare class PDWVectorStore extends VectorStore {
+    private clientMemoryManager;
+    private memoryIndexService;
+    private storageService;
+    private embeddingService;
+    private config;
+    private documentMapping;
+    constructor(embeddings: EmbeddingsInterface, config: PDWVectorStoreConfig);
+    private initializeServices;
+    /**
+     * Create PDWVectorStore from documents (LangChain factory pattern)
+     *
+     * @param docs - Documents to add
+     * @param embeddings - Embeddings interface
+     * @param dbConfig - Combined config object containing both PDWVectorStoreConfig and PDWAddDocumentOptions
+     */
+    static fromDocuments(docs: Document[], embeddings: EmbeddingsInterface, dbConfig: PDWVectorStoreConfig & {
+        addOptions?: PDWAddDocumentOptions;
+    }): Promise<PDWVectorStore>;
+    /**
+     * Create PDWVectorStore from texts (LangChain factory pattern)
+     *
+     * @param texts - Text strings to add
+     * @param metadatas - Metadata for each text
+     * @param embeddings - Embeddings interface
+     * @param dbConfig - Combined config object containing both PDWVectorStoreConfig and PDWAddDocumentOptions
+     */
+    static fromTexts(texts: string[], metadatas: object[] | object, embeddings: EmbeddingsInterface, dbConfig: PDWVectorStoreConfig & {
+        addOptions?: PDWAddDocumentOptions;
+    }): Promise<PDWVectorStore>;
+    /**
+     * Add documents to the vector store
+     *
+     * This creates memories on Sui blockchain and stores them on Walrus.
+     * Requires wallet signing via PDWAddDocumentOptions.
+     */
+    addDocuments(documents: Document[], options?: PDWAddDocumentOptions): Promise<string[]>;
+    /**
+     * Add vectors directly to the store
+     *
+     * This is useful when you already have embeddings and want to avoid regenerating them.
+     */
+    addVectors(vectors: number[][], documents: Document[], options?: PDWAddDocumentOptions): Promise<string[]>;
+    /**
+     * Similarity search with score (core LangChain method)
+     */
+    similaritySearchVectorWithScore(query: number[], k: number, filter?: Record<string, any>): Promise<[Document, number][]>;
+    /**
+     * Similarity search (standard LangChain method)
+     */
+    similaritySearch(query: string, k?: number, filter?: Record<string, any>): Promise<Document[]>;
+    /**
+     * Similarity search with score using query string
+     */
+    similaritySearchWithScore(query: string, k?: number, filter?: Record<string, any>): Promise<[Document, number][]>;
+    /**
+     * Delete documents by IDs (blob IDs)
+     */
+    delete(params: {
+        ids: string[];
+    }): Promise<void>;
+    /**
+     * Maximum marginal relevance search
+     * Finds documents that are similar to the query but diverse from each other
+     */
+    maxMarginalRelevanceSearch(query: string, options: MaxMarginalRelevanceSearchOptions<this["FilterType"]>, _callbacks?: Callbacks): Promise<Document[]>;
+    /**
+     * Get statistics about the vector store
+     */
+    getStats(): Promise<{
+        totalMemories: number;
+        categoryCounts: Record<string, number>;
+        importanceDistribution: Record<number, number>;
+        averageImportance: number;
+        oldestMemory: Date | null;
+        newestMemory: Date | null;
+        indexSize: number;
+    }>;
+    /**
+     * Clear all memories for the current user
+     * Note: This only clears the local index, not blockchain or Walrus
+     */
+    clear(): Promise<void>;
+    _vectorstoreType(): string;
+}
+//# sourceMappingURL=PDWVectorStore.d.ts.map

package/dist/langchain/PDWVectorStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"PDWVectorStore.d.ts","sourceRoot":"","sources":["../../src/langchain/PDWVectorStore.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAiCG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,8BAA8B,CAAC;AAC3D,OAAO,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AACrD,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACtE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,mCAAmC,CAAC;AACnE,OAAO,KAAK,EAAE,iCAAiC,EAAE,MAAM,8BAA8B,CAAC;AAOtF,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAEpD,MAAM,WAAW,oBAAoB;IAEnC,WAAW,EAAE,MAAM,CAAC;IAGpB,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,OAAO,CAAC,EAAE,SAAS,GAAG,SAAS,GAAG,QAAQ,CAAC;IAG3C,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,aAAa,CAAC,EAAE,SAAS,GAAG,SAAS,CAAC;IAGtC,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAG7B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,CAAC,CAAC,EAAE,MAAM,CAAC;IAGX,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAG5B,mBAAmB,CAAC,EAAE,MAAM,EAAE,CAAC;CAChC;AAED,MAAM,WAAW,qBAAqB;IAEpC,OAAO,EAAE;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC;IAC7B,cAAc,EAAE,CAAC,MAAM,EAAE;QAAE,WAAW,EAAE,WAAW,CAAA;KAAE,EAAE,SAAS,EAAE;QAChE,SAAS,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,IAAI,CAAC;QACjC,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;KACjC,KAAK,IAAI,CAAC;IACX,MAAM,EAAE,SAAS,CAAC;IAGlB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;CACvC;AAED;;;;;;;;;GASG;AACH,qBAAa,cAAe,SAAQ,WAAW;IAC7C,OAAO,CAAC,mBAAmB,CAAuB;IAClD,OAAO,CAAC,kBAAkB,CAAsB;IAChD,OAAO,CAAC,cAAc,CAAkB;IACxC,OAAO,CAAC,gBAAgB,CAAoB;IAC5C,OAAO,CAAC,MAAM,CAAuB;IAGrC,OAAO,CAAC,eAAe,CAA6B;gBAGlD,UAAU,EAAE,mBAAmB,EAC/B,MAAM,EAAE,oBAAoB;IAS9B,OAAO,CAAC,kBAAkB;IAyC1B;;;;;;OAMG;WACU,aAAa,CACxB,IAAI,EAAE,QAAQ,EAAE,EAChB,UAAU,EAAE,mBAAmB,EAC/B,QAAQ,EAAE,oBAAoB,GAAG;QAAE,UAAU,CAAC,EAAE,qBAAqB,CAAA;KAAE,GACtE,OAAO,CAAC,cAAc,CAAC;IAW1B;;;;;;;OAOG;WACU,SAAS,CACpB,KAAK,EAAE,MAAM,EAAE,EACf,SAAS,EAAE,MAAM,EAAE,GAAG,MAAM,EAC5B,UAAU,EAAE,mBAAmB,EAC/B,QAAQ,EAAE,oBAAoB,GAAG;QAAE,UAAU,CAAC,EAAE,qBAAqB,CAAA;KAAE,GACtE,OAAO,CAAC,cAAc,CAAC;IAQ1B;;;;;OAKG;IACG,YAAY,CAChB,SAAS,EAAE,QAAQ,EAAE,EACrB,OAAO,CAAC,EAAE,qBAAqB,GAC9B,OAAO,CAAC,MAAM,EAAE,CAAC;IA4CpB;;;;OAIG;IACG,UAAU,CACd,OAAO,EAAE,MAAM,EAAE,EAAE,EACnB,SAAS,EAAE,QAAQ,EAAE,EACrB,OAAO,CAAC,EAAE,qBAAqB,GAC9B,OAAO,CAAC,MAAM,EAAE,CAAC;IAWpB;;OAEG;IACG,+BAA+B,CACnC,KAAK,EAAE,MAAM,EAAE,EACf,CAAC,EAAE,MAAM,EACT,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAC3B,OAAO,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC;IAqChC;;OAEG;IACG,gBAAgB,CACpB,KAAK,EAAE,MAAM,EACb,CAAC,GAAE,MAAU,EACb,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAC3B,OAAO,CAAC,QAAQ,EAAE,CAAC;IAKtB;;OAEG;IACG,yBAAyB,CAC7B,KAAK,EAAE,MAAM,EACb,CAAC,GAAE,MAAU,EACb,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAC3B,OAAO,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC;IAMhC;;OAEG;IACG,MAAM,CAAC,MAAM,EAAE;QAAE,GAAG,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IActD;;;OAGG;IACG,0BAA0B,CAC9B,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,iCAAiC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,EAC9D,UAAU,CAAC,EAAE,SAAS,GACrB,OAAO,CAAC,QAAQ,EAAE,CAAC;IA8CtB;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC;QACxB,aAAa,EAAE,MAAM,CAAC;QACtB,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACvC,sBAAsB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC/C,iBAAiB,EAAE,MAAM,CAAC;QAC1B,YAAY,EAAE,IAAI,GAAG,IAAI,CAAC;QAC1B,YAAY,EAAE,IAAI,GAAG,IAAI,CAAC;QAC1B,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;IAIF;;;OAGG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAM5B,gBAAgB,IAAI,MAAM;CAG3B"}

package/dist/langchain/PDWVectorStore.js

@@ -0,0 +1,304 @@
+/**
+ * PDWVectorStore - LangChain VectorStore implementation for Personal Data Wallet
+ *
+ * Integrates PDW's decentralized storage (Walrus) and vector search (HNSW) with LangChain.
+ * Enables RAG workflows with memories stored on Sui blockchain and Walrus.
+ *
+ * Features:
+ * - Decentralized vector storage on Walrus
+ * - HNSW-powered similarity search (browser-compatible WebAssembly)
+ * - Blockchain-backed ownership via Sui
+ * - SEAL encryption for privacy (optional)
+ * - Memory metadata and filtering
+ *
+ * @example
+ * ```typescript
+ * import { PDWVectorStore, PDWEmbeddings } from 'personal-data-wallet-sdk/langchain';
+ * import { useCurrentAccount, useSuiClient } from '@mysten/dapp-kit';
+ *
+ * const embeddings = new PDWEmbeddings({ geminiApiKey });
+ * const vectorStore = new PDWVectorStore(embeddings, {
+ *   userAddress: account.address,
+ *   packageId: '0x...',
+ *   walrusAggregator: 'https://aggregator.walrus-testnet.walrus.space',
+ * });
+ *
+ * // Add documents (requires Sui wallet signing)
+ * await vectorStore.addDocuments([
+ *   { pageContent: 'Hello world', metadata: { category: 'greeting' } }
+ * ], { account, signAndExecute, client });
+ *
+ * // Search
+ * const results = await vectorStore.similaritySearch('hi', 5);
+ * ```
+ */
+import { VectorStore } from '@langchain/core/vectorstores';
+import { Document } from '@langchain/core/documents';
+import { ClientMemoryManager } from '../client/ClientMemoryManager.js';
+import { MemoryIndexService } from '../services/MemoryIndexService.js';
+import { StorageService } from '../services/StorageService.js';
+import { EmbeddingService } from '../services/EmbeddingService.js';
+/**
+ * LangChain VectorStore implementation using Personal Data Wallet
+ *
+ * This class wraps PDW's ClientMemoryManager and MemoryIndexService to provide
+ * a standard LangChain VectorStore interface while preserving all PDW features:
+ * - Decentralized storage on Walrus
+ * - Blockchain ownership via Sui
+ * - SEAL encryption
+ * - HNSW vector search
+ */
+export class PDWVectorStore extends VectorStore {
+    constructor(embeddings, config) {
+        super(embeddings, {});
+        // Track document to blobId mapping
+        this.documentMapping = new Map(); // blobId -> documentId
+        this.config = config;
+        // Initialize PDW services
+        this.initializeServices();
+    }
+    initializeServices() {
+        // Initialize ClientMemoryManager
+        const managerConfig = {
+            packageId: this.config.packageId,
+            accessRegistryId: this.config.accessRegistryId || '',
+            walrusAggregator: this.config.walrusAggregator || 'https://aggregator.walrus-testnet.walrus.space',
+            geminiApiKey: this.config.geminiApiKey,
+            sealServerObjectIds: this.config.sealServerObjectIds,
+            walrusNetwork: this.config.walrusNetwork || 'testnet',
+            enableLocalIndexing: true,
+        };
+        this.clientMemoryManager = new ClientMemoryManager(managerConfig);
+        // Initialize services
+        this.embeddingService = new EmbeddingService({
+            apiKey: this.config.geminiApiKey,
+            model: this.config.embeddingModel || 'text-embedding-004',
+            dimensions: this.config.embeddingDimensions || 3072,
+        });
+        this.storageService = new StorageService({
+            walrusPublisherUrl: this.config.walrusPublisher,
+            walrusAggregatorUrl: this.config.walrusAggregator || 'https://aggregator.walrus-testnet.walrus.space',
+            network: (this.config.network === 'devnet' ? 'testnet' : this.config.network) || 'testnet',
+        });
+        this.memoryIndexService = new MemoryIndexService(this.storageService, {
+            maxElements: this.config.maxElements || 10000,
+            dimension: this.config.embeddingDimensions || 3072,
+            efConstruction: this.config.efConstruction || 200,
+            m: this.config.m || 16,
+        });
+        // Connect embedding service to index service
+        this.memoryIndexService.initialize(this.embeddingService, this.storageService);
+    }
+    /**
+     * Create PDWVectorStore from documents (LangChain factory pattern)
+     *
+     * @param docs - Documents to add
+     * @param embeddings - Embeddings interface
+     * @param dbConfig - Combined config object containing both PDWVectorStoreConfig and PDWAddDocumentOptions
+     */
+    static async fromDocuments(docs, embeddings, dbConfig) {
+        const { addOptions, ...config } = dbConfig;
+        const instance = new PDWVectorStore(embeddings, config);
+        if (addOptions) {
+            await instance.addDocuments(docs, addOptions);
+        }
+        return instance;
+    }
+    /**
+     * Create PDWVectorStore from texts (LangChain factory pattern)
+     *
+     * @param texts - Text strings to add
+     * @param metadatas - Metadata for each text
+     * @param embeddings - Embeddings interface
+     * @param dbConfig - Combined config object containing both PDWVectorStoreConfig and PDWAddDocumentOptions
+     */
+    static async fromTexts(texts, metadatas, embeddings, dbConfig) {
+        const docs = texts.map((text, i) => {
+            const metadata = Array.isArray(metadatas) ? metadatas[i] : metadatas;
+            return new Document({ pageContent: text, metadata });
+        });
+        return PDWVectorStore.fromDocuments(docs, embeddings, dbConfig);
+    }
+    /**
+     * Add documents to the vector store
+     *
+     * This creates memories on Sui blockchain and stores them on Walrus.
+     * Requires wallet signing via PDWAddDocumentOptions.
+     */
+    async addDocuments(documents, options) {
+        if (!options) {
+            throw new Error('PDWVectorStore.addDocuments requires options with wallet signing. ' +
+                'Pass { account, signAndExecute, client } to sign transactions.');
+        }
+        const blobIds = [];
+        for (const doc of documents) {
+            try {
+                if (options.onProgress) {
+                    options.onProgress(`Creating memory: ${doc.pageContent.substring(0, 50)}...`);
+                }
+                // Create memory using ClientMemoryManager
+                // This handles: embedding → encryption → Walrus upload → on-chain registration
+                const blobId = await this.clientMemoryManager.createMemory({
+                    content: doc.pageContent,
+                    category: options.category || doc.metadata?.category || this.config.defaultCategory || 'general',
+                    account: options.account,
+                    signAndExecute: options.signAndExecute,
+                    client: options.client,
+                    onProgress: options.onProgress,
+                });
+                blobIds.push(blobId);
+                this.documentMapping.set(blobId, doc.id || blobId);
+                if (options.onProgress) {
+                    options.onProgress(`✅ Memory created: ${blobId}`);
+                }
+            }
+            catch (error) {
+                console.error(`Failed to add document:`, error);
+                throw new Error(`Failed to add document: ${error instanceof Error ? error.message : String(error)}`);
+            }
+        }
+        return blobIds;
+    }
+    /**
+     * Add vectors directly to the store
+     *
+     * This is useful when you already have embeddings and want to avoid regenerating them.
+     */
+    async addVectors(vectors, documents, options) {
+        if (vectors.length !== documents.length) {
+            throw new Error('Vectors and documents must have the same length');
+        }
+        // Note: Currently ClientMemoryManager generates embeddings internally
+        // For now, we'll call addDocuments which will regenerate embeddings
+        // TODO: Add support for passing pre-computed embeddings to ClientMemoryManager
+        return this.addDocuments(documents, options);
+    }
+    /**
+     * Similarity search with score (core LangChain method)
+     */
+    async similaritySearchVectorWithScore(query, k, filter) {
+        // Build search query for MemoryIndexService
+        const searchQuery = {
+            vector: query,
+            userAddress: this.config.userAddress,
+            k: k || 5,
+            threshold: filter?.minSimilarity || 0.0,
+            categories: filter?.category ? [filter.category] : filter?.categories,
+            tags: filter?.tags,
+            dateRange: filter?.dateRange,
+            importanceRange: filter?.importanceRange,
+            includeContent: false, // Don't fetch content by default (can be slow with decryption)
+        };
+        // Search using MemoryIndexService
+        const results = await this.memoryIndexService.searchMemories(searchQuery);
+        // Convert to LangChain format
+        return results.map((result) => {
+            // Use metadata topic as pageContent if content not included
+            const pageContent = result.metadata?.topic || result.metadata?.category || '';
+            const doc = new Document({
+                pageContent,
+                metadata: {
+                    ...result.metadata,
+                    blobId: result.blobId,
+                    memoryId: result.memoryId,
+                    similarity: result.similarity,
+                    relevanceScore: result.relevanceScore,
+                },
+            });
+            return [doc, result.similarity];
+        });
+    }
+    /**
+     * Similarity search (standard LangChain method)
+     */
+    async similaritySearch(query, k = 5, filter) {
+        const resultsWithScore = await this.similaritySearchWithScore(query, k, filter);
+        return resultsWithScore.map(([doc]) => doc);
+    }
+    /**
+     * Similarity search with score using query string
+     */
+    async similaritySearchWithScore(query, k = 5, filter) {
+        // Generate query embedding
+        const embedding = await this.embeddings.embedQuery(query);
+        return this.similaritySearchVectorWithScore(embedding, k, filter);
+    }
+    /**
+     * Delete documents by IDs (blob IDs)
+     */
+    async delete(params) {
+        for (const blobId of params.ids) {
+            try {
+                // Remove from memory index
+                // Note: This doesn't delete from Walrus or blockchain
+                // Those are immutable - we just remove from local index
+                await this.memoryIndexService.removeMemory(this.config.userAddress, blobId);
+                this.documentMapping.delete(blobId);
+            }
+            catch (error) {
+                console.error(`Failed to delete document ${blobId}:`, error);
+            }
+        }
+    }
+    /**
+     * Maximum marginal relevance search
+     * Finds documents that are similar to the query but diverse from each other
+     */
+    async maxMarginalRelevanceSearch(query, options, _callbacks) {
+        const k = options.k ?? 4;
+        const fetchK = options.fetchK ?? 20;
+        const lambda = options.lambda ?? 0.5;
+        // Generate query embedding
+        const queryEmbedding = await this.embeddings.embedQuery(query);
+        // Handle filter (can be string or object from base class)
+        const filterObj = typeof options.filter === 'object' && options.filter !== null
+            ? options.filter
+            : {};
+        // Fetch more results than needed
+        const searchQuery = {
+            vector: queryEmbedding,
+            userAddress: this.config.userAddress,
+            k: fetchK,
+            threshold: filterObj.minSimilarity || 0.0,
+            categories: filterObj.categories,
+            tags: filterObj.tags,
+            includeContent: false,
+            diversityFactor: 1 - lambda, // Higher diversity factor = more diverse results
+        };
+        const results = await this.memoryIndexService.searchMemories(searchQuery);
+        // Convert to documents and apply MMR
+        const documents = results.slice(0, k).map((result) => {
+            const pageContent = result.metadata?.topic || result.metadata?.category || '';
+            return new Document({
+                pageContent,
+                metadata: {
+                    ...result.metadata,
+                    blobId: result.blobId,
+                    memoryId: result.memoryId,
+                    similarity: result.similarity,
+                    relevanceScore: result.relevanceScore,
+                },
+            });
+        });
+        return documents;
+    }
+    /**
+     * Get statistics about the vector store
+     */
+    async getStats() {
+        return this.memoryIndexService.getIndexStats(this.config.userAddress);
+    }
+    /**
+     * Clear all memories for the current user
+     * Note: This only clears the local index, not blockchain or Walrus
+     */
+    async clear() {
+        await this.memoryIndexService.clearUserIndex(this.config.userAddress);
+        this.documentMapping.clear();
+    }
+    // Required by VectorStore abstract class
+    _vectorstoreType() {
+        return 'pdw';
+    }
+}
+//# sourceMappingURL=PDWVectorStore.js.map

package/dist/langchain/PDWVectorStore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"PDWVectorStore.js","sourceRoot":"","sources":["../../src/langchain/PDWVectorStore.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAiCG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,8BAA8B,CAAC;AAC3D,OAAO,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AAIrD,OAAO,EAAE,mBAAmB,EAAE,MAAM,+BAA+B,CAAC;AAEpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEpE,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAsDhE;;;;;;;;;GASG;AACH,MAAM,OAAO,cAAe,SAAQ,WAAW;IAU7C,YACE,UAA+B,EAC/B,MAA4B;QAE5B,KAAK,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;QAPxB,mCAAmC;QAC3B,oBAAe,GAAG,IAAI,GAAG,EAAkB,CAAC,CAAC,uBAAuB;QAO1E,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,0BAA0B;QAC1B,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;IAEO,kBAAkB;QACxB,iCAAiC;QACjC,MAAM,aAAa,GAA8B;YAC/C,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;YAChC,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,EAAE;YACpD,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,gDAAgD;YAClG,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;YACtC,mBAAmB,EAAE,IAAI,CAAC,MAAM,CAAC,mBAAmB;YACpD,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,SAAS;YACrD,mBAAmB,EAAE,IAAI;SAC1B,CAAC;QAEF,IAAI,CAAC,mBAAmB,GAAG,IAAI,mBAAmB,CAAC,aAAa,CAAC,CAAC;QAElE,sBAAsB;QACtB,IAAI,CAAC,gBAAgB,GAAG,IAAI,gBAAgB,CAAC;YAC3C,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;YAChC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,IAAI,oBAAoB;YACzD,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,mBAAmB,IAAI,IAAI;SACpD,CAAC,CAAC;QAEH,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CAAC;YACvC,kBAAkB,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;YAC/C,mBAAmB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,gDAAgD;YACrG,OAAO,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,SAAS;SAC3F,CAAC,CAAC;QAEH,IAAI,CAAC,kBAAkB,GAAG,IAAI,kBAAkB,CAC9C,IAAI,CAAC,cAAc,EACnB;YACE,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,KAAK;YAC7C,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,mBAAmB,IAAI,IAAI;YAClD,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,IAAI,GAAG;YACjD,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE;SACvB,CACF,CAAC;QAEF,6CAA6C;QAC7C,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;IACjF,CAAC;IAED;;;;;;OAMG;IACH,MAAM,CAAC,KAAK,CAAC,aAAa,CACxB,IAAgB,EAChB,UAA+B,EAC/B,QAAuE;QAEvE,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,EAAE,GAAG,QAAQ,CAAC;QAC3C,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC,UAAU,EAAE,MAA8B,CAAC,CAAC;QAEhF,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,QAAQ,CAAC,YAAY,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;;;;OAOG;IACH,MAAM,CAAC,KAAK,CAAC,SAAS,CACpB,KAAe,EACf,SAA4B,EAC5B,UAA+B,EAC/B,QAAuE;QAEvE,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;YACjC,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACrE,OAAO,IAAI,QAAQ,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QACH,OAAO,cAAc,CAAC,aAAa,CAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;IAClE,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,YAAY,CAChB,SAAqB,EACrB,OAA+B;QAE/B,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CACb,oEAAoE;gBACpE,gEAAgE,CACjE,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAa,EAAE,CAAC;QAE7B,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;YAC5B,IAAI,CAAC;gBACH,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;oBACvB,OAAO,CAAC,UAAU,CAAC,oBAAoB,GAAG,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC;gBAChF,CAAC;gBAED,0CAA0C;gBAC1C,+EAA+E;gBAC/E,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,YAAY,CAAC;oBACzD,OAAO,EAAE,GAAG,CAAC,WAAW;oBACxB,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,GAAG,CAAC,QAAQ,EAAE,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe,IAAI,SAAS;oBAChG,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,cAAc,EAAE,OAAO,CAAC,cAAc;oBACtC,MAAM,EAAE,OAAO,CAAC,MAAM;oBACtB,UAAU,EAAE,OAAO,CAAC,UAAU;iBAC/B,CAAC,CAAC;gBAEH,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACrB,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,IAAI,MAAM,CAAC,CAAC;gBAEnD,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;oBACvB,OAAO,CAAC,UAAU,CAAC,qBAAqB,MAAM,EAAE,CAAC,CAAC;gBACpD,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;gBAChD,MAAM,IAAI,KAAK,CACb,2BAA2B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CACpF,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,UAAU,CACd,OAAmB,EACnB,SAAqB,EACrB,OAA+B;QAE/B,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,MAAM,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;QAED,sEAAsE;QACtE,oEAAoE;QACpE,+EAA+E;QAC/E,OAAO,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC/C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,+BAA+B,CACnC,KAAe,EACf,CAAS,EACT,MAA4B;QAE5B,4CAA4C;QAC5C,MAAM,WAAW,GAAsB;YACrC,MAAM,EAAE,KAAK;YACb,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;YACpC,CAAC,EAAE,CAAC,IAAI,CAAC;YACT,SAAS,EAAE,MAAM,EAAE,aAAa,IAAI,GAAG;YACvC,UAAU,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,UAAU;YACrE,IAAI,EAAE,MAAM,EAAE,IAAI;YAClB,SAAS,EAAE,MAAM,EAAE,SAAS;YAC5B,eAAe,EAAE,MAAM,EAAE,eAAe;YACxC,cAAc,EAAE,KAAK,EAAE,+DAA+D;SACvF,CAAC;QAEF,kCAAkC;QAClC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;QAE1E,8BAA8B;QAC9B,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAA0B,EAAE,EAAE;YAChD,4DAA4D;YAC5D,MAAM,WAAW,GAAG,MAAM,CAAC,QAAQ,EAAE,KAAK,IAAI,MAAM,CAAC,QAAQ,EAAE,QAAQ,IAAI,EAAE,CAAC;YAE9E,MAAM,GAAG,GAAG,IAAI,QAAQ,CAAC;gBACvB,WAAW;gBACX,QAAQ,EAAE;oBACR,GAAG,MAAM,CAAC,QAAQ;oBAClB,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,cAAc,EAAE,MAAM,CAAC,cAAc;iBACtC;aACF,CAAC,CAAC;YAEH,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,UAAU,CAAuB,CAAC;QACxD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CACpB,KAAa,EACb,IAAY,CAAC,EACb,MAA4B;QAE5B,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,yBAAyB,CAAC,KAAK,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAChF,OAAO,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;IAC9C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,yBAAyB,CAC7B,KAAa,EACb,IAAY,CAAC,EACb,MAA4B;QAE5B,2BAA2B;QAC3B,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAC1D,OAAO,IAAI,CAAC,+BAA+B,CAAC,SAAS,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;IACpE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,MAAyB;QACpC,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,GAAG,EAAE,CAAC;YAChC,IAAI,CAAC;gBACH,2BAA2B;gBAC3B,sDAAsD;gBACtD,wDAAwD;gBACxD,MAAM,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;gBAC5E,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACtC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,MAAM,GAAG,EAAE,KAAK,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,0BAA0B,CAC9B,KAAa,EACb,OAA8D,EAC9D,UAAsB;QAEtB,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC;QACzB,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC;QACpC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,GAAG,CAAC;QAErC,2BAA2B;QAC3B,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAE/D,0DAA0D;QAC1D,MAAM,SAAS,GAAG,OAAO,OAAO,CAAC,MAAM,KAAK,QAAQ,IAAI,OAAO,CAAC,MAAM,KAAK,IAAI;YAC7E,CAAC,CAAC,OAAO,CAAC,MAA6B;YACvC,CAAC,CAAC,EAAE,CAAC;QAEP,iCAAiC;QACjC,MAAM,WAAW,GAAsB;YACrC,MAAM,EAAE,cAAc;YACtB,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;YACpC,CAAC,EAAE,MAAM;YACT,SAAS,EAAE,SAAS,CAAC,aAAa,IAAI,GAAG;YACzC,UAAU,EAAE,SAAS,CAAC,UAAU;YAChC,IAAI,EAAE,SAAS,CAAC,IAAI;YACpB,cAAc,EAAE,KAAK;YACrB,eAAe,EAAE,CAAC,GAAG,MAAM,EAAE,iDAAiD;SAC/E,CAAC;QAEF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;QAE1E,qCAAqC;QACrC,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,MAA0B,EAAE,EAAE;YACvE,MAAM,WAAW,GAAG,MAAM,CAAC,QAAQ,EAAE,KAAK,IAAI,MAAM,CAAC,QAAQ,EAAE,QAAQ,IAAI,EAAE,CAAC;YAE9E,OAAO,IAAI,QAAQ,CAAC;gBAClB,WAAW;gBACX,QAAQ,EAAE;oBACR,GAAG,MAAM,CAAC,QAAQ;oBAClB,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,cAAc,EAAE,MAAM,CAAC,cAAc;iBACtC;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QASZ,OAAO,IAAI,CAAC,kBAAkB,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IACxE,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,kBAAkB,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACtE,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;IAC/B,CAAC;IAED,yCAAyC;IACzC,gBAAgB;QACd,OAAO,KAAK,CAAC;IACf,CAAC;CACF"}

package/dist/langchain/createPDWRAG.d.ts

@@ -0,0 +1,115 @@
+/**
+ * createPDWRAG - Simple RAG Helper for Personal Data Wallet
+ *
+ * A one-function RAG setup that combines PDWVectorStore with any LangChain LLM.
+ * Provides the quickest way to build RAG applications with decentralized storage.
+ *
+ * @example
+ * ```typescript
+ * import { createPDWRAG, PDWEmbeddings, PDWVectorStore } from 'personal-data-wallet-sdk/langchain';
+ * import { ChatGoogleGenerativeAI } from '@langchain/google-genai';
+ *
+ * const embeddings = new PDWEmbeddings({ geminiApiKey });
+ * const vectorStore = new PDWVectorStore(embeddings, config);
+ * const llm = new ChatGoogleGenerativeAI({ apiKey: geminiApiKey });
+ *
+ * const ragChain = await createPDWRAG({
+ *   vectorStore,
+ *   llm,
+ *   systemPrompt: 'You are my personal AI assistant.'
+ * });
+ *
+ * const answer = await ragChain.invoke({ question: 'What did I do last week?' });
+ * ```
+ */
+import { ChatPromptTemplate } from '@langchain/core/prompts';
+import { RunnableSequence } from '@langchain/core/runnables';
+import type { Runnable } from '@langchain/core/runnables';
+import { VectorStoreRetriever } from '@langchain/core/vectorstores';
+import type { BaseMessage } from '@langchain/core/messages';
+import type { PDWVectorStore } from './PDWVectorStore';
+export interface PDWRAGConfig {
+    /**
+     * PDWVectorStore instance for retrieval
+     */
+    vectorStore: PDWVectorStore;
+    /**
+     * LangChain LLM for generation
+     * Can be any LLM: ChatGoogleGenerativeAI, ChatOpenAI, ChatAnthropic, etc.
+     */
+    llm: Runnable<any, BaseMessage>;
+    /**
+     * System prompt for the RAG chain
+     * @default 'You are a helpful assistant with access to a knowledge base.'
+     */
+    systemPrompt?: string;
+    /**
+     * Number of documents to retrieve
+     * @default 5
+     */
+    k?: number;
+    /**
+     * Minimum similarity threshold (0-1)
+     * @default 0.5
+     */
+    minSimilarity?: number;
+    /**
+     * Custom retriever (overrides vectorStore if provided)
+     */
+    retriever?: VectorStoreRetriever;
+    /**
+     * Custom prompt template (overrides systemPrompt)
+     */
+    promptTemplate?: ChatPromptTemplate;
+    /**
+     * Whether to include source documents in the output
+     * @default false
+     */
+    returnSourceDocuments?: boolean;
+    /**
+     * Metadata filters for retrieval
+     */
+    filter?: Record<string, any>;
+}
+export interface PDWRAGResult {
+    /**
+     * Generated answer from the LLM
+     */
+    answer: string;
+    /**
+     * Source documents used for context (if returnSourceDocuments is true)
+     */
+    sourceDocuments?: Array<{
+        content: string;
+        metadata: any;
+        similarity?: number;
+    }>;
+}
+/**
+ * Create a RAG chain with PDWVectorStore
+ *
+ * This helper function sets up a complete RAG pipeline:
+ * 1. Retriever: Fetches relevant documents from PDWVectorStore
+ * 2. Prompt: Formats context and question for the LLM
+ * 3. LLM: Generates answer based on context
+ * 4. Parser: Extracts text from LLM response
+ *
+ * @param config - RAG configuration
+ * @returns A runnable chain that takes { question: string } and returns answer
+ */
+export declare function createPDWRAG(config: PDWRAGConfig): Promise<RunnableSequence>;
+/**
+ * Create a RAG chain that returns both answer and source documents
+ *
+ * @param config - RAG configuration with returnSourceDocuments enabled
+ * @returns A runnable chain that returns { answer, sourceDocuments }
+ */
+export declare function createPDWRAGWithSources(config: Omit<PDWRAGConfig, 'returnSourceDocuments'>): Promise<RunnableSequence>;
+/**
+ * Create a conversational RAG chain that maintains chat history
+ *
+ * @param config - RAG configuration
+ * @returns A runnable chain that takes { question, chatHistory } and returns answer
+ */
+export declare function createConversationalPDWRAG(config: PDWRAGConfig): Promise<RunnableSequence>;
+//# sourceMappingURL=createPDWRAG.d.ts.map

package/dist/langchain/createPDWRAG.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"createPDWRAG.d.ts","sourceRoot":"","sources":["../../src/langchain/createPDWRAG.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AAEH,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC7D,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AAE1D,OAAO,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AACpE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAEvD,MAAM,WAAW,YAAY;IAC3B;;OAEG;IACH,WAAW,EAAE,cAAc,CAAC;IAE5B;;;OAGG;IACH,GAAG,EAAE,QAAQ,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;IAEhC;;;OAGG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IAEtB;;;OAGG;IACH,CAAC,CAAC,EAAE,MAAM,CAAC;IAEX;;;OAGG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;IAEvB;;OAEG;IACH,SAAS,CAAC,EAAE,oBAAoB,CAAC;IAEjC;;OAEG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IAEpC;;;OAGG;IACH,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAEhC;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC9B;AAED,MAAM,WAAW,YAAY;IAC3B;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,eAAe,CAAC,EAAE,KAAK,CAAC;QACtB,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,EAAE,GAAG,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,CAAC,CAAC;CACJ;AAED;;;;;;;;;;;GAWG;AACH,wBAAsB,YAAY,CAChC,MAAM,EAAE,YAAY,GACnB,OAAO,CAAC,gBAAgB,CAAC,CAgD3B;AAED;;;;;GAKG;AACH,wBAAsB,uBAAuB,CAC3C,MAAM,EAAE,IAAI,CAAC,YAAY,EAAE,uBAAuB,CAAC,GAClD,OAAO,CAAC,gBAAgB,CAAC,CAmE3B;AAED;;;;;GAKG;AACH,wBAAsB,0BAA0B,CAC9C,MAAM,EAAE,YAAY,GACnB,OAAO,CAAC,gBAAgB,CAAC,CAoD3B"}