@cellaware/utils 1.0.2 → 2.0.1

package/README.md

@@ -26,12 +26,6 @@ Azure Cosmos database key.
 
 Connection string of Azure Storage account.
 
-### CHATWMS_ENVIRONMENT
-
-Indicates the environment of ChatWMS instance. This is used mainly for Cosmos and Storage interactions. For example, we log conversation history in a development database if the environment is `dev`.
-
-Expected Values: `dev` | `prod`
-
 ### CHATWMS_URL
 
 ChatWMS URL.

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

@@ -1,4 +1,4 @@
-export declare function chatwmsCosmosSelect(collectionId: string, query: string, environment?: string): Promise<any[]>;
-export declare function chatwmsCosmosInsert(collectionId: string, data: any, environment?: string): Promise<boolean>;
-export declare function chatwmsCosmosDelete(collectionId: string, query: string, environment?: string): Promise<boolean>;
-export declare function chatwmsCosmosUpdate(collectionId: string, query: string, data: any, environment?: string): Promise<boolean>;
+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 +1,15 @@
 import { cosmosDelete, cosmosInsert, cosmosSelect, cosmosUpdate } from "../../azure/cosmos.js";
-import { getChatWMSEnvironment } from "../environment.js";
+const DATABASE_ID = 'chatwms';
 const PARTITION_KEY = '/clientId';
-export async function chatwmsCosmosSelect(collectionId, query, environment) {
-    return cosmosSelect(environment ?? getChatWMSEnvironment(), collectionId, PARTITION_KEY, query);
+export async function chatwmsCosmosSelect(collectionId, query) {
+    return cosmosSelect(DATABASE_ID, collectionId, PARTITION_KEY, query);
 }
-export async function chatwmsCosmosInsert(collectionId, data, environment) {
-    return cosmosInsert(environment ?? getChatWMSEnvironment(), collectionId, PARTITION_KEY, data);
+export async function chatwmsCosmosInsert(collectionId, data) {
+    return cosmosInsert(DATABASE_ID, collectionId, PARTITION_KEY, data);
 }
-export async function chatwmsCosmosDelete(collectionId, query, environment) {
-    return cosmosDelete(environment ?? getChatWMSEnvironment(), collectionId, PARTITION_KEY, query);
+export async function chatwmsCosmosDelete(collectionId, query) {
+    return cosmosDelete(DATABASE_ID, collectionId, PARTITION_KEY, query);
 }
-export async function chatwmsCosmosUpdate(collectionId, query, data, environment) {
-    return cosmosUpdate(environment ?? getChatWMSEnvironment(), collectionId, PARTITION_KEY, query, data);
+export async function chatwmsCosmosUpdate(collectionId, query, data) {
+    return cosmosUpdate(DATABASE_ID, collectionId, PARTITION_KEY, query, data);
 }

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

@@ -1,5 +1,5 @@
 export declare function getBlobId(clientId: string, blobName: string): string;
-export declare function chatwmsContainerCreate(environment?: string): Promise<void>;
-export declare function chatwmsBlobUpload(blobId: string, data: any, environment?: string): Promise<void>;
-export declare function chatwmsBlobDelete(blobId: string, environment?: string): Promise<void>;
-export declare function chatwmsBlobDownload(blobId: string, environment?: string): Promise<any[]>;
+export declare function chatwmsContainerCreate(): Promise<void>;
+export declare function chatwmsBlobUpload(blobId: string, data: any): Promise<void>;
+export declare function chatwmsBlobDelete(blobId: string): Promise<void>;
+export declare function chatwmsBlobDownload(blobId: string): Promise<any[]>;

package/dist/chatwms/azure/storage.js

@@ -1,17 +1,17 @@
 import { blobDelete, blobDownload, blobUpload, containerCreate } from "../../azure/storage.js";
-import { getChatWMSEnvironment } from "../environment.js";
+const CONTAINER_CLIENT_ID = 'chatwms';
 export function getBlobId(clientId, blobName) {
     return clientId.replace('_', '-') + '-' + blobName;
 }
-export async function chatwmsContainerCreate(environment) {
-    return containerCreate(environment ?? getChatWMSEnvironment());
+export async function chatwmsContainerCreate() {
+    return containerCreate(CONTAINER_CLIENT_ID);
 }
-export async function chatwmsBlobUpload(blobId, data, environment) {
-    return blobUpload(environment ?? getChatWMSEnvironment(), blobId, data);
+export async function chatwmsBlobUpload(blobId, data) {
+    return blobUpload(CONTAINER_CLIENT_ID, blobId, data);
 }
-export async function chatwmsBlobDelete(blobId, environment) {
-    return blobDelete(environment ?? getChatWMSEnvironment(), blobId);
+export async function chatwmsBlobDelete(blobId) {
+    return blobDelete(CONTAINER_CLIENT_ID, blobId);
 }
-export async function chatwmsBlobDownload(blobId, environment) {
-    return blobDownload(environment ?? getChatWMSEnvironment(), blobId);
+export async function chatwmsBlobDownload(blobId) {
+    return blobDownload(CONTAINER_CLIENT_ID, blobId);
 }

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

@@ -1,4 +1,5 @@
 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`.
@@ -15,10 +16,10 @@ export declare class SingleActionChain extends BaseChain {
     _chainType(): string;
     get inputKeys(): string[];
     get outputKeys(): string[];
-    getModelName(): string;
+    getModelName(): ModelName;
     getPrompt(): string;
 }
-export declare function createSingleActionChain(name: string, modelName: string, inputKeys: string[], prompt: string, temperature?: number, verbose?: boolean): SingleActionChain;
+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.
@@ -30,11 +31,11 @@ export declare function createSingleActionChain(name: string, modelName: string,
 export declare class ChainStore {
     private promptsPath;
     private chains;
-    constructor(promptsPath: string, defaultModelName?: string);
+    constructor(promptsPath: string, defaultModelName?: ModelName);
     private initBuiltinChains;
     private static getTokenUsage;
     private getPrompt;
-    addChain(name: string, modelName: string, temperature?: number, verbose?: boolean): void;
+    addChain(name: string, modelName: ModelName, temperature?: number, verbose?: boolean): 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/cost.d.ts

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

package/dist/llm/model.d.ts

@@ -0,0 +1 @@
+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

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

package/package.json

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

package/dist/chatwms/environment.d.ts

@@ -1,7 +0,0 @@
-export declare const ENVIRONMENT_DEV = "dev";
-export declare const ENVIRONMENT_PROD = "prod";
-export declare const ENVIRONMENT_ALL = "all";
-export declare function coalesceEnvironment(environment: string): "dev" | "prod";
-export declare function getEnvironmentDescription(environment: string): "DEVELOPMENT" | "PRODUCTION" | "ALL";
-export declare function printEnvironment(environment: string): void;
-export declare function getChatWMSEnvironment(): string;

package/dist/chatwms/environment.js

@@ -1,26 +0,0 @@
-export const ENVIRONMENT_DEV = 'dev';
-export const ENVIRONMENT_PROD = 'prod';
-export const ENVIRONMENT_ALL = 'all';
-export function coalesceEnvironment(environment) {
-    return environment === ENVIRONMENT_PROD ? ENVIRONMENT_PROD : ENVIRONMENT_DEV;
-}
-export function getEnvironmentDescription(environment) {
-    switch (environment) {
-        case ENVIRONMENT_DEV:
-            return 'DEVELOPMENT';
-        case ENVIRONMENT_PROD:
-            return 'PRODUCTION';
-        case ENVIRONMENT_ALL:
-            return 'ALL';
-        default:
-            return 'DEVELOPMENT';
-    }
-}
-export function printEnvironment(environment) {
-    console.log(`============================================================
-                        ${getEnvironmentDescription(environment)}    
-============================================================`);
-}
-export function getChatWMSEnvironment() {
-    return coalesceEnvironment(process.env.CHATWMS_ENVIRONMENT ?? 'dev');
-}