@cellaware/utils 3.0.12 → 3.0.13

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

@@ -35,7 +35,14 @@ export declare class ChainStore {
     private initBuiltinChains;
     private static getTokenUsage;
     private getPrompt;
-    addChain(name: string, modelName: ModelName, temperature?: number, verbose?: boolean): void;
+    /**
+     * - `name`: Chain name
+     * - `modelName`: OpenAI model name
+     * - `temperature`: OpenAI model temperature
+     * - `verbose`: OpenAI verbose parameter
+     * - `templateName`: Template chain name to copy prompt from
+     */
+    addChain(name: string, modelName: ModelName, temperature?: number, verbose?: boolean, templateName?: string): void;
     addExistingChain(chain: SingleActionChain): void;
     callChain(name: string, args: any, tokenUsages: any[]): Promise<import("langchain/dist/schema/index.js").ChainValues>;
     translate(statement: string, language: string, tokenUsages?: any[]): Promise<string>;

package/dist/llm/chain-store.js

@@ -156,12 +156,12 @@ Your translation here:
             cost: getLLMTransactionCost(tokenUsage, modelName)
         };
     }
-    getPrompt(chainName) {
+    getPrompt(chainName, templateChainName) {
         let prompt = {
             content: '',
             variables: []
         };
-        const path = `${this.promptsPath}/${chainName}.md`;
+        const path = !!templateChainName ? `${this.promptsPath}/${templateChainName}.md` : `${this.promptsPath}/${chainName}.md`;
         if (!fs.existsSync(path)) {
             throw new Error(`CHAIN_STORE: Prompt file '${path}' not found`);
         }
@@ -182,8 +182,15 @@ Your translation here:
         prompt.variables = Array.from(distinctMatches);
         return prompt;
     }
-    addChain(name, modelName, temperature, verbose) {
-        const prompt = this.getPrompt(name);
+    /**
+     * - `name`: Chain name
+     * - `modelName`: OpenAI model name
+     * - `temperature`: OpenAI model temperature
+     * - `verbose`: OpenAI verbose parameter
+     * - `templateName`: Template chain name to copy prompt from
+     */
+    addChain(name, modelName, temperature, verbose, templateName) {
+        const prompt = this.getPrompt(name, templateName);
         const llm = new ChatOpenAI({
             modelName, temperature: temperature ?? 0, configuration: {
                 organization: process.env.OPENAI_ORGANIZATION,

package/package.json

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