langchain 0.0.62 → 0.0.63

package/dist/chains/base.cjs

@@ -21,11 +21,11 @@ class BaseChain extends index_js_2.BaseLangChain {
     async run(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     input, callbacks) {
-        const isKeylessInput = this.inputKeys.length === 1;
+        const isKeylessInput = this.inputKeys.length <= 1;
         if (!isKeylessInput) {
             throw new Error(`Chain ${this._chainType()} expects multiple inputs, cannot use 'run' `);
         }
-        const values = { [this.inputKeys[0]]: input };
+        const values = this.inputKeys.length ? { [this.inputKeys[0]]: input } : {};
         const returnValues = await this.call(values, callbacks);
         const keys = Object.keys(returnValues);
         if (keys.length === 1) {
@@ -63,6 +63,7 @@ class BaseChain extends index_js_2.BaseLangChain {
         // add the runManager's currentRunId to the outputValues
         Object.defineProperty(outputValues, index_js_1.RUN_KEY, {
             value: runManager ? { runId: runManager?.runId } : undefined,
+            configurable: true,
         });
         return outputValues;
     }

package/dist/chains/base.js

@@ -18,11 +18,11 @@ export class BaseChain extends BaseLangChain {
     async run(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     input, callbacks) {
-        const isKeylessInput = this.inputKeys.length === 1;
+        const isKeylessInput = this.inputKeys.length <= 1;
         if (!isKeylessInput) {
             throw new Error(`Chain ${this._chainType()} expects multiple inputs, cannot use 'run' `);
         }
-        const values = { [this.inputKeys[0]]: input };
+        const values = this.inputKeys.length ? { [this.inputKeys[0]]: input } : {};
         const returnValues = await this.call(values, callbacks);
         const keys = Object.keys(returnValues);
         if (keys.length === 1) {
@@ -60,6 +60,7 @@ export class BaseChain extends BaseLangChain {
         // add the runManager's currentRunId to the outputValues
         Object.defineProperty(outputValues, RUN_KEY, {
             value: runManager ? { runId: runManager?.runId } : undefined,
+            configurable: true,
         });
         return outputValues;
     }

package/dist/chat_models/base.cjs

@@ -37,6 +37,7 @@ class BaseChatModel extends index_js_2.BaseLanguageModel {
         await runManager?.handleLLMEnd(output);
         Object.defineProperty(output, index_js_1.RUN_KEY, {
             value: runManager ? { runId: runManager?.runId } : undefined,
+            configurable: true,
         });
         return output;
     }

package/dist/chat_models/base.js

@@ -34,6 +34,7 @@ export class BaseChatModel extends BaseLanguageModel {
         await runManager?.handleLLMEnd(output);
         Object.defineProperty(output, RUN_KEY, {
             value: runManager ? { runId: runManager?.runId } : undefined,
+            configurable: true,
         });
         return output;
     }

package/dist/llms/base.cjs

@@ -58,6 +58,7 @@ class BaseLLM extends index_js_3.BaseLanguageModel {
         // it isnt included when listing the keys of the output object.
         Object.defineProperty(output, index_js_2.RUN_KEY, {
             value: runManager ? { runId: runManager?.runId } : undefined,
+            configurable: true,
         });
         return output;
     }

package/dist/llms/base.js

@@ -55,6 +55,7 @@ export class BaseLLM extends BaseLanguageModel {
         // it isnt included when listing the keys of the output object.
         Object.defineProperty(output, RUN_KEY, {
             value: runManager ? { runId: runManager?.runId } : undefined,
+            configurable: true,
         });
         return output;
     }

package/dist/vectorstores/chroma.cjs

@@ -13,6 +13,12 @@ class Chroma extends base_js_1.VectorStore {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "collection", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "collectionName", {
             enumerable: true,
             configurable: true,
@@ -31,44 +37,36 @@ class Chroma extends base_js_1.VectorStore {
             writable: true,
             value: void 0
         });
-        this.index = args.index;
         this.numDimensions = args.numDimensions;
         this.embeddings = embeddings;
         this.collectionName = ensureCollectionName(args.collectionName);
-        this.url = args.url || "http://localhost:8000";
+        if ("index" in args) {
+            this.index = args.index;
+        }
+        else if ("url" in args) {
+            this.url = args.url || "http://localhost:8000";
+        }
     }
     async addDocuments(documents) {
         const texts = documents.map(({ pageContent }) => pageContent);
         await this.addVectors(await this.embeddings.embedDocuments(texts), documents);
     }
     async ensureCollection() {
-        if (!this.index) {
-            const { ChromaClient } = await Chroma.imports();
-            this.index = new ChromaClient(this.url);
-            try {
-                await this.index.createCollection(this.collectionName);
-            }
-            catch {
-                // ignore error
+        if (!this.collection) {
+            if (!this.index) {
+                const { ChromaClient } = await Chroma.imports();
+                this.index = new ChromaClient(this.url);
             }
+            this.collection = await this.index.getOrCreateCollection(this.collectionName);
         }
+        return this.collection;
     }
     async addVectors(vectors, documents) {
         if (vectors.length === 0) {
             return;
         }
-        if (!this.index) {
-            if (this.numDimensions === undefined) {
-                this.numDimensions = vectors[0].length;
-            }
-            const { ChromaClient } = await Chroma.imports();
-            this.index = new ChromaClient(this.url);
-            try {
-                await this.index.createCollection(this.collectionName);
-            }
-            catch {
-                // ignore error
-            }
+        if (this.numDimensions === undefined) {
+            this.numDimensions = vectors[0].length;
         }
         if (vectors.length !== documents.length) {
             throw new Error(`Vectors and metadatas must have the same length`);
@@ -76,19 +74,19 @@ class Chroma extends base_js_1.VectorStore {
         if (vectors[0].length !== this.numDimensions) {
             throw new Error(`Vectors must have the same length as the number of dimensions (${this.numDimensions})`);
         }
-        const collection = await this.index.getCollection(this.collectionName);
+        const collection = await this.ensureCollection();
         const docstoreSize = await collection.count();
         await collection.add(Array.from({ length: vectors.length }, (_, i) => (docstoreSize + i).toString()), vectors, documents.map(({ metadata }) => metadata), documents.map(({ pageContent }) => pageContent));
     }
     async similaritySearchVectorWithScore(query, k) {
-        if (!this.index) {
-            throw new Error("Vector store not initialised yet. Try calling `addTexts` first.");
-        }
-        const collection = await this.index.getCollection(this.collectionName);
+        const collection = await this.ensureCollection();
         // similaritySearchVectorWithScore supports one query vector at a time
         // chroma supports multiple query vectors at a time
         const result = await collection.query(query, k);
         const { ids, distances, documents, metadatas } = result;
+        if (!ids || !distances || !documents || !metadatas) {
+            return [];
+        }
         // get the result data from the first and only query vector
         const [firstIds] = ids;
         const [firstDistances] = distances;

package/dist/vectorstores/chroma.d.ts

@@ -1,21 +1,25 @@
-import type { ChromaClient as ChromaClientT } from "chromadb";
+import type { ChromaClient as ChromaClientT, Collection } from "chromadb";
 import { Embeddings } from "../embeddings/base.js";
 import { VectorStore } from "./base.js";
 import { Document } from "../document.js";
-export interface ChromaLibArgs {
+export type ChromaLibArgs = {
     url?: string;
     numDimensions?: number;
     collectionName?: string;
+} | {
     index?: ChromaClientT;
-}
+    numDimensions?: number;
+    collectionName?: string;
+};
 export declare class Chroma extends VectorStore {
     index?: ChromaClientT;
+    collection?: Collection;
     collectionName: string;
     numDimensions?: number;
     url: string;
     constructor(embeddings: Embeddings, args: ChromaLibArgs);
     addDocuments(documents: Document[]): Promise<void>;
-    ensureCollection(): Promise<void>;
+    ensureCollection(): Promise<Collection>;
     addVectors(vectors: number[][], documents: Document[]): Promise<void>;
     similaritySearchVectorWithScore(query: number[], k: number): Promise<[Document<Record<string, any>>, number][]>;
     static fromTexts(texts: string[], metadatas: object[] | object, embeddings: Embeddings, dbConfig: {

package/dist/vectorstores/chroma.js

@@ -10,6 +10,12 @@ export class Chroma extends VectorStore {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "collection", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "collectionName", {
             enumerable: true,
             configurable: true,
@@ -28,44 +34,36 @@ export class Chroma extends VectorStore {
             writable: true,
             value: void 0
         });
-        this.index = args.index;
         this.numDimensions = args.numDimensions;
         this.embeddings = embeddings;
         this.collectionName = ensureCollectionName(args.collectionName);
-        this.url = args.url || "http://localhost:8000";
+        if ("index" in args) {
+            this.index = args.index;
+        }
+        else if ("url" in args) {
+            this.url = args.url || "http://localhost:8000";
+        }
     }
     async addDocuments(documents) {
         const texts = documents.map(({ pageContent }) => pageContent);
         await this.addVectors(await this.embeddings.embedDocuments(texts), documents);
     }
     async ensureCollection() {
-        if (!this.index) {
-            const { ChromaClient } = await Chroma.imports();
-            this.index = new ChromaClient(this.url);
-            try {
-                await this.index.createCollection(this.collectionName);
-            }
-            catch {
-                // ignore error
+        if (!this.collection) {
+            if (!this.index) {
+                const { ChromaClient } = await Chroma.imports();
+                this.index = new ChromaClient(this.url);
             }
+            this.collection = await this.index.getOrCreateCollection(this.collectionName);
         }
+        return this.collection;
     }
     async addVectors(vectors, documents) {
         if (vectors.length === 0) {
             return;
         }
-        if (!this.index) {
-            if (this.numDimensions === undefined) {
-                this.numDimensions = vectors[0].length;
-            }
-            const { ChromaClient } = await Chroma.imports();
-            this.index = new ChromaClient(this.url);
-            try {
-                await this.index.createCollection(this.collectionName);
-            }
-            catch {
-                // ignore error
-            }
+        if (this.numDimensions === undefined) {
+            this.numDimensions = vectors[0].length;
         }
         if (vectors.length !== documents.length) {
             throw new Error(`Vectors and metadatas must have the same length`);
@@ -73,19 +71,19 @@ export class Chroma extends VectorStore {
         if (vectors[0].length !== this.numDimensions) {
             throw new Error(`Vectors must have the same length as the number of dimensions (${this.numDimensions})`);
         }
-        const collection = await this.index.getCollection(this.collectionName);
+        const collection = await this.ensureCollection();
         const docstoreSize = await collection.count();
         await collection.add(Array.from({ length: vectors.length }, (_, i) => (docstoreSize + i).toString()), vectors, documents.map(({ metadata }) => metadata), documents.map(({ pageContent }) => pageContent));
     }
     async similaritySearchVectorWithScore(query, k) {
-        if (!this.index) {
-            throw new Error("Vector store not initialised yet. Try calling `addTexts` first.");
-        }
-        const collection = await this.index.getCollection(this.collectionName);
+        const collection = await this.ensureCollection();
         // similaritySearchVectorWithScore supports one query vector at a time
         // chroma supports multiple query vectors at a time
         const result = await collection.query(query, k);
         const { ids, distances, documents, metadatas } = result;
+        if (!ids || !distances || !documents || !metadatas) {
+            return [];
+        }
         // get the result data from the first and only query vector
         const [firstIds] = ids;
         const [firstDistances] = distances;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langchain",
-  "version": "0.0.62",
+  "version": "0.0.63",
   "description": "Typescript bindings for langchain",
   "type": "module",
   "engines": {
@@ -302,6 +302,7 @@
     "@typescript-eslint/parser": "^5.51.0",
     "axios": "^0.26.0",
     "cheerio": "^1.0.0-rc.12",
+    "chromadb": "^1.4.0",
     "cohere-ai": "^5.0.2",
     "d3-dsv": "^2.0.0",
     "dotenv": "^16.0.3",
@@ -346,7 +347,7 @@
     "@zilliz/milvus2-sdk-node": "^2.2.0",
     "axios": "^0.26.0",
     "cheerio": "^1.0.0-rc.12",
-    "chromadb": "^1.3.0",
+    "chromadb": "^1.4.0",
     "cohere-ai": "^5.0.2",
     "d3-dsv": "^2.0.0",
     "epub2": "^3.0.1",