langchain 0.0.149 → 0.0.150

package/dist/llms/replicate.cjs

@@ -44,6 +44,12 @@ class Replicate extends base_js_1.LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "promptKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         const apiKey = fields?.apiKey ??
             (0, env_js_1.getEnvironmentVariable)("REPLICATE_API_KEY") ?? // previous environment variable for backwards compatibility
             (0, env_js_1.getEnvironmentVariable)("REPLICATE_API_TOKEN"); // current environment variable, matching the Python library
@@ -53,6 +59,7 @@ class Replicate extends base_js_1.LLM {
         this.apiKey = apiKey;
         this.model = fields.model;
         this.input = fields.input ?? {};
+        this.promptKey = fields.promptKey;
     }
     _llmType() {
         return "replicate";
@@ -64,11 +71,30 @@ class Replicate extends base_js_1.LLM {
             userAgent: "langchain",
             auth: this.apiKey,
         });
+        if (this.promptKey === undefined) {
+            const [modelString, versionString] = this.model.split(":");
+            const version = await replicate.models.versions.get(modelString.split("/")[0], modelString.split("/")[1], versionString);
+            const openapiSchema = version.openapi_schema;
+            const inputProperties = 
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            openapiSchema?.components?.schemas?.Input?.properties;
+            if (inputProperties === undefined) {
+                this.promptKey = "prompt";
+            }
+            else {
+                const sortedInputProperties = Object.entries(inputProperties).sort(([_keyA, valueA], [_keyB, valueB]) => {
+                    const orderA = valueA["x-order"] || 0;
+                    const orderB = valueB["x-order"] || 0;
+                    return orderA - orderB;
+                });
+                this.promptKey = sortedInputProperties[0][0] ?? "prompt";
+            }
+        }
         const output = await this.caller.callWithOptions({ signal: options.signal }, () => replicate.run(this.model, {
-            wait: true,
             input: {
+                // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+                [this.promptKey]: prompt,
                 ...this.input,
-                prompt,
             },
         }));
         if (typeof output === "string") {

package/dist/llms/replicate.d.ts

@@ -10,6 +10,8 @@ export interface ReplicateInput {
         [key: string]: string | number | boolean;
     };
     apiKey?: string;
+    /** The key used to pass prompts to the model. */
+    promptKey?: string;
 }
 /**
  * Class responsible for managing the interaction with the Replicate API.
@@ -26,6 +28,7 @@ export declare class Replicate extends LLM implements ReplicateInput {
     model: ReplicateInput["model"];
     input: ReplicateInput["input"];
     apiKey: string;
+    promptKey?: string;
     constructor(fields: ReplicateInput & BaseLLMParams);
     _llmType(): string;
     /** @ignore */

package/dist/llms/replicate.js

@@ -41,6 +41,12 @@ export class Replicate extends LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "promptKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         const apiKey = fields?.apiKey ??
             getEnvironmentVariable("REPLICATE_API_KEY") ?? // previous environment variable for backwards compatibility
             getEnvironmentVariable("REPLICATE_API_TOKEN"); // current environment variable, matching the Python library
@@ -50,6 +56,7 @@ export class Replicate extends LLM {
         this.apiKey = apiKey;
         this.model = fields.model;
         this.input = fields.input ?? {};
+        this.promptKey = fields.promptKey;
     }
     _llmType() {
         return "replicate";
@@ -61,11 +68,30 @@ export class Replicate extends LLM {
             userAgent: "langchain",
             auth: this.apiKey,
         });
+        if (this.promptKey === undefined) {
+            const [modelString, versionString] = this.model.split(":");
+            const version = await replicate.models.versions.get(modelString.split("/")[0], modelString.split("/")[1], versionString);
+            const openapiSchema = version.openapi_schema;
+            const inputProperties = 
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            openapiSchema?.components?.schemas?.Input?.properties;
+            if (inputProperties === undefined) {
+                this.promptKey = "prompt";
+            }
+            else {
+                const sortedInputProperties = Object.entries(inputProperties).sort(([_keyA, valueA], [_keyB, valueB]) => {
+                    const orderA = valueA["x-order"] || 0;
+                    const orderB = valueB["x-order"] || 0;
+                    return orderA - orderB;
+                });
+                this.promptKey = sortedInputProperties[0][0] ?? "prompt";
+            }
+        }
         const output = await this.caller.callWithOptions({ signal: options.signal }, () => replicate.run(this.model, {
-            wait: true,
             input: {
+                // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+                [this.promptKey]: prompt,
                 ...this.input,
-                prompt,
             },
         }));
         if (typeof output === "string") {

package/dist/prompts/prompt.cjs

@@ -99,6 +99,8 @@ class PromptTemplate extends base_js_1.BaseStringPromptTemplate {
             }
         });
         return new PromptTemplate({
+            // Rely on extracted types
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
             inputVariables: [...names],
             templateFormat,
             template,

package/dist/prompts/prompt.d.ts

@@ -32,7 +32,7 @@ type NonAlphanumeric = " " | "\t" | "\n" | "\r" | '"' | "'" | "{" | "[" | "(" |
  */
 type ExtractTemplateParamsRecursive<T extends string, Result extends string[] = []> = T extends `${string}{${infer Param}}${infer Rest}` ? Param extends `${NonAlphanumeric}${string}` ? ExtractTemplateParamsRecursive<Rest, Result> : ExtractTemplateParamsRecursive<Rest, [...Result, Param]> : Result;
 export type ParamsFromFString<T extends string> = {
-    [Key in ExtractTemplateParamsRecursive<T>[number]]: string;
+    [Key in ExtractTemplateParamsRecursive<T>[number] | (string & Record<never, never>)]: string;
 };
 /**
  * Schema to represent a basic prompt for an LLM.

package/dist/prompts/prompt.js

@@ -96,6 +96,8 @@ export class PromptTemplate extends BaseStringPromptTemplate {
             }
         });
         return new PromptTemplate({
+            // Rely on extracted types
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
             inputVariables: [...names],
             templateFormat,
             template,

package/dist/schema/runnable/base.cjs

@@ -700,6 +700,7 @@ class RunnableSequence extends Runnable {
     static isRunnableSequence(thing) {
         return Array.isArray(thing.middle) && Runnable.isRunnable(thing);
     }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     static from([first, ...runnables]) {
         return new RunnableSequence({
             first: _coerceToRunnable(first),

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

@@ -242,7 +242,7 @@ export declare class RunnableSequence<RunInput = any, RunOutput = any> extends R
     _streamIterator(input: RunInput, options?: RunnableConfig): AsyncGenerator<RunOutput>;
     pipe<NewRunOutput>(coerceable: RunnableLike<RunOutput, NewRunOutput>): RunnableSequence<RunInput, Exclude<NewRunOutput, Error>>;
     static isRunnableSequence(thing: any): thing is RunnableSequence;
-    static from<RunInput, RunOutput>([first, ...runnables]: [
+    static from<RunInput = any, RunOutput = any>([first, ...runnables]: [
         RunnableLike<RunInput>,
         ...RunnableLike[],
         RunnableLike<any, RunOutput>

package/dist/schema/runnable/base.js

@@ -690,6 +690,7 @@ export class RunnableSequence extends Runnable {
     static isRunnableSequence(thing) {
         return Array.isArray(thing.middle) && Runnable.isRunnable(thing);
     }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     static from([first, ...runnables]) {
         return new RunnableSequence({
             first: _coerceToRunnable(first),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langchain",
-  "version": "0.0.149",
+  "version": "0.0.150",
   "description": "Typescript bindings for langchain",
   "type": "module",
   "engines": {
@@ -728,7 +728,7 @@
     "puppeteer": "^19.7.2",
     "redis": "^4.6.6",
     "release-it": "^15.10.1",
-    "replicate": "^0.12.3",
+    "replicate": "^0.18.0",
     "rimraf": "^5.0.1",
     "rollup": "^3.19.1",
     "sonix-speech-recognition": "^2.1.1",
@@ -815,7 +815,7 @@
     "playwright": "^1.32.1",
     "puppeteer": "^19.7.2",
     "redis": "^4.6.4",
-    "replicate": "^0.12.3",
+    "replicate": "^0.18.0",
     "sonix-speech-recognition": "^2.1.1",
     "srt-parser-2": "^1.2.2",
     "typeorm": "^0.3.12",