langchain 0.0.188 → 0.0.190

package/dist/embeddings/cohere.cjs

@@ -27,12 +27,6 @@ class CohereEmbeddings extends base_js_1.Embeddings {
             writable: true,
             value: 48
         });
-        Object.defineProperty(this, "inputType", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
         Object.defineProperty(this, "apiKey", {
             enumerable: true,
             configurable: true,
@@ -51,7 +45,6 @@ class CohereEmbeddings extends base_js_1.Embeddings {
         }
         this.modelName = fieldsWithDefaults?.modelName ?? this.modelName;
         this.batchSize = fieldsWithDefaults?.batchSize ?? this.batchSize;
-        this.inputType = fieldsWithDefaults?.inputType;
         this.apiKey = apiKey;
     }
     /**
@@ -65,15 +58,14 @@ class CohereEmbeddings extends base_js_1.Embeddings {
         const batchRequests = batches.map((batch) => this.embeddingWithRetry({
             model: this.modelName,
             texts: batch,
-            inputType: this.inputType,
         }));
         const batchResponses = await Promise.all(batchRequests);
         const embeddings = [];
         for (let i = 0; i < batchResponses.length; i += 1) {
             const batch = batches[i];
-            const { embeddings: batchResponse } = batchResponses[i];
+            const { body: batchResponse } = batchResponses[i];
             for (let j = 0; j < batch.length; j += 1) {
-                embeddings.push(batchResponse[j]);
+                embeddings.push(batchResponse.embeddings[j]);
             }
         }
         return embeddings;
@@ -85,11 +77,11 @@ class CohereEmbeddings extends base_js_1.Embeddings {
      */
     async embedQuery(text) {
         await this.maybeInitClient();
-        const { embeddings } = await this.embeddingWithRetry({
+        const { body } = await this.embeddingWithRetry({
             model: this.modelName,
             texts: [text],
         });
-        return embeddings[0];
+        return body.embeddings[0];
     }
     /**
      * Generates embeddings with retry capabilities.
@@ -105,17 +97,16 @@ class CohereEmbeddings extends base_js_1.Embeddings {
      */
     async maybeInitClient() {
         if (!this.client) {
-            const { CohereClient } = await CohereEmbeddings.imports();
-            this.client = new CohereClient({
-                token: this.apiKey,
-            });
+            const { cohere } = await CohereEmbeddings.imports();
+            this.client = cohere;
+            this.client.init(this.apiKey);
         }
     }
     /** @ignore */
     static async imports() {
         try {
-            const { CohereClient } = await import("cohere-ai");
-            return { CohereClient };
+            const { default: cohere } = await import("cohere-ai");
+            return { cohere };
         }
         catch (e) {
             throw new Error("Please install cohere-ai as a dependency with, e.g. `yarn add cohere-ai`");

package/dist/embeddings/cohere.d.ts

@@ -10,17 +10,6 @@ export interface CohereEmbeddingsParams extends EmbeddingsParams {
      * limited by the Cohere API to a maximum of 96.
      */
     batchSize?: number;
-    /**
-     * Specifies the type of input you're giving to the model.
-     * Not required for older versions of the embedding models (i.e. anything lower than v3),
-     * but is required for more recent versions (i.e. anything bigger than v2).
-     *
-     * * `search_document` - Use this when you encode documents for embeddings that you store in a vector database for search use-cases.
-     * * `search_query` - Use this when you query your vector DB to find relevant documents.
-     * * `classification` - Use this when you use the embeddings as an input to a text classifier.
-     * * `clustering` - Use this when you want to cluster the embeddings.
-     */
-    inputType?: string;
 }
 /**
  * A class for generating embeddings using the Cohere API.
@@ -28,7 +17,6 @@ export interface CohereEmbeddingsParams extends EmbeddingsParams {
 export declare class CohereEmbeddings extends Embeddings implements CohereEmbeddingsParams {
     modelName: string;
     batchSize: number;
-    inputType: string | undefined;
     private apiKey;
     private client;
     /**
@@ -63,6 +51,6 @@ export declare class CohereEmbeddings extends Embeddings implements CohereEmbedd
     private maybeInitClient;
     /** @ignore */
     static imports(): Promise<{
-        CohereClient: typeof import("cohere-ai").CohereClient;
+        cohere: typeof import("cohere-ai");
     }>;
 }

package/dist/embeddings/cohere.js

@@ -24,12 +24,6 @@ export class CohereEmbeddings extends Embeddings {
             writable: true,
             value: 48
         });
-        Object.defineProperty(this, "inputType", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
         Object.defineProperty(this, "apiKey", {
             enumerable: true,
             configurable: true,
@@ -48,7 +42,6 @@ export class CohereEmbeddings extends Embeddings {
         }
         this.modelName = fieldsWithDefaults?.modelName ?? this.modelName;
         this.batchSize = fieldsWithDefaults?.batchSize ?? this.batchSize;
-        this.inputType = fieldsWithDefaults?.inputType;
         this.apiKey = apiKey;
     }
     /**
@@ -62,15 +55,14 @@ export class CohereEmbeddings extends Embeddings {
         const batchRequests = batches.map((batch) => this.embeddingWithRetry({
             model: this.modelName,
             texts: batch,
-            inputType: this.inputType,
         }));
         const batchResponses = await Promise.all(batchRequests);
         const embeddings = [];
         for (let i = 0; i < batchResponses.length; i += 1) {
             const batch = batches[i];
-            const { embeddings: batchResponse } = batchResponses[i];
+            const { body: batchResponse } = batchResponses[i];
             for (let j = 0; j < batch.length; j += 1) {
-                embeddings.push(batchResponse[j]);
+                embeddings.push(batchResponse.embeddings[j]);
             }
         }
         return embeddings;
@@ -82,11 +74,11 @@ export class CohereEmbeddings extends Embeddings {
      */
     async embedQuery(text) {
         await this.maybeInitClient();
-        const { embeddings } = await this.embeddingWithRetry({
+        const { body } = await this.embeddingWithRetry({
             model: this.modelName,
             texts: [text],
         });
-        return embeddings[0];
+        return body.embeddings[0];
     }
     /**
      * Generates embeddings with retry capabilities.
@@ -102,17 +94,16 @@ export class CohereEmbeddings extends Embeddings {
      */
     async maybeInitClient() {
         if (!this.client) {
-            const { CohereClient } = await CohereEmbeddings.imports();
-            this.client = new CohereClient({
-                token: this.apiKey,
-            });
+            const { cohere } = await CohereEmbeddings.imports();
+            this.client = cohere;
+            this.client.init(this.apiKey);
         }
     }
     /** @ignore */
     static async imports() {
         try {
-            const { CohereClient } = await import("cohere-ai");
-            return { CohereClient };
+            const { default: cohere } = await import("cohere-ai");
+            return { cohere };
         }
         catch (e) {
             throw new Error("Please install cohere-ai as a dependency with, e.g. `yarn add cohere-ai`");

package/dist/llms/cohere.cjs

@@ -67,20 +67,18 @@ class Cohere extends base_js_1.LLM {
     }
     /** @ignore */
     async _call(prompt, options) {
-        const { CohereClient } = await Cohere.imports();
-        const cohere = new CohereClient({
-            token: this.apiKey,
-        });
+        const { cohere } = await Cohere.imports();
+        cohere.init(this.apiKey);
         // Hit the `generate` endpoint on the `large` model
         const generateResponse = await this.caller.callWithOptions({ signal: options.signal }, cohere.generate.bind(cohere), {
             prompt,
             model: this.model,
-            maxTokens: this.maxTokens,
+            max_tokens: this.maxTokens,
             temperature: this.temperature,
-            endSequences: options.stop,
+            end_sequences: options.stop,
         });
         try {
-            return generateResponse.generations[0].text;
+            return generateResponse.body.generations[0].text;
         }
         catch {
             console.log(generateResponse);
@@ -90,8 +88,8 @@ class Cohere extends base_js_1.LLM {
     /** @ignore */
     static async imports() {
         try {
-            const { CohereClient } = await import("cohere-ai");
-            return { CohereClient };
+            const { default: cohere } = await import("cohere-ai");
+            return { cohere };
         }
         catch (e) {
             throw new Error("Please install cohere-ai as a dependency with, e.g. `yarn add cohere-ai`");

package/dist/llms/cohere.d.ts

@@ -36,6 +36,6 @@ export declare class Cohere extends LLM implements CohereInput {
     _call(prompt: string, options: this["ParsedCallOptions"]): Promise<string>;
     /** @ignore */
     static imports(): Promise<{
-        CohereClient: typeof import("cohere-ai").CohereClient;
+        cohere: typeof import("cohere-ai");
     }>;
 }

package/dist/llms/cohere.js

@@ -64,20 +64,18 @@ export class Cohere extends LLM {
     }
     /** @ignore */
     async _call(prompt, options) {
-        const { CohereClient } = await Cohere.imports();
-        const cohere = new CohereClient({
-            token: this.apiKey,
-        });
+        const { cohere } = await Cohere.imports();
+        cohere.init(this.apiKey);
         // Hit the `generate` endpoint on the `large` model
         const generateResponse = await this.caller.callWithOptions({ signal: options.signal }, cohere.generate.bind(cohere), {
             prompt,
             model: this.model,
-            maxTokens: this.maxTokens,
+            max_tokens: this.maxTokens,
             temperature: this.temperature,
-            endSequences: options.stop,
+            end_sequences: options.stop,
         });
         try {
-            return generateResponse.generations[0].text;
+            return generateResponse.body.generations[0].text;
         }
         catch {
             console.log(generateResponse);
@@ -87,8 +85,8 @@ export class Cohere extends LLM {
     /** @ignore */
     static async imports() {
         try {
-            const { CohereClient } = await import("cohere-ai");
-            return { CohereClient };
+            const { default: cohere } = await import("cohere-ai");
+            return { cohere };
         }
         catch (e) {
             throw new Error("Please install cohere-ai as a dependency with, e.g. `yarn add cohere-ai`");

package/dist/schema/runnable/passthrough.cjs

@@ -32,7 +32,30 @@ class RunnableAssign extends base_js_1.Runnable {
 }
 exports.RunnableAssign = RunnableAssign;
 /**
- * A runnable that passes through the input.
+ * A runnable to passthrough inputs unchanged or with additional keys.
+ *
+ * This runnable behaves almost like the identity function, except that it
+ * can be configured to add additional keys to the output, if the input is
+ * an object.
+ *
+ * The example below demonstrates how to use `RunnablePassthrough to
+ * passthrough the input from the `.invoke()`
+ *
+ * @example
+ * ```typescript
+ * const chain = RunnableSequence.from([
+ *   {
+ *     question: new RunnablePassthrough(),
+ *     context: async () => loadContextFromStore(),
+ *   },
+ *   prompt,
+ *   llm,
+ *   outputParser,
+ * ]);
+ * const response = await chain.invoke(
+ *   "I can pass a single string instead of an object since I'm using `RunnablePassthrough`."
+ * );
+ * ```
  */
 class RunnablePassthrough extends base_js_1.Runnable {
     constructor() {
@@ -56,6 +79,34 @@ class RunnablePassthrough extends base_js_1.Runnable {
     async invoke(input, options) {
         return this._callWithConfig((input) => Promise.resolve(input), input, options);
     }
+    /**
+     * A runnable that assigns key-value pairs to the input.
+     *
+     * The example below shows how you could use it with an inline function.
+     *
+     * @example
+     * ```typescript
+     * const prompt =
+     *   PromptTemplate.fromTemplate(`Write a SQL query to answer the question using the following schema: {schema}
+     * Question: {question}
+     * SQL Query:`);
+     *
+     * // The `RunnablePassthrough.assign()` is used here to passthrough the input from the `.invoke()`
+     * // call (in this example it's the question), along with any inputs passed to the `.assign()` method.
+     * // In this case, we're passing the schema.
+     * const sqlQueryGeneratorChain = RunnableSequence.from([
+     *   RunnablePassthrough.assign({
+     *     schema: async () => db.getTableInfo(),
+     *   }),
+     *   prompt,
+     *   new ChatOpenAI({}).bind({ stop: ["\nSQLResult:"] }),
+     *   new StringOutputParser(),
+     * ]);
+     * const result = await sqlQueryGeneratorChain.invoke({
+     *   question: "How many employees are there?",
+     * });
+     * ```
+     */
     static assign(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     mapping) {

package/dist/schema/runnable/passthrough.d.ts

@@ -10,12 +10,63 @@ export declare class RunnableAssign<RunInput extends Record<string, any> = any,
     invoke(input: RunInput, options?: Partial<CallOptions>): Promise<RunOutput>;
 }
 /**
- * A runnable that passes through the input.
+ * A runnable to passthrough inputs unchanged or with additional keys.
+ *
+ * This runnable behaves almost like the identity function, except that it
+ * can be configured to add additional keys to the output, if the input is
+ * an object.
+ *
+ * The example below demonstrates how to use `RunnablePassthrough to
+ * passthrough the input from the `.invoke()`
+ *
+ * @example
+ * ```typescript
+ * const chain = RunnableSequence.from([
+ *   {
+ *     question: new RunnablePassthrough(),
+ *     context: async () => loadContextFromStore(),
+ *   },
+ *   prompt,
+ *   llm,
+ *   outputParser,
+ * ]);
+ * const response = await chain.invoke(
+ *   "I can pass a single string instead of an object since I'm using `RunnablePassthrough`."
+ * );
+ * ```
  */
 export declare class RunnablePassthrough<RunInput> extends Runnable<RunInput, RunInput> {
     static lc_name(): string;
     lc_namespace: string[];
     lc_serializable: boolean;
     invoke(input: RunInput, options?: Partial<RunnableConfig>): Promise<RunInput>;
+    /**
+     * A runnable that assigns key-value pairs to the input.
+     *
+     * The example below shows how you could use it with an inline function.
+     *
+     * @example
+     * ```typescript
+     * const prompt =
+     *   PromptTemplate.fromTemplate(`Write a SQL query to answer the question using the following schema: {schema}
+     * Question: {question}
+     * SQL Query:`);
+     *
+     * // The `RunnablePassthrough.assign()` is used here to passthrough the input from the `.invoke()`
+     * // call (in this example it's the question), along with any inputs passed to the `.assign()` method.
+     * // In this case, we're passing the schema.
+     * const sqlQueryGeneratorChain = RunnableSequence.from([
+     *   RunnablePassthrough.assign({
+     *     schema: async () => db.getTableInfo(),
+     *   }),
+     *   prompt,
+     *   new ChatOpenAI({}).bind({ stop: ["\nSQLResult:"] }),
+     *   new StringOutputParser(),
+     * ]);
+     * const result = await sqlQueryGeneratorChain.invoke({
+     *   question: "How many employees are there?",
+     * });
+     * ```
+     */
     static assign(mapping: Record<string, RunnableLike<Record<string, unknown>, any>>): RunnableAssign<Record<string, unknown>, Record<string, unknown>>;
 }

package/dist/schema/runnable/passthrough.js

@@ -28,7 +28,30 @@ export class RunnableAssign extends Runnable {
     }
 }
 /**
- * A runnable that passes through the input.
+ * A runnable to passthrough inputs unchanged or with additional keys.
+ *
+ * This runnable behaves almost like the identity function, except that it
+ * can be configured to add additional keys to the output, if the input is
+ * an object.
+ *
+ * The example below demonstrates how to use `RunnablePassthrough to
+ * passthrough the input from the `.invoke()`
+ *
+ * @example
+ * ```typescript
+ * const chain = RunnableSequence.from([
+ *   {
+ *     question: new RunnablePassthrough(),
+ *     context: async () => loadContextFromStore(),
+ *   },
+ *   prompt,
+ *   llm,
+ *   outputParser,
+ * ]);
+ * const response = await chain.invoke(
+ *   "I can pass a single string instead of an object since I'm using `RunnablePassthrough`."
+ * );
+ * ```
  */
 export class RunnablePassthrough extends Runnable {
     constructor() {
@@ -52,6 +75,34 @@ export class RunnablePassthrough extends Runnable {
     async invoke(input, options) {
         return this._callWithConfig((input) => Promise.resolve(input), input, options);
     }
+    /**
+     * A runnable that assigns key-value pairs to the input.
+     *
+     * The example below shows how you could use it with an inline function.
+     *
+     * @example
+     * ```typescript
+     * const prompt =
+     *   PromptTemplate.fromTemplate(`Write a SQL query to answer the question using the following schema: {schema}
+     * Question: {question}
+     * SQL Query:`);
+     *
+     * // The `RunnablePassthrough.assign()` is used here to passthrough the input from the `.invoke()`
+     * // call (in this example it's the question), along with any inputs passed to the `.assign()` method.
+     * // In this case, we're passing the schema.
+     * const sqlQueryGeneratorChain = RunnableSequence.from([
+     *   RunnablePassthrough.assign({
+     *     schema: async () => db.getTableInfo(),
+     *   }),
+     *   prompt,
+     *   new ChatOpenAI({}).bind({ stop: ["\nSQLResult:"] }),
+     *   new StringOutputParser(),
+     * ]);
+     * const result = await sqlQueryGeneratorChain.invoke({
+     *   question: "How many employees are there?",
+     * });
+     * ```
+     */
     static assign(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     mapping) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langchain",
-  "version": "0.0.188",
+  "version": "0.0.190",
   "description": "Typescript bindings for langchain",
   "type": "module",
   "engines": {
@@ -902,7 +902,7 @@
     "closevector-common": "0.1.0-alpha.1",
     "closevector-node": "0.1.0-alpha.10",
     "closevector-web": "0.1.0-alpha.15",
-    "cohere-ai": "^7.2.0",
+    "cohere-ai": ">=6.0.0",
     "convex": "^1.3.1",
     "d3-dsv": "^2.0.0",
     "dotenv": "^16.0.3",
@@ -1020,7 +1020,7 @@
     "closevector-common": "0.1.0-alpha.1",
     "closevector-node": "0.1.0-alpha.10",
     "closevector-web": "0.1.0-alpha.16",
-    "cohere-ai": "^7.2.0",
+    "cohere-ai": ">=6.0.0",
     "convex": "^1.3.1",
     "d3-dsv": "^2.0.0",
     "epub2": "^3.0.1",