langchain 0.0.22 → 0.0.23

package/chat_models.d.ts

@@ -0,0 +1 @@
+export * from './dist/chat_models/index.js'

package/chat_models.js

@@ -0,0 +1 @@
+export * from './dist/chat_models/index.js'

package/dist/cache.d.ts

@@ -1,4 +1,4 @@
-import { Generation } from "./llms/index.js";
+import { Generation } from "./schema/index.js";
 export declare const getKey: (...strings: string[]) => string;
 export declare abstract class BaseCache<T = Generation[]> {
     abstract lookup(key: string): T | undefined;

package/dist/chains/llm_chain.d.ts

@@ -1,12 +1,13 @@
-import { BaseChain, ChainValues, ChainInputs } from "./index.js";
+import { BaseChain, ChainInputs, ChainValues } from "./index.js";
 import { BaseLLM, SerializedLLM } from "../llms/index.js";
+import { BaseLanguageModel } from "../schema/index.js";
 import { BaseMemory } from "../memory/index.js";
 import { BasePromptTemplate, SerializedBasePromptTemplate } from "../prompts/index.js";
 export interface LLMChainInput extends ChainInputs {
     /** Prompt object to use */
     prompt: BasePromptTemplate;
     /** LLM Wrapper to use */
-    llm: BaseLLM;
+    llm: BaseLanguageModel;
     /** @ignore */
     outputKey: string;
 }
@@ -31,12 +32,12 @@ export type SerializedLLMChain = {
  */
 export declare class LLMChain extends BaseChain implements LLMChainInput {
     prompt: BasePromptTemplate;
-    llm: BaseLLM;
+    llm: BaseLanguageModel;
     outputKey: string;
     get inputKeys(): string[];
     constructor(fields: {
         prompt: BasePromptTemplate;
-        llm: BaseLLM;
+        llm: BaseLanguageModel;
         outputKey?: string;
         memory?: BaseMemory;
     });

package/dist/chains/llm_chain.js

@@ -48,10 +48,9 @@ export class LLMChain extends BaseChain {
         if ("stop" in values && Array.isArray(values.stop)) {
             stop = values.stop;
         }
-        const formattedString = await this.prompt.format(values);
-        const llmResult = await this.llm.call(formattedString, stop);
-        const result = { [this.outputKey]: llmResult };
-        return result;
+        const promptValue = await this.prompt.formatPromptValue(values);
+        const { generations } = await this.llm.generatePrompt([promptValue], stop);
+        return { [this.outputKey]: generations[0][0].text };
     }
     /**
      * Format prompt with values and pass to LLM
@@ -82,7 +81,7 @@ export class LLMChain extends BaseChain {
     serialize() {
         return {
             _type: this._chainType(),
-            llm: this.llm.serialize(),
+            // llm: this.llm.serialize(), TODO fix this now that llm is BaseLanguageModel
             prompt: this.prompt.serialize(),
         };
     }

package/dist/chains/llm_chain.js.map

@@ -1 +1 @@
-{"version":3,"file":"llm_chain.js","sourceRoot":"","sources":["../../src/chains/llm_chain.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAA4B,MAAM,YAAY,CAAC;AAEjE,OAAO,EAAE,OAAO,EAAiB,MAAM,kBAAkB,CAAC;AAE1D,OAAO,EAAc,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EACL,kBAAkB,EAElB,cAAc,GACf,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AAoBzD;;;;;;;;;;;GAWG;AACH,MAAM,OAAO,QAAS,SAAQ,SAAS;IAOrC,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC;IACpC,CAAC;IAED,YAAY,MAKX;QACC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAhBvB;;;;;WAA2B;QAE3B;;;;;WAAa;QAEb;;;;mBAAY,MAAM;WAAC;QAajB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,IAAI,CAAC;QACT,IAAI,MAAM,IAAI,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;YAClD,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;SACpB;QACD,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACzD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC;QAC/C,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,OAAO,CAAC,MAAmB;QAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvC,OAAO,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAChC,CAAC;IAED,UAAU;QACR,OAAO,WAAoB,CAAC;IAC9B,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAwB;QAC/C,MAAM,aAAa,GAAG,MAAM,qBAAqB,CAC/C,KAAK,EACL,IAAI,CACL,CAAC;QACF,MAAM,gBAAgB,GAAG,MAAM,qBAAqB,CAGlD,QAAQ,EAAE,IAAI,CAAC,CAAC;QAElB,OAAO,IAAI,QAAQ,CAAC;YAClB,GAAG,EAAE,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC;YAC7C,MAAM,EAAE,MAAM,kBAAkB,CAAC,WAAW,CAAC,gBAAgB,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;SAChC,CAAC;IACJ,CAAC;CACF;AAED,mCAAmC;AACnC,MAAM,eAAe,GAAG;;;;;IAKpB,CAAC;AAEL,MAAM,aAAa,GAAG,IAAI,cAAc,CAAC;IACvC,QAAQ,EAAE,eAAe;IACzB,cAAc,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC;CACrC,CAAC,CAAC;AAEH,MAAM,OAAO,iBAAkB,SAAQ,QAAQ;IAC7C,YAAY,MAKX;QACC,KAAK,CAAC;YACJ,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,aAAa;YACtC,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,UAAU;SAC1C,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;IACpD,CAAC;CACF"}
+{"version":3,"file":"llm_chain.js","sourceRoot":"","sources":["../../src/chains/llm_chain.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAA4B,MAAM,YAAY,CAAC;AAEjE,OAAO,EAAE,OAAO,EAAiB,MAAM,kBAAkB,CAAC;AAG1D,OAAO,EAAc,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EACL,kBAAkB,EAClB,cAAc,GAEf,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AAoBzD;;;;;;;;;;;GAWG;AACH,MAAM,OAAO,QAAS,SAAQ,SAAS;IAOrC,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC;IACpC,CAAC;IAED,YAAY,MAKX;QACC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAhBvB;;;;;WAA2B;QAE3B;;;;;WAAuB;QAEvB;;;;mBAAY,MAAM;WAAC;QAajB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,IAAI,CAAC;QACT,IAAI,MAAM,IAAI,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;YAClD,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;SACpB;QACD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAChE,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,CAAC;QAC3E,OAAO,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IACtD,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,OAAO,CAAC,MAAmB;QAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvC,OAAO,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAChC,CAAC;IAED,UAAU;QACR,OAAO,WAAoB,CAAC;IAC9B,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAwB;QAC/C,MAAM,aAAa,GAAG,MAAM,qBAAqB,CAC/C,KAAK,EACL,IAAI,CACL,CAAC;QACF,MAAM,gBAAgB,GAAG,MAAM,qBAAqB,CAGlD,QAAQ,EAAE,IAAI,CAAC,CAAC;QAElB,OAAO,IAAI,QAAQ,CAAC;YAClB,GAAG,EAAE,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC;YAC7C,MAAM,EAAE,MAAM,kBAAkB,CAAC,WAAW,CAAC,gBAAgB,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,6EAA6E;YAC7E,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;SAChC,CAAC;IACJ,CAAC;CACF;AAED,mCAAmC;AACnC,MAAM,eAAe,GAAG;;;;;IAKpB,CAAC;AAEL,MAAM,aAAa,GAAG,IAAI,cAAc,CAAC;IACvC,QAAQ,EAAE,eAAe;IACzB,cAAc,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC;CACrC,CAAC,CAAC;AAEH,MAAM,OAAO,iBAAkB,SAAQ,QAAQ;IAC7C,YAAY,MAKX;QACC,KAAK,CAAC;YACJ,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,aAAa;YACtC,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,UAAU;SAC1C,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;IACpD,CAAC;CACF"}

package/dist/chains/tests/combine_docs_chain.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"combine_docs_chain.test.js","sourceRoot":"","sources":["../../../src/chains/tests/combine_docs_chain.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,OAAO,EAAa,MAAM,qBAAqB,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,MAAM,+BAA+B,CAAC;AAE5D,IAAI,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;IAC9C,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,aAAa,GAAG,CAAC,CAAC;IAEtB,MAAM,OAAQ,SAAQ,OAAO;QAC3B,QAAQ;YACN,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,KAAK,CAAC,SAAS,CAAC,OAAiB,EAAE,CAAY;YAC7C,OAAO;gBACL,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;oBAClC,IAAI,UAAU,GAAG,EAAE,CAAC;oBACpB,IAAI,MAAM,CAAC,UAAU,CAAC,2BAA2B,CAAC,EAAE;wBAClD,UAAU,IAAI,CAAC,CAAC;wBAChB,UAAU,GAAG,sBAAsB,CAAC;qBACrC;yBAAM,IAAI,MAAM,CAAC,UAAU,CAAC,+BAA+B,CAAC,EAAE;wBAC7D,aAAa,IAAI,CAAC,CAAC;wBACnB,UAAU,GAAG,gBAAgB,CAAC;qBAC/B;oBACD,OAAO;wBACL;4BACE,IAAI,EAAE,UAAU;4BAChB,KAAK,EAAE,CAAC;yBACT;qBACF,CAAC;gBACJ,CAAC,CAAC;aACH,CAAC;QACJ,CAAC;KACF;IAED,MAAM,KAAK,GAAG,IAAI,OAAO,CAAC,EAAE,CAAC,CAAC;IAC9B,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;IACzD,MAAM,IAAI,GAAG;QACX,IAAI,QAAQ,CAAC,EAAE,WAAW,EAAE,0BAA0B,EAAE,CAAC;QACzD,IAAI,QAAQ,CAAC,EAAE,WAAW,EAAE,0BAA0B,EAAE,CAAC;KAC1D,CAAC;IACF,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC;QAC3B,eAAe,EAAE,IAAI;QACrB,QAAQ,EAAE,kCAAkC;KAC7C,CAAC,CAAC;IACH,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;IAErB,MAAM,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC;QAClB,IAAI,EAAE,gBAAgB;KACvB,CAAC,CAAC;IACH,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,kBAAkB;IAC9C,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAChC,CAAC,CAAC,CAAC"}
+{"version":3,"file":"combine_docs_chain.test.js","sourceRoot":"","sources":["../../../src/chains/tests/combine_docs_chain.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,+BAA+B,CAAC;AAG5D,IAAI,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;IAC9C,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,aAAa,GAAG,CAAC,CAAC;IAEtB,MAAM,OAAQ,SAAQ,OAAO;QAC3B,QAAQ;YACN,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,KAAK,CAAC,SAAS,CAAC,OAAiB,EAAE,CAAY;YAC7C,OAAO;gBACL,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;oBAClC,IAAI,UAAU,GAAG,EAAE,CAAC;oBACpB,IAAI,MAAM,CAAC,UAAU,CAAC,2BAA2B,CAAC,EAAE;wBAClD,UAAU,IAAI,CAAC,CAAC;wBAChB,UAAU,GAAG,sBAAsB,CAAC;qBACrC;yBAAM,IAAI,MAAM,CAAC,UAAU,CAAC,+BAA+B,CAAC,EAAE;wBAC7D,aAAa,IAAI,CAAC,CAAC;wBACnB,UAAU,GAAG,gBAAgB,CAAC;qBAC/B;oBACD,OAAO;wBACL;4BACE,IAAI,EAAE,UAAU;4BAChB,KAAK,EAAE,CAAC;yBACT;qBACF,CAAC;gBACJ,CAAC,CAAC;aACH,CAAC;QACJ,CAAC;KACF;IAED,MAAM,KAAK,GAAG,IAAI,OAAO,CAAC,EAAE,CAAC,CAAC;IAC9B,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;IACzD,MAAM,IAAI,GAAG;QACX,IAAI,QAAQ,CAAC,EAAE,WAAW,EAAE,0BAA0B,EAAE,CAAC;QACzD,IAAI,QAAQ,CAAC,EAAE,WAAW,EAAE,0BAA0B,EAAE,CAAC;KAC1D,CAAC;IACF,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC;QAC3B,eAAe,EAAE,IAAI;QACrB,QAAQ,EAAE,kCAAkC;KAC7C,CAAC,CAAC;IACH,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;IAErB,MAAM,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC;QAClB,IAAI,EAAE,gBAAgB;KACvB,CAAC,CAAC;IACH,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,kBAAkB;IAC9C,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAChC,CAAC,CAAC,CAAC"}

package/dist/chains/tests/llm_chain.int.test.js

@@ -1,6 +1,7 @@
 import { test } from "@jest/globals";
 import { OpenAI } from "../../llms/openai.js";
-import { PromptTemplate } from "../../prompts/index.js";
+import { ChatOpenAI } from "../../chat_models/index.js";
+import { ChatPromptTemplate, HumanMessagePromptTemplate, PromptTemplate, } from "../../prompts/index.js";
 import { LLMChain, ConversationChain } from "../llm_chain.js";
 import { loadChain } from "../load.js";
 test("Test OpenAI", async () => {
@@ -44,4 +45,16 @@ test("Test ConversationChain", async () => {
     const res = await chain.call({ input: "my favorite color" });
     console.log({ res });
 });
+test("Test LLMChain with ChatOpenAI", async () => {
+    const model = new ChatOpenAI({ temperature: 0.9 });
+    const template = "What is a good name for a company that makes {product}?";
+    const prompt = new PromptTemplate({ template, inputVariables: ["product"] });
+    const humanMessagePrompt = new HumanMessagePromptTemplate(prompt);
+    const chatPromptTemplate = ChatPromptTemplate.fromPromptMessages([
+        humanMessagePrompt,
+    ]);
+    const chatChain = new LLMChain({ llm: model, prompt: chatPromptTemplate });
+    const res = await chatChain.call({ product: "colorful socks" });
+    console.log({ res });
+});
 //# sourceMappingURL=llm_chain.int.test.js.map

package/dist/chains/tests/llm_chain.int.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"llm_chain.int.test.js","sourceRoot":"","sources":["../../../src/chains/tests/llm_chain.int.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EAAE,QAAQ,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEvC,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;IAC7B,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;IACxD,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC3D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,iBAAiB,EAAE,KAAK,IAAI,EAAE;IACjC,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;IACxD,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;IACjD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE;IAC5B,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;IACxD,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,EAAE,mBAAmB,EAAE,CAAC,CAAC,CAAC;IAC9D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;IACrC,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,oCAAoC,CAAC,CAAC;IACpE,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;IACxC,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;IACxD,MAAM,KAAK,GAAG,IAAI,iBAAiB,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACpD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC"}
+{"version":3,"file":"llm_chain.int.test.js","sourceRoot":"","sources":["../../../src/chains/tests/llm_chain.int.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AACxD,OAAO,EACL,kBAAkB,EAClB,0BAA0B,EAC1B,cAAc,GACf,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,QAAQ,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEvC,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;IAC7B,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;IACxD,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC3D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,iBAAiB,EAAE,KAAK,IAAI,EAAE;IACjC,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;IACxD,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;IACjD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE;IAC5B,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;IACxD,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,EAAE,mBAAmB,EAAE,CAAC,CAAC,CAAC;IAC9D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;IACrC,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,oCAAoC,CAAC,CAAC;IACpE,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;IACxC,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;IACxD,MAAM,KAAK,GAAG,IAAI,iBAAiB,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACpD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;IAC/C,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;IACnD,MAAM,QAAQ,GAAG,yDAAyD,CAAC;IAC3E,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IAC7E,MAAM,kBAAkB,GAAG,IAAI,0BAA0B,CAAC,MAAM,CAAC,CAAC;IAClE,MAAM,kBAAkB,GAAG,kBAAkB,CAAC,kBAAkB,CAAC;QAC/D,kBAAkB;KACnB,CAAC,CAAC;IACH,MAAM,SAAS,GAAG,IAAI,QAAQ,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,kBAAkB,EAAE,CAAC,CAAC;IAC3E,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC;IAChE,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC"}

package/dist/chat_models/base.d.ts

@@ -0,0 +1,32 @@
+import { BaseChatMessage, BaseLanguageModel, BasePromptValue, ChatResult, LLMCallbackManager, LLMResult } from "../schema/index.js";
+export type SerializedChatModel = {
+    _model: string;
+    _type: string;
+} & Record<string, any>;
+export declare abstract class BaseChatModel extends BaseLanguageModel {
+    callbackManager: LLMCallbackManager;
+    verbose: boolean;
+    protected constructor(callbackManager?: LLMCallbackManager, verbose?: boolean);
+    generate(messages: BaseChatMessage[][], stop?: string[]): Promise<LLMResult>;
+    /**
+     * Get the identifying parameters of the LLM.
+     */
+    _identifyingParams(): Record<string, any>;
+    _modelType(): string;
+    abstract _llmType(): string;
+    /**
+     * Return a json-like object representing this Chat model.
+     */
+    serialize(): SerializedChatModel;
+    private _tokenizer?;
+    getNumTokens(text: string): number;
+    generatePrompt(promptValues: BasePromptValue[], stop?: string[]): Promise<LLMResult>;
+    abstract _generate(messages: BaseChatMessage[], stop?: string[]): Promise<ChatResult>;
+    call(messages: BaseChatMessage[], stop?: string[]): Promise<BaseChatMessage>;
+    callPrompt(promptValue: BasePromptValue, stop?: string[]): Promise<BaseChatMessage>;
+}
+export declare abstract class SimpleChatModel extends BaseChatModel {
+    protected constructor(callbackManager?: LLMCallbackManager, verbose?: boolean);
+    abstract _call(messages: BaseChatMessage[], stop?: string[]): Promise<string>;
+    _generate(messages: BaseChatMessage[], stop?: string[]): Promise<ChatResult>;
+}

package/dist/chat_models/base.js

@@ -0,0 +1,110 @@
+import GPT3Tokenizer from "gpt3-tokenizer";
+import { AIChatMessage, BaseLanguageModel, } from "../schema/index.js";
+const getCallbackManager = () => ({
+    handleStart: (..._args) => {
+        // console.log(args);
+    },
+    handleEnd: (..._args) => {
+        // console.log(args);
+    },
+    handleError: (..._args) => {
+        // console.log(args);
+    },
+});
+const getVerbosity = () => true;
+export class BaseChatModel extends BaseLanguageModel {
+    constructor(callbackManager, verbose) {
+        super();
+        Object.defineProperty(this, "callbackManager", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "verbose", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        // TODO deserialize
+        Object.defineProperty(this, "_tokenizer", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.callbackManager = callbackManager ?? getCallbackManager();
+        this.verbose = verbose ?? getVerbosity();
+    }
+    async generate(messages, stop) {
+        const generations = [];
+        for (const message of messages) {
+            const result = await this._generate(message, stop);
+            generations.push(result.generations);
+        }
+        return {
+            generations,
+        };
+    }
+    /**
+     * Get the identifying parameters of the LLM.
+     */
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    _identifyingParams() {
+        return {};
+    }
+    _modelType() {
+        return "base_chat_model";
+    }
+    /**
+     * Return a json-like object representing this Chat model.
+     */
+    serialize() {
+        return {
+            ...this._identifyingParams(),
+            _type: this._llmType(),
+            _model: this._modelType(),
+        };
+    }
+    getNumTokens(text) {
+        // TODOs copied from py implementation
+        // TODO: this method may not be exact.
+        // TODO: this method may differ based on model (eg codex, gpt-3.5).
+        if (this._tokenizer === undefined) {
+            const Constructor = GPT3Tokenizer.default;
+            this._tokenizer = new Constructor({ type: "gpt3" });
+        }
+        return this._tokenizer.encode(text).bpe.length;
+    }
+    async generatePrompt(promptValues, stop) {
+        const promptMessages = promptValues.map((promptValue) => promptValue.toChatMessages());
+        return this.generate(promptMessages, stop);
+    }
+    async call(messages, stop) {
+        const { generations } = await this._generate(messages, stop);
+        return generations[0].message;
+    }
+    async callPrompt(promptValue, stop) {
+        const promptMessages = promptValue.toChatMessages();
+        return this.call(promptMessages, stop);
+    }
+}
+export class SimpleChatModel extends BaseChatModel {
+    constructor(callbackManager, verbose) {
+        super(callbackManager, verbose);
+    }
+    async _generate(messages, stop) {
+        const text = await this._call(messages, stop);
+        const message = new AIChatMessage(text);
+        return {
+            generations: [
+                {
+                    text: message.text,
+                    message,
+                },
+            ],
+        };
+    }
+}
+//# sourceMappingURL=base.js.map

package/dist/chat_models/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/chat_models/base.ts"],"names":[],"mappings":"AAAA,OAAO,aAAa,MAAM,gBAAgB,CAAC;AAC3C,OAAO,EACL,aAAa,EAEb,iBAAiB,GAMlB,MAAM,oBAAoB,CAAC;AAE5B,MAAM,kBAAkB,GAAG,GAAuB,EAAE,CAAC,CAAC;IACpD,WAAW,EAAE,CAAC,GAAG,KAAK,EAAE,EAAE;QACxB,qBAAqB;IACvB,CAAC;IACD,SAAS,EAAE,CAAC,GAAG,KAAK,EAAE,EAAE;QACtB,qBAAqB;IACvB,CAAC;IACD,WAAW,EAAE,CAAC,GAAG,KAAK,EAAE,EAAE;QACxB,qBAAqB;IACvB,CAAC;CACF,CAAC,CAAC;AAEH,MAAM,YAAY,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC;AAQhC,MAAM,OAAgB,aAAc,SAAQ,iBAAiB;IAK3D,YACE,eAAoC,EACpC,OAAiB;QAEjB,KAAK,EAAE,CAAC;QARV;;;;;WAAoC;QAEpC;;;;;WAAiB;QAkDjB,mBAAmB;QAEnB;;;;;WAA2C;QA7CzC,IAAI,CAAC,eAAe,GAAG,eAAe,IAAI,kBAAkB,EAAE,CAAC;QAC/D,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,YAAY,EAAE,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,QAA6B,EAC7B,IAAe;QAEf,MAAM,WAAW,GAAuB,EAAE,CAAC;QAC3C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;YAC9B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YACnD,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;SACtC;QACD,OAAO;YACL,WAAW;SACZ,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,8DAA8D;IAC9D,kBAAkB;QAChB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,UAAU;QACR,OAAO,iBAA0B,CAAC;IACpC,CAAC;IAID;;OAEG;IACH,SAAS;QACP,OAAO;YACL,GAAG,IAAI,CAAC,kBAAkB,EAAE;YAC5B,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE;YACtB,MAAM,EAAE,IAAI,CAAC,UAAU,EAAE;SAC1B,CAAC;IACJ,CAAC;IAMD,YAAY,CAAC,IAAY;QACvB,sCAAsC;QACtC,sCAAsC;QACtC,mEAAmE;QACnE,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,EAAE;YACjC,MAAM,WAAW,GAAG,aAAa,CAAC,OAAO,CAAC;YAC1C,IAAI,CAAC,UAAU,GAAG,IAAI,WAAW,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;SACrD;QACD,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,YAA+B,EAC/B,IAAe;QAEf,MAAM,cAAc,GAAwB,YAAY,CAAC,GAAG,CAC1D,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,cAAc,EAAE,CAC9C,CAAC;QACF,OAAO,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;IAC7C,CAAC;IAOD,KAAK,CAAC,IAAI,CACR,QAA2B,EAC3B,IAAe;QAEf,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC7D,OAAO,WAAW,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,UAAU,CACd,WAA4B,EAC5B,IAAe;QAEf,MAAM,cAAc,GAAsB,WAAW,CAAC,cAAc,EAAE,CAAC;QACvE,OAAO,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;IACzC,CAAC;CACF;AAED,MAAM,OAAgB,eAAgB,SAAQ,aAAa;IACzD,YACE,eAAoC,EACpC,OAAiB;QAEjB,KAAK,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;IAClC,CAAC;IAID,KAAK,CAAC,SAAS,CACb,QAA2B,EAC3B,IAAe;QAEf,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC;QACxC,OAAO;YACL,WAAW,EAAE;gBACX;oBACE,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,OAAO;iBACR;aACF;SACF,CAAC;IACJ,CAAC;CACF"}

package/dist/chat_models/index.d.ts

@@ -0,0 +1 @@
+export { ChatOpenAI } from "./openai.js";

package/dist/chat_models/index.js

@@ -0,0 +1,2 @@
+export { ChatOpenAI } from "./openai.js";
+//# sourceMappingURL=index.js.map

package/dist/chat_models/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/chat_models/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC"}

package/dist/chat_models/openai.d.ts

@@ -0,0 +1,150 @@
+import { CreateChatCompletionRequest, ConfigurationParameters } from "openai";
+import { BaseChatModel } from "./base.js";
+import { BaseChatMessage, ChatResult, LLMCallbackManager } from "../schema/index.js";
+interface ModelParams {
+    /** Sampling temperature to use, between 0 and 2, defaults to 1 */
+    temperature: number;
+    /** Total probability mass of tokens to consider at each step, between 0 and 1, defaults to 1 */
+    topP: number;
+    /** Penalizes repeated tokens according to frequency */
+    frequencyPenalty: number;
+    /** Penalizes repeated tokens */
+    presencePenalty: number;
+    /** Number of chat completions to generate for each prompt */
+    n: number;
+    /** Dictionary used to adjust the probability of specific tokens being generated */
+    logitBias?: Record<string, number>;
+    /** Whether to stream the results or not */
+    streaming: boolean;
+    /**
+     * Maximum number of tokens to generate in the completion. -1 returns as many
+     * tokens as possible given the prompt and the model's maximum context size.
+     */
+    maxTokens: number;
+}
+/**
+ * Input to OpenAI class.
+ * @augments ModelParams
+ */
+interface OpenAIInput extends ModelParams {
+    /** Model name to use */
+    modelName: string;
+    /** Holds any additional parameters that are valid to pass to {@link
+     * https://platform.openai.com/docs/api-reference/completions/create |
+     * `openai.create`} that are not explicitly specified on this class.
+     */
+    modelKwargs?: Kwargs;
+    /** Maximum number of retries to make when generating */
+    maxRetries: number;
+    /** List of stop words to use when generating */
+    stop?: string[];
+}
+type Kwargs = Record<string, any>;
+/**
+ * Wrapper around OpenAI large language models that use the Chat endpoint.
+ *
+ * To use you should have the `openai` package installed, with the
+ * `OPENAI_API_KEY` environment variable set.
+ *
+ * @remarks
+ * Any parameters that are valid to be passed to {@link
+ * https://platform.openai.com/docs/api-reference/chat/create |
+ * `openai.createCompletion`} can be passed through {@link modelKwargs}, even
+ * if not explicitly available on this class.
+ *
+ * @augments BaseLLM
+ * @augments OpenAIInput
+ */
+export declare class ChatOpenAI extends BaseChatModel implements OpenAIInput {
+    temperature: number;
+    topP: number;
+    frequencyPenalty: number;
+    presencePenalty: number;
+    n: number;
+    logitBias?: Record<string, number>;
+    modelName: string;
+    modelKwargs?: Kwargs;
+    maxRetries: number;
+    stop?: string[];
+    streaming: boolean;
+    maxTokens: number;
+    private batchClient;
+    private streamingClient;
+    private clientConfig;
+    constructor(fields?: Partial<OpenAIInput> & {
+        callbackManager?: LLMCallbackManager;
+        concurrency?: number;
+        cache?: boolean;
+        verbose?: boolean;
+        openAIApiKey?: string;
+    }, configuration?: ConfigurationParameters);
+    /**
+     * Get the parameters used to invoke the model
+     */
+    invocationParams(): Omit<CreateChatCompletionRequest, "messages"> & Kwargs;
+    _identifyingParams(): {
+        apiKey?: string | Promise<string> | ((name: string) => string) | ((name: string) => Promise<string>) | undefined;
+        organization?: string | undefined;
+        username?: string | undefined;
+        password?: string | undefined;
+        accessToken?: string | Promise<string> | ((name?: string | undefined, scopes?: string[] | undefined) => string) | ((name?: string | undefined, scopes?: string[] | undefined) => Promise<string>) | undefined;
+        basePath?: string | undefined;
+        baseOptions?: any;
+        formDataCtor?: (new () => any) | undefined;
+        stop?: import("openai").CreateChatCompletionRequestStop | undefined;
+        model: string;
+        temperature?: number | null | undefined;
+        top_p?: number | null | undefined;
+        n?: number | null | undefined;
+        stream?: boolean | null | undefined;
+        presence_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
+        logit_bias?: object | null | undefined;
+        user?: string | undefined;
+        model_name: string;
+    };
+    /**
+     * Get the identifying parameters for the model
+     */
+    identifyingParams(): {
+        apiKey?: string | Promise<string> | ((name: string) => string) | ((name: string) => Promise<string>) | undefined;
+        organization?: string | undefined;
+        username?: string | undefined;
+        password?: string | undefined;
+        accessToken?: string | Promise<string> | ((name?: string | undefined, scopes?: string[] | undefined) => string) | ((name?: string | undefined, scopes?: string[] | undefined) => Promise<string>) | undefined;
+        basePath?: string | undefined;
+        baseOptions?: any;
+        formDataCtor?: (new () => any) | undefined;
+        stop?: import("openai").CreateChatCompletionRequestStop | undefined;
+        model: string;
+        temperature?: number | null | undefined;
+        top_p?: number | null | undefined;
+        n?: number | null | undefined;
+        stream?: boolean | null | undefined;
+        presence_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
+        logit_bias?: object | null | undefined;
+        user?: string | undefined;
+        model_name: string;
+    };
+    /**
+     * Call out to OpenAI's endpoint with k unique prompts
+     *
+     * @param messages - The messages to pass into the model.
+     * @param [stop] - Optional list of stop words to use when generating.
+     *
+     * @returns The full LLM output.
+     *
+     * @example
+     * ```ts
+     * import { OpenAI } from "langchain/llms";
+     * const openai = new OpenAI();
+     * const response = await openai.generate(["Tell me a joke."]);
+     * ```
+     */
+    _generate(messages: BaseChatMessage[], stop?: string[]): Promise<ChatResult>;
+    /** @ignore */
+    completionWithRetry(request: CreateChatCompletionRequest): Promise<import("axios").AxiosResponse<import("openai").CreateChatCompletionResponse, any>>;
+    _llmType(): string;
+}
+export {};