langchain 0.0.165 → 0.0.166

package/dist/vectorstores/supabase.cjs

@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.SupabaseVectorStore = void 0;
 const base_js_1 = require("./base.cjs");
 const document_js_1 = require("../document.cjs");
+const math_js_1 = require("../util/math.cjs");
 /**
  * Class for interacting with a Supabase database to store and manage
  * vectors.
@@ -103,14 +104,7 @@ class SupabaseVectorStore extends base_js_1.VectorStore {
             await this.client.from(this.tableName).delete().eq("id", id);
         }
     }
-    /**
-     * Performs a similarity search on the vector store.
-     * @param query The query vector.
-     * @param k The number of results to return.
-     * @param filter Optional filter to apply to the search.
-     * @returns A promise that resolves with the search results when the search is complete.
-     */
-    async similaritySearchVectorWithScore(query, k, filter) {
+    async _searchSupabase(query, k, filter) {
         if (filter && this.filter) {
             throw new Error("cannot provide both `filter` and `this.filter`");
         }
@@ -135,6 +129,17 @@ class SupabaseVectorStore extends base_js_1.VectorStore {
         if (error) {
             throw new Error(`Error searching for documents: ${error.code} ${error.message} ${error.details}`);
         }
+        return searches;
+    }
+    /**
+     * Performs a similarity search on the vector store.
+     * @param query The query vector.
+     * @param k The number of results to return.
+     * @param filter Optional filter to apply to the search.
+     * @returns A promise that resolves with the search results when the search is complete.
+     */
+    async similaritySearchVectorWithScore(query, k, filter) {
+        const searches = await this._searchSupabase(query, k, filter);
         const result = searches.map((resp) => [
             new document_js_1.Document({
                 metadata: resp.metadata,
@@ -144,6 +149,30 @@ class SupabaseVectorStore extends base_js_1.VectorStore {
         ]);
         return result;
     }
+    /**
+     * Return documents selected using the maximal marginal relevance.
+     * Maximal marginal relevance optimizes for similarity to the query AND diversity
+     * among selected documents.
+     *
+     * @param {string} query - Text to look up documents similar to.
+     * @param {number} options.k - Number of documents to return.
+     * @param {number} options.fetchK=20- Number of documents to fetch before passing to the MMR algorithm.
+     * @param {number} options.lambda=0.5 - Number between 0 and 1 that determines the degree of diversity among the results,
+     *                 where 0 corresponds to maximum diversity and 1 to minimum diversity.
+     * @param {SupabaseLibArgs} options.filter - Optional filter to apply to the search.
+     *
+     * @returns {Promise<Document[]>} - List of documents selected by maximal marginal relevance.
+     */
+    async maxMarginalRelevanceSearch(query, options) {
+        const queryEmbedding = await this.embeddings.embedQuery(query);
+        const searches = await this._searchSupabase(queryEmbedding, options.fetchK ?? 20, options.filter);
+        const embeddingList = searches.map((searchResp) => searchResp.embedding);
+        const mmrIndexes = (0, math_js_1.maximalMarginalRelevance)(queryEmbedding, embeddingList, options.lambda, options.k);
+        return mmrIndexes.map((idx) => new document_js_1.Document({
+            metadata: searches[idx].metadata,
+            pageContent: searches[idx].content,
+        }));
+    }
     /**
      * Creates a new SupabaseVectorStore instance from an array of texts.
      * @param texts The texts to create documents from.

package/dist/vectorstores/supabase.d.ts

@@ -1,11 +1,21 @@
 import type { SupabaseClient } from "@supabase/supabase-js";
 import type { PostgrestFilterBuilder } from "@supabase/postgrest-js";
-import { VectorStore } from "./base.js";
+import { MaxMarginalRelevanceSearchOptions, VectorStore } from "./base.js";
 import { Embeddings } from "../embeddings/base.js";
 import { Document } from "../document.js";
 export type SupabaseMetadata = Record<string, any>;
 export type SupabaseFilter = PostgrestFilterBuilder<any, any, any>;
 export type SupabaseFilterRPCCall = (rpcCall: SupabaseFilter) => SupabaseFilter;
+/**
+ * Interface for the response returned when searching embeddings.
+ */
+interface SearchEmbeddingsResponse {
+    id: number;
+    content: string;
+    metadata: object;
+    embedding: number[];
+    similarity: number;
+}
 /**
  * Interface for the arguments required to initialize a Supabase library.
  */
@@ -56,6 +66,7 @@ export declare class SupabaseVectorStore extends VectorStore {
     delete(params: {
         ids: string[] | number[];
     }): Promise<void>;
+    protected _searchSupabase(query: number[], k: number, filter?: this["FilterType"]): Promise<SearchEmbeddingsResponse[]>;
     /**
      * Performs a similarity search on the vector store.
      * @param query The query vector.
@@ -64,6 +75,21 @@ export declare class SupabaseVectorStore extends VectorStore {
      * @returns A promise that resolves with the search results when the search is complete.
      */
     similaritySearchVectorWithScore(query: number[], k: number, filter?: this["FilterType"]): Promise<[Document, number][]>;
+    /**
+     * Return documents selected using the maximal marginal relevance.
+     * Maximal marginal relevance optimizes for similarity to the query AND diversity
+     * among selected documents.
+     *
+     * @param {string} query - Text to look up documents similar to.
+     * @param {number} options.k - Number of documents to return.
+     * @param {number} options.fetchK=20- Number of documents to fetch before passing to the MMR algorithm.
+     * @param {number} options.lambda=0.5 - Number between 0 and 1 that determines the degree of diversity among the results,
+     *                 where 0 corresponds to maximum diversity and 1 to minimum diversity.
+     * @param {SupabaseLibArgs} options.filter - Optional filter to apply to the search.
+     *
+     * @returns {Promise<Document[]>} - List of documents selected by maximal marginal relevance.
+     */
+    maxMarginalRelevanceSearch(query: string, options: MaxMarginalRelevanceSearchOptions<this["FilterType"]>): Promise<Document[]>;
     /**
      * Creates a new SupabaseVectorStore instance from an array of texts.
      * @param texts The texts to create documents from.
@@ -89,3 +115,4 @@ export declare class SupabaseVectorStore extends VectorStore {
      */
     static fromExistingIndex(embeddings: Embeddings, dbConfig: SupabaseLibArgs): Promise<SupabaseVectorStore>;
 }
+export {};

package/dist/vectorstores/supabase.js

@@ -1,5 +1,6 @@
 import { VectorStore } from "./base.js";
 import { Document } from "../document.js";
+import { maximalMarginalRelevance } from "../util/math.js";
 /**
  * Class for interacting with a Supabase database to store and manage
  * vectors.
@@ -100,14 +101,7 @@ export class SupabaseVectorStore extends VectorStore {
             await this.client.from(this.tableName).delete().eq("id", id);
         }
     }
-    /**
-     * Performs a similarity search on the vector store.
-     * @param query The query vector.
-     * @param k The number of results to return.
-     * @param filter Optional filter to apply to the search.
-     * @returns A promise that resolves with the search results when the search is complete.
-     */
-    async similaritySearchVectorWithScore(query, k, filter) {
+    async _searchSupabase(query, k, filter) {
         if (filter && this.filter) {
             throw new Error("cannot provide both `filter` and `this.filter`");
         }
@@ -132,6 +126,17 @@ export class SupabaseVectorStore extends VectorStore {
         if (error) {
             throw new Error(`Error searching for documents: ${error.code} ${error.message} ${error.details}`);
         }
+        return searches;
+    }
+    /**
+     * Performs a similarity search on the vector store.
+     * @param query The query vector.
+     * @param k The number of results to return.
+     * @param filter Optional filter to apply to the search.
+     * @returns A promise that resolves with the search results when the search is complete.
+     */
+    async similaritySearchVectorWithScore(query, k, filter) {
+        const searches = await this._searchSupabase(query, k, filter);
         const result = searches.map((resp) => [
             new Document({
                 metadata: resp.metadata,
@@ -141,6 +146,30 @@ export class SupabaseVectorStore extends VectorStore {
         ]);
         return result;
     }
+    /**
+     * Return documents selected using the maximal marginal relevance.
+     * Maximal marginal relevance optimizes for similarity to the query AND diversity
+     * among selected documents.
+     *
+     * @param {string} query - Text to look up documents similar to.
+     * @param {number} options.k - Number of documents to return.
+     * @param {number} options.fetchK=20- Number of documents to fetch before passing to the MMR algorithm.
+     * @param {number} options.lambda=0.5 - Number between 0 and 1 that determines the degree of diversity among the results,
+     *                 where 0 corresponds to maximum diversity and 1 to minimum diversity.
+     * @param {SupabaseLibArgs} options.filter - Optional filter to apply to the search.
+     *
+     * @returns {Promise<Document[]>} - List of documents selected by maximal marginal relevance.
+     */
+    async maxMarginalRelevanceSearch(query, options) {
+        const queryEmbedding = await this.embeddings.embedQuery(query);
+        const searches = await this._searchSupabase(queryEmbedding, options.fetchK ?? 20, options.filter);
+        const embeddingList = searches.map((searchResp) => searchResp.embedding);
+        const mmrIndexes = maximalMarginalRelevance(queryEmbedding, embeddingList, options.lambda, options.k);
+        return mmrIndexes.map((idx) => new Document({
+            metadata: searches[idx].metadata,
+            pageContent: searches[idx].content,
+        }));
+    }
     /**
      * Creates a new SupabaseVectorStore instance from an array of texts.
      * @param texts The texts to create documents from.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langchain",
-  "version": "0.0.165",
+  "version": "0.0.166",
   "description": "Typescript bindings for langchain",
   "type": "module",
   "engines": {
@@ -217,6 +217,9 @@
     "vectorstores/lancedb.cjs",
     "vectorstores/lancedb.js",
     "vectorstores/lancedb.d.ts",
+    "vectorstores/momento_vector_index.cjs",
+    "vectorstores/momento_vector_index.js",
+    "vectorstores/momento_vector_index.d.ts",
     "vectorstores/mongo.cjs",
     "vectorstores/mongo.js",
     "vectorstores/mongo.d.ts",
@@ -637,6 +640,9 @@
     "util/math.cjs",
     "util/math.js",
     "util/math.d.ts",
+    "util/time.cjs",
+    "util/time.js",
+    "util/time.d.ts",
     "experimental/autogpt.cjs",
     "experimental/autogpt.js",
     "experimental/autogpt.d.ts",
@@ -667,6 +673,9 @@
     "evaluation.cjs",
     "evaluation.js",
     "evaluation.d.ts",
+    "runnables/remote.cjs",
+    "runnables/remote.js",
+    "runnables/remote.d.ts",
     "index.cjs",
     "index.js",
     "index.d.ts"
@@ -714,7 +723,8 @@
     "@faker-js/faker": "^7.6.0",
     "@getmetal/metal-sdk": "^4.0.0",
     "@getzep/zep-js": "^0.7.0",
-    "@gomomento/sdk": "^1.23.0",
+    "@gomomento/sdk": "^1.41.2",
+    "@gomomento/sdk-core": "^1.41.2",
     "@google-ai/generativelanguage": "^0.2.1",
     "@google-cloud/storage": "^6.10.1",
     "@huggingface/inference": "^1.5.1",
@@ -843,7 +853,9 @@
     "@elastic/elasticsearch": "^8.4.0",
     "@getmetal/metal-sdk": "*",
     "@getzep/zep-js": "^0.7.0",
-    "@gomomento/sdk": "^1.23.0",
+    "@gomomento/sdk": "^1.41.2",
+    "@gomomento/sdk-core": "^1.41.2",
+    "@gomomento/sdk-web": "^1.41.2",
     "@google-ai/generativelanguage": "^0.2.1",
     "@google-cloud/storage": "^6.10.1",
     "@huggingface/inference": "^1.5.1",
@@ -966,6 +978,12 @@
     "@gomomento/sdk": {
       "optional": true
     },
+    "@gomomento/sdk-core": {
+      "optional": true
+    },
+    "@gomomento/sdk-web": {
+      "optional": true
+    },
     "@google-ai/generativelanguage": {
       "optional": true
     },
@@ -1570,6 +1588,11 @@
       "import": "./vectorstores/lancedb.js",
       "require": "./vectorstores/lancedb.cjs"
     },
+    "./vectorstores/momento_vector_index": {
+      "types": "./vectorstores/momento_vector_index.d.ts",
+      "import": "./vectorstores/momento_vector_index.js",
+      "require": "./vectorstores/momento_vector_index.cjs"
+    },
     "./vectorstores/mongo": {
       "types": "./vectorstores/mongo.d.ts",
       "import": "./vectorstores/mongo.js",
@@ -2270,6 +2293,11 @@
       "import": "./util/math.js",
       "require": "./util/math.cjs"
     },
+    "./util/time": {
+      "types": "./util/time.d.ts",
+      "import": "./util/time.js",
+      "require": "./util/time.cjs"
+    },
     "./experimental/autogpt": {
       "types": "./experimental/autogpt.d.ts",
       "import": "./experimental/autogpt.js",
@@ -2320,6 +2348,11 @@
       "import": "./evaluation.js",
       "require": "./evaluation.cjs"
     },
+    "./runnables/remote": {
+      "types": "./runnables/remote.d.ts",
+      "import": "./runnables/remote.js",
+      "require": "./runnables/remote.cjs"
+    },
     "./package.json": "./package.json"
   }
 }

package/runnables/remote.cjs

@@ -0,0 +1 @@
+module.exports = require('../dist/runnables/remote.cjs');

package/runnables/remote.d.ts

@@ -0,0 +1 @@
+export * from '../dist/runnables/remote.js'

package/runnables/remote.js

@@ -0,0 +1 @@
+export * from '../dist/runnables/remote.js'

package/util/time.cjs

@@ -0,0 +1 @@
+module.exports = require('../dist/util/time.cjs');

package/util/time.d.ts

@@ -0,0 +1 @@
+export * from '../dist/util/time.js'

package/util/time.js

@@ -0,0 +1 @@
+export * from '../dist/util/time.js'

package/vectorstores/momento_vector_index.cjs

@@ -0,0 +1 @@
+module.exports = require('../dist/vectorstores/momento_vector_index.cjs');

package/vectorstores/momento_vector_index.d.ts

@@ -0,0 +1 @@
+export * from '../dist/vectorstores/momento_vector_index.js'

package/vectorstores/momento_vector_index.js

@@ -0,0 +1 @@
+export * from '../dist/vectorstores/momento_vector_index.js'