@just-every/ensemble 0.2.45 → 0.2.47

package/dist/utils/model_class_config.d.ts

@@ -0,0 +1,12 @@
+import { ModelClass } from '../types/types.js';
+export declare function getModelClass(className: string): ModelClass | undefined;
+export declare function getModelClassNames(): string[];
+export declare function overrideModelClass(className: string, config: Partial<ModelClass>): void;
+export declare function setModelClassModels(className: string, models: string[], random?: boolean): void;
+export declare function addModelToClass(className: string, modelId: string): void;
+export declare function removeModelFromClass(className: string, modelId: string): void;
+export declare function setModelClassRandom(className: string, random: boolean): void;
+export declare function resetModelClass(className: string): void;
+export declare function getAllModelClasses(): Record<string, ModelClass>;
+export declare function updateModelClasses(updates: Record<string, Partial<ModelClass>>): void;
+//# sourceMappingURL=model_class_config.d.ts.map

package/dist/utils/model_class_config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model_class_config.d.ts","sourceRoot":"","sources":["../../utils/model_class_config.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,UAAU,EAAgB,MAAM,mBAAmB,CAAC;AAW7D,wBAAgB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,UAAU,GAAG,SAAS,CAwBvE;AAMD,wBAAgB,kBAAkB,IAAI,MAAM,EAAE,CAE7C;AAOD,wBAAgB,kBAAkB,CAC9B,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,OAAO,CAAC,UAAU,CAAC,GAC5B,IAAI,CAkBN;AAQD,wBAAgB,mBAAmB,CAC/B,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,EAAE,EAChB,MAAM,CAAC,EAAE,OAAO,GACjB,IAAI,CAUN;AAOD,wBAAgB,eAAe,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAcxE;AAOD,wBAAgB,oBAAoB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAa7E;AAOD,wBAAgB,mBAAmB,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,GAAG,IAAI,CAU5E;AAMD,wBAAgB,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAIvD;AAMD,wBAAgB,kBAAkB,IAAI,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAW/D;AAMD,wBAAgB,kBAAkB,CAC9B,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC,GAC7C,IAAI,CAIN"}

package/dist/utils/model_class_config.js

@@ -0,0 +1,93 @@
+import { MODEL_CLASSES, findModel } from '../data/model_data.js';
+import { overrideModelClass as _overrideModelClass, getModelClassOverride, } from './external_models.js';
+export function getModelClass(className) {
+    if (!(className in MODEL_CLASSES)) {
+        return undefined;
+    }
+    const baseConfig = MODEL_CLASSES[className];
+    const override = getModelClassOverride(className);
+    if (!override) {
+        return { ...baseConfig };
+    }
+    return {
+        ...baseConfig,
+        ...override,
+        models: override.models || baseConfig.models,
+    };
+}
+export function getModelClassNames() {
+    return Object.keys(MODEL_CLASSES);
+}
+export function overrideModelClass(className, config) {
+    if (!(className in MODEL_CLASSES)) {
+        throw new Error(`Model class '${className}' does not exist`);
+    }
+    if (config.models) {
+        for (const modelId of config.models) {
+            if (!findModel(modelId)) {
+                console.warn(`Model '${modelId}' not found in registry, but adding to class '${className}' anyway`);
+            }
+        }
+    }
+    _overrideModelClass(className, config);
+}
+export function setModelClassModels(className, models, random) {
+    const currentConfig = getModelClass(className);
+    if (!currentConfig) {
+        throw new Error(`Model class '${className}' does not exist`);
+    }
+    overrideModelClass(className, {
+        models,
+        random: random !== undefined ? random : currentConfig.random,
+    });
+}
+export function addModelToClass(className, modelId) {
+    const currentConfig = getModelClass(className);
+    if (!currentConfig) {
+        throw new Error(`Model class '${className}' does not exist`);
+    }
+    if (currentConfig.models.includes(modelId)) {
+        return;
+    }
+    const newModels = [...currentConfig.models, modelId];
+    setModelClassModels(className, newModels);
+}
+export function removeModelFromClass(className, modelId) {
+    const currentConfig = getModelClass(className);
+    if (!currentConfig) {
+        throw new Error(`Model class '${className}' does not exist`);
+    }
+    const newModels = currentConfig.models.filter(id => id !== modelId);
+    if (newModels.length < currentConfig.models.length) {
+        setModelClassModels(className, newModels);
+    }
+}
+export function setModelClassRandom(className, random) {
+    const currentConfig = getModelClass(className);
+    if (!currentConfig) {
+        throw new Error(`Model class '${className}' does not exist`);
+    }
+    overrideModelClass(className, {
+        models: currentConfig.models,
+        random,
+    });
+}
+export function resetModelClass(className) {
+    overrideModelClass(className, {});
+}
+export function getAllModelClasses() {
+    const result = {};
+    for (const className of getModelClassNames()) {
+        const config = getModelClass(className);
+        if (config) {
+            result[className] = config;
+        }
+    }
+    return result;
+}
+export function updateModelClasses(updates) {
+    for (const [className, config] of Object.entries(updates)) {
+        overrideModelClass(className, config);
+    }
+}
+//# sourceMappingURL=model_class_config.js.map

package/dist/utils/model_class_config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model_class_config.js","sourceRoot":"","sources":["../../utils/model_class_config.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAEjE,OAAO,EACH,kBAAkB,IAAI,mBAAmB,EACzC,qBAAqB,GACxB,MAAM,sBAAsB,CAAC;AAO9B,MAAM,UAAU,aAAa,CAAC,SAAiB;IAE3C,IAAI,CAAC,CAAC,SAAS,IAAI,aAAa,CAAC,EAAE,CAAC;QAChC,OAAO,SAAS,CAAC;IACrB,CAAC;IAGD,MAAM,UAAU,GAAG,aAAa,CAAC,SAAuC,CAAC,CAAC;IAG1E,MAAM,QAAQ,GAAG,qBAAqB,CAAC,SAAS,CAAC,CAAC;IAGlD,IAAI,CAAC,QAAQ,EAAE,CAAC;QACZ,OAAO,EAAE,GAAG,UAAU,EAAE,CAAC;IAC7B,CAAC;IAGD,OAAO;QACH,GAAG,UAAU;QACb,GAAG,QAAQ;QAEX,MAAM,EAAE,QAAQ,CAAC,MAAM,IAAI,UAAU,CAAC,MAAM;KAC/C,CAAC;AACN,CAAC;AAMD,MAAM,UAAU,kBAAkB;IAC9B,OAAO,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;AACtC,CAAC;AAOD,MAAM,UAAU,kBAAkB,CAC9B,SAAiB,EACjB,MAA2B;IAG3B,IAAI,CAAC,CAAC,SAAS,IAAI,aAAa,CAAC,EAAE,CAAC;QAChC,MAAM,IAAI,KAAK,CAAC,gBAAgB,SAAS,kBAAkB,CAAC,CAAC;IACjE,CAAC;IAGD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QAChB,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;gBACtB,OAAO,CAAC,IAAI,CACR,UAAU,OAAO,iDAAiD,SAAS,UAAU,CACxF,CAAC;YACN,CAAC;QACL,CAAC;IACL,CAAC;IAED,mBAAmB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;AAC3C,CAAC;AAQD,MAAM,UAAU,mBAAmB,CAC/B,SAAiB,EACjB,MAAgB,EAChB,MAAgB;IAEhB,MAAM,aAAa,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;IAC/C,IAAI,CAAC,aAAa,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,gBAAgB,SAAS,kBAAkB,CAAC,CAAC;IACjE,CAAC;IAED,kBAAkB,CAAC,SAAS,EAAE;QAC1B,MAAM;QACN,MAAM,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC,MAAM;KAC/D,CAAC,CAAC;AACP,CAAC;AAOD,MAAM,UAAU,eAAe,CAAC,SAAiB,EAAE,OAAe;IAC9D,MAAM,aAAa,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;IAC/C,IAAI,CAAC,aAAa,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,gBAAgB,SAAS,kBAAkB,CAAC,CAAC;IACjE,CAAC;IAGD,IAAI,aAAa,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QACzC,OAAO;IACX,CAAC;IAGD,MAAM,SAAS,GAAG,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrD,mBAAmB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;AAC9C,CAAC;AAOD,MAAM,UAAU,oBAAoB,CAAC,SAAiB,EAAE,OAAe;IACnE,MAAM,aAAa,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;IAC/C,IAAI,CAAC,aAAa,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,gBAAgB,SAAS,kBAAkB,CAAC,CAAC;IACjE,CAAC;IAGD,MAAM,SAAS,GAAG,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,OAAO,CAAC,CAAC;IAGpE,IAAI,SAAS,CAAC,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACjD,mBAAmB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;IAC9C,CAAC;AACL,CAAC;AAOD,MAAM,UAAU,mBAAmB,CAAC,SAAiB,EAAE,MAAe;IAClE,MAAM,aAAa,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;IAC/C,IAAI,CAAC,aAAa,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,gBAAgB,SAAS,kBAAkB,CAAC,CAAC;IACjE,CAAC;IAED,kBAAkB,CAAC,SAAS,EAAE;QAC1B,MAAM,EAAE,aAAa,CAAC,MAAM;QAC5B,MAAM;KACT,CAAC,CAAC;AACP,CAAC;AAMD,MAAM,UAAU,eAAe,CAAC,SAAiB;IAG7C,kBAAkB,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;AACtC,CAAC;AAMD,MAAM,UAAU,kBAAkB;IAC9B,MAAM,MAAM,GAA+B,EAAE,CAAC;IAE9C,KAAK,MAAM,SAAS,IAAI,kBAAkB,EAAE,EAAE,CAAC;QAC3C,MAAM,MAAM,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;QACxC,IAAI,MAAM,EAAE,CAAC;YACT,MAAM,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC;QAC/B,CAAC;IACL,CAAC;IAED,OAAO,MAAM,CAAC;AAClB,CAAC;AAMD,MAAM,UAAU,kBAAkB,CAC9B,OAA4C;IAE5C,KAAK,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACxD,kBAAkB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IAC1C,CAAC;AACL,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@just-every/ensemble",
-    "version": "0.2.45",
+    "version": "0.2.47",
     "description": "LLM provider abstraction layer with unified streaming interface",
     "type": "module",
     "main": "./dist/index.js",