llm-exe 0.0.7 → 1.0.1

package/dist/executor/_base.d.ts

@@ -23,37 +23,38 @@ export declare abstract class BaseExecutor<I extends PlainObject, O = any, H ext
      */
     name: string;
     /**
-     * @property executions - Prompt type (chat)
+     * @property executions -
      */
     executions: number;
+    traceId: string | null;
     /**
      * @property hooks - hooks to be ran during execution
      */
     hooks: any;
     readonly allowedHooks: any[];
     constructor(name: string, type: string, options?: CoreExecutorExecuteOptions<H>);
-    abstract handler(input: I): Promise<any>;
+    abstract handler(input: I, _options?: any): Promise<any>;
     /**
      *
      * Used to filter the input of the handler
      * @param _input
      * @returns original input formatted for handler
      */
-    getHandlerInput(_input: I, _metadata: ExecutorExecutionMetadata<I, any>): any;
+    getHandlerInput(_input: I, _metadata: ExecutorExecutionMetadata<I, any>, _options?: any): any;
     /**
      *
      * Used to filter the output of the handler
      * @param _input
      * @returns output O
      */
-    getHandlerOutput(out: any, _metadata: ExecutorExecutionMetadata<any, O>): O;
+    getHandlerOutput(out: any, _metadata: ExecutorExecutionMetadata<any, O>, _options?: any): O;
     /**
      *
      * execute - Runs the executor
      * @param _input
      * @returns handler output
      */
-    execute(_input: I): Promise<O>;
+    execute(_input: I, _options?: any): Promise<O>;
     metadata(): Record<string, any>;
     getMetadata(metadata?: Record<string, any>): ExecutorMetadata;
     runHook(hook: keyof H, _metadata: ExecutorExecutionMetadata): void;
@@ -62,4 +63,6 @@ export declare abstract class BaseExecutor<I extends PlainObject, O = any, H ext
     on(eventName: keyof H, fn: ListenerFunction): this;
     off(eventName: keyof H, fn: ListenerFunction): this;
     once(eventName: keyof H, fn: ListenerFunction): this;
+    withTraceId(traceId: string): this;
+    getTraceId(): string | null;
 }

package/dist/executor/_base.js

@@ -49,7 +49,7 @@ class BaseExecutor {
             value: void 0
         });
         /**
-         * @property executions - Prompt type (chat)
+         * @property executions -
          */
         Object.defineProperty(this, "executions", {
             enumerable: true,
@@ -57,6 +57,12 @@ class BaseExecutor {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "traceId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: null
+        });
         /**
          * @property hooks - hooks to be ran during execution
          */
@@ -92,7 +98,7 @@ class BaseExecutor {
      * @param _input
      * @returns original input formatted for handler
      */
-    getHandlerInput(_input, _metadata) {
+    getHandlerInput(_input, _metadata, _options) {
         return (0, utils_1.ensureInputIsObject)(_input);
     }
     /**
@@ -101,7 +107,7 @@ class BaseExecutor {
      * @param _input
      * @returns output O
      */
-    getHandlerOutput(out, _metadata) {
+    getHandlerOutput(out, _metadata, _options) {
         return out;
     }
     /**
@@ -110,18 +116,18 @@ class BaseExecutor {
      * @param _input
      * @returns handler output
      */
-    async execute(_input) {
+    async execute(_input, _options) {
         this.executions++;
         const _metadata = (0, _metadata_1.createMetadataState)({
             start: new Date().getTime(),
             input: _input,
         });
         try {
-            const input = this.getHandlerInput(_input, _metadata.asPlainObject());
+            const input = this.getHandlerInput(_input, _metadata.asPlainObject(), _options);
             _metadata.setItem({ handlerInput: input });
-            let result = await this.handler(input);
+            let result = await this.handler(input, _options);
             _metadata.setItem({ handlerOutput: result });
-            const output = this.getHandlerOutput(result, _metadata.asPlainObject());
+            const output = this.getHandlerOutput(result, _metadata.asPlainObject(), _options);
             _metadata.setItem({ output });
             this.runHook("onSuccess", _metadata.asPlainObject());
             return output;
@@ -141,6 +147,7 @@ class BaseExecutor {
     }
     getMetadata(metadata) {
         return Object.assign({}, this.metadata(), {
+            traceId: this.getTraceId(),
             id: this.id,
             type: this.type,
             name: this.name,
@@ -210,6 +217,13 @@ class BaseExecutor {
         this.hooks[eventName].push(onceWrapper);
         return this;
     }
+    withTraceId(traceId) {
+        this.traceId = traceId;
+        return this;
+    }
+    getTraceId() {
+        return this.traceId;
+    }
 }
 exports.BaseExecutor = BaseExecutor;
 //# sourceMappingURL=_base.js.map

package/dist/executor/_base.js.map

@@ -1 +1 @@
-{"version":3,"file":"_base.js","sourceRoot":"src/","sources":["executor/_base.ts"],"names":[],"mappings":";;;AAUA,mCAA0D;AAC1D,2CAAkD;AAClD,yCAA2E;AAE3E;;;;;GAKG;AACH,MAAsB,YAAY;IAoChC,YACE,IAAY,EACZ,IAAY,EACZ,OAAuC;QAlCzC;;WAEG;QACH;;;;;WAAY;QAEZ;;WAEG;QACH;;;;;WAAY;QAEZ;;WAEG;QACH;;;;;WAAiB;QAEjB;;WAEG;QACH;;;;;WAAY;QAEZ;;WAEG;QACH;;;;;WAAkB;QAElB;;WAEG;QACH;;;;;WAAkB;QAClB;;;;mBAA+B,CAAC,sBAAc,EAAE,mBAAW,EAAE,qBAAa,CAAC;WAAC;QAO1E,IAAI,CAAC,EAAE,GAAG,IAAA,YAAI,GAAE,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC;QACpC,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC;QACpB,IAAI,CAAC,KAAK,GAAG;YACX,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,EAAE;YACX,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,IAAI,OAAO,EAAE,KAAK,EAAE;YAClB,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;SAC9B;IACH,CAAC;IAID;;;;;OAKG;IACH,eAAe,CACb,MAAS,EACT,SAA4C;QAE5C,OAAO,IAAA,2BAAmB,EAAC,MAAM,CAAC,CAAC;IACrC,CAAC;IAED;;;;;OAKG;IACH,gBAAgB,CAAC,GAAQ,EAAE,SAA4C;QACrE,OAAO,GAAQ,CAAC;IAClB,CAAC;IACD;;;;;OAKG;IACH,KAAK,CAAC,OAAO,CAAC,MAAS;QACrB,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,SAAS,GAAG,IAAA,+BAAmB,EAAC;YACpC,KAAK,EAAE,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE;YAC3B,KAAK,EAAE,MAAM;SACd,CAAC,CAAC;QAEH,IAAI;YACF,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,SAAS,CAAC,aAAa,EAAE,CAAC,CAAC;YAEtE,SAAS,CAAC,OAAO,CAAC,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;YAE3C,IAAI,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAEvC,SAAS,CAAC,OAAO,CAAC,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC,CAAC;YAE7C,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,SAAS,CAAC,aAAa,EAAE,CAAC,CAAC;YACxE,SAAS,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;YAE9B,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,SAAS,CAAC,aAAa,EAAE,CAAC,CAAC;YACrD,OAAO,MAAM,CAAC;SACf;QAAC,OAAO,KAAU,EAAE;YACnB,SAAS,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAC1D,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,aAAa,EAAE,CAAC,CAAC;YACnD,MAAM,KAAK,CAAC;SACb;gBAAS;YACR,SAAS,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACjD,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,SAAS,CAAC,aAAa,EAAE,CAAC,CAAC;SACvD;IACH,CAAC;IAED,QAAQ;QACN,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,WAAW,CAAC,QAA8B;QACxC,OAAO,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE;YACxC,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,QAAQ;SACT,CAAC,CAAC;IACL,CAAC;IAED,OAAO,CAAC,IAAa,EAAE,SAAoC;QACzD,0BAA0B;QAC1B,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,GAAG,EAAE,EAAE,GAAG,IAAA,YAAI,EAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACnE,KAAK,MAAM,MAAM,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC,kBAAkB,EAAE;YAClD,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE;gBAChC,IAAI;oBACF,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;iBACvC;gBAAC,OAAO,KAAK,EAAE;oBACd,sCAAsC;iBACvC;aACF;SACF;IACH,CAAC;IACD,QAAQ,CAAC,QAAkC,EAAE;QAC3C,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAA2B,CAAC;QAC9D,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAC1C,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAC9B,EAAE;YACD,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC;YACjC,IAAI,SAAS,EAAE;gBACb,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;gBAClE,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE;oBACzB,IACE,IAAI;wBACJ,OAAO,IAAI,KAAK,UAAU;wBAC1B,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAmB,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,EAC9D;wBACA,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;qBAChC;iBACF;aACF;SACF;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,UAAU,CAAC,SAAkB,EAAE,EAAoB;QACjD,IAAI,OAAO,EAAE,KAAK,UAAU;YAAE,OAAO,IAAI,CAAC;QAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAC;QACtB,KAAK,IAAI,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE;YACpC,IAAI,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,EAAE;gBACjB,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBACjB,MAAM;aACP;SACF;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,EAAE,CAAC,SAAkB,EAAE,EAAoB;QACzC,OAAO,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,EAAE,EAA8B,CAAC,CAAC;IACxE,CAAC;IAED,GAAG,CAAC,SAAkB,EAAE,EAAoB;QAC1C,OAAO,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;IACxC,CAAC;IAED,IAAI,CAAC,SAAkB,EAAE,EAAoB;QAC3C,IAAI,OAAO,EAAE,KAAK,UAAU;YAAE,OAAO,IAAI,CAAC;QAC1C,MAAM,WAAW,GAAqB,CAAC,GAAG,IAAW,EAAE,EAAE;YACvD,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;YACZ,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;QACnC,CAAC,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAtMD,oCAsMC"}
+{"version":3,"file":"_base.js","sourceRoot":"src/","sources":["executor/_base.ts"],"names":[],"mappings":";;;AAUA,mCAA0D;AAC1D,2CAAkD;AAClD,yCAA2E;AAE3E;;;;;GAKG;AACH,MAAsB,YAAY;IAqChC,YACE,IAAY,EACZ,IAAY,EACZ,OAAuC;QAnCzC;;WAEG;QACH;;;;;WAAY;QAEZ;;WAEG;QACH;;;;;WAAY;QAEZ;;WAEG;QACH;;;;;WAAiB;QAEjB;;WAEG;QACH;;;;;WAAY;QAEZ;;WAEG;QACH;;;;;WAAkB;QAElB;;;;mBAAgC,IAAI;WAAC;QACrC;;WAEG;QACH;;;;;WAAkB;QAClB;;;;mBAA+B,CAAC,sBAAc,EAAE,mBAAW,EAAE,qBAAa,CAAC;WAAC;QAO1E,IAAI,CAAC,EAAE,GAAG,IAAA,YAAI,GAAE,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC;QACpC,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC;QACpB,IAAI,CAAC,KAAK,GAAG;YACX,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,EAAE;YACX,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,IAAI,OAAO,EAAE,KAAK,EAAE;YAClB,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;SAC9B;IACH,CAAC;IAID;;;;;OAKG;IACH,eAAe,CACb,MAAS,EACT,SAA4C,EAC5C,QAAc;QAEd,OAAO,IAAA,2BAAmB,EAAC,MAAM,CAAC,CAAC;IACrC,CAAC;IAED;;;;;OAKG;IACH,gBAAgB,CACd,GAAQ,EACR,SAA4C,EAC5C,QAAc;QAEd,OAAO,GAAQ,CAAC;IAClB,CAAC;IACD;;;;;OAKG;IACH,KAAK,CAAC,OAAO,CAAC,MAAS,EAAE,QAAc;QACrC,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,SAAS,GAAG,IAAA,+BAAmB,EAAC;YACpC,KAAK,EAAE,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE;YAC3B,KAAK,EAAE,MAAM;SACd,CAAC,CAAC;QAEH,IAAI;YACF,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAChC,MAAM,EACN,SAAS,CAAC,aAAa,EAAE,EACzB,QAAQ,CACT,CAAC;YAEF,SAAS,CAAC,OAAO,CAAC,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;YAE3C,IAAI,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;YAEjD,SAAS,CAAC,OAAO,CAAC,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC,CAAC;YAE7C,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAClC,MAAM,EACN,SAAS,CAAC,aAAa,EAAE,EACzB,QAAQ,CACT,CAAC;YACF,SAAS,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;YAE9B,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,SAAS,CAAC,aAAa,EAAE,CAAC,CAAC;YACrD,OAAO,MAAM,CAAC;SACf;QAAC,OAAO,KAAU,EAAE;YACnB,SAAS,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAC1D,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,aAAa,EAAE,CAAC,CAAC;YACnD,MAAM,KAAK,CAAC;SACb;gBAAS;YACR,SAAS,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACjD,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,SAAS,CAAC,aAAa,EAAE,CAAC,CAAC;SACvD;IACH,CAAC;IAED,QAAQ;QACN,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,WAAW,CAAC,QAA8B;QACxC,OAAO,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE;YACxC,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE;YAC1B,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,QAAQ;SACT,CAAC,CAAC;IACL,CAAC;IAED,OAAO,CAAC,IAAa,EAAE,SAAoC;QACzD,0BAA0B;QAC1B,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,GAAG,EAAE,EAAE,GAAG,IAAA,YAAI,EAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACnE,KAAK,MAAM,MAAM,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC,kBAAkB,EAAE;YAClD,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE;gBAChC,IAAI;oBACF,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;iBACvC;gBAAC,OAAO,KAAK,EAAE;oBACd,sCAAsC;iBACvC;aACF;SACF;IACH,CAAC;IACD,QAAQ,CAAC,QAAkC,EAAE;QAC3C,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAA2B,CAAC;QAC9D,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAC1C,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAC9B,EAAE;YACD,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC;YACjC,IAAI,SAAS,EAAE;gBACb,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;gBAClE,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE;oBACzB,IACE,IAAI;wBACJ,OAAO,IAAI,KAAK,UAAU;wBAC1B,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAmB,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,EAC9D;wBACA,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;qBAChC;iBACF;aACF;SACF;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,UAAU,CAAC,SAAkB,EAAE,EAAoB;QACjD,IAAI,OAAO,EAAE,KAAK,UAAU;YAAE,OAAO,IAAI,CAAC;QAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAC;QACtB,KAAK,IAAI,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE;YACpC,IAAI,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,EAAE;gBACjB,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBACjB,MAAM;aACP;SACF;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,EAAE,CAAC,SAAkB,EAAE,EAAoB;QACzC,OAAO,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,EAAE,EAA8B,CAAC,CAAC;IACxE,CAAC;IAED,GAAG,CAAC,SAAkB,EAAE,EAAoB;QAC1C,OAAO,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;IACxC,CAAC;IAED,IAAI,CAAC,SAAkB,EAAE,EAAoB;QAC3C,IAAI,OAAO,EAAE,KAAK,UAAU;YAAE,OAAO,IAAI,CAAC;QAC1C,MAAM,WAAW,GAAqB,CAAC,GAAG,IAAW,EAAE,EAAE;YACvD,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;YACZ,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;QACnC,CAAC,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC;IACd,CAAC;IACD,WAAW,CAAC,OAAgB;QAC1B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAA;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAA;IACrB,CAAC;CACF;AA5ND,oCA4NC"}

package/dist/executor/_functions.js.map

@@ -1 +1 @@
-{"version":3,"file":"_functions.js","sourceRoot":"src/","sources":["executor/_functions.ts"],"names":[],"mappings":";;;AAUA,iCAAsC;AACtC,+BAAoC;AAEpC;;;;;;GAMG;AACH,SAAgB,kBAAkB,CAChC,OAAqC,EACrC,OAAoC;IAEpC,OAAO,IAAI,mBAAY,CAAO,EAAE,OAAO,EAAE,EAAE,OAAO,CAAC,CAAC;AACtD,CAAC;AALD,gDAKC;AAED;;;;;;;;GAQG;AACH,SAAgB,iBAAiB,CAM/B,gBAAoE,EACpE,OAAsD;IAEtD,OAAO,IAAI,iBAAW,CAA6B,gBAAgB,EAAE,OAAO,CAAC,CAAC;AAChF,CAAC;AAVD,8CAUC"}
+{"version":3,"file":"_functions.js","sourceRoot":"src/","sources":["executor/_functions.ts"],"names":[],"mappings":";;;AAUA,iCAAsC;AACtC,+BAAoC;AAGpC;;;;;;GAMG;AACH,SAAgB,kBAAkB,CAChC,OAAqC,EACrC,OAAoC;IAEpC,OAAO,IAAI,mBAAY,CAAO,EAAE,OAAO,EAAE,EAAE,OAAO,CAAC,CAAC;AACtD,CAAC;AALD,gDAKC;AAED;;;;;;;;GAQG;AACH,SAAgB,iBAAiB,CAM/B,gBAAoE,EACpE,OAAsD;IAEtD,OAAO,IAAI,iBAAW,CAA6B,gBAAgB,EAAE,OAAO,CAAC,CAAC;AAChF,CAAC;AAVD,8CAUC"}

package/dist/executor/index.d.ts

@@ -1,4 +1,5 @@
 export { BaseExecutor } from "./_base";
 export { CoreExecutor } from "./core";
 export { LlmExecutor } from "./llm";
+export { LlmExecutorOpenAiFunctions } from "./llm-openai-function";
 export { createCoreExecutor, createLlmExecutor } from "./_functions";

package/dist/executor/index.js

@@ -1,12 +1,14 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.createLlmExecutor = exports.createCoreExecutor = exports.LlmExecutor = exports.CoreExecutor = exports.BaseExecutor = void 0;
+exports.createLlmExecutor = exports.createCoreExecutor = exports.LlmExecutorOpenAiFunctions = exports.LlmExecutor = exports.CoreExecutor = exports.BaseExecutor = void 0;
 var _base_1 = require("./_base");
 Object.defineProperty(exports, "BaseExecutor", { enumerable: true, get: function () { return _base_1.BaseExecutor; } });
 var core_1 = require("./core");
 Object.defineProperty(exports, "CoreExecutor", { enumerable: true, get: function () { return core_1.CoreExecutor; } });
 var llm_1 = require("./llm");
 Object.defineProperty(exports, "LlmExecutor", { enumerable: true, get: function () { return llm_1.LlmExecutor; } });
+var llm_openai_function_1 = require("./llm-openai-function");
+Object.defineProperty(exports, "LlmExecutorOpenAiFunctions", { enumerable: true, get: function () { return llm_openai_function_1.LlmExecutorOpenAiFunctions; } });
 var _functions_1 = require("./_functions");
 Object.defineProperty(exports, "createCoreExecutor", { enumerable: true, get: function () { return _functions_1.createCoreExecutor; } });
 Object.defineProperty(exports, "createLlmExecutor", { enumerable: true, get: function () { return _functions_1.createLlmExecutor; } });

package/dist/executor/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"src/","sources":["executor/index.ts"],"names":[],"mappings":";;;AAAA,iCAAuC;AAA9B,qGAAA,YAAY,OAAA;AACrB,+BAAsC;AAA7B,oGAAA,YAAY,OAAA;AACrB,6BAAoC;AAA3B,kGAAA,WAAW,OAAA;AACpB,2CAAqE;AAA5D,gHAAA,kBAAkB,OAAA;AAAE,+GAAA,iBAAiB,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["executor/index.ts"],"names":[],"mappings":";;;AAAA,iCAAuC;AAA9B,qGAAA,YAAY,OAAA;AACrB,+BAAsC;AAA7B,oGAAA,YAAY,OAAA;AACrB,6BAAoC;AAA3B,kGAAA,WAAW,OAAA;AACpB,6DAAmE;AAA1D,iIAAA,0BAA0B,OAAA;AACnC,2CAAqE;AAA5D,gHAAA,kBAAkB,OAAA;AAAE,+GAAA,iBAAiB,OAAA"}

package/dist/executor/llm-openai-function.d.ts

@@ -0,0 +1,13 @@
+import { PromptInput, ParserOutput, CoreExecutorExecuteOptions, ExecutorWithLlmOptions, LlmExecutorHooks, OpenAiLlmExecutorOptions, OpenAiFunctionCall } from "../types";
+import { OpenAI } from "../llm";
+import { BasePrompt } from "../prompt";
+import { BaseState } from "../state";
+import { BaseParser } from "../parser";
+import { LlmExecutor } from "./llm";
+/**
+ * Core Executor With LLM
+ */
+export declare class LlmExecutorOpenAiFunctions<Llm extends OpenAI, Prompt extends BasePrompt<Record<string, any>>, Parser extends BaseParser, State extends BaseState> extends LlmExecutor<Llm, Prompt, Parser, State> {
+    constructor(llmConfiguration: ExecutorWithLlmOptions<Llm, Prompt, Parser, State>, options?: CoreExecutorExecuteOptions<LlmExecutorHooks>);
+    execute<T extends OpenAiFunctionCall>(_input: PromptInput<Prompt>, _options: OpenAiLlmExecutorOptions<T>): Promise<ParserOutput<Parser>>;
+}

package/dist/executor/llm-openai-function.js

@@ -0,0 +1,23 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LlmExecutorOpenAiFunctions = void 0;
+const OpenAiFunctionParser_1 = require("../parser/parsers/OpenAiFunctionParser");
+const parser_1 = require("../parser");
+const llm_1 = require("./llm");
+/**
+ * Core Executor With LLM
+ */
+class LlmExecutorOpenAiFunctions extends llm_1.LlmExecutor {
+    constructor(llmConfiguration, options) {
+        super(Object.assign({}, llmConfiguration, {
+            parser: new OpenAiFunctionParser_1.OpenAiFunctionParser({
+                parser: llmConfiguration.parser || new parser_1.StringParser(),
+            }),
+        }), options);
+    }
+    async execute(_input, _options) {
+        return super.execute(_input, _options);
+    }
+}
+exports.LlmExecutorOpenAiFunctions = LlmExecutorOpenAiFunctions;
+//# sourceMappingURL=llm-openai-function.js.map

package/dist/executor/llm-openai-function.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm-openai-function.js","sourceRoot":"src/","sources":["executor/llm-openai-function.ts"],"names":[],"mappings":";;;AAYA,gFAA6E;AAC7E,qCAAoD;AACpD,+BAAoC;AAEpC;;GAEG;AACH,MAAa,0BAKX,SAAQ,iBAAuC;IAC/C,YACE,gBAAoE,EACpE,OAAsD;QAEtD,KAAK,CACH,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,gBAAgB,EAAE;YAClC,MAAM,EAAE,IAAI,2CAAoB,CAAC;gBAC/B,MAAM,EAAE,gBAAgB,CAAC,MAAM,IAAI,IAAI,qBAAY,EAAE;aACtD,CAAC;SACH,CAAC,EACF,OAAO,CACR,CAAC;IACJ,CAAC;IAKD,KAAK,CAAC,OAAO,CACX,MAA2B,EAC3B,QAAqC;QAErC,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;IACxC,CAAC;CACF;AA7BD,gEA6BC"}

package/dist/executor/llm.d.ts

@@ -1,4 +1,4 @@
-import { PromptInput, ParserOutput, CoreExecutorExecuteOptions, ExecutorWithLlmOptions, ExecutorExecutionMetadata, LlmExecutorHooks } from "../types";
+import { PromptInput, ParserOutput, CoreExecutorExecuteOptions, ExecutorWithLlmOptions, ExecutorExecutionMetadata, LlmExecutorHooks, LlmExecutorExecuteOptions } from "../types";
 import { BaseLlm } from "../llm";
 import { BaseParser, StringParser } from "../parser";
 import { BasePrompt } from "../prompt";
@@ -13,11 +13,13 @@ export declare class LlmExecutor<Llm extends BaseLlm<any>, Prompt extends BasePr
     promptFn: any;
     parser: StringParser | Parser;
     constructor(llmConfiguration: ExecutorWithLlmOptions<Llm, Prompt, Parser, State>, options?: CoreExecutorExecuteOptions<LlmExecutorHooks>);
-    handler(_input: PromptInput<Prompt>): Promise<any>;
+    execute(_input: PromptInput<Prompt>, _options?: LlmExecutorExecuteOptions): Promise<ParserOutput<Parser>>;
+    handler(_input: PromptInput<Prompt>, ..._args: any[]): Promise<any>;
     getHandlerInput(_input: PromptInput<Prompt>): any;
     getHandlerOutput(out: any, _metadata: ExecutorExecutionMetadata<PromptInput<Prompt>, ParserOutput<Parser>>): ParserOutput<Parser>;
     metadata(): {
         llm: {
+            traceId: string | null;
             promptType: import("../types").PromptType;
             timeout: number;
             jitter: "none" | "full";
@@ -26,4 +28,5 @@ export declare class LlmExecutor<Llm extends BaseLlm<any>, Prompt extends BasePr
             metrics: any;
         };
     };
+    getTraceId(): string | null;
 }

package/dist/executor/llm.js

@@ -44,8 +44,11 @@ class LlmExecutor extends _base_1.BaseExecutor {
             this.promptFn = null;
         }
     }
-    async handler(_input) {
-        const call = await this.llm.call(_input);
+    async execute(_input, _options) {
+        return super.execute(_input, _options);
+    }
+    async handler(_input, ..._args) {
+        const call = await this.llm.call(_input, ..._args);
         const result = call.getResultContent();
         return result;
     }
@@ -68,6 +71,12 @@ class LlmExecutor extends _base_1.BaseExecutor {
             llm: this.llm.getMetadata(),
         };
     }
+    getTraceId() {
+        if (this.traceId) {
+            return this.traceId;
+        }
+        return this.llm.getTraceId();
+    }
 }
 exports.LlmExecutor = LlmExecutor;
 //# sourceMappingURL=llm.js.map

package/dist/executor/llm.js.map

@@ -1 +1 @@
-{"version":3,"file":"llm.js","sourceRoot":"src/","sources":["executor/llm.ts"],"names":[],"mappings":";;;AASA,qCAAoD;AAGpD,mCAAuC;AAEvC;;GAEG;AACH,MAAa,WAKX,SAAQ,oBAIT;IAMC,YACE,gBAAoE,EACpE,OAAsD;QAEtD,KAAK,CACH,gBAAgB,CAAC,IAAI,IAAI,wBAAwB,EACjD,cAAc,EACd,OAAO,CACR,CAAC;QAbJ;;;;;WAAW;QACX;;;;;WAAc;QACd;;;;;WAAqB;QACrB;;;;;WAAc;QAYZ,IAAI,CAAC,GAAG,GAAG,gBAAgB,CAAC,GAAG,CAAC;QAChC,IAAI,CAAC,MAAM,GAAG,gBAAgB,EAAE,MAAM,IAAI,IAAI,qBAAY,EAAE,CAAC;QAE7D,IAAI,OAAO,gBAAgB,CAAC,MAAM,KAAK,UAAU,EAAE;YACjD,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;YACxB,IAAI,CAAC,QAAQ,GAAG,gBAAgB,CAAC,MAAM,CAAC;SACzC;aAAM;YACL,IAAI,CAAC,MAAM,GAAG,gBAAgB,CAAC,MAAM,CAAC;YACtC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;SACtB;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAA2B;QACvC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,eAAe,CAAC,MAA2B;QACzC,IAAI,IAAI,CAAC,MAAM,EAAE;YACf,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;SACnC;QAED,IAAI,IAAI,CAAC,QAAQ,EAAE;YACjB,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YACrC,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC9C,OAAO,eAAe,CAAC;SACxB;QACD,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;IACpC,CAAC;IAED,gBAAgB,CACd,GAAQ,EACR,SAGC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;IAC3C,CAAC;IAED,QAAQ;QACN,OAAO;YACL,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE;SAC5B,CAAC;IACJ,CAAC;CACF;AAvED,kCAuEC"}
+{"version":3,"file":"llm.js","sourceRoot":"src/","sources":["executor/llm.ts"],"names":[],"mappings":";;;AAUA,qCAAoD;AAGpD,mCAAuC;AAEvC;;GAEG;AACH,MAAa,WAKX,SAAQ,oBAIT;IAMC,YACE,gBAAoE,EACpE,OAAsD;QAEtD,KAAK,CACH,gBAAgB,CAAC,IAAI,IAAI,wBAAwB,EACjD,cAAc,EACd,OAAO,CACR,CAAC;QAbJ;;;;;WAAW;QACX;;;;;WAAc;QACd;;;;;WAAqB;QACrB;;;;;WAAc;QAYZ,IAAI,CAAC,GAAG,GAAG,gBAAgB,CAAC,GAAG,CAAC;QAChC,IAAI,CAAC,MAAM,GAAG,gBAAgB,EAAE,MAAM,IAAI,IAAI,qBAAY,EAAE,CAAC;QAE7D,IAAI,OAAO,gBAAgB,CAAC,MAAM,KAAK,UAAU,EAAE;YACjD,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;YACxB,IAAI,CAAC,QAAQ,GAAG,gBAAgB,CAAC,MAAM,CAAC;SACzC;aAAM;YACL,IAAI,CAAC,MAAM,GAAG,gBAAgB,CAAC,MAAM,CAAC;YACtC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;SACtB;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CACX,MAA2B,EAC3B,QAAoC;QAEpC,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAA2B,EAAE,GAAG,KAAY;QACxD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,eAAe,CAAC,MAA2B;QACzC,IAAI,IAAI,CAAC,MAAM,EAAE;YACf,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;SACnC;QAED,IAAI,IAAI,CAAC,QAAQ,EAAE;YACjB,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YACrC,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC9C,OAAO,eAAe,CAAC;SACxB;QACD,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;IACpC,CAAC;IAED,gBAAgB,CACd,GAAQ,EACR,SAGC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;IAC3C,CAAC;IAED,QAAQ;QACN,OAAO;YACL,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE;SAC5B,CAAC;IACJ,CAAC;IACD,UAAU;QACR,IAAI,IAAI,CAAC,OAAO,EAAE;YAChB,OAAO,IAAI,CAAC,OAAO,CAAC;SACrB;QACD,OAAO,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC;IAC/B,CAAC;CACF;AApFD,kCAoFC"}

package/dist/index.d.ts

@@ -8,6 +8,7 @@ import { PineconeVectorStore } from "./vector";
 import { BasePrompt, ChatPrompt, TextPrompt, createPrompt, createChatPrompt } from "./prompt";
 import { BaseParser, CustomParser, createParser, createCustomParser } from "./parser";
 import { DefaultState, BaseStateItem, DefaultStateItem, createState, createStateItem } from "./state";
+import { LlmExecutorOpenAiFunctions } from "./executor/llm-openai-function";
 export declare const llmExe: {
     /**
      * Utilities
@@ -40,6 +41,7 @@ export declare const llmExe: {
     BaseExecutor: typeof BaseExecutor;
     createCoreExecutor: typeof createCoreExecutor;
     createLlmExecutor: typeof createLlmExecutor;
+    LlmExecutorOpenAiFunctions: typeof LlmExecutorOpenAiFunctions;
     /**
      * Callable
      */

package/dist/index.js

@@ -24,6 +24,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.llmExe = void 0;
+// import { LlmExecutorExecuteOptions } from './interfaces/openai';
 const callable_1 = require("./plugins/callable");
 const _functions_1 = require("./executor/_functions");
 const _base_1 = require("./executor/_base");
@@ -34,6 +35,7 @@ const vector_1 = require("./vector");
 const prompt_1 = require("./prompt");
 const parser_1 = require("./parser");
 const state_1 = require("./state");
+const llm_openai_function_1 = require("./executor/llm-openai-function");
 exports.llmExe = {
     /**
      * Utilities
@@ -66,6 +68,7 @@ exports.llmExe = {
     BaseExecutor: _base_1.BaseExecutor,
     createCoreExecutor: _functions_1.createCoreExecutor,
     createLlmExecutor: _functions_1.createLlmExecutor,
+    LlmExecutorOpenAiFunctions: llm_openai_function_1.LlmExecutorOpenAiFunctions,
     /**
      * Callable
      */

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"src/","sources":["index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,iDAA0E;AAC1E,sDAA8E;AAC9E,4CAAgD;AAChD,+CAAiC;AACjC,+BAAyD;AACzD,+CAAqD;AACrD,qCAA+C;AAE/C,qCAMkB;AAElB,qCAKkB;AAElB,mCAMiB;AAEJ,QAAA,MAAM,GAAG;IACpB;;OAEG;IACH,KAAK;IACL;;OAEG;IACH,OAAO,EAAP,aAAO;IACP,MAAM,EAAN,YAAM;IACN,eAAe,EAAf,qBAAe;IACf;;OAEG;IACH,UAAU,EAAV,mBAAU;IACV,UAAU,EAAV,mBAAU;IACV,UAAU,EAAV,mBAAU;IACV,YAAY,EAAZ,qBAAY;IACZ,gBAAgB,EAAhB,yBAAgB;IAChB;;OAEG;IACH,UAAU,EAAV,mBAAU;IACV,YAAY,EAAZ,qBAAY;IACZ,YAAY,EAAZ,qBAAY;IACZ,kBAAkB,EAAlB,2BAAkB;IAClB;;OAEG;IACH,YAAY,EAAZ,oBAAY;IACZ,kBAAkB,EAAlB,+BAAkB;IAClB,iBAAiB,EAAjB,8BAAiB;IAEjB;;OAEG;IACH,YAAY,EAAZ,uBAAY;IACZ,sBAAsB,EAAtB,iCAAsB;IACtB;;OAEG;IACH,eAAe,EAAf,wBAAe;IACf,mBAAmB,EAAnB,4BAAmB;IACnB;;OAEG;IACH,YAAY,EAAZ,oBAAY;IACZ,aAAa,EAAb,qBAAa;IACb,gBAAgB,EAAhB,wBAAgB;IAChB,WAAW,EAAX,mBAAW;IACX,eAAe,EAAf,uBAAe;CAChB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,mEAAmE;AACnE,iDAA0E;AAC1E,sDAA8E;AAC9E,4CAAgD;AAChD,+CAAiC;AACjC,+BAAyD;AACzD,+CAAqD;AACrD,qCAA+C;AAE/C,qCAMkB;AAElB,qCAKkB;AAElB,mCAMiB;AAEjB,wEAA4E;AAE/D,QAAA,MAAM,GAAG;IACpB;;OAEG;IACH,KAAK;IACL;;OAEG;IACH,OAAO,EAAP,aAAO;IACP,MAAM,EAAN,YAAM;IACN,eAAe,EAAf,qBAAe;IACf;;OAEG;IACH,UAAU,EAAV,mBAAU;IACV,UAAU,EAAV,mBAAU;IACV,UAAU,EAAV,mBAAU;IACV,YAAY,EAAZ,qBAAY;IACZ,gBAAgB,EAAhB,yBAAgB;IAChB;;OAEG;IACH,UAAU,EAAV,mBAAU;IACV,YAAY,EAAZ,qBAAY;IACZ,YAAY,EAAZ,qBAAY;IACZ,kBAAkB,EAAlB,2BAAkB;IAClB;;OAEG;IACH,YAAY,EAAZ,oBAAY;IACZ,kBAAkB,EAAlB,+BAAkB;IAClB,iBAAiB,EAAjB,8BAAiB;IACjB,0BAA0B,EAA1B,gDAA0B;IAC1B;;OAEG;IACH,YAAY,EAAZ,uBAAY;IACZ,sBAAsB,EAAtB,iCAAsB;IACtB;;OAEG;IACH,eAAe,EAAf,wBAAe;IACf,mBAAmB,EAAnB,4BAAmB;IACnB;;OAEG;IACH,YAAY,EAAZ,oBAAY;IACZ,aAAa,EAAb,qBAAa;IACb,gBAAgB,EAAhB,wBAAgB;IAChB,WAAW,EAAX,mBAAW;IACX,eAAe,EAAf,uBAAe;CAChB,CAAC"}

package/dist/interfaces/chat.d.ts

@@ -1,16 +1,31 @@
-export type IChatMessageRole = "system" | "assistant" | "user";
+export type IChatMessageRole = "system" | "assistant" | "user" | "function" | "function_call";
+export type FinishReasons = "function_call" | "stop";
 export interface IChatMessageBase {
     role: IChatMessageRole;
-    content: string;
+    content: string | null;
 }
 export interface IChatUserMessage extends IChatMessageBase {
     role: Extract<IChatMessageRole, "user">;
     content: string;
     name?: string;
 }
+export interface IChatFunctionMessage extends IChatMessageBase {
+    role: Extract<IChatMessageRole, "function">;
+    content: string;
+    name: string;
+}
 export interface IChatAssistantMessage extends IChatMessageBase {
     role: Extract<IChatMessageRole, "assistant">;
     content: string;
+    function_call?: undefined;
+}
+export interface IChatAssistantFunctionCallMessage extends IChatMessageBase {
+    role: Extract<IChatMessageRole, "assistant">;
+    content: null;
+    function_call?: {
+        name: string;
+        arguments: string;
+    };
 }
 export interface IChatSystemMessage extends IChatMessageBase {
     role: Extract<IChatMessageRole, "system">;
@@ -21,7 +36,39 @@ export interface IChatMessagesPlaceholder {
     content: string;
 }
 export type IPromptMessages = (IChatSystemMessage | IChatMessagesPlaceholder)[];
-export type IPromptChatMessages = (IChatUserMessage | IChatAssistantMessage | IChatSystemMessage | IChatMessagesPlaceholder)[];
-export type IChatMessage = (IChatUserMessage | IChatAssistantMessage | IChatSystemMessage);
-export type IChatMessages = (IChatUserMessage | IChatAssistantMessage | IChatSystemMessage)[];
+export type IPromptChatMessages = (IChatUserMessage | IChatAssistantMessage | IChatAssistantFunctionCallMessage | IChatSystemMessage | IChatMessagesPlaceholder | IChatFunctionMessage)[];
+export type IChatMessage = IChatUserMessage | IChatAssistantMessage | IChatAssistantFunctionCallMessage | IChatSystemMessage | IChatFunctionMessage;
+export type IChatMessages = IChatMessage[];
 export type PromptTemplateHistoryToken = `{{>DialogueHistory key='${string}'}}`;
+interface OutputOpenAIChatChoiceBase {
+    message: {
+        role: Extract<IChatMessageRole, "assistant">;
+        content: string | null;
+        function_call: null | {
+            name: string;
+            arguments: string;
+        };
+    };
+    finish_reason: FinishReasons;
+}
+export interface OutputOpenAIChatChoiceFunction extends OutputOpenAIChatChoiceBase {
+    message: {
+        role: Extract<IChatMessageRole, "assistant">;
+        content: null;
+        function_call: {
+            name: string;
+            arguments: string;
+        };
+    };
+    finish_reason: Extract<FinishReasons, "function_call">;
+}
+export interface OutputOpenAIChatChoiceMessage extends OutputOpenAIChatChoiceBase {
+    message: {
+        role: Extract<IChatMessageRole, "assistant">;
+        content: string;
+        function_call: null;
+    };
+    finish_reason: Exclude<FinishReasons, "function_call">;
+}
+export type OutputOpenAIChatChoice = OutputOpenAIChatChoiceFunction | OutputOpenAIChatChoiceMessage;
+export {};

package/dist/interfaces/functions.d.ts

@@ -60,3 +60,8 @@ export type CoreExecutorHookInput<H = BaseExecutorHooks> = {
 export interface CoreExecutorExecuteOptions<T = BaseExecutorHooks> {
     hooks?: CoreExecutorHookInput<T>;
 }
+export interface CallableExecutorCore {
+    name: string;
+    description: string;
+    parameters?: Record<string, any>;
+}

package/dist/interfaces/openai.d.ts

@@ -1,17 +1,19 @@
+import { CallableExecutorCore } from "./index";
 import { PromptType } from "./prompt";
 export interface BaseLlmOptions {
+    traceId?: null | string;
     timeout?: number;
     maxDelay?: number;
     numOfAttempts?: number;
     jitter?: "none" | "full";
     promptType?: PromptType;
 }
-export type OpenAIChatModelName = "gpt-3.5-turbo" | "gpt-4";
+export type OpenAIChatModelName = "gpt-3.5-turbo" | "gpt-3.5-turbo-0613" | "gpt-3.5-turbo-16k" | "gpt-4-0613" | "gpt-4" | "gpt-4-0613" | "gpt-4-32k-0613";
 export type OpenAIConversationModelName = "davinci" | "text-curie-001" | "text-babbage-001" | "text-ada-001";
 export type OpenAIEmbeddingModelName = "text-embedding-ada-002";
 export type OpenAIModelName = OpenAIChatModelName | OpenAIConversationModelName | OpenAIEmbeddingModelName;
 export interface OpenAIOptions extends BaseLlmOptions {
-    openAIApiKey: string;
+    openAIApiKey?: string;
     modelName: OpenAIModelName;
     maxTokens?: number;
     temperature?: number;
@@ -24,6 +26,8 @@ export interface OpenAIOptions extends BaseLlmOptions {
     frequencyPenalty?: number | null;
     logitBias?: object | null;
     user?: string;
+    function_call?: OpenAiFunctionCall;
+    functions?: CallableExecutorCore[];
 }
 export interface EmbedOpenAIOptions extends OpenAIOptions {
     openAIApiKey: string;
@@ -31,3 +35,12 @@ export interface EmbedOpenAIOptions extends OpenAIOptions {
     batchSize?: number;
     stripNewLines?: boolean;
 }
+export type OpenAiFunctionCall = "auto" | "none" | {
+    name: string;
+};
+export interface LlmExecutorExecuteOptions {
+}
+export interface OpenAiLlmExecutorOptions<T extends OpenAiFunctionCall = "auto"> extends LlmExecutorExecuteOptions {
+    functions: CallableExecutorCore[];
+    function_call: T;
+}

package/dist/interfaces/parser.d.ts

@@ -1,9 +1,8 @@
 import { JSONSchema7 } from "json-schema-to-ts";
 export type CreateParserType = "json" | "string" | "boolean" | "number" | "stringExtract" | "listToArray" | "listToJson" | "listToKeyValue" | "replaceStringTemplate" | "markdownCodeBlocks" | "markdownCodeBlock";
 export interface BaseParserOptions {
-    schema?: JSONSchema7;
-    validateSchema?: boolean;
 }
 export interface BaseParserOptionsWithSchema<S extends JSONSchema7 | undefined = undefined> extends BaseParserOptions {
     schema?: S;
+    validateSchema?: boolean;
 }

package/dist/interfaces/utils.d.ts

@@ -13,3 +13,6 @@ export interface Serializable {
     serialize?(): Record<string, any>;
     deserialize?(): void;
 }
+export interface EmptyObject {
+    [key: string]: never;
+}

package/dist/llm/_base.d.ts

@@ -30,6 +30,7 @@ export declare abstract class BaseLlm<C = any> {
      * The jitter strategy to use between retries. Options are "none" or "full".
      */
     protected jitter: "none" | "full";
+    protected traceId: null | string;
     /**
      * An object to store metrics related to the LLM calls.
      */
@@ -39,6 +40,8 @@ export declare abstract class BaseLlm<C = any> {
      * @param {BaseLlmOptions} options - The options object for LLM configuration.
      */
     constructor(options: BaseLlmOptions);
+    shouldRetry(_error: any, _stepNumber: number): boolean;
+    handleError(_error: any): void;
     getPromptType(): PromptType;
     /**
      * getMetrics returns the metrics object for the LLM.
@@ -48,6 +51,7 @@ export declare abstract class BaseLlm<C = any> {
      * getMetadata - Used to return metrics and data about executions.
      */
     getMetadata(): {
+        traceId: string | null;
         promptType: PromptType;
         timeout: number;
         jitter: "none" | "full";
@@ -61,18 +65,20 @@ export declare abstract class BaseLlm<C = any> {
      * @param _input - The input to the LLM.
      * @return The output from the LLM.
      */
-    call(_input: any, attributes?: Record<string, any>): Promise<BaseLlmOutput>;
+    call(_input: any, ...args: any[]): Promise<BaseLlmOutput>;
     /**
      * _callWithRetry is a private method that handles retries and timeouts for the LLM call.
      * @param _input - The input to the LLM.
      * @return The result of the LLM call.
      * @throws An error if the maximum number of retries is reached.
      */
-    _callWithRetry(_input: any, attributes?: Record<string, any>): Promise<any>;
+    _callWithRetry(_input: any, ...args: any[]): Promise<any>;
     /**
-     * _call is an abstract method that must be implemented by subclasses to make the actual LLM call.
+     * _call is an method that must be implemented by subclasses to make the actual LLM call.
      * @param {any} _input - The input to the LLM.
      * @return The output from the LLM.
      */
-    _call(_input: any, _attributes?: Record<string, any>): Promise<BaseLlmOutput>;
+    _call(_input: any, ..._args: any[]): Promise<BaseLlmOutput>;
+    withTraceId(traceId: string): this;
+    getTraceId(): string | null;
 }

package/dist/llm/_base.js

@@ -69,6 +69,12 @@ class BaseLlm {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "traceId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         /**
          * An object to store metrics related to the LLM calls.
          */
@@ -84,6 +90,7 @@ class BaseLlm {
                 history: [],
             }
         });
+        this.traceId = options?.traceId || null;
         this.client = null;
         this.promptType = options?.promptType || "text";
         this.timeout = options.timeout || 30000;
@@ -91,6 +98,12 @@ class BaseLlm {
         this.maxDelay = options.maxDelay || 5000;
         this.numOfAttempts = options.numOfAttempts || 5;
     }
+    shouldRetry(_error, _stepNumber) {
+        return true;
+    }
+    handleError(_error) {
+        throw _error;
+    }
     getPromptType() {
         return this.promptType;
     }
@@ -105,6 +118,7 @@ class BaseLlm {
      */
     getMetadata() {
         return {
+            traceId: this.getTraceId(),
             promptType: this.promptType,
             timeout: this.timeout,
             jitter: this.jitter,
@@ -119,9 +133,8 @@ class BaseLlm {
      * @param _input - The input to the LLM.
      * @return The output from the LLM.
      */
-    async call(_input, attributes) {
-        const called = await this._callWithRetry(_input, attributes);
-        return called;
+    async call(_input, ...args) {
+        return this._callWithRetry(_input, ...args);
     }
     /**
      * _callWithRetry is a private method that handles retries and timeouts for the LLM call.
@@ -129,15 +142,15 @@ class BaseLlm {
      * @return The result of the LLM call.
      * @throws An error if the maximum number of retries is reached.
      */
-    async _callWithRetry(_input, attributes) {
+    async _callWithRetry(_input, ...args) {
         try {
             this.metrics.total_calls++;
-            const result = await (0, exponential_backoff_1.backOff)(() => (0, utils_1.asyncCallWithTimeout)(this._call(_input, attributes), this.timeout), {
+            const result = await (0, exponential_backoff_1.backOff)(() => (0, utils_1.asyncCallWithTimeout)(this._call(_input, ...args), this.timeout), {
                 startingDelay: 0,
                 maxDelay: this.maxDelay,
                 numOfAttempts: this.numOfAttempts,
                 jitter: this.jitter,
-                retry: () => {
+                retry: (_error, _stepNumber) => {
                     this.metrics.total_call_retry++;
                     console.log(`Handler timeout after ${this.timeout}. Retrying...`);
                     return true;
@@ -148,17 +161,24 @@ class BaseLlm {
         }
         catch (error) {
             this.metrics.total_call_error++;
-            throw error;
+            this.handleError(error);
         }
     }
     /**
-     * _call is an abstract method that must be implemented by subclasses to make the actual LLM call.
+     * _call is an method that must be implemented by subclasses to make the actual LLM call.
      * @param {any} _input - The input to the LLM.
      * @return The output from the LLM.
      */
-    async _call(_input, _attributes) {
+    async _call(_input, ..._args) {
         return new output_1.OutputDefault(undefined);
     }
+    withTraceId(traceId) {
+        this.traceId = traceId;
+        return this;
+    }
+    getTraceId() {
+        return this.traceId;
+    }
 }
 exports.BaseLlm = BaseLlm;
 //# sourceMappingURL=_base.js.map