@cellaware/utils 3.0.13 → 3.1.13

package/package.json

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

package/dist/azure/cosmos.d.ts

@@ -1,13 +0,0 @@
-export declare const COSMOS_CURRENT_YEAR_CLAUSE = "DateTimePart('yyyy', GetCurrentDateTime())";
-export declare const COSMOS_RECORD_YEAR_CLAUSE = "DateTimePart('yyyy', TimestampToDateTime(c._ts*1000))";
-export declare const COSMOS_CURRENT_MONTH_CLAUSE = "DateTimePart('mm', GetCurrentDateTime())";
-export declare const COSMOS_RECORD_MONTH_CLAUSE = "DateTimePart('mm', TimestampToDateTime(c._ts*1000))";
-export declare const COSMOS_CURRENT_DAY_CLAUSE = "DateTimePart('dd', GetCurrentDateTime())";
-export declare const COSMOS_RECORD_DAY_CLAUSE = "DateTimePart('dd', TimestampToDateTime(c._ts*1000))";
-export declare const COSMOS_CURRENT_DAY_CONDITION_CLAUSE = "DateTimeDiff('dd', TimestampToDateTime(c._ts*1000), GetCurrentDateTime()) = 0";
-export declare const COSMOS_CURRENT_MONTH_CONDITION_CLAUSE = "DateTimeDiff('mm', TimestampToDateTime(c._ts*1000), GetCurrentDateTime()) = 0";
-export declare const COSMOS_CURRENT_YEAR_CONDITION_CLAUSE = "DateTimeDiff('yyyy', TimestampToDateTime(c._ts*1000), GetCurrentDateTime()) = 0";
-export declare function cosmosSelect(databaseId: string, collectionId: string, partitionKey: string, query: string): Promise<any[]>;
-export declare function cosmosInsert(databaseId: string, collectionId: string, partitionKey: string, data: any): Promise<boolean>;
-export declare function cosmosDelete(databaseId: string, collectionId: string, partitionKey: string, query: string): Promise<boolean>;
-export declare function cosmosUpdate(databaseId: string, collectionId: string, partitionKey: string, query: string, data: any): Promise<boolean>;

package/dist/azure/cosmos.js

@@ -1,83 +0,0 @@
-import { CosmosClient } from '@azure/cosmos';
-export const COSMOS_CURRENT_YEAR_CLAUSE = `DateTimePart('yyyy', GetCurrentDateTime())`;
-export const COSMOS_RECORD_YEAR_CLAUSE = `DateTimePart('yyyy', TimestampToDateTime(c._ts*1000))`;
-export const COSMOS_CURRENT_MONTH_CLAUSE = `DateTimePart('mm', GetCurrentDateTime())`;
-export const COSMOS_RECORD_MONTH_CLAUSE = `DateTimePart('mm', TimestampToDateTime(c._ts*1000))`;
-export const COSMOS_CURRENT_DAY_CLAUSE = `DateTimePart('dd', GetCurrentDateTime())`;
-export const COSMOS_RECORD_DAY_CLAUSE = `DateTimePart('dd', TimestampToDateTime(c._ts*1000))`;
-export const COSMOS_CURRENT_DAY_CONDITION_CLAUSE = `DateTimeDiff('dd', TimestampToDateTime(c._ts*1000), GetCurrentDateTime()) = 0`;
-export const COSMOS_CURRENT_MONTH_CONDITION_CLAUSE = `DateTimeDiff('mm', TimestampToDateTime(c._ts*1000), GetCurrentDateTime()) = 0`;
-export const COSMOS_CURRENT_YEAR_CONDITION_CLAUSE = `DateTimeDiff('yyyy', TimestampToDateTime(c._ts*1000), GetCurrentDateTime()) = 0`;
-export async function cosmosSelect(databaseId, collectionId, partitionKey, query) {
-    try {
-        const cosmosClient = new CosmosClient({
-            endpoint: process.env.COSMOS_DB_ENDPOINT ?? '',
-            key: process.env.COSMOS_DB_KEY ?? ''
-        });
-        const { database } = await cosmosClient.databases.createIfNotExists({ id: databaseId });
-        const { container } = await database.containers.createIfNotExists({
-            id: collectionId,
-            partitionKey
-        });
-        const { resources } = await container.items.query({
-            query
-        }).fetchAll();
-        return resources;
-    }
-    catch (err) {
-        console.log(`COSMOS: Read error: ${err.message}`);
-        return [];
-    }
-}
-export async function cosmosInsert(databaseId, collectionId, partitionKey, data) {
-    try {
-        const cosmosClient = new CosmosClient({
-            endpoint: process.env.COSMOS_DB_ENDPOINT ?? '',
-            key: process.env.COSMOS_DB_KEY ?? ''
-        });
-        const { database } = await cosmosClient.databases.createIfNotExists({ id: databaseId });
-        const { container } = await database.containers.createIfNotExists({
-            id: collectionId,
-            partitionKey
-        });
-        await container.items.create(data);
-        return true;
-    }
-    catch (err) {
-        console.log(`COSMOS: Write error: ${err.message}`);
-        return false;
-    }
-}
-export async function cosmosDelete(databaseId, collectionId, partitionKey, query) {
-    try {
-        const cosmosClient = new CosmosClient({
-            endpoint: process.env.COSMOS_DB_ENDPOINT ?? '',
-            key: process.env.COSMOS_DB_KEY ?? ''
-        });
-        const { database } = await cosmosClient.databases.createIfNotExists({ id: databaseId });
-        const { container } = await database.containers.createIfNotExists({
-            id: collectionId,
-            partitionKey
-        });
-        const { resources } = await container.items.query({
-            query
-        }).fetchAll();
-        resources.forEach(doc => {
-            container.item(doc.id, doc.clientId).delete();
-        });
-        return true;
-    }
-    catch (err) {
-        console.log(`COSMOS: Delete error: ${err.message}`);
-        return false;
-    }
-}
-// Delete then insert -- partial item updates not possible.
-export async function cosmosUpdate(databaseId, collectionId, partitionKey, query, data) {
-    const delRes = await cosmosDelete(databaseId, collectionId, partitionKey, query);
-    if (!delRes) {
-        return delRes;
-    }
-    const insRes = await cosmosInsert(databaseId, collectionId, partitionKey, data);
-    return insRes;
-}

package/dist/azure/function.d.ts

@@ -1,2 +0,0 @@
-export declare function functionFetchJson(url: string, method: string, key: string, body?: any): Promise<any>;
-export declare function functionFetchText(url: string, method: string, key: string, text: string): Promise<any>;

package/dist/azure/function.js

@@ -1,56 +0,0 @@
-import { sleep } from "../util.js";
-const STATUS_INTERNAL_SERVER_ERROR = 500;
-const STATUS_SERVICE_UNAVAILABLE = 503;
-const RETRY_SLEEP_MS = 3000;
-const MAX_RETRIES = 3;
-async function functionFetch(url, req) {
-    let res;
-    let status = STATUS_SERVICE_UNAVAILABLE;
-    let retryNum = 0;
-    do {
-        if (retryNum > 0) {
-            await sleep(RETRY_SLEEP_MS);
-        }
-        try {
-            res = await fetch(url, req);
-            status = res.status;
-        }
-        catch (err) {
-            // if at max retries throw the error
-            if (retryNum === MAX_RETRIES) {
-                throw err;
-            }
-        }
-        retryNum++;
-    } while (retryNum < MAX_RETRIES &&
-        (status === STATUS_INTERNAL_SERVER_ERROR || status === STATUS_SERVICE_UNAVAILABLE));
-    return res;
-}
-export async function functionFetchJson(url, method, key, body) {
-    const headers = {
-        'Content-Type': 'application/json',
-        'Accept': 'application/json',
-        'x-functions-key': key
-    };
-    let req = {
-        method,
-        headers
-    };
-    if (body != null) {
-        req.body = JSON.stringify(body);
-    }
-    return functionFetch(url, req);
-}
-export async function functionFetchText(url, method, key, text) {
-    const headers = {
-        'Content-Type': 'application/text',
-        'Accept': 'application/json',
-        'x-functions-key': key
-    };
-    let req = {
-        method,
-        headers,
-        body: text
-    };
-    return functionFetch(url, req);
-}

package/dist/azure/storage.d.ts

@@ -1,4 +0,0 @@
-export declare function storageContainerCreate(containerClientId: string): Promise<void>;
-export declare function storageBlobUpload(containerClientId: string, blobId: string, data: any): Promise<void>;
-export declare function storageBlobDelete(containerClientId: string, blobId: string): Promise<void>;
-export declare function storageBlobDownload(containerClientId: string, blobId: string): Promise<any[]>;

package/dist/azure/storage.js

@@ -1,67 +0,0 @@
-import { BlobServiceClient } from '@azure/storage-blob';
-function streamToBuffer(stream) {
-    return new Promise((resolve, reject) => {
-        const chunks = [];
-        stream.on("data", (data) => {
-            chunks.push(Buffer.isBuffer(data) ? data : Buffer.from(data));
-        });
-        stream.on("end", () => {
-            resolve(Buffer.concat(chunks));
-        });
-        stream.on("error", reject);
-    });
-}
-export async function storageContainerCreate(containerClientId) {
-    try {
-        const blobServiceClient = BlobServiceClient.fromConnectionString(process.env.STORAGE_CONNECTION_STRING ?? '');
-        const containerClient = blobServiceClient.getContainerClient(containerClientId);
-        await containerClient.create();
-    }
-    catch (err) {
-        console.log(`STORAGE: CONTAINER create error: ${err.message}`);
-    }
-}
-export async function storageBlobUpload(containerClientId, blobId, data) {
-    try {
-        const blobServiceClient = BlobServiceClient.fromConnectionString(process.env.STORAGE_CONNECTION_STRING ?? '');
-        const containerClient = blobServiceClient.getContainerClient(containerClientId);
-        const blockBlobClient = containerClient.getBlockBlobClient(blobId);
-        await blockBlobClient.upload(data, data.length);
-    }
-    catch (err) {
-        console.log(`STORAGE: BLOB upload error: ${err.message}`);
-    }
-}
-export async function storageBlobDelete(containerClientId, blobId) {
-    try {
-        const blobServiceClient = BlobServiceClient.fromConnectionString(process.env.STORAGE_CONNECTION_STRING ?? '');
-        const containerClient = blobServiceClient.getContainerClient(containerClientId);
-        const blockBlobClient = containerClient.getBlockBlobClient(blobId);
-        await blockBlobClient.delete();
-    }
-    catch (err) {
-        console.log(`STORAGE: BLOB delete error: ${err.message}`);
-    }
-}
-export async function storageBlobDownload(containerClientId, blobId) {
-    if (!process.env.STORAGE_CONNECTION_STRING) {
-        console.log(`STORAGE: No connection string set`);
-        return [];
-    }
-    try {
-        const blobServiceClient = BlobServiceClient.fromConnectionString(process.env.STORAGE_CONNECTION_STRING ?? '');
-        const containerClient = blobServiceClient.getContainerClient(containerClientId);
-        const blockBlobClient = containerClient.getBlockBlobClient(blobId);
-        const downloadBlockBlobResponse = await blockBlobClient.download(0);
-        let data;
-        if (downloadBlockBlobResponse.readableStreamBody) {
-            let buf = await streamToBuffer(downloadBlockBlobResponse.readableStreamBody);
-            data = buf.toString();
-        }
-        return JSON.parse(data);
-    }
-    catch (err) {
-        console.log(`STORAGE: BLOB download error: ${err.message}`);
-    }
-    return [];
-}

package/dist/chatwms/azure/cosmos.d.ts

@@ -1,4 +0,0 @@
-export declare function chatwmsCosmosSelect(collectionId: string, query: string): Promise<any[]>;
-export declare function chatwmsCosmosInsert(collectionId: string, data: any): Promise<boolean>;
-export declare function chatwmsCosmosDelete(collectionId: string, query: string): Promise<boolean>;
-export declare function chatwmsCosmosUpdate(collectionId: string, query: string, data: any): Promise<boolean>;

package/dist/chatwms/azure/cosmos.js

@@ -1,15 +0,0 @@
-import { cosmosDelete, cosmosInsert, cosmosSelect, cosmosUpdate } from "../../azure/cosmos.js";
-const DATABASE_ID = 'chatwms';
-const PARTITION_KEY = '/clientId';
-export async function chatwmsCosmosSelect(collectionId, query) {
-    return cosmosSelect(DATABASE_ID, collectionId, PARTITION_KEY, query);
-}
-export async function chatwmsCosmosInsert(collectionId, data) {
-    return cosmosInsert(DATABASE_ID, collectionId, PARTITION_KEY, data);
-}
-export async function chatwmsCosmosDelete(collectionId, query) {
-    return cosmosDelete(DATABASE_ID, collectionId, PARTITION_KEY, query);
-}
-export async function chatwmsCosmosUpdate(collectionId, query, data) {
-    return cosmosUpdate(DATABASE_ID, collectionId, PARTITION_KEY, query, data);
-}

package/dist/chatwms/azure/function.d.ts

@@ -1,2 +0,0 @@
-export declare function chatwmsFunctionFetch(path: string, method: string, body?: any): Promise<any>;
-export declare function chatwmsFunctionQuery(path: string, method: string, query: string): Promise<any>;

package/dist/chatwms/azure/function.js

@@ -1,9 +0,0 @@
-import { functionFetchJson, functionFetchText } from "../../azure/function.js";
-export function chatwmsFunctionFetch(path, method, body) {
-    const url = `${process.env['CHATWMS_URL']}/${path}`;
-    return functionFetchJson(url, method, process.env['CHATWMS_KEY'] ?? '', body);
-}
-export function chatwmsFunctionQuery(path, method, query) {
-    const url = `${process.env['CHATWMS_URL']}/${path}`;
-    return functionFetchText(url, method, process.env['CHATWMS_KEY'] ?? '', query);
-}

package/dist/chatwms/azure/storage.d.ts

@@ -1,5 +0,0 @@
-export declare function chatwmsStorageGetBlobId(clientId: string, blobName: string): string;
-export declare function chatwmsStorageContainerCreate(): Promise<void>;
-export declare function chatwmsStorageBlobUpload(blobId: string, data: any): Promise<void>;
-export declare function chatwmsStorageBlobDelete(blobId: string): Promise<void>;
-export declare function chatwmsStorageBlobDownload(blobId: string): Promise<any[]>;

package/dist/chatwms/azure/storage.js

@@ -1,17 +0,0 @@
-import { storageBlobDelete, storageBlobDownload, storageBlobUpload, storageContainerCreate } from "../../azure/storage.js";
-const CONTAINER_CLIENT_ID = 'chatwms';
-export function chatwmsStorageGetBlobId(clientId, blobName) {
-    return clientId.replace('_', '-') + '-' + blobName;
-}
-export async function chatwmsStorageContainerCreate() {
-    return storageContainerCreate(CONTAINER_CLIENT_ID);
-}
-export async function chatwmsStorageBlobUpload(blobId, data) {
-    return storageBlobUpload(CONTAINER_CLIENT_ID, blobId, data);
-}
-export async function chatwmsStorageBlobDelete(blobId) {
-    return storageBlobDelete(CONTAINER_CLIENT_ID, blobId);
-}
-export async function chatwmsStorageBlobDownload(blobId) {
-    return storageBlobDownload(CONTAINER_CLIENT_ID, blobId);
-}

package/dist/chatwms/client.d.ts

@@ -1,2 +0,0 @@
-export declare const CHATWMS_GENERIC_CLIENT_ID = "chatwms";
-export declare function chatwmsGetClientId(customer: string, warehouse: string): string;

package/dist/chatwms/client.js

@@ -1,4 +0,0 @@
-export const CHATWMS_GENERIC_CLIENT_ID = 'chatwms';
-export function chatwmsGetClientId(customer, warehouse) {
-    return `${customer}_${warehouse}`;
-}

package/dist/chatwms/github/issue.d.ts

@@ -1 +0,0 @@
-export declare function chatwmsGithubCreateIssue(title: string, body: any, labels: string[]): Promise<boolean>;

package/dist/chatwms/github/issue.js

@@ -1,4 +0,0 @@
-import { githubCreateIssue } from "../../github/issue.js";
-export async function chatwmsGithubCreateIssue(title, body, labels) {
-    return githubCreateIssue(`https://api.github.com/repos/cellaware/chatwms-feedback/issues`, title, body, labels);
-}

package/dist/chatwms/product.d.ts

@@ -1 +0,0 @@
-export declare function chatwmsGetProductId(vendor: string, version: string): string;

package/dist/chatwms/product.js

@@ -1,3 +0,0 @@
-export function chatwmsGetProductId(vendor, version) {
-    return `${vendor}_${version}`;
-}

package/dist/chatwms/user.d.ts

@@ -1,6 +0,0 @@
-export interface ChatWMSUserData {
-    userId: string;
-    userDetails: string;
-    identityProvider: string;
-}
-export declare function chatwmsInitUserData(): ChatWMSUserData;

package/dist/chatwms/user.js

@@ -1,7 +0,0 @@
-export function chatwmsInitUserData() {
-    return {
-        userId: '',
-        userDetails: '',
-        identityProvider: ''
-    };
-}

package/dist/github/issue.d.ts

@@ -1 +0,0 @@
-export declare function githubCreateIssue(issueUrl: string, issueTitle: string, issueBody: any, issueLabels: string[]): Promise<boolean>;

package/dist/github/issue.js

@@ -1,23 +0,0 @@
-export async function githubCreateIssue(issueUrl, issueTitle, issueBody, issueLabels) {
-    const token = process.env.GITHUB_TOKEN ?? '';
-    if (token.length === 0) {
-        throw new Error('GITHUB: `GITHUB_TOKEN` environment variable is not set');
-    }
-    const headers = {
-        'Authorization': `token ${token}`,
-        'X-GitHub-Api-Version': '2022-11-28',
-        'Accept': 'application/vnd.github+json'
-    };
-    const body = {
-        title: issueTitle,
-        body: issueBody,
-        labels: issueLabels
-    };
-    const req = {
-        method: 'POST',
-        headers,
-        body: JSON.stringify(body)
-    };
-    let res = await fetch(issueUrl, req);
-    return res.ok;
-}

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

@@ -1,50 +0,0 @@
-import { BaseChain } from 'langchain/chains';
-import { ModelName } from './model.js';
-/**
- * `SingleActionChain` only supports single interaction with LLM, and only 1
- * output key is allowed. Output key is defaulted to `answer`.
- */
-export declare class SingleActionChain extends BaseChain {
-    private name;
-    private llm;
-    private _inputKeys;
-    private _outputKey;
-    private prompt;
-    private promptTemplate;
-    constructor(fields: any, name: string, inputKeys: string[]);
-    _call(values: any, runManager: any): Promise<any>;
-    _chainType(): string;
-    get inputKeys(): string[];
-    get outputKeys(): string[];
-    getModelName(): ModelName;
-    getPrompt(): string;
-}
-export declare function createSingleActionChain(name: string, modelName: ModelName, inputKeys: string[], prompt: string, temperature?: number, verbose?: boolean): SingleActionChain;
-/**
- * A `ChainStore` is a chain registry and calling interface. The `ChainStore`
- * keeps a map of all registered chains.
- *
- * The following chains are built-in:
- *  - `translation_chain`: `translate()`
- *
- */
-export declare class ChainStore {
-    private promptsPath;
-    private chains;
-    constructor(promptsPath: string, defaultModelName?: ModelName);
-    private initBuiltinChains;
-    private static getTokenUsage;
-    private getPrompt;
-    /**
-     * - `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>;
-    getPrompts(): string;
-}

package/dist/llm/chain-store.js

@@ -1,270 +0,0 @@
-import fs from 'fs';
-import { ChatOpenAI } from 'langchain/chat_models/openai';
-import { PromptTemplate } from 'langchain/prompts';
-import { LLMChain, BaseChain } from 'langchain/chains';
-import { getLLMTransactionCost } from './cost.js';
-const CHAIN_TIMEOUT_MS = 150_000;
-const TRANSLATION_CHAIN_NAME = 'translation_chain';
-/**
- * `SingleActionChain` only supports single interaction with LLM, and only 1
- * output key is allowed. Output key is defaulted to `answer`.
- */
-export class SingleActionChain extends BaseChain {
-    constructor(fields, name, inputKeys) {
-        super(fields);
-        Object.defineProperty(this, "name", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "llm", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "_inputKeys", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "_outputKey", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "prompt", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "promptTemplate", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        if (inputKeys.length == 0) {
-            throw new Error(`SINGLE_ACTION_CHAIN: Chain '${name}' needs at least one input key!`);
-        }
-        this.name = name;
-        this.llm = fields.llm;
-        this._inputKeys = inputKeys;
-        this._outputKey = 'answer';
-        this.prompt = fields.prompt;
-        this.promptTemplate = new PromptTemplate({
-            template: fields.prompt,
-            inputVariables: this.inputKeys
-        });
-    }
-    async _call(values, runManager) {
-        let output = {
-            [this._outputKey]: ''
-        };
-        const llmChain = new LLMChain({
-            prompt: this.promptTemplate,
-            llm: this.llm,
-            outputKey: this._outputKey,
-            memory: this.memory,
-        });
-        let llmInputs = {};
-        for (const inputKey of this._inputKeys) {
-            llmInputs[inputKey] = values[inputKey];
-        }
-        const llmAnswer = await llmChain.predict(llmInputs, runManager?.getChild());
-        output[this._outputKey] = llmAnswer;
-        return output;
-    }
-    _chainType() {
-        return this.name;
-    }
-    get inputKeys() {
-        return this._inputKeys;
-    }
-    get outputKeys() {
-        return [this._outputKey];
-    }
-    getModelName() {
-        return this.llm.modelName;
-    }
-    getPrompt() {
-        return this.prompt;
-    }
-}
-export function createSingleActionChain(name, modelName, inputKeys, prompt, temperature, verbose) {
-    const llm = new ChatOpenAI({
-        modelName, temperature: temperature ?? 0, configuration: {
-            organization: process.env.OPENAI_ORGANIZATION,
-            timeout: CHAIN_TIMEOUT_MS
-        }
-    });
-    return new SingleActionChain({
-        llm,
-        prompt,
-        verbose: verbose ?? false
-    }, name, inputKeys);
-}
-/**
- * A `ChainStore` is a chain registry and calling interface. The `ChainStore`
- * keeps a map of all registered chains.
- *
- * The following chains are built-in:
- *  - `translation_chain`: `translate()`
- *
- */
-export class ChainStore {
-    constructor(promptsPath, defaultModelName) {
-        Object.defineProperty(this, "promptsPath", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "chains", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.promptsPath = promptsPath;
-        this.chains = new Map();
-        this.initBuiltinChains(defaultModelName);
-    }
-    initBuiltinChains(defaultModelName) {
-        // Translation Chain:
-        {
-            this.addExistingChain(createSingleActionChain(TRANSLATION_CHAIN_NAME, defaultModelName ?? 'gpt-4o-mini', ['statement', 'language'], `You are a helpful AI translator who translates an English statement to {language}.
-
-Given an English statement, translate the English statement into {language}. 
-
-Here is the English statement that you need to translate to {language}:
-{statement}
-
-Your translation here:
-`));
-        }
-    }
-    static getTokenUsage(chainName, modelName, tokenUsage) {
-        return {
-            chain: chainName,
-            model: modelName,
-            tokenUsage,
-            cost: getLLMTransactionCost(tokenUsage, modelName)
-        };
-    }
-    getPrompt(chainName, templateChainName) {
-        let prompt = {
-            content: '',
-            variables: []
-        };
-        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`);
-        }
-        const content = fs.readFileSync(path, 'utf8');
-        if (content.length === 0) {
-            throw new Error(`CHAIN_STORE: Prompt file '${path}' is empty`);
-        }
-        let matches = content.matchAll(/{.*?}/gm);
-        let distinctMatches = new Set();
-        for (const match of Array.from(matches)) {
-            let matchStr = match[0];
-            // Exclude escaped curly braces.
-            if (!matchStr.startsWith('{{')) {
-                distinctMatches.add(matchStr.substring(1, matchStr.length - 1));
-            }
-        }
-        prompt.content = content;
-        prompt.variables = Array.from(distinctMatches);
-        return prompt;
-    }
-    /**
-     * - `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,
-                timeout: CHAIN_TIMEOUT_MS
-            }
-        });
-        const chain = new SingleActionChain({
-            llm,
-            prompt: prompt.content,
-            verbose: verbose ?? false
-        }, name, prompt.variables);
-        this.chains.set(name, chain);
-    }
-    addExistingChain(chain) {
-        this.chains.set(chain._chainType(), chain);
-    }
-    async callChain(name, args, tokenUsages) {
-        let chain = this.chains.get(name);
-        if (!chain) {
-            throw new Error(`CHAIN_STORE: Chain '${name}' does not exist`);
-        }
-        let presentInputKeySet = new Set();
-        for (const inputKey of Object.keys(args)) {
-            presentInputKeySet.add(inputKey);
-        }
-        let presentInputKeys = Array.from(presentInputKeySet);
-        // Make sure all input keys are present.
-        for (const inputKey of chain.inputKeys) {
-            if (!presentInputKeys.includes(inputKey)) {
-                throw new Error(`CHAIN_STORE: Input key '${inputKey}' not present for chain '${chain._chainType()}'`);
-            }
-        }
-        return await chain.call(args, [
-            {
-                handleLLMEnd: async (cbOutput) => {
-                    tokenUsages.push(ChainStore.getTokenUsage(chain._chainType(), chain.getModelName(), cbOutput.llmOutput?.tokenUsage));
-                },
-            }
-        ]);
-    }
-    async translate(statement, language, tokenUsages) {
-        // No need to translate if requested language is in default language.
-        const DEFAULT_LANGUAGE = 'English';
-        if (language.toLowerCase() === DEFAULT_LANGUAGE.toLowerCase()) {
-            return statement;
-        }
-        const translationAnswer = await this.callChain(TRANSLATION_CHAIN_NAME, { statement, language }, tokenUsages ?? []);
-        let translationAnswerStr = translationAnswer.answer;
-        /*
-            Need to check if AI indicator exists in the answer. It is possible that the
-            translation chain will include this in an answer. Pretty sure this happens
-            because the translation chain can see conversation history, and messages from
-            AI will be prefixed as such.
-        */
-        const AI_PREFIX = 'AI: ';
-        if (translationAnswerStr.includes(AI_PREFIX)) {
-            console.log(`ANALYST: Removing AI indicator from translation answer`);
-            translationAnswerStr = translationAnswerStr.substring(translationAnswerStr.indexOf(AI_PREFIX) + AI_PREFIX.length);
-        }
-        // Remove 'Your translation here:' if it is there.
-        if (translationAnswerStr.includes('Your translation here:')) {
-            translationAnswerStr = translationAnswerStr.substring(translationAnswerStr.indexOf('Your translation here:') + 'Your translation here:'.length);
-        }
-        return translationAnswerStr.trim();
-    }
-    getPrompts() {
-        let buf = '';
-        for (const chain of Array.from(this.chains.values())) {
-            buf += `================================================= =================================================
-                                            ${chain._chainType()}
-================================================= =================================================
-${chain.getPrompt()}
-`;
-        }
-        return buf;
-    }
-}

package/dist/llm/cost.d.ts

@@ -1,3 +0,0 @@
-import { ModelName } from "./model";
-export declare function getLLMCostPerToken(modelName: ModelName): any;
-export declare function getLLMTransactionCost(tokenUsage: any, modelName: ModelName): number;

package/dist/llm/cost.js

@@ -1,59 +0,0 @@
-// https://openai.com/pricing
-export function getLLMCostPerToken(modelName) {
-    let inputCost = 0.0;
-    let outputCost = 0.0;
-    switch (modelName) {
-        case 'o1-preview':
-            inputCost = 0.015;
-            outputCost = 0.060;
-            break;
-        case 'o1-mini':
-            inputCost = 0.003;
-            outputCost = 0.012;
-            break;
-        case 'gpt-4o':
-            inputCost = 0.005;
-            outputCost = 0.015;
-            break;
-        case 'gpt-4o-2024-08-06':
-            inputCost = 0.0025;
-            outputCost = 0.010;
-            break;
-        case 'gpt-4o-mini':
-            inputCost = 0.00015;
-            outputCost = 0.0006;
-            break;
-        case 'gpt-4-turbo':
-            inputCost = 0.01;
-            outputCost = 0.03;
-            break;
-        case 'gpt-4':
-            inputCost = 0.03;
-            outputCost = 0.06;
-            break;
-        case 'gpt-4-32k':
-            inputCost = 0.06;
-            outputCost = 0.12;
-            break;
-        case 'gpt-3.5-turbo':
-            inputCost = 0.0005;
-            outputCost = 0.0015;
-            break;
-        default:
-            inputCost = 0.0;
-            outputCost = 0.0;
-            break;
-    }
-    // OpenAI model costs are measured in thousands of tokens -- therefore we need to divide each figure by 1000 before returning.
-    inputCost /= 1000.0;
-    outputCost /= 1000.0;
-    return { inputCost, outputCost };
-}
-export function getLLMTransactionCost(tokenUsage, modelName) {
-    let cost = 0.0;
-    if (!!tokenUsage) {
-        const { inputCost, outputCost } = getLLMCostPerToken(modelName);
-        cost = (tokenUsage.promptTokens * inputCost) + (tokenUsage.completionTokens * outputCost);
-    }
-    return cost;
-}

package/dist/llm/model.d.ts

@@ -1 +0,0 @@
-export type ModelName = 'o1-preview' | 'o1-mini' | 'gpt-4o' | 'gpt-4o-2024-08-06' | 'gpt-4o-mini' | 'gpt-4-turbo' | 'gpt-4' | 'gpt-4-32k' | 'gpt-3.5-turbo';

package/dist/llm/model.js

@@ -1 +0,0 @@
-export {};

package/dist/util.d.ts

@@ -1,9 +0,0 @@
-export declare function sleep(ms: number): Promise<any>;
-export declare function removeMarkdownIndicators(input: string): string;
-export declare function removePrefixIndicators(input: string, prefixes: string[]): string;
-export declare function isLeapYear(): boolean;
-export declare function getDaysInMonth(): number;
-export declare function getCurrentDayInMonth(): number;
-export declare function getDaysInYear(): 366 | 365;
-export declare function getCurrentMonth(): number;
-export declare function getCurrentYear(): number;

package/dist/util.js

@@ -1,49 +0,0 @@
-export async function sleep(ms) {
-    return new Promise(resolve => setTimeout(resolve, ms));
-}
-export function removeMarkdownIndicators(input) {
-    let output = input;
-    // Make sure markdown indicator exists.
-    if (output.includes('```')) {
-        // Remove first markdown indicator.
-        output = output.substring(output.indexOf('```'));
-        output = output.substring(output.indexOf('\n'));
-        // First markdown indicator removed, now do the last.
-        output = output.substring(0, output.indexOf('```')).trim();
-    }
-    return output;
-}
-export function removePrefixIndicators(input, prefixes) {
-    let output = input;
-    for (const prefix of prefixes) {
-        if (output.includes(prefix)) {
-            output = output.substring(output.indexOf(prefix) + prefix.length);
-        }
-    }
-    return output;
-}
-export function isLeapYear() {
-    let year = new Date().getFullYear();
-    return ((year % 4 == 0) && (year % 100 != 0)) || (year % 400 == 0);
-}
-export function getDaysInMonth() {
-    let date = new Date();
-    // NOTE: month is 0-based.
-    return new Date(date.getFullYear(), date.getMonth() + 1, 0).getDate();
-}
-export function getCurrentDayInMonth() {
-    let date = new Date();
-    return date.getDate();
-}
-export function getDaysInYear() {
-    return isLeapYear() ? 366 : 365;
-}
-export function getCurrentMonth() {
-    let date = new Date();
-    // NOTE: month is 0-based.
-    return date.getMonth() + 1;
-}
-export function getCurrentYear() {
-    let date = new Date();
-    return date.getFullYear();
-}