@cellaware/utils 7.0.0 → 7.0.2

package/dist/llm/chain-store.d.ts

@@ -5,7 +5,7 @@ import { ModelName, ModelOptions } from './model.js';
  * output key is allowed. Output key is defaulted to `answer`.
  */
 export declare class SingleActionChain extends BaseChain {
-    private displayName;
+    private _name;
     private llm;
     private _inputKeys;
     private _outputKey;

package/dist/llm/chain-store.js

@@ -13,7 +13,7 @@ const TRANSLATION_CHAIN_NAME = 'translation_chain';
 export class SingleActionChain extends BaseChain {
     constructor(fields, name, inputKeys) {
         super(fields);
-        Object.defineProperty(this, "displayName", {
+        Object.defineProperty(this, "_name", {
             enumerable: true,
             configurable: true,
             writable: true,
@@ -52,7 +52,7 @@ export class SingleActionChain extends BaseChain {
         if (inputKeys.length == 0) {
             throw new Error(`SINGLE_ACTION_CHAIN: Chain '${name}' needs at least one input key!`);
         }
-        this.displayName = name;
+        this._name = name;
         this.llm = fields.llm;
         this._inputKeys = inputKeys;
         this._outputKey = 'answer';
@@ -77,7 +77,7 @@ export class SingleActionChain extends BaseChain {
         return output;
     }
     _chainType() {
-        return this.displayName;
+        return this._name;
     }
     get inputKeys() {
         return this._inputKeys;
@@ -233,8 +233,10 @@ Your translation here:
         return chain.invoke(args, {
             callbacks: [
                 {
-                    handleLLMEnd: async (cbOutput) => {
-                        tokenUsages.push(ChainStore.getTokenUsage(chain._chainType(), chain.getModelName(), cbOutput.llmOutput?.tokenUsage));
+                    handleLLMEnd: (output) => {
+                        console.log(output);
+                        const usage = output?.llmOutput?.tokenUsage;
+                        tokenUsages.push(ChainStore.getTokenUsage(chain._chainType(), chain.getModelName(), usage));
                     },
                 }
             ]

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@cellaware/utils",
-    "version": "7.0.0",
+    "version": "7.0.2",
     "description": "Cellaware Utilities for Node.js",
     "author": "Cellaware Technologies",
     "type": "module",