langchain 0.0.44 → 0.0.45

package/dist/chat_models/openai.d.ts

@@ -45,6 +45,10 @@ interface OpenAIInput extends ModelParams {
     modelKwargs?: Kwargs;
     /** List of stop words to use when generating */
     stop?: string[];
+    /**
+     * Timeout to use when making requests to OpenAI.
+     */
+    timeout?: number;
 }
 type Kwargs = Record<string, any>;
 /**
@@ -72,6 +76,7 @@ export declare class ChatOpenAI extends BaseChatModel implements OpenAIInput {
     modelName: string;
     modelKwargs?: Kwargs;
     stop?: string[];
+    timeout?: number;
     streaming: boolean;
     maxTokens?: number;
     private client;

package/dist/chat_models/openai.js

@@ -98,6 +98,12 @@ export class ChatOpenAI extends BaseChatModel {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "timeout", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "streaming", {
             enumerable: true,
             configurable: true,
@@ -110,7 +116,6 @@ export class ChatOpenAI extends BaseChatModel {
             writable: true,
             value: void 0
         });
-        // Used for non-streaming requests
         Object.defineProperty(this, "client", {
             enumerable: true,
             configurable: true,
@@ -129,6 +134,7 @@ export class ChatOpenAI extends BaseChatModel {
         }
         this.modelName = fields?.modelName ?? this.modelName;
         this.modelKwargs = fields?.modelKwargs ?? {};
+        this.timeout = fields?.timeout;
         this.temperature = fields?.temperature ?? this.temperature;
         this.topP = fields?.topP ?? this.topP;
         this.frequencyPenalty = fields?.frequencyPenalty ?? this.frequencyPenalty;
@@ -293,7 +299,11 @@ export class ChatOpenAI extends BaseChatModel {
         if (!this.client) {
             const clientConfig = new Configuration({
                 ...this.clientConfig,
-                baseOptions: { adapter: fetchAdapter },
+                baseOptions: {
+                    timeout: this.timeout,
+                    adapter: fetchAdapter,
+                    ...this.clientConfig.baseOptions,
+                },
             });
             this.client = new OpenAIApi(clientConfig);
         }

package/dist/chat_models/openai.js.map

@@ -1 +1 @@
-{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/chat_models/openai.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,aAAa,EACb,SAAS,GAKV,MAAM,QAAQ,CAAC;AAEhB,OAAO,YAAY,MAAM,gCAAgC,CAAC;AAC1D,OAAO,EAAE,aAAa,EAAuB,MAAM,WAAW,CAAC;AAC/D,OAAO,EACL,aAAa,EAGb,WAAW,EAEX,gBAAgB,EAEhB,iBAAiB,GAClB,MAAM,oBAAoB,CAAC;AAY5B,SAAS,uBAAuB,CAC9B,IAAiB;IAEjB,QAAQ,IAAI,EAAE;QACZ,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB,KAAK,IAAI;YACP,OAAO,WAAW,CAAC;QACrB,KAAK,OAAO;YACV,OAAO,MAAM,CAAC;QAChB;YACE,MAAM,IAAI,KAAK,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;KACpD;AACH,CAAC;AAED,SAAS,2BAA2B,CAClC,IAAuD,EACvD,IAAY;IAEZ,QAAQ,IAAI,EAAE;QACZ,KAAK,MAAM;YACT,OAAO,IAAI,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACpC,KAAK,WAAW;YACd,OAAO,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC;QACjC,KAAK,QAAQ;YACX,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACrC;YACE,OAAO,IAAI,WAAW,CAAC,IAAI,EAAE,IAAI,IAAI,SAAS,CAAC,CAAC;KACnD;AACH,CAAC;AAoDD;;;;;;;;;;;;;;GAcG;AACH,MAAM,OAAO,UAAW,SAAQ,aAAa;IA4B3C,YACE,MAKG,EACH,aAAuC;QAEvC,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QApCtB;;;;mBAAc,CAAC;WAAC;QAEhB;;;;mBAAO,CAAC;WAAC;QAET;;;;mBAAmB,CAAC;WAAC;QAErB;;;;mBAAkB,CAAC;WAAC;QAEpB;;;;mBAAI,CAAC;WAAC;QAEN;;;;;WAAmC;QAEnC;;;;mBAAY,eAAe;WAAC;QAE5B;;;;;WAAqB;QAErB;;;;;WAAgB;QAEhB;;;;mBAAY,KAAK;WAAC;QAElB;;;;;WAAmB;QAEnB,kCAAkC;QAClC;;;;;WAA0B;QAE1B;;;;;WAA8C;QAa5C,MAAM,MAAM,GAAG,MAAM,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAClE,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;SAC7C;QAED,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,EAAE,CAAC;QAE7C,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC;QAC3D,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QACtC,IAAI,CAAC,gBAAgB,GAAG,MAAM,EAAE,gBAAgB,IAAI,IAAI,CAAC,gBAAgB,CAAC;QAC1E,IAAI,CAAC,eAAe,GAAG,MAAM,EAAE,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC;QACvE,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,CAAC;QACnC,IAAI,CAAC,CAAC,GAAG,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,CAAC;QACnC,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,CAAC;QAEzB,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,KAAK,CAAC;QAE5C,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE;YAChC,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;SACrD;QAED,IAAI,CAAC,YAAY,GAAG;YAClB,MAAM,EAAE,MAAM,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc;YAC1D,GAAG,aAAa;SACjB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,KAAK,EAAE,IAAI,CAAC,IAAI;YAChB,iBAAiB,EAAE,IAAI,CAAC,gBAAgB;YACxC,gBAAgB,EAAE,IAAI,CAAC,eAAe;YACtC,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,CAAC,EAAE,IAAI,CAAC,CAAC;YACT,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,SAAS;YACtB,GAAG,IAAI,CAAC,WAAW;SACpB,CAAC;IACJ,CAAC;IAED,kBAAkB;QAChB,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,GAAG,IAAI,CAAC,gBAAgB,EAAE;YAC1B,GAAG,IAAI,CAAC,YAAY;SACrB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO,IAAI,CAAC,kBAAkB,EAAE,CAAC;IACnC,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,SAAS,CACb,QAA2B,EAC3B,IAAe;QAEf,MAAM,UAAU,GAAe,EAAE,CAAC;QAClC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,EAAE;YACrB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;SAC3D;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,IAAI,GAAG,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC;QAClC,MAAM,cAAc,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YAChD,IAAI,EAAE,uBAAuB,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjD,OAAO,EAAE,OAAO,CAAC,IAAI;SACtB,CAAC,CAAC,CAAC;QAEJ,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM;YACxB,CAAC,CAAC,MAAM,IAAI,OAAO,CAA+B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAClE,IAAI,QAAsC,CAAC;gBAC3C,IAAI,QAAQ,GAAG,KAAK,CAAC;gBACrB,IAAI,CAAC,mBAAmB,CACtB;oBACE,GAAG,MAAM;oBACT,QAAQ,EAAE,cAAc;iBACzB,EACD;oBACE,YAAY,EAAE,QAAQ;oBACtB,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE;wBACnB,IAAI,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,KAAK,QAAQ,EAAE;4BACrC,OAAO,CAAC,QAAQ,CAAC,CAAC;yBACnB;6BAAM;4BACL,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAUpC,CAAC;4BAEF,mDAAmD;4BACnD,IAAI,CAAC,QAAQ,EAAE;gCACb,QAAQ,GAAG;oCACT,EAAE,EAAE,OAAO,CAAC,EAAE;oCACd,MAAM,EAAE,OAAO,CAAC,MAAM;oCACtB,OAAO,EAAE,OAAO,CAAC,OAAO;oCACxB,KAAK,EAAE,OAAO,CAAC,KAAK;oCACpB,OAAO,EAAE,EAAE;iCACZ,CAAC;6BACH;4BAED,iCAAiC;4BACjC,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;4BAChC,IAAI,IAAI,IAAI,IAAI,EAAE;gCAChB,IAAI,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAChC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK,CAC9B,CAAC;gCAEF,IAAI,CAAC,MAAM,EAAE;oCACX,MAAM,GAAG;wCACP,KAAK,EAAE,IAAI,CAAC,KAAK;wCACjB,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,SAAS;qCAC/C,CAAC;oCACF,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;iCAC/B;gCAED,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE;oCACnB,MAAM,CAAC,OAAO,GAAG;wCACf,IAAI,EAAE,IAAI,CAAC,KAAK;4CACd,EAAE,IAA6C;wCACjD,OAAO,EAAE,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE;qCACnC,CAAC;iCACH;gCAED,MAAM,CAAC,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;gCACpD,mCAAmC;gCACnC,KAAK,IAAI,CAAC,eAAe,CAAC,iBAAiB,CACzC,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE,EACzB,IAAI,CACL,CAAC;6BACH;yBACF;oBACH,CAAC;iBACF,CACF,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBAChB,IAAI,CAAC,QAAQ,EAAE;wBACb,QAAQ,GAAG,IAAI,CAAC;wBAChB,MAAM,CAAC,KAAK,CAAC,CAAC;qBACf;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;YACJ,CAAC,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC;gBAC7B,GAAG,MAAM;gBACT,QAAQ,EAAE,cAAc;aACzB,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAE/B,MAAM,EACJ,iBAAiB,EAAE,gBAAgB,EACnC,aAAa,EAAE,YAAY,EAC3B,YAAY,EAAE,WAAW,GAC1B,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;QAErB,IAAI,gBAAgB,EAAE;YACpB,UAAU,CAAC,gBAAgB;gBACzB,CAAC,UAAU,CAAC,gBAAgB,IAAI,CAAC,CAAC,GAAG,gBAAgB,CAAC;SACzD;QAED,IAAI,YAAY,EAAE;YAChB,UAAU,CAAC,YAAY,GAAG,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC,CAAC,GAAG,YAAY,CAAC;SACzE;QAED,IAAI,WAAW,EAAE;YACf,UAAU,CAAC,WAAW,GAAG,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,WAAW,CAAC;SACtE;QAED,MAAM,WAAW,GAAqB,EAAE,CAAC;QACzC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,OAAO,EAAE;YAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,IAAI,IAAI,SAAS,CAAC;YAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;YACzC,WAAW,CAAC,IAAI,CAAC;gBACf,IAAI;gBACJ,OAAO,EAAE,2BAA2B,CAAC,IAAI,EAAE,IAAI,CAAC;aACjD,CAAC,CAAC;SACJ;QACD,OAAO;YACL,WAAW;YACX,SAAS,EAAE,EAAE,UAAU,EAAE;SAC1B,CAAC;IACJ,CAAC;IAED,cAAc;IACd,KAAK,CAAC,mBAAmB,CACvB,OAAoC,EACpC,OAAqC;QAErC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YAChB,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC;gBACrC,GAAG,IAAI,CAAC,YAAY;gBACpB,WAAW,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE;aACvC,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC;SAC3C;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CACrB,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAClD,OAAO,EACP,OAAO,CACR,CAAC;IACJ,CAAC;IAED,QAAQ;QACN,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,iBAAiB,CAAC,GAAG,UAA6B;QAChD,OAAO,UAAU,CAAC,MAAM,CAGtB,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE;YACjB,IAAI,SAAS,IAAI,SAAS,CAAC,UAAU,EAAE;gBACrC,GAAG,CAAC,UAAU,CAAC,gBAAgB;oBAC7B,SAAS,CAAC,UAAU,CAAC,gBAAgB,IAAI,CAAC,CAAC;gBAC7C,GAAG,CAAC,UAAU,CAAC,YAAY,IAAI,SAAS,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC,CAAC;gBACtE,GAAG,CAAC,UAAU,CAAC,WAAW,IAAI,SAAS,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC;aACrE;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EACD;YACE,UAAU,EAAE;gBACV,gBAAgB,EAAE,CAAC;gBACnB,YAAY,EAAE,CAAC;gBACf,WAAW,EAAE,CAAC;aACf;SACF,CACF,CAAC;IACJ,CAAC;CACF"}
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/chat_models/openai.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,aAAa,EACb,SAAS,GAKV,MAAM,QAAQ,CAAC;AAEhB,OAAO,YAAY,MAAM,gCAAgC,CAAC;AAC1D,OAAO,EAAE,aAAa,EAAuB,MAAM,WAAW,CAAC;AAC/D,OAAO,EACL,aAAa,EAGb,WAAW,EAEX,gBAAgB,EAEhB,iBAAiB,GAClB,MAAM,oBAAoB,CAAC;AAY5B,SAAS,uBAAuB,CAC9B,IAAiB;IAEjB,QAAQ,IAAI,EAAE;QACZ,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB,KAAK,IAAI;YACP,OAAO,WAAW,CAAC;QACrB,KAAK,OAAO;YACV,OAAO,MAAM,CAAC;QAChB;YACE,MAAM,IAAI,KAAK,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;KACpD;AACH,CAAC;AAED,SAAS,2BAA2B,CAClC,IAAuD,EACvD,IAAY;IAEZ,QAAQ,IAAI,EAAE;QACZ,KAAK,MAAM;YACT,OAAO,IAAI,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACpC,KAAK,WAAW;YACd,OAAO,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC;QACjC,KAAK,QAAQ;YACX,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACrC;YACE,OAAO,IAAI,WAAW,CAAC,IAAI,EAAE,IAAI,IAAI,SAAS,CAAC,CAAC;KACnD;AACH,CAAC;AAyDD;;;;;;;;;;;;;;GAcG;AACH,MAAM,OAAO,UAAW,SAAQ,aAAa;IA6B3C,YACE,MAKG,EACH,aAAuC;QAEvC,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QArCtB;;;;mBAAc,CAAC;WAAC;QAEhB;;;;mBAAO,CAAC;WAAC;QAET;;;;mBAAmB,CAAC;WAAC;QAErB;;;;mBAAkB,CAAC;WAAC;QAEpB;;;;mBAAI,CAAC;WAAC;QAEN;;;;;WAAmC;QAEnC;;;;mBAAY,eAAe;WAAC;QAE5B;;;;;WAAqB;QAErB;;;;;WAAgB;QAEhB;;;;;WAAiB;QAEjB;;;;mBAAY,KAAK;WAAC;QAElB;;;;;WAAmB;QAEnB;;;;;WAA0B;QAE1B;;;;;WAA8C;QAa5C,MAAM,MAAM,GAAG,MAAM,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAClE,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;SAC7C;QAED,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,EAAE,CAAC;QAC7C,IAAI,CAAC,OAAO,GAAG,MAAM,EAAE,OAAO,CAAC;QAE/B,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC;QAC3D,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QACtC,IAAI,CAAC,gBAAgB,GAAG,MAAM,EAAE,gBAAgB,IAAI,IAAI,CAAC,gBAAgB,CAAC;QAC1E,IAAI,CAAC,eAAe,GAAG,MAAM,EAAE,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC;QACvE,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,CAAC;QACnC,IAAI,CAAC,CAAC,GAAG,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,CAAC;QACnC,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,CAAC;QAEzB,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,KAAK,CAAC;QAE5C,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE;YAChC,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;SACrD;QAED,IAAI,CAAC,YAAY,GAAG;YAClB,MAAM,EAAE,MAAM,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc;YAC1D,GAAG,aAAa;SACjB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,KAAK,EAAE,IAAI,CAAC,IAAI;YAChB,iBAAiB,EAAE,IAAI,CAAC,gBAAgB;YACxC,gBAAgB,EAAE,IAAI,CAAC,eAAe;YACtC,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,CAAC,EAAE,IAAI,CAAC,CAAC;YACT,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,SAAS;YACtB,GAAG,IAAI,CAAC,WAAW;SACpB,CAAC;IACJ,CAAC;IAED,kBAAkB;QAChB,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,GAAG,IAAI,CAAC,gBAAgB,EAAE;YAC1B,GAAG,IAAI,CAAC,YAAY;SACrB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO,IAAI,CAAC,kBAAkB,EAAE,CAAC;IACnC,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,SAAS,CACb,QAA2B,EAC3B,IAAe;QAEf,MAAM,UAAU,GAAe,EAAE,CAAC;QAClC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,EAAE;YACrB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;SAC3D;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,IAAI,GAAG,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC;QAClC,MAAM,cAAc,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YAChD,IAAI,EAAE,uBAAuB,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjD,OAAO,EAAE,OAAO,CAAC,IAAI;SACtB,CAAC,CAAC,CAAC;QAEJ,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM;YACxB,CAAC,CAAC,MAAM,IAAI,OAAO,CAA+B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAClE,IAAI,QAAsC,CAAC;gBAC3C,IAAI,QAAQ,GAAG,KAAK,CAAC;gBACrB,IAAI,CAAC,mBAAmB,CACtB;oBACE,GAAG,MAAM;oBACT,QAAQ,EAAE,cAAc;iBACzB,EACD;oBACE,YAAY,EAAE,QAAQ;oBACtB,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE;wBACnB,IAAI,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,KAAK,QAAQ,EAAE;4BACrC,OAAO,CAAC,QAAQ,CAAC,CAAC;yBACnB;6BAAM;4BACL,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAUpC,CAAC;4BAEF,mDAAmD;4BACnD,IAAI,CAAC,QAAQ,EAAE;gCACb,QAAQ,GAAG;oCACT,EAAE,EAAE,OAAO,CAAC,EAAE;oCACd,MAAM,EAAE,OAAO,CAAC,MAAM;oCACtB,OAAO,EAAE,OAAO,CAAC,OAAO;oCACxB,KAAK,EAAE,OAAO,CAAC,KAAK;oCACpB,OAAO,EAAE,EAAE;iCACZ,CAAC;6BACH;4BAED,iCAAiC;4BACjC,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;4BAChC,IAAI,IAAI,IAAI,IAAI,EAAE;gCAChB,IAAI,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAChC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK,CAC9B,CAAC;gCAEF,IAAI,CAAC,MAAM,EAAE;oCACX,MAAM,GAAG;wCACP,KAAK,EAAE,IAAI,CAAC,KAAK;wCACjB,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,SAAS;qCAC/C,CAAC;oCACF,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;iCAC/B;gCAED,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE;oCACnB,MAAM,CAAC,OAAO,GAAG;wCACf,IAAI,EAAE,IAAI,CAAC,KAAK;4CACd,EAAE,IAA6C;wCACjD,OAAO,EAAE,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE;qCACnC,CAAC;iCACH;gCAED,MAAM,CAAC,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;gCACpD,mCAAmC;gCACnC,KAAK,IAAI,CAAC,eAAe,CAAC,iBAAiB,CACzC,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE,EACzB,IAAI,CACL,CAAC;6BACH;yBACF;oBACH,CAAC;iBACF,CACF,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBAChB,IAAI,CAAC,QAAQ,EAAE;wBACb,QAAQ,GAAG,IAAI,CAAC;wBAChB,MAAM,CAAC,KAAK,CAAC,CAAC;qBACf;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;YACJ,CAAC,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC;gBAC7B,GAAG,MAAM;gBACT,QAAQ,EAAE,cAAc;aACzB,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAE/B,MAAM,EACJ,iBAAiB,EAAE,gBAAgB,EACnC,aAAa,EAAE,YAAY,EAC3B,YAAY,EAAE,WAAW,GAC1B,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;QAErB,IAAI,gBAAgB,EAAE;YACpB,UAAU,CAAC,gBAAgB;gBACzB,CAAC,UAAU,CAAC,gBAAgB,IAAI,CAAC,CAAC,GAAG,gBAAgB,CAAC;SACzD;QAED,IAAI,YAAY,EAAE;YAChB,UAAU,CAAC,YAAY,GAAG,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC,CAAC,GAAG,YAAY,CAAC;SACzE;QAED,IAAI,WAAW,EAAE;YACf,UAAU,CAAC,WAAW,GAAG,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,WAAW,CAAC;SACtE;QAED,MAAM,WAAW,GAAqB,EAAE,CAAC;QACzC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,OAAO,EAAE;YAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,IAAI,IAAI,SAAS,CAAC;YAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;YACzC,WAAW,CAAC,IAAI,CAAC;gBACf,IAAI;gBACJ,OAAO,EAAE,2BAA2B,CAAC,IAAI,EAAE,IAAI,CAAC;aACjD,CAAC,CAAC;SACJ;QACD,OAAO;YACL,WAAW;YACX,SAAS,EAAE,EAAE,UAAU,EAAE;SAC1B,CAAC;IACJ,CAAC;IAED,cAAc;IACd,KAAK,CAAC,mBAAmB,CACvB,OAAoC,EACpC,OAAqC;QAErC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YAChB,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC;gBACrC,GAAG,IAAI,CAAC,YAAY;gBACpB,WAAW,EAAE;oBACX,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,OAAO,EAAE,YAAY;oBACrB,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW;iBACjC;aACF,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC;SAC3C;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CACrB,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAClD,OAAO,EACP,OAAO,CACR,CAAC;IACJ,CAAC;IAED,QAAQ;QACN,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,iBAAiB,CAAC,GAAG,UAA6B;QAChD,OAAO,UAAU,CAAC,MAAM,CAGtB,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE;YACjB,IAAI,SAAS,IAAI,SAAS,CAAC,UAAU,EAAE;gBACrC,GAAG,CAAC,UAAU,CAAC,gBAAgB;oBAC7B,SAAS,CAAC,UAAU,CAAC,gBAAgB,IAAI,CAAC,CAAC;gBAC7C,GAAG,CAAC,UAAU,CAAC,YAAY,IAAI,SAAS,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC,CAAC;gBACtE,GAAG,CAAC,UAAU,CAAC,WAAW,IAAI,SAAS,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC;aACrE;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EACD;YACE,UAAU,EAAE;gBACV,gBAAgB,EAAE,CAAC;gBACnB,YAAY,EAAE,CAAC;gBACf,WAAW,EAAE,CAAC;aACf;SACF,CACF,CAAC;IACJ,CAAC;CACF"}

package/dist/llms/openai-chat.d.ts

@@ -33,6 +33,10 @@ interface OpenAIInput extends ModelParams {
     modelKwargs?: Kwargs;
     /** List of stop words to use when generating */
     stop?: string[];
+    /**
+     * Timeout to use when making requests to OpenAI.
+     */
+    timeout?: number;
     /**
      * Maximum number of tokens to generate in the completion.  If not specified,
      * defaults to the maximum number of tokens allowed by the model.
@@ -66,6 +70,7 @@ export declare class OpenAIChat extends LLM implements OpenAIInput {
     modelName: string;
     prefixMessages?: ChatCompletionRequestMessage[];
     modelKwargs?: Kwargs;
+    timeout?: number;
     stop?: string[];
     streaming: boolean;
     private client;

package/dist/llms/openai-chat.js

@@ -79,6 +79,12 @@ export class OpenAIChat extends LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "timeout", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "stop", {
             enumerable: true,
             configurable: true,
@@ -110,6 +116,7 @@ export class OpenAIChat extends LLM {
         this.modelName = fields?.modelName ?? this.modelName;
         this.prefixMessages = fields?.prefixMessages ?? this.prefixMessages;
         this.modelKwargs = fields?.modelKwargs ?? {};
+        this.timeout = fields?.timeout;
         this.temperature = fields?.temperature ?? this.temperature;
         this.topP = fields?.topP ?? this.topP;
         this.frequencyPenalty = fields?.frequencyPenalty ?? this.frequencyPenalty;
@@ -258,8 +265,9 @@ export class OpenAIChat extends LLM {
             const clientConfig = new Configuration({
                 ...this.clientConfig,
                 baseOptions: {
-                    ...this.clientConfig.baseOptions,
+                    timeout: this.timeout,
                     adapter: fetchAdapter,
+                    ...this.clientConfig.baseOptions,
                 },
             });
             this.client = new OpenAIApi(clientConfig);

package/dist/llms/openai-chat.js.map

@@ -1 +1 @@
-{"version":3,"file":"openai-chat.js","sourceRoot":"","sources":["../../src/llms/openai-chat.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,aAAa,EACb,SAAS,GAMV,MAAM,QAAQ,CAAC;AAEhB,OAAO,YAAY,MAAM,gCAAgC,CAAC;AAC1D,OAAO,EAAiB,GAAG,EAAE,MAAM,WAAW,CAAC;AAuD/C;;;;;;;;;;;;;;GAcG;AACH,MAAM,OAAO,UAAW,SAAQ,GAAG;IA6BjC,YACE,MAGG,EACH,aAAuC;QAEvC,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QAnCtB;;;;mBAAc,CAAC;WAAC;QAEhB;;;;mBAAO,CAAC;WAAC;QAET;;;;mBAAmB,CAAC;WAAC;QAErB;;;;mBAAkB,CAAC;WAAC;QAEpB;;;;mBAAI,CAAC;WAAC;QAEN;;;;;WAAmC;QAEnC;;;;;WAAmB;QAEnB;;;;mBAAY,eAAe;WAAC;QAE5B;;;;;WAAgD;QAEhD;;;;;WAAqB;QAErB;;;;;WAAgB;QAEhB;;;;mBAAY,KAAK;WAAC;QAElB;;;;;WAA0B;QAE1B;;;;;WAA8C;QAW5C,MAAM,MAAM,GAAG,MAAM,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAClE,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;SAC7C;QAED,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,cAAc,GAAG,MAAM,EAAE,cAAc,IAAI,IAAI,CAAC,cAAc,CAAC;QACpE,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,EAAE,CAAC;QAE7C,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC;QAC3D,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QACtC,IAAI,CAAC,gBAAgB,GAAG,MAAM,EAAE,gBAAgB,IAAI,IAAI,CAAC,gBAAgB,CAAC;QAC1E,IAAI,CAAC,eAAe,GAAG,MAAM,EAAE,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC;QACvE,IAAI,CAAC,CAAC,GAAG,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,CAAC;QACnC,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,CAAC;QACnC,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,CAAC;QAEzB,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,KAAK,CAAC;QAE5C,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE;YAChC,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;SACrD;QAED,IAAI,CAAC,YAAY,GAAG;YAClB,MAAM;YACN,GAAG,aAAa;SACjB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,KAAK,EAAE,IAAI,CAAC,IAAI;YAChB,iBAAiB,EAAE,IAAI,CAAC,gBAAgB;YACxC,gBAAgB,EAAE,IAAI,CAAC,eAAe;YACtC,CAAC,EAAE,IAAI,CAAC,CAAC;YACT,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,SAAS;YACtB,GAAG,IAAI,CAAC,WAAW;SACpB,CAAC;IACJ,CAAC;IAED,kBAAkB;QAChB,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,GAAG,IAAI,CAAC,gBAAgB,EAAE;YAC1B,GAAG,IAAI,CAAC,YAAY;SACrB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,GAAG,IAAI,CAAC,gBAAgB,EAAE;YAC1B,GAAG,IAAI,CAAC,YAAY;SACrB,CAAC;IACJ,CAAC;IAEO,cAAc,CAAC,MAAc;QACnC,MAAM,OAAO,GAAiC;YAC5C,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,MAAM;SAChB,CAAC;QACF,OAAO,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IAC7E,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,KAAK,CAAC,MAAc,EAAE,IAAe;QACzC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,EAAE;YACrB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;SAC3D;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,IAAI,GAAG,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC;QAElC,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM;YACxB,CAAC,CAAC,MAAM,IAAI,OAAO,CAA+B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAClE,IAAI,QAAsC,CAAC;gBAC3C,IAAI,QAAQ,GAAG,KAAK,CAAC;gBACrB,IAAI,CAAC,mBAAmB,CACtB;oBACE,GAAG,MAAM;oBACT,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC;iBACtC,EACD;oBACE,YAAY,EAAE,QAAQ;oBACtB,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE;wBACnB,IAAI,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,KAAK,QAAQ,EAAE;4BACrC,OAAO,CAAC,QAAQ,CAAC,CAAC;yBACnB;6BAAM;4BACL,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAUpC,CAAC;4BAEF,mDAAmD;4BACnD,IAAI,CAAC,QAAQ,EAAE;gCACb,QAAQ,GAAG;oCACT,EAAE,EAAE,OAAO,CAAC,EAAE;oCACd,MAAM,EAAE,OAAO,CAAC,MAAM;oCACtB,OAAO,EAAE,OAAO,CAAC,OAAO;oCACxB,KAAK,EAAE,OAAO,CAAC,KAAK;oCACpB,OAAO,EAAE,EAAE;iCACZ,CAAC;6BACH;4BAED,iCAAiC;4BACjC,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;4BAChC,IAAI,IAAI,IAAI,IAAI,EAAE;gCAChB,IAAI,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAChC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK,CAC9B,CAAC;gCAEF,IAAI,CAAC,MAAM,EAAE;oCACX,MAAM,GAAG;wCACP,KAAK,EAAE,IAAI,CAAC,KAAK;wCACjB,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,SAAS;qCAC/C,CAAC;oCACF,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;iCAC/B;gCAED,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE;oCACnB,MAAM,CAAC,OAAO,GAAG;wCACf,IAAI,EAAE,IAAI,CAAC,KAAK;4CACd,EAAE,IAA6C;wCACjD,OAAO,EAAE,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE;qCACnC,CAAC;iCACH;gCAED,MAAM,CAAC,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;gCACpD,mCAAmC;gCACnC,KAAK,IAAI,CAAC,eAAe,CAAC,iBAAiB,CACzC,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE,EACzB,IAAI,CACL,CAAC;6BACH;yBACF;oBACH,CAAC;iBACF,CACF,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBAChB,IAAI,CAAC,QAAQ,EAAE;wBACb,QAAQ,GAAG,IAAI,CAAC;wBAChB,MAAM,CAAC,KAAK,CAAC,CAAC;qBACf;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;YACJ,CAAC,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC;gBAC7B,GAAG,MAAM;gBACT,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC;aACtC,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAE/B,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;IAChD,CAAC;IAED,cAAc;IACd,KAAK,CAAC,mBAAmB,CACvB,OAAoC,EACpC,OAAqC;QAErC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YAChB,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC;gBACrC,GAAG,IAAI,CAAC,YAAY;gBACpB,WAAW,EAAE;oBACX,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW;oBAChC,OAAO,EAAE,YAAY;iBACtB;aACF,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC;SAC3C;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CACrB,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAClD,OAAO,EACP,OAAO,CACR,CAAC;IACJ,CAAC;IAED,QAAQ;QACN,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}
+{"version":3,"file":"openai-chat.js","sourceRoot":"","sources":["../../src/llms/openai-chat.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,aAAa,EACb,SAAS,GAMV,MAAM,QAAQ,CAAC;AAEhB,OAAO,YAAY,MAAM,gCAAgC,CAAC;AAC1D,OAAO,EAAiB,GAAG,EAAE,MAAM,WAAW,CAAC;AA4D/C;;;;;;;;;;;;;;GAcG;AACH,MAAM,OAAO,UAAW,SAAQ,GAAG;IA+BjC,YACE,MAGG,EACH,aAAuC;QAEvC,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QArCtB;;;;mBAAc,CAAC;WAAC;QAEhB;;;;mBAAO,CAAC;WAAC;QAET;;;;mBAAmB,CAAC;WAAC;QAErB;;;;mBAAkB,CAAC;WAAC;QAEpB;;;;mBAAI,CAAC;WAAC;QAEN;;;;;WAAmC;QAEnC;;;;;WAAmB;QAEnB;;;;mBAAY,eAAe;WAAC;QAE5B;;;;;WAAgD;QAEhD;;;;;WAAqB;QAErB;;;;;WAAiB;QAEjB;;;;;WAAgB;QAEhB;;;;mBAAY,KAAK;WAAC;QAElB;;;;;WAA0B;QAE1B;;;;;WAA8C;QAW5C,MAAM,MAAM,GAAG,MAAM,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAClE,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;SAC7C;QAED,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,cAAc,GAAG,MAAM,EAAE,cAAc,IAAI,IAAI,CAAC,cAAc,CAAC;QACpE,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,EAAE,CAAC;QAC7C,IAAI,CAAC,OAAO,GAAG,MAAM,EAAE,OAAO,CAAC;QAE/B,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC;QAC3D,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QACtC,IAAI,CAAC,gBAAgB,GAAG,MAAM,EAAE,gBAAgB,IAAI,IAAI,CAAC,gBAAgB,CAAC;QAC1E,IAAI,CAAC,eAAe,GAAG,MAAM,EAAE,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC;QACvE,IAAI,CAAC,CAAC,GAAG,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,CAAC;QACnC,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,CAAC;QACnC,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,CAAC;QAEzB,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,KAAK,CAAC;QAE5C,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE;YAChC,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;SACrD;QAED,IAAI,CAAC,YAAY,GAAG;YAClB,MAAM;YACN,GAAG,aAAa;SACjB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,KAAK,EAAE,IAAI,CAAC,IAAI;YAChB,iBAAiB,EAAE,IAAI,CAAC,gBAAgB;YACxC,gBAAgB,EAAE,IAAI,CAAC,eAAe;YACtC,CAAC,EAAE,IAAI,CAAC,CAAC;YACT,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,SAAS;YACtB,GAAG,IAAI,CAAC,WAAW;SACpB,CAAC;IACJ,CAAC;IAED,kBAAkB;QAChB,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,GAAG,IAAI,CAAC,gBAAgB,EAAE;YAC1B,GAAG,IAAI,CAAC,YAAY;SACrB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,GAAG,IAAI,CAAC,gBAAgB,EAAE;YAC1B,GAAG,IAAI,CAAC,YAAY;SACrB,CAAC;IACJ,CAAC;IAEO,cAAc,CAAC,MAAc;QACnC,MAAM,OAAO,GAAiC;YAC5C,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,MAAM;SAChB,CAAC;QACF,OAAO,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IAC7E,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,KAAK,CAAC,MAAc,EAAE,IAAe;QACzC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,EAAE;YACrB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;SAC3D;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,IAAI,GAAG,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC;QAElC,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM;YACxB,CAAC,CAAC,MAAM,IAAI,OAAO,CAA+B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAClE,IAAI,QAAsC,CAAC;gBAC3C,IAAI,QAAQ,GAAG,KAAK,CAAC;gBACrB,IAAI,CAAC,mBAAmB,CACtB;oBACE,GAAG,MAAM;oBACT,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC;iBACtC,EACD;oBACE,YAAY,EAAE,QAAQ;oBACtB,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE;wBACnB,IAAI,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,KAAK,QAAQ,EAAE;4BACrC,OAAO,CAAC,QAAQ,CAAC,CAAC;yBACnB;6BAAM;4BACL,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAUpC,CAAC;4BAEF,mDAAmD;4BACnD,IAAI,CAAC,QAAQ,EAAE;gCACb,QAAQ,GAAG;oCACT,EAAE,EAAE,OAAO,CAAC,EAAE;oCACd,MAAM,EAAE,OAAO,CAAC,MAAM;oCACtB,OAAO,EAAE,OAAO,CAAC,OAAO;oCACxB,KAAK,EAAE,OAAO,CAAC,KAAK;oCACpB,OAAO,EAAE,EAAE;iCACZ,CAAC;6BACH;4BAED,iCAAiC;4BACjC,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;4BAChC,IAAI,IAAI,IAAI,IAAI,EAAE;gCAChB,IAAI,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAChC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK,CAC9B,CAAC;gCAEF,IAAI,CAAC,MAAM,EAAE;oCACX,MAAM,GAAG;wCACP,KAAK,EAAE,IAAI,CAAC,KAAK;wCACjB,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,SAAS;qCAC/C,CAAC;oCACF,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;iCAC/B;gCAED,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE;oCACnB,MAAM,CAAC,OAAO,GAAG;wCACf,IAAI,EAAE,IAAI,CAAC,KAAK;4CACd,EAAE,IAA6C;wCACjD,OAAO,EAAE,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE;qCACnC,CAAC;iCACH;gCAED,MAAM,CAAC,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;gCACpD,mCAAmC;gCACnC,KAAK,IAAI,CAAC,eAAe,CAAC,iBAAiB,CACzC,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE,EACzB,IAAI,CACL,CAAC;6BACH;yBACF;oBACH,CAAC;iBACF,CACF,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBAChB,IAAI,CAAC,QAAQ,EAAE;wBACb,QAAQ,GAAG,IAAI,CAAC;wBAChB,MAAM,CAAC,KAAK,CAAC,CAAC;qBACf;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;YACJ,CAAC,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC;gBAC7B,GAAG,MAAM;gBACT,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC;aACtC,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAE/B,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;IAChD,CAAC;IAED,cAAc;IACd,KAAK,CAAC,mBAAmB,CACvB,OAAoC,EACpC,OAAqC;QAErC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YAChB,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC;gBACrC,GAAG,IAAI,CAAC,YAAY;gBACpB,WAAW,EAAE;oBACX,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,OAAO,EAAE,YAAY;oBACrB,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW;iBACjC;aACF,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC;SAC3C;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CACrB,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAClD,OAAO,EACP,OAAO,CACR,CAAC;IACJ,CAAC;IAED,QAAQ;QACN,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}

package/dist/llms/openai.d.ts

@@ -41,6 +41,10 @@ interface OpenAIInput extends ModelParams {
     batchSize: number;
     /** List of stop words to use when generating */
     stop?: string[];
+    /**
+     * Timeout to use when making requests to OpenAI.
+     */
+    timeout?: number;
 }
 type Kwargs = Record<string, any>;
 /**
@@ -70,6 +74,7 @@ export declare class OpenAI extends BaseLLM implements OpenAIInput {
     modelName: string;
     modelKwargs?: Kwargs;
     batchSize: number;
+    timeout?: number;
     stop?: string[];
     streaming: boolean;
     private client;

package/dist/llms/openai.js

@@ -93,6 +93,12 @@ export class OpenAI extends BaseLLM {
             writable: true,
             value: 20
         });
+        Object.defineProperty(this, "timeout", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "stop", {
             enumerable: true,
             configurable: true,
@@ -105,7 +111,6 @@ export class OpenAI extends BaseLLM {
             writable: true,
             value: false
         });
-        // Used for non-streaming requests
         Object.defineProperty(this, "client", {
             enumerable: true,
             configurable: true,
@@ -125,6 +130,7 @@ export class OpenAI extends BaseLLM {
         this.modelName = fields?.modelName ?? this.modelName;
         this.modelKwargs = fields?.modelKwargs ?? {};
         this.batchSize = fields?.batchSize ?? this.batchSize;
+        this.timeout = fields?.timeout;
         this.temperature = fields?.temperature ?? this.temperature;
         this.maxTokens = fields?.maxTokens ?? this.maxTokens;
         this.topP = fields?.topP ?? this.topP;
@@ -294,8 +300,9 @@ export class OpenAI extends BaseLLM {
             const clientConfig = new Configuration({
                 ...this.clientConfig,
                 baseOptions: {
-                    ...this.clientConfig.baseOptions,
+                    timeout: this.timeout,
                     adapter: fetchAdapter,
+                    ...this.clientConfig.baseOptions,
                 },
             });
             this.client = new OpenAIApi(clientConfig);

package/dist/llms/openai.js.map

@@ -1 +1 @@
-{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/llms/openai.ts"],"names":[],"mappings":"AACA,OAAO,EACL,aAAa,EAKb,SAAS,GACV,MAAM,QAAQ,CAAC;AAChB,OAAO,YAAY,MAAM,gCAAgC,CAAC;AAE1D,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAiB,MAAM,WAAW,CAAC;AACnD,OAAO,EAAE,kBAAkB,EAAE,MAAM,kCAAkC,CAAC;AACtE,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAiE9C;;;;;;;;;;;;;;GAcG;AACH,MAAM,OAAO,MAAO,SAAQ,OAAO;IAgCjC,YACE,MAGG,EACH,aAAuC;QAEvC,IACE,MAAM,EAAE,SAAS,EAAE,UAAU,CAAC,eAAe,CAAC;YAC9C,MAAM,EAAE,SAAS,EAAE,UAAU,CAAC,OAAO,CAAC,EACtC;YACA,qFAAqF;YACrF,OAAO,IAAI,UAAU,CAAC,MAAM,EAAE,aAAa,CAAkB,CAAC;SAC/D;QACD,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QA7CtB;;;;mBAAc,GAAG;WAAC;QAElB;;;;mBAAY,GAAG;WAAC;QAEhB;;;;mBAAO,CAAC;WAAC;QAET;;;;mBAAmB,CAAC;WAAC;QAErB;;;;mBAAkB,CAAC;WAAC;QAEpB;;;;mBAAI,CAAC;WAAC;QAEN;;;;mBAAS,CAAC;WAAC;QAEX;;;;;WAAmC;QAEnC;;;;mBAAY,kBAAkB;WAAC;QAE/B;;;;;WAAqB;QAErB;;;;mBAAY,EAAE;WAAC;QAEf;;;;;WAAgB;QAEhB;;;;mBAAY,KAAK;WAAC;QAElB,kCAAkC;QAClC;;;;;WAA0B;QAE1B;;;;;WAA8C;QAkB5C,MAAM,MAAM,GAAG,MAAM,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAClE,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;SAC7C;QAED,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,EAAE,CAAC;QAC7C,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QAErD,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC;QAC3D,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QACtC,IAAI,CAAC,gBAAgB,GAAG,MAAM,EAAE,gBAAgB,IAAI,IAAI,CAAC,gBAAgB,CAAC;QAC1E,IAAI,CAAC,eAAe,GAAG,MAAM,EAAE,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC;QACvE,IAAI,CAAC,CAAC,GAAG,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC;QAC5C,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,CAAC;QACnC,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,CAAC;QAEzB,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,KAAK,CAAC;QAE5C,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE;YAChC,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;SACrD;QAED,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE;YACrC,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;SAC1D;QAED,IAAI,CAAC,YAAY,GAAG;YAClB,MAAM;YACN,GAAG,aAAa;SACjB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,KAAK,EAAE,IAAI,CAAC,IAAI;YAChB,iBAAiB,EAAE,IAAI,CAAC,gBAAgB;YACxC,gBAAgB,EAAE,IAAI,CAAC,eAAe;YACtC,CAAC,EAAE,IAAI,CAAC,CAAC;YACT,OAAO,EAAE,IAAI,CAAC,MAAM;YACpB,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,SAAS;YACtB,GAAG,IAAI,CAAC,WAAW;SACpB,CAAC;IACJ,CAAC;IAED,kBAAkB;QAChB,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,GAAG,IAAI,CAAC,gBAAgB,EAAE;YAC1B,GAAG,IAAI,CAAC,YAAY;SACrB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO,IAAI,CAAC,kBAAkB,EAAE,CAAC;IACnC,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,SAAS,CAAC,OAAiB,EAAE,IAAe;QAChD,MAAM,UAAU,GAAG,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QACvD,MAAM,OAAO,GAA2C,EAAE,CAAC;QAC3D,MAAM,UAAU,GAAe,EAAE,CAAC;QAElC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,EAAE;YACrB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;SAC3D;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,IAAI,GAAG,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC;QAElC,IAAI,MAAM,CAAC,UAAU,KAAK,CAAC,CAAC,EAAE;YAC5B,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE;gBACxB,MAAM,IAAI,KAAK,CACb,wDAAwD,CACzD,CAAC;aACH;YACD,MAAM,CAAC,UAAU,GAAG,MAAM,kBAAkB,CAAC;gBAC3C,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;gBAClB,iEAAiE;gBACjE,SAAS,EAAE,IAAI,CAAC,SAA0B;aAC3C,CAAC,CAAC;SACJ;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;YAC7C,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM;gBACxB,CAAC,CAAC,MAAM,IAAI,OAAO,CAA2B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBAC9D,MAAM,MAAM,GAAyC,EAAE,CAAC;oBACxD,IAAI,QAAmD,CAAC;oBACxD,IAAI,QAAQ,GAAG,KAAK,CAAC;oBACrB,IAAI,CAAC,mBAAmB,CACtB;wBACE,GAAG,MAAM;wBACT,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;qBACtB,EACD;wBACE,YAAY,EAAE,QAAQ;wBACtB,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE;4BACnB,IAAI,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,KAAK,QAAQ,EAAE;gCACrC,OAAO,CAAC;oCACN,GAAG,QAAQ;oCACX,OAAO,EAAE,CAAC,MAAM,CAAC;iCAClB,CAAC,CAAC;6BACJ;iCAAM;gCACL,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAGpC,CAAC;gCAEF,mDAAmD;gCACnD,IAAI,CAAC,QAAQ,EAAE;oCACb,QAAQ,GAAG;wCACT,EAAE,EAAE,OAAO,CAAC,EAAE;wCACd,MAAM,EAAE,OAAO,CAAC,MAAM;wCACtB,OAAO,EAAE,OAAO,CAAC,OAAO;wCACxB,KAAK,EAAE,OAAO,CAAC,KAAK;qCACrB,CAAC;iCACH;gCAED,iCAAiC;gCACjC,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gCAChC,IAAI,IAAI,IAAI,IAAI,EAAE;oCAChB,MAAM,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;oCACtD,MAAM,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;oCAC1C,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;oCAChC,mCAAmC;oCACnC,KAAK,IAAI,CAAC,eAAe,CAAC,iBAAiB,CACzC,IAAI,CAAC,IAAI,IAAI,EAAE,EACf,IAAI,CACL,CAAC;iCACH;6BACF;wBACH,CAAC;qBACF,CACF,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;wBAChB,IAAI,CAAC,QAAQ,EAAE;4BACb,QAAQ,GAAG,IAAI,CAAC;4BAChB,MAAM,CAAC,KAAK,CAAC,CAAC;yBACf;oBACH,CAAC,CAAC,CAAC;gBACL,CAAC,CAAC;gBACJ,CAAC,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC;oBAC7B,GAAG,MAAM;oBACT,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;iBACtB,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAE/B,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;YAE9B,MAAM,EACJ,iBAAiB,EAAE,gBAAgB,EACnC,aAAa,EAAE,YAAY,EAC3B,YAAY,EAAE,WAAW,GAC1B,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;YAErB,IAAI,gBAAgB,EAAE;gBACpB,UAAU,CAAC,gBAAgB;oBACzB,CAAC,UAAU,CAAC,gBAAgB,IAAI,CAAC,CAAC,GAAG,gBAAgB,CAAC;aACzD;YAED,IAAI,YAAY,EAAE;gBAChB,UAAU,CAAC,YAAY,GAAG,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC,CAAC,GAAG,YAAY,CAAC;aACzE;YAED,IAAI,WAAW,EAAE;gBACf,UAAU,CAAC,WAAW,GAAG,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,WAAW,CAAC;aACtE;SACF;QAED,MAAM,WAAW,GAAG,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE,CACpE,aAAa,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAC7B,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,EAAE;YACvB,cAAc,EAAE;gBACd,YAAY,EAAE,MAAM,CAAC,aAAa;gBAClC,QAAQ,EAAE,MAAM,CAAC,QAAQ;aAC1B;SACF,CAAC,CAAC,CACJ,CAAC;QACF,OAAO;YACL,WAAW;YACX,SAAS,EAAE,EAAE,UAAU,EAAE;SAC1B,CAAC;IACJ,CAAC;IAED,cAAc;IACd,KAAK,CAAC,mBAAmB,CACvB,OAAgC,EAChC,OAAqC;QAErC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YAChB,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC;gBACrC,GAAG,IAAI,CAAC,YAAY;gBACpB,WAAW,EAAE;oBACX,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW;oBAChC,OAAO,EAAE,YAAY;iBACtB;aACF,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC;SAC3C;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CACrB,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAC9C,OAAO,EACP,OAAO,CACR,CAAC;IACJ,CAAC;IAED,QAAQ;QACN,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,iBAAkB,SAAQ,MAAM;IAK3C,YACE,MAGC;QAED,KAAK,CAAC,MAAM,CAAC,CAAC;QAVhB;;;;;WAA2B;QAE3B;;;;;WAAkB;QAUhB,IAAI,CAAC,MAAM,GAAG,MAAM,EAAE,MAAM,IAAI,EAAE,CAAC;QACnC,IAAI,CAAC,iBAAiB;YACpB,MAAM,EAAE,iBAAiB,IAAI,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;QAE/D,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE;YAC3B,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;SAChD;IACH,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,OAAgC,EAChC,OAAqC;QAErC,IAAI,OAAO,CAAC,MAAM,EAAE;YAClB,OAAO,KAAK,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;SACpD;QAED,MAAM,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACpC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QAC1D,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAElC,qDAAqD;QACrD,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,2CAA2C,EAAE;YACzE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,MAAM,EAAE,kBAAkB;aAC3B;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,aAAa,EAAE,0BAA0B;gBACzC,IAAI,EAAE,EAAE;gBACR,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE;gBACzD,IAAI,EAAE,IAAI,CAAC,MAAM,IAAI,EAAE;gBACvB,gBAAgB,EAAE,QAAQ,CAAC,IAAI;gBAC/B,kBAAkB,EAAE,IAAI,CAAC,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC;gBACvD,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC;gBACnD,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,mBAAmB;aACzC,CAAC;SACH,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/llms/openai.ts"],"names":[],"mappings":"AACA,OAAO,EACL,aAAa,EAKb,SAAS,GACV,MAAM,QAAQ,CAAC;AAChB,OAAO,YAAY,MAAM,gCAAgC,CAAC;AAE1D,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAiB,MAAM,WAAW,CAAC;AACnD,OAAO,EAAE,kBAAkB,EAAE,MAAM,kCAAkC,CAAC;AACtE,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAsE9C;;;;;;;;;;;;;;GAcG;AACH,MAAM,OAAO,MAAO,SAAQ,OAAO;IAiCjC,YACE,MAGG,EACH,aAAuC;QAEvC,IACE,MAAM,EAAE,SAAS,EAAE,UAAU,CAAC,eAAe,CAAC;YAC9C,MAAM,EAAE,SAAS,EAAE,UAAU,CAAC,OAAO,CAAC,EACtC;YACA,qFAAqF;YACrF,OAAO,IAAI,UAAU,CAAC,MAAM,EAAE,aAAa,CAAkB,CAAC;SAC/D;QACD,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QA9CtB;;;;mBAAc,GAAG;WAAC;QAElB;;;;mBAAY,GAAG;WAAC;QAEhB;;;;mBAAO,CAAC;WAAC;QAET;;;;mBAAmB,CAAC;WAAC;QAErB;;;;mBAAkB,CAAC;WAAC;QAEpB;;;;mBAAI,CAAC;WAAC;QAEN;;;;mBAAS,CAAC;WAAC;QAEX;;;;;WAAmC;QAEnC;;;;mBAAY,kBAAkB;WAAC;QAE/B;;;;;WAAqB;QAErB;;;;mBAAY,EAAE;WAAC;QAEf;;;;;WAAiB;QAEjB;;;;;WAAgB;QAEhB;;;;mBAAY,KAAK;WAAC;QAElB;;;;;WAA0B;QAE1B;;;;;WAA8C;QAkB5C,MAAM,MAAM,GAAG,MAAM,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAClE,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;SAC7C;QAED,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,EAAE,CAAC;QAC7C,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,OAAO,GAAG,MAAM,EAAE,OAAO,CAAC;QAE/B,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC;QAC3D,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QACtC,IAAI,CAAC,gBAAgB,GAAG,MAAM,EAAE,gBAAgB,IAAI,IAAI,CAAC,gBAAgB,CAAC;QAC1E,IAAI,CAAC,eAAe,GAAG,MAAM,EAAE,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC;QACvE,IAAI,CAAC,CAAC,GAAG,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC;QAC5C,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,CAAC;QACnC,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,CAAC;QAEzB,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,KAAK,CAAC;QAE5C,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE;YAChC,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;SACrD;QAED,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE;YACrC,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;SAC1D;QAED,IAAI,CAAC,YAAY,GAAG;YAClB,MAAM;YACN,GAAG,aAAa;SACjB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,KAAK,EAAE,IAAI,CAAC,IAAI;YAChB,iBAAiB,EAAE,IAAI,CAAC,gBAAgB;YACxC,gBAAgB,EAAE,IAAI,CAAC,eAAe;YACtC,CAAC,EAAE,IAAI,CAAC,CAAC;YACT,OAAO,EAAE,IAAI,CAAC,MAAM;YACpB,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,SAAS;YACtB,GAAG,IAAI,CAAC,WAAW;SACpB,CAAC;IACJ,CAAC;IAED,kBAAkB;QAChB,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,GAAG,IAAI,CAAC,gBAAgB,EAAE;YAC1B,GAAG,IAAI,CAAC,YAAY;SACrB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO,IAAI,CAAC,kBAAkB,EAAE,CAAC;IACnC,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,SAAS,CAAC,OAAiB,EAAE,IAAe;QAChD,MAAM,UAAU,GAAG,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QACvD,MAAM,OAAO,GAA2C,EAAE,CAAC;QAC3D,MAAM,UAAU,GAAe,EAAE,CAAC;QAElC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,EAAE;YACrB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;SAC3D;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,IAAI,GAAG,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC;QAElC,IAAI,MAAM,CAAC,UAAU,KAAK,CAAC,CAAC,EAAE;YAC5B,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE;gBACxB,MAAM,IAAI,KAAK,CACb,wDAAwD,CACzD,CAAC;aACH;YACD,MAAM,CAAC,UAAU,GAAG,MAAM,kBAAkB,CAAC;gBAC3C,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;gBAClB,iEAAiE;gBACjE,SAAS,EAAE,IAAI,CAAC,SAA0B;aAC3C,CAAC,CAAC;SACJ;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;YAC7C,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM;gBACxB,CAAC,CAAC,MAAM,IAAI,OAAO,CAA2B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBAC9D,MAAM,MAAM,GAAyC,EAAE,CAAC;oBACxD,IAAI,QAAmD,CAAC;oBACxD,IAAI,QAAQ,GAAG,KAAK,CAAC;oBACrB,IAAI,CAAC,mBAAmB,CACtB;wBACE,GAAG,MAAM;wBACT,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;qBACtB,EACD;wBACE,YAAY,EAAE,QAAQ;wBACtB,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE;4BACnB,IAAI,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,KAAK,QAAQ,EAAE;gCACrC,OAAO,CAAC;oCACN,GAAG,QAAQ;oCACX,OAAO,EAAE,CAAC,MAAM,CAAC;iCAClB,CAAC,CAAC;6BACJ;iCAAM;gCACL,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAGpC,CAAC;gCAEF,mDAAmD;gCACnD,IAAI,CAAC,QAAQ,EAAE;oCACb,QAAQ,GAAG;wCACT,EAAE,EAAE,OAAO,CAAC,EAAE;wCACd,MAAM,EAAE,OAAO,CAAC,MAAM;wCACtB,OAAO,EAAE,OAAO,CAAC,OAAO;wCACxB,KAAK,EAAE,OAAO,CAAC,KAAK;qCACrB,CAAC;iCACH;gCAED,iCAAiC;gCACjC,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gCAChC,IAAI,IAAI,IAAI,IAAI,EAAE;oCAChB,MAAM,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;oCACtD,MAAM,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;oCAC1C,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;oCAChC,mCAAmC;oCACnC,KAAK,IAAI,CAAC,eAAe,CAAC,iBAAiB,CACzC,IAAI,CAAC,IAAI,IAAI,EAAE,EACf,IAAI,CACL,CAAC;iCACH;6BACF;wBACH,CAAC;qBACF,CACF,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;wBAChB,IAAI,CAAC,QAAQ,EAAE;4BACb,QAAQ,GAAG,IAAI,CAAC;4BAChB,MAAM,CAAC,KAAK,CAAC,CAAC;yBACf;oBACH,CAAC,CAAC,CAAC;gBACL,CAAC,CAAC;gBACJ,CAAC,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC;oBAC7B,GAAG,MAAM;oBACT,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;iBACtB,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAE/B,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;YAE9B,MAAM,EACJ,iBAAiB,EAAE,gBAAgB,EACnC,aAAa,EAAE,YAAY,EAC3B,YAAY,EAAE,WAAW,GAC1B,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;YAErB,IAAI,gBAAgB,EAAE;gBACpB,UAAU,CAAC,gBAAgB;oBACzB,CAAC,UAAU,CAAC,gBAAgB,IAAI,CAAC,CAAC,GAAG,gBAAgB,CAAC;aACzD;YAED,IAAI,YAAY,EAAE;gBAChB,UAAU,CAAC,YAAY,GAAG,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC,CAAC,GAAG,YAAY,CAAC;aACzE;YAED,IAAI,WAAW,EAAE;gBACf,UAAU,CAAC,WAAW,GAAG,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,WAAW,CAAC;aACtE;SACF;QAED,MAAM,WAAW,GAAG,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE,CACpE,aAAa,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAC7B,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,EAAE;YACvB,cAAc,EAAE;gBACd,YAAY,EAAE,MAAM,CAAC,aAAa;gBAClC,QAAQ,EAAE,MAAM,CAAC,QAAQ;aAC1B;SACF,CAAC,CAAC,CACJ,CAAC;QACF,OAAO;YACL,WAAW;YACX,SAAS,EAAE,EAAE,UAAU,EAAE;SAC1B,CAAC;IACJ,CAAC;IAED,cAAc;IACd,KAAK,CAAC,mBAAmB,CACvB,OAAgC,EAChC,OAAqC;QAErC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YAChB,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC;gBACrC,GAAG,IAAI,CAAC,YAAY;gBACpB,WAAW,EAAE;oBACX,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,OAAO,EAAE,YAAY;oBACrB,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW;iBACjC;aACF,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC;SAC3C;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CACrB,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAC9C,OAAO,EACP,OAAO,CACR,CAAC;IACJ,CAAC;IAED,QAAQ;QACN,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,iBAAkB,SAAQ,MAAM;IAK3C,YACE,MAGC;QAED,KAAK,CAAC,MAAM,CAAC,CAAC;QAVhB;;;;;WAA2B;QAE3B;;;;;WAAkB;QAUhB,IAAI,CAAC,MAAM,GAAG,MAAM,EAAE,MAAM,IAAI,EAAE,CAAC;QACnC,IAAI,CAAC,iBAAiB;YACpB,MAAM,EAAE,iBAAiB,IAAI,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;QAE/D,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE;YAC3B,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;SAChD;IACH,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,OAAgC,EAChC,OAAqC;QAErC,IAAI,OAAO,CAAC,MAAM,EAAE;YAClB,OAAO,KAAK,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;SACpD;QAED,MAAM,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACpC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QAC1D,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAElC,qDAAqD;QACrD,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,2CAA2C,EAAE;YACzE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,MAAM,EAAE,kBAAkB;aAC3B;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,aAAa,EAAE,0BAA0B;gBACzC,IAAI,EAAE,EAAE;gBACR,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE;gBACzD,IAAI,EAAE,IAAI,CAAC,MAAM,IAAI,EAAE;gBACvB,gBAAgB,EAAE,QAAQ,CAAC,IAAI;gBAC/B,kBAAkB,EAAE,IAAI,CAAC,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC;gBACvD,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC;gBACnD,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,mBAAmB;aACzC,CAAC;SACH,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}

package/dist/memory/index.d.ts

@@ -1,4 +1,4 @@
 export { BufferMemory } from "./buffer_memory.js";
 export { BaseMemory } from "./base.js";
 export { BufferWindowMemory } from "./buffer_window_memory.js";
-export { BaseChatMemory } from "./chat_memory.js";
+export { BaseChatMemory, ChatMessageHistory } from "./chat_memory.js";

package/dist/memory/index.js

@@ -1,5 +1,5 @@
 export { BufferMemory } from "./buffer_memory.js";
 export { BaseMemory } from "./base.js";
 export { BufferWindowMemory } from "./buffer_window_memory.js";
-export { BaseChatMemory } from "./chat_memory.js";
+export { BaseChatMemory, ChatMessageHistory } from "./chat_memory.js";
 //# sourceMappingURL=index.js.map

package/dist/memory/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/memory/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACvC,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/memory/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACvC,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC"}

package/dist/output_parsers/structured.d.ts

@@ -1,9 +1,9 @@
 import { z } from "zod";
 import { BaseOutputParser } from "../schema/index.js";
-export declare class StructuredOutputParser<T extends z.AnyZodObject> extends BaseOutputParser {
+export declare class StructuredOutputParser<T extends z.ZodTypeAny> extends BaseOutputParser {
     schema: T;
     constructor(schema: T);
-    static fromZodSchema<T extends z.AnyZodObject>(schema: T): StructuredOutputParser<T>;
+    static fromZodSchema<T extends z.ZodTypeAny>(schema: T): StructuredOutputParser<T>;
     static fromNamesAndDescriptions<S extends {
         [key: string]: string;
     }>(schemas: S): StructuredOutputParser<z.ZodObject<{

package/dist/output_parsers/structured.js

@@ -1,30 +1,40 @@
-/* eslint-disable no-else-return */
 import { z } from "zod";
 import { BaseOutputParser, OutputParserException } from "../schema/index.js";
 function printSchema(schema, depth = 0) {
+    if (schema instanceof z.ZodString &&
+        schema._def.checks.some((check) => check.kind === "datetime")) {
+        return "datetime";
+    }
     if (schema instanceof z.ZodString) {
         return "string";
     }
-    else if (schema instanceof z.ZodArray) {
+    if (schema instanceof z.ZodNumber) {
+        return "number";
+    }
+    if (schema instanceof z.ZodBoolean) {
+        return "boolean";
+    }
+    if (schema instanceof z.ZodDate) {
+        return "date";
+    }
+    if (schema instanceof z.ZodOptional) {
+        return `${printSchema(schema._def.innerType, depth)} // Optional`;
+    }
+    if (schema instanceof z.ZodArray) {
         return `${printSchema(schema._def.type, depth)}[]`;
     }
-    else if (schema instanceof z.ZodObject) {
+    if (schema instanceof z.ZodObject) {
         const indent = "\t".repeat(depth);
         const indentIn = "\t".repeat(depth + 1);
         return `{${schema._def.description ? ` // ${schema._def.description}` : ""}
 ${Object.entries(schema.shape)
-            .map(([key, value]) => 
-        // eslint-disable-next-line prefer-template
-        `${indentIn}"${key}": ${printSchema(value, depth + 1)}` +
-            (value._def.description
-                ? ` // ${value._def.description}`
-                : ""))
+            .map(([key, value]) => `${indentIn}"${key}": ${printSchema(value, depth + 1)}${value._def.description
+            ? ` // ${value._def.description}`
+            : ""}`)
             .join("\n")}
 ${indent}}`;
     }
-    else {
-        throw new Error(`Unsupported type: ${schema}`);
-    }
+    throw new Error(`Unsupported type: ${schema._def.innerType}`);
 }
 export class StructuredOutputParser extends BaseOutputParser {
     constructor(schema) {
@@ -48,7 +58,7 @@ export class StructuredOutputParser extends BaseOutputParser {
 
 \`\`\`json
 ${printSchema(this.schema)}
-\`\`\` 
+\`\`\`
 `;
     }
     async parse(text) {

package/dist/output_parsers/structured.js.map

@@ -1 +1 @@
-{"version":3,"file":"structured.js","sourceRoot":"","sources":["../../src/output_parsers/structured.ts"],"names":[],"mappings":"AAAA,mCAAmC;AACnC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAE7E,SAAS,WAAW,CAAC,MAAoB,EAAE,KAAK,GAAG,CAAC;IAClD,IAAI,MAAM,YAAY,CAAC,CAAC,SAAS,EAAE;QACjC,OAAO,QAAQ,CAAC;KACjB;SAAM,IAAI,MAAM,YAAY,CAAC,CAAC,QAAQ,EAAE;QACvC,OAAO,GAAG,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC;KACpD;SAAM,IAAI,MAAM,YAAY,CAAC,CAAC,SAAS,EAAE;QACxC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;QACxC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE;EAC5E,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;aAC3B,GAAG,CACF,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;QACf,2CAA2C;QAC3C,GAAG,QAAQ,IAAI,GAAG,MAAM,WAAW,CAAC,KAAqB,EAAE,KAAK,GAAG,CAAC,CAAC,EAAE;YACvE,CAAE,KAAsB,CAAC,IAAI,CAAC,WAAW;gBACvC,CAAC,CAAC,OAAQ,KAAsB,CAAC,IAAI,CAAC,WAAW,EAAE;gBACnD,CAAC,CAAC,EAAE,CAAC,CACV;aACA,IAAI,CAAC,IAAI,CAAC;EACX,MAAM,GAAG,CAAC;KACT;SAAM;QACL,MAAM,IAAI,KAAK,CAAC,qBAAqB,MAAM,EAAE,CAAC,CAAC;KAChD;AACH,CAAC;AAED,MAAM,OAAO,sBAEX,SAAQ,gBAAgB;IACxB,YAAmB,MAAS;QAC1B,KAAK,EAAE,CAAC;;;;;mBADS;;IAEnB,CAAC;IAED,MAAM,CAAC,aAAa,CAA2B,MAAS;QACtD,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC;IAC1B,CAAC;IAED,MAAM,CAAC,wBAAwB,CAC7B,OAAU;QAEV,MAAM,SAAS,GAAG,CAAC,CAAC,MAAM,CACxB,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,CACzB,CAAC,CAAC,IAAI,EAAE,WAAW,CAAC,EAAE,EAAE,CACtB,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAU,CACpD,CACF,CACF,CAAC;QAEF,OAAO,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,qBAAqB;QACnB,OAAO;;;EAGT,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC;;CAEzB,CAAC;IACA,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,IAAI;YACF,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YACpE,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;SAC5C;QAAC,OAAO,CAAC,EAAE;YACV,MAAM,IAAI,qBAAqB,CAC7B,0BAA0B,IAAI,YAAY,CAAC,EAAE,CAC9C,CAAC;SACH;IACH,CAAC;CACF"}
+{"version":3,"file":"structured.js","sourceRoot":"","sources":["../../src/output_parsers/structured.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAE7E,SAAS,WAAW,CAAC,MAAoB,EAAE,KAAK,GAAG,CAAC;IAClD,IACE,MAAM,YAAY,CAAC,CAAC,SAAS;QAC7B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,CAAC,EAC7D;QACA,OAAO,UAAU,CAAC;KACnB;IACD,IAAI,MAAM,YAAY,CAAC,CAAC,SAAS,EAAE;QACjC,OAAO,QAAQ,CAAC;KACjB;IACD,IAAI,MAAM,YAAY,CAAC,CAAC,SAAS,EAAE;QACjC,OAAO,QAAQ,CAAC;KACjB;IACD,IAAI,MAAM,YAAY,CAAC,CAAC,UAAU,EAAE;QAClC,OAAO,SAAS,CAAC;KAClB;IACD,IAAI,MAAM,YAAY,CAAC,CAAC,OAAO,EAAE;QAC/B,OAAO,MAAM,CAAC;KACf;IACD,IAAI,MAAM,YAAY,CAAC,CAAC,WAAW,EAAE;QACnC,OAAO,GAAG,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,cAAc,CAAC;KACnE;IACD,IAAI,MAAM,YAAY,CAAC,CAAC,QAAQ,EAAE;QAChC,OAAO,GAAG,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC;KACpD;IACD,IAAI,MAAM,YAAY,CAAC,CAAC,SAAS,EAAE;QACjC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;QACxC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE;EAC5E,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;aAC3B,GAAG,CACF,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CACf,GAAG,QAAQ,IAAI,GAAG,MAAM,WAAW,CAAC,KAAqB,EAAE,KAAK,GAAG,CAAC,CAAC,GAClE,KAAsB,CAAC,IAAI,CAAC,WAAW;YACtC,CAAC,CAAC,OAAQ,KAAsB,CAAC,IAAI,CAAC,WAAW,EAAE;YACnD,CAAC,CAAC,EACN,EAAE,CACL;aACA,IAAI,CAAC,IAAI,CAAC;EACX,MAAM,GAAG,CAAC;KACT;IACD,MAAM,IAAI,KAAK,CAAC,qBAAqB,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;AAChE,CAAC;AAED,MAAM,OAAO,sBAEX,SAAQ,gBAAgB;IACxB,YAAmB,MAAS;QAC1B,KAAK,EAAE,CAAC;;;;;mBADS;;IAEnB,CAAC;IAED,MAAM,CAAC,aAAa,CAAyB,MAAS;QACpD,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC;IAC1B,CAAC;IAED,MAAM,CAAC,wBAAwB,CAC7B,OAAU;QAEV,MAAM,SAAS,GAAG,CAAC,CAAC,MAAM,CACxB,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,CACzB,CAAC,CAAC,IAAI,EAAE,WAAW,CAAC,EAAE,EAAE,CACtB,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAU,CACpD,CACF,CACF,CAAC;QAEF,OAAO,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,qBAAqB;QACnB,OAAO;;;EAGT,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC;;CAEzB,CAAC;IACA,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,IAAI;YACF,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YACpE,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;SAC5C;QAAC,OAAO,CAAC,EAAE;YACV,MAAM,IAAI,qBAAqB,CAC7B,0BAA0B,IAAI,YAAY,CAAC,EAAE,CAC9C,CAAC;SACH;IACH,CAAC;CACF"}

package/dist/output_parsers/tests/structured.test.js

@@ -9,15 +9,15 @@ test("StructuredOutputParser.fromNamesAndDescriptions", async () => {
         url: "value",
     });
     expect(parser.getFormatInstructions()).toMatchInlineSnapshot(`
-    "The output should be a markdown code snippet formatted in the following schema:
+"The output should be a markdown code snippet formatted in the following schema:
 
-    \`\`\`json
-    {
-    	"url": string // A link to the resource
-    }
-    \`\`\` 
-    "
-  `);
+\`\`\`json
+{
+	"url": string // A link to the resource
+}
+\`\`\`
+"
+`);
 });
 test("StructuredOutputParser.fromZodSchema", async () => {
     const parser = StructuredOutputParser.fromZodSchema(z.object({ url: z.string().describe("A link to the resource") }));
@@ -25,46 +25,89 @@ test("StructuredOutputParser.fromZodSchema", async () => {
         url: "value",
     });
     expect(parser.getFormatInstructions()).toMatchInlineSnapshot(`
-    "The output should be a markdown code snippet formatted in the following schema:
+"The output should be a markdown code snippet formatted in the following schema:
 
-    \`\`\`json
-    {
-    	"url": string // A link to the resource
-    }
-    \`\`\` 
-    "
-  `);
+\`\`\`json
+{
+	"url": string // A link to the resource
+}
+\`\`\`
+"
+`);
+});
+test("StructuredOutputParser.fromZodSchema", async () => {
+    const parser = StructuredOutputParser.fromZodSchema(z.object({
+        answer: z.string().describe("answer to the user's question"),
+        sources: z
+            .array(z.string())
+            .describe("sources used to answer the question, should be websites."),
+    }));
+    expect(await parser.parse('```json\n{"answer": "value", "sources": ["this-source"]}```')).toEqual({
+        answer: "value",
+        sources: ["this-source"],
+    });
+    expect(parser.getFormatInstructions()).toMatchInlineSnapshot(`
+"The output should be a markdown code snippet formatted in the following schema:
+
+\`\`\`json
+{
+	"answer": string // answer to the user's question
+	"sources": string[] // sources used to answer the question, should be websites.
+}
+\`\`\`
+"
+`);
 });
 test("StructuredOutputParser.fromZodSchema", async () => {
     const parser = StructuredOutputParser.fromZodSchema(z
         .object({
         url: z.string().describe("A link to the resource"),
         title: z.string().describe("A title for the resource"),
+        year: z.number().describe("The year the resource was created"),
+        createdAt: z
+            .string()
+            .datetime()
+            .describe("The date and time the resource was created"),
+        createdAtDate: z.coerce
+            .date()
+            .describe("The date the resource was created")
+            .optional(),
         authors: z.array(z.object({
             name: z.string().describe("The name of the author"),
             email: z.string().describe("The email of the author"),
+            address: z
+                .string()
+                .optional()
+                .describe("The address of the author"),
         })),
     })
         .describe("Only One object"));
-    expect(await parser.parse('```json\n{"url": "value", "title": "value", "authors": [{"name": "value", "email": "value"}]}```')).toEqual({
+    expect(await parser.parse('```json\n{"url": "value", "title": "value", "year": 2011, "createdAt": "2023-03-29T16:07:09.600Z", "createdAtDate": "2023-03-29", "authors": [{"name": "value", "email": "value"}]}```')).toEqual({
         url: "value",
         title: "value",
+        year: 2011,
+        createdAt: "2023-03-29T16:07:09.600Z",
+        createdAtDate: new Date("2023-03-29T00:00:00.000Z"),
         authors: [{ name: "value", email: "value" }],
     });
     expect(parser.getFormatInstructions()).toMatchInlineSnapshot(`
-    "The output should be a markdown code snippet formatted in the following schema:
+"The output should be a markdown code snippet formatted in the following schema:
 
-    \`\`\`json
-    { // Only One object
-    	"url": string // A link to the resource
-    	"title": string // A title for the resource
-    	"authors": {
-    		"name": string // The name of the author
-    		"email": string // The email of the author
-    	}[]
-    }
-    \`\`\` 
-    "
-  `);
+\`\`\`json
+{ // Only One object
+	"url": string // A link to the resource
+	"title": string // A title for the resource
+	"year": number // The year the resource was created
+	"createdAt": datetime // The date and time the resource was created
+	"createdAtDate": date // Optional // The date the resource was created
+	"authors": {
+		"name": string // The name of the author
+		"email": string // The email of the author
+		"address": string // Optional // The address of the author
+	}[]
+}
+\`\`\`
+"
+`);
 });
 //# sourceMappingURL=structured.test.js.map

package/dist/output_parsers/tests/structured.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"structured.test.js","sourceRoot":"","sources":["../../../src/output_parsers/tests/structured.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAE1D,IAAI,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;IACjE,MAAM,MAAM,GAAG,sBAAsB,CAAC,wBAAwB,CAAC;QAC7D,GAAG,EAAE,wBAAwB;KAC9B,CAAC,CAAC;IAEH,MAAM,CAAC,MAAM,MAAM,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC,CAAC,OAAO,CAAC;QACjE,GAAG,EAAE,OAAO;KACb,CAAC,CAAC;IAEH,MAAM,CAAC,MAAM,CAAC,qBAAqB,EAAE,CAAC,CAAC,qBAAqB,CAAC;;;;;;;;;GAS5D,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;IACtD,MAAM,MAAM,GAAG,sBAAsB,CAAC,aAAa,CACjD,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,CAAC,CACjE,CAAC;IAEF,MAAM,CAAC,MAAM,MAAM,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC,CAAC,OAAO,CAAC;QACjE,GAAG,EAAE,OAAO;KACb,CAAC,CAAC;IAEH,MAAM,CAAC,MAAM,CAAC,qBAAqB,EAAE,CAAC,CAAC,qBAAqB,CAAC;;;;;;;;;GAS5D,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;IACtD,MAAM,MAAM,GAAG,sBAAsB,CAAC,aAAa,CACjD,CAAC;SACE,MAAM,CAAC;QACN,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;QAClD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,0BAA0B,CAAC;QACtD,OAAO,EAAE,CAAC,CAAC,KAAK,CACd,CAAC,CAAC,MAAM,CAAC;YACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;YACnD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;SACtD,CAAC,CACH;KACF,CAAC;SACD,QAAQ,CAAC,iBAAiB,CAAC,CAC/B,CAAC;IAEF,MAAM,CACJ,MAAM,MAAM,CAAC,KAAK,CAChB,kGAAkG,CACnG,CACF,CAAC,OAAO,CAAC;QACR,GAAG,EAAE,OAAO;QACZ,KAAK,EAAE,OAAO;QACd,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;KAC7C,CAAC,CAAC;IAEH,MAAM,CAAC,MAAM,CAAC,qBAAqB,EAAE,CAAC,CAAC,qBAAqB,CAAC;;;;;;;;;;;;;;GAc5D,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"structured.test.js","sourceRoot":"","sources":["../../../src/output_parsers/tests/structured.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAE1D,IAAI,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;IACjE,MAAM,MAAM,GAAG,sBAAsB,CAAC,wBAAwB,CAAC;QAC7D,GAAG,EAAE,wBAAwB;KAC9B,CAAC,CAAC;IAEH,MAAM,CAAC,MAAM,MAAM,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC,CAAC,OAAO,CAAC;QACjE,GAAG,EAAE,OAAO;KACb,CAAC,CAAC;IAEH,MAAM,CAAC,MAAM,CAAC,qBAAqB,EAAE,CAAC,CAAC,qBAAqB,CAAC;;;;;;;;;CAS9D,CAAC,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;IACtD,MAAM,MAAM,GAAG,sBAAsB,CAAC,aAAa,CACjD,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,CAAC,CACjE,CAAC;IAEF,MAAM,CAAC,MAAM,MAAM,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC,CAAC,OAAO,CAAC;QACjE,GAAG,EAAE,OAAO;KACb,CAAC,CAAC;IAEH,MAAM,CAAC,MAAM,CAAC,qBAAqB,EAAE,CAAC,CAAC,qBAAqB,CAAC;;;;;;;;;CAS9D,CAAC,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;IACtD,MAAM,MAAM,GAAG,sBAAsB,CAAC,aAAa,CACjD,CAAC,CAAC,MAAM,CAAC;QACP,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,+BAA+B,CAAC;QAC5D,OAAO,EAAE,CAAC;aACP,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;aACjB,QAAQ,CAAC,0DAA0D,CAAC;KACxE,CAAC,CACH,CAAC;IAEF,MAAM,CACJ,MAAM,MAAM,CAAC,KAAK,CAChB,6DAA6D,CAC9D,CACF,CAAC,OAAO,CAAC;QACR,MAAM,EAAE,OAAO;QACf,OAAO,EAAE,CAAC,aAAa,CAAC;KACzB,CAAC,CAAC;IAEH,MAAM,CAAC,MAAM,CAAC,qBAAqB,EAAE,CAAC,CAAC,qBAAqB,CAAC;;;;;;;;;;CAU9D,CAAC,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;IACtD,MAAM,MAAM,GAAG,sBAAsB,CAAC,aAAa,CACjD,CAAC;SACE,MAAM,CAAC;QACN,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;QAClD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,0BAA0B,CAAC;QACtD,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;QAC9D,SAAS,EAAE,CAAC;aACT,MAAM,EAAE;aACR,QAAQ,EAAE;aACV,QAAQ,CAAC,4CAA4C,CAAC;QACzD,aAAa,EAAE,CAAC,CAAC,MAAM;aACpB,IAAI,EAAE;aACN,QAAQ,CAAC,mCAAmC,CAAC;aAC7C,QAAQ,EAAE;QACb,OAAO,EAAE,CAAC,CAAC,KAAK,CACd,CAAC,CAAC,MAAM,CAAC;YACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;YACnD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;YACrD,OAAO,EAAE,CAAC;iBACP,MAAM,EAAE;iBACR,QAAQ,EAAE;iBACV,QAAQ,CAAC,2BAA2B,CAAC;SACzC,CAAC,CACH;KACF,CAAC;SACD,QAAQ,CAAC,iBAAiB,CAAC,CAC/B,CAAC;IAEF,MAAM,CACJ,MAAM,MAAM,CAAC,KAAK,CAChB,wLAAwL,CACzL,CACF,CAAC,OAAO,CAAC;QACR,GAAG,EAAE,OAAO;QACZ,KAAK,EAAE,OAAO;QACd,IAAI,EAAE,IAAI;QACV,SAAS,EAAE,0BAA0B;QACrC,aAAa,EAAE,IAAI,IAAI,CAAC,0BAA0B,CAAC;QACnD,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;KAC7C,CAAC,CAAC;IAEH,MAAM,CAAC,MAAM,CAAC,qBAAqB,EAAE,CAAC,CAAC,qBAAqB,CAAC;;;;;;;;;;;;;;;;;;CAkB9D,CAAC,CAAC;AACH,CAAC,CAAC,CAAC"}

package/dist/retrievers/index.d.ts

@@ -1 +1,2 @@
 export { ChatGPTPluginRetriever } from "./chatgpt-plugin.js";
+export { RemoteLangChainRetriever } from "./remote-retriever.js";

package/dist/retrievers/index.js

@@ -1,2 +1,3 @@
 export { ChatGPTPluginRetriever } from "./chatgpt-plugin.js";
+export { RemoteLangChainRetriever } from "./remote-retriever.js";
 //# sourceMappingURL=index.js.map

package/dist/retrievers/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/retrievers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/retrievers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAC7D,OAAO,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC"}

package/dist/retrievers/remote-retriever.d.ts

@@ -0,0 +1,23 @@
+import { Document } from "../document.js";
+import { BaseRetriever } from "../schema/index.js";
+import { AsyncCaller, AsyncCallerParams } from "../util/async_caller.js";
+interface RemoteRetrieverParams extends AsyncCallerParams {
+    url: string;
+    auth?: Record<string, string>;
+    inputKey?: string;
+    responseKey?: string;
+    pageContentKey?: string;
+    metadataKey?: string;
+}
+export declare class RemoteLangChainRetriever extends BaseRetriever {
+    url: string;
+    auth?: Record<string, string>;
+    inputKey: string;
+    responseKey: string;
+    asyncCaller: AsyncCaller;
+    pageContentKey: string;
+    metadataKey: string;
+    constructor({ url, auth, inputKey, responseKey, pageContentKey, metadataKey, ...rest }: RemoteRetrieverParams);
+    getRelevantDocuments(query: string): Promise<Document[]>;
+}
+export {};

package/dist/retrievers/remote-retriever.js

@@ -0,0 +1,81 @@
+import { Document } from "../document.js";
+import { BaseRetriever } from "../schema/index.js";
+import { AsyncCaller } from "../util/async_caller.js";
+export class RemoteLangChainRetriever extends BaseRetriever {
+    constructor({ url, auth = undefined, inputKey = "message", responseKey = "response", pageContentKey = "page_content", metadataKey = "metadata", ...rest }) {
+        super();
+        Object.defineProperty(this, "url", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "auth", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "inputKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "responseKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "asyncCaller", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "pageContentKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "metadataKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.url = url;
+        this.auth = auth;
+        this.inputKey = inputKey;
+        this.responseKey = responseKey;
+        this.pageContentKey = pageContentKey;
+        this.metadataKey = metadataKey;
+        this.asyncCaller = new AsyncCaller(rest);
+    }
+    async getRelevantDocuments(query) {
+        const res = await this.asyncCaller.call(fetch, this.url, {
+            method: "POST",
+            body: JSON.stringify({
+                [this.inputKey]: query,
+            }),
+            headers: {
+                Accept: "application/json",
+                "Content-Type": "application/json",
+                ...(this.auth ? this.auth : {}),
+            },
+        });
+        if (!res.ok) {
+            throw new Error(`Error calling RemoteLangChainRetriever: ${res.status}: ${res.statusText}`);
+        }
+        const body = await res.json();
+        return body[this.responseKey].map(
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        (r) => new Document({
+            pageContent: r[this.pageContentKey],
+            metadata: r[this.metadataKey],
+        }));
+    }
+}
+//# sourceMappingURL=remote-retriever.js.map

package/dist/retrievers/remote-retriever.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"remote-retriever.js","sourceRoot":"","sources":["../../src/retrievers/remote-retriever.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAqB,MAAM,yBAAyB,CAAC;AAazE,MAAM,OAAO,wBAAyB,SAAQ,aAAa;IAezD,YAAY,EACV,GAAG,EACH,IAAI,GAAG,SAAS,EAChB,QAAQ,GAAG,SAAS,EACpB,WAAW,GAAG,UAAU,EACxB,cAAc,GAAG,cAAc,EAC/B,WAAW,GAAG,UAAU,EACxB,GAAG,IAAI,EACe;QACtB,KAAK,EAAE,CAAC;QAvBV;;;;;WAAY;QAEZ;;;;;WAA8B;QAE9B;;;;;WAAiB;QAEjB;;;;;WAAoB;QAEpB;;;;;WAAyB;QAEzB;;;;;WAAuB;QAEvB;;;;;WAAoB;QAYlB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,KAAa;QACtC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE;YACvD,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,KAAK;aACvB,CAAC;YACF,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;gBAClC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;aAChC;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE;YACX,MAAM,IAAI,KAAK,CACb,2CAA2C,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,UAAU,EAAE,CAC3E,CAAC;SACH;QAED,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,GAAG;QAC/B,8DAA8D;QAC9D,CAAC,CAAM,EAAE,EAAE,CACT,IAAI,QAAQ,CAAC;YACX,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC;YACnC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC;SAC9B,CAAC,CACL,CAAC;IACJ,CAAC;CACF"}

package/dist/util/async_caller.js

@@ -41,6 +41,7 @@ export class AsyncCaller {
     call(callable, ...args) {
         return this.queue.add(() => backOff(() => callable(...args), {
             numOfAttempts: this.maxRetries,
+            jitter: "full",
             // If needed we can change some of the defaults here,
             // but they're quite sensible.
         }), { throwOnTimeout: true });

package/dist/util/async_caller.js.map

@@ -1 +1 @@
-{"version":3,"file":"async_caller.js","sourceRoot":"","sources":["../../src/util/async_caller.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,MAAM,MAAM,SAAS,CAAC;AAe7B;;;;;;;;;;;;GAYG;AACH,MAAM,OAAO,WAAW;IAOtB,YAAY,MAAyB;QANrC;;;;;WAA8D;QAE9D;;;;;WAAsD;QAEtD;;;;;WAAwB;QAGtB,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,QAAQ,CAAC;QACxD,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,CAAC,CAAC;QAEzC,IAAI,CAAC,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;IAChE,CAAC;IAED,8DAA8D;IAC9D,IAAI,CACF,QAAW,EACX,GAAG,IAAmB;QAEtB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CACnB,GAAG,EAAE,CACH,OAAO,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,EAAE;YAC/B,aAAa,EAAE,IAAI,CAAC,UAAU;YAC9B,qDAAqD;YACrD,8BAA8B;SAC/B,CAAC,EACJ,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,CAAC;IACJ,CAAC;CACF"}
+{"version":3,"file":"async_caller.js","sourceRoot":"","sources":["../../src/util/async_caller.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,MAAM,MAAM,SAAS,CAAC;AAe7B;;;;;;;;;;;;GAYG;AACH,MAAM,OAAO,WAAW;IAOtB,YAAY,MAAyB;QANrC;;;;;WAA8D;QAE9D;;;;;WAAsD;QAEtD;;;;;WAAwB;QAGtB,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,QAAQ,CAAC;QACxD,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,CAAC,CAAC;QAEzC,IAAI,CAAC,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;IAChE,CAAC;IAED,8DAA8D;IAC9D,IAAI,CACF,QAAW,EACX,GAAG,IAAmB;QAEtB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CACnB,GAAG,EAAE,CACH,OAAO,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,EAAE;YAC/B,aAAa,EAAE,IAAI,CAAC,UAAU;YAC9B,MAAM,EAAE,MAAM;YACd,qDAAqD;YACrD,8BAA8B;SAC/B,CAAC,EACJ,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,CAAC;IACJ,CAAC;CACF"}

package/dist/vectorstores/base.js

@@ -18,7 +18,7 @@ export class VectorStoreRetriever extends BaseRetriever {
         this.k = fields.k ?? this.k;
     }
     async getRelevantDocuments(query) {
-        const results = await this.vectorStore.similaritySearch(query, 4);
+        const results = await this.vectorStore.similaritySearch(query, this.k);
         return results;
     }
 }

package/dist/vectorstores/base.js.map

@@ -1 +1 @@
-{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/vectorstores/base.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,MAAM,OAAO,oBAAqB,SAAQ,aAAa;IAKrD,YAAY,MAAgD;QAC1D,KAAK,EAAE,CAAC;QALV;;;;;WAAyB;QAEzB;;;;mBAAI,CAAC;WAAC;QAIJ,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,KAAa;QACtC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAClE,OAAO,OAAO,CAAC;IACjB,CAAC;CACF;AAED,MAAM,OAAgB,WAAW;IAG/B,8DAA8D;IAC9D,YAAY,UAAsB,EAAE,SAA8B;QAHlE;;;;;WAAuB;QAIrB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAeD,KAAK,CAAC,gBAAgB,CACpB,KAAa,EACb,CAAC,GAAG,CAAC,EACL,SAA6B,SAAS;QAEtC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,+BAA+B,CACxD,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EACvC,CAAC,EACD,MAAM,CACP,CAAC;QAEF,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,yBAAyB,CAC7B,KAAa,EACb,CAAC,GAAG,CAAC,EACL,SAA6B,SAAS;QAEtC,OAAO,IAAI,CAAC,+BAA+B,CACzC,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EACvC,CAAC,EACD,MAAM,CACP,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,SAAS,CACd,MAAgB,EAChB,UAA6B,EAC7B,WAAuB;IACvB,8DAA8D;IAC9D,SAA8B;QAE9B,MAAM,IAAI,KAAK,CACb,qGAAqG,CACtG,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,aAAa,CAClB,KAAiB,EACjB,WAAuB;IACvB,8DAA8D;IAC9D,SAA8B;QAE9B,MAAM,IAAI,KAAK,CACb,qGAAqG,CACtG,CAAC;IACJ,CAAC;IAED,WAAW,CAAC,CAAU;QACpB,OAAO,IAAI,oBAAoB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;IAC5D,CAAC;CACF;AAED,MAAM,OAAgB,mBAAoB,SAAQ,WAAW;IAG3D,MAAM,CAAC,IAAI,CACT,UAAkB,EAClB,WAAuB;QAEvB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF"}
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/vectorstores/base.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,MAAM,OAAO,oBAAqB,SAAQ,aAAa;IAKrD,YAAY,MAAgD;QAC1D,KAAK,EAAE,CAAC;QALV;;;;;WAAyB;QAEzB;;;;mBAAI,CAAC;WAAC;QAIJ,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,KAAa;QACtC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;QACvE,OAAO,OAAO,CAAC;IACjB,CAAC;CACF;AAED,MAAM,OAAgB,WAAW;IAG/B,8DAA8D;IAC9D,YAAY,UAAsB,EAAE,SAA8B;QAHlE;;;;;WAAuB;QAIrB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAeD,KAAK,CAAC,gBAAgB,CACpB,KAAa,EACb,CAAC,GAAG,CAAC,EACL,SAA6B,SAAS;QAEtC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,+BAA+B,CACxD,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EACvC,CAAC,EACD,MAAM,CACP,CAAC;QAEF,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,yBAAyB,CAC7B,KAAa,EACb,CAAC,GAAG,CAAC,EACL,SAA6B,SAAS;QAEtC,OAAO,IAAI,CAAC,+BAA+B,CACzC,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EACvC,CAAC,EACD,MAAM,CACP,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,SAAS,CACd,MAAgB,EAChB,UAA6B,EAC7B,WAAuB;IACvB,8DAA8D;IAC9D,SAA8B;QAE9B,MAAM,IAAI,KAAK,CACb,qGAAqG,CACtG,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,aAAa,CAClB,KAAiB,EACjB,WAAuB;IACvB,8DAA8D;IAC9D,SAA8B;QAE9B,MAAM,IAAI,KAAK,CACb,qGAAqG,CACtG,CAAC;IACJ,CAAC;IAED,WAAW,CAAC,CAAU;QACpB,OAAO,IAAI,oBAAoB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;IAC5D,CAAC;CACF;AAED,MAAM,OAAgB,mBAAoB,SAAQ,WAAW;IAG3D,MAAM,CAAC,IAAI,CACT,UAAkB,EAClB,WAAuB;QAEvB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF"}

package/dist/vectorstores/tests/pinecone.test.js

@@ -12,8 +12,28 @@ test("PineconeStore with external ids", async () => {
     const embeddings = new FakeEmbeddings();
     const store = new PineconeStore(embeddings, { pineconeIndex: client });
     expect(store).toBeDefined();
-    await store.addDocuments([{ pageContent: "hello", metadata: { a: 1 } }], ["id1"]);
+    await store.addDocuments([
+        {
+            pageContent: "hello",
+            metadata: {
+                a: 1,
+                b: { nested: [1, { a: 4 }] },
+            },
+        },
+    ], ["id1"]);
     expect(client.upsert).toHaveBeenCalledTimes(1);
+    expect(client.upsert).toHaveBeenCalledWith({
+        upsertRequest: {
+            namespace: undefined,
+            vectors: [
+                {
+                    id: "id1",
+                    metadata: { a: 1, "b.nested.0": 1, "b.nested.1.a": 4, text: "hello" },
+                    values: [0.1, 0.2, 0.3, 0.4],
+                },
+            ],
+        },
+    });
     const results = await store.similaritySearch("hello", 1);
     expect(results).toHaveLength(0);
 });

package/dist/vectorstores/tests/pinecone.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"pinecone.test.js","sourceRoot":"","sources":["../../../src/vectorstores/tests/pinecone.test.ts"],"names":[],"mappings":"AAAA,uDAAuD;AACvD,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAE1D,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE/C,IAAI,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;IACjD,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAO,CAAC,iBAAiB,CAAC;YACtC,OAAO,EAAE,EAAE;SACZ,CAAC;KACH,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;IAExC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,UAAU,EAAE,EAAE,aAAa,EAAE,MAAa,EAAE,CAAC,CAAC;IAE9E,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CACtB,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,EAC9C,CAAC,KAAK,CAAC,CACR,CAAC;IAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAE/C,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAEzD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;IAClD,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAO,CAAC,iBAAiB,CAAC;YACtC,OAAO,EAAE,EAAE;SACZ,CAAC;KACH,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;IAExC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,UAAU,EAAE,EAAE,aAAa,EAAE,MAAa,EAAE,CAAC,CAAC;IAE9E,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAEzE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAE/C,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAEzD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC"}
+{"version":3,"file":"pinecone.test.js","sourceRoot":"","sources":["../../../src/vectorstores/tests/pinecone.test.ts"],"names":[],"mappings":"AAAA,uDAAuD;AACvD,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAE1D,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE/C,IAAI,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;IACjD,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAO,CAAC,iBAAiB,CAAC;YACtC,OAAO,EAAE,EAAE;SACZ,CAAC;KACH,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;IAExC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,UAAU,EAAE,EAAE,aAAa,EAAE,MAAa,EAAE,CAAC,CAAC;IAE9E,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CACtB;QACE;YACE,WAAW,EAAE,OAAO;YACpB,QAAQ,EAAE;gBACR,CAAC,EAAE,CAAC;gBACJ,CAAC,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;aAC7B;SACF;KACF,EACD,CAAC,KAAK,CAAC,CACR,CAAC;IAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAE/C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;QACzC,aAAa,EAAE;YACb,SAAS,EAAE,SAAS;YACpB,OAAO,EAAE;gBACP;oBACE,EAAE,EAAE,KAAK;oBACT,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE;oBACrE,MAAM,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;iBAC7B;aACF;SACF;KACF,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAEzD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;IAClD,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAO,CAAC,iBAAiB,CAAC;YACtC,OAAO,EAAE,EAAE;SACZ,CAAC;KACH,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;IAExC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,UAAU,EAAE,EAAE,aAAa,EAAE,MAAa,EAAE,CAAC,CAAC;IAE9E,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAEzE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAE/C,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAEzD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langchain",
-  "version": "0.0.44",
+  "version": "0.0.45",
   "description": "Typescript bindings for langchain",
   "type": "module",
   "engines": {