jaidaken
f09734b0ee
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Execution Tests / test (macos-latest) (push) Has been cancelled
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Execution Tests / test (windows-latest) (push) Has been cancelled
Test server launches without errors / test (push) Has been cancelled
Unit Tests / test (macos-latest) (push) Has been cancelled
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Unit Tests / test (windows-2022) (push) Has been cancelled
Includes 30 custom nodes committed directly, 7 Civitai-exclusive loras stored via Git LFS, and a setup script that installs all dependencies and downloads HuggingFace-hosted models on vast.ai. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
73 lines
2.5 KiB
JavaScript
73 lines
2.5 KiB
JavaScript
import { rgthreeApi } from "./rgthree_api.js";
|
|
import { api } from "../../scripts/api.js";
|
|
class BaseModelInfoService extends EventTarget {
|
|
constructor() {
|
|
super();
|
|
this.fileToInfo = new Map();
|
|
this.init();
|
|
}
|
|
init() {
|
|
api.addEventListener(this.apiRefreshEventString, this.handleAsyncUpdate.bind(this));
|
|
}
|
|
async getInfo(file, refresh, light) {
|
|
if (this.fileToInfo.has(file) && !refresh) {
|
|
return this.fileToInfo.get(file);
|
|
}
|
|
return this.fetchInfo(file, refresh, light);
|
|
}
|
|
async refreshInfo(file) {
|
|
return this.fetchInfo(file, true);
|
|
}
|
|
async clearFetchedInfo(file) {
|
|
await rgthreeApi.clearModelsInfo({ type: this.modelInfoType, files: [file] });
|
|
this.fileToInfo.delete(file);
|
|
return null;
|
|
}
|
|
async savePartialInfo(file, data) {
|
|
let info = await rgthreeApi.saveModelInfo(this.modelInfoType, file, data);
|
|
this.fileToInfo.set(file, info);
|
|
return info;
|
|
}
|
|
handleAsyncUpdate(event) {
|
|
var _a;
|
|
const info = (_a = event.detail) === null || _a === void 0 ? void 0 : _a.data;
|
|
if (info === null || info === void 0 ? void 0 : info.file) {
|
|
this.setFreshInfo(info.file, info);
|
|
}
|
|
}
|
|
async fetchInfo(file, refresh = false, light = false) {
|
|
var _a;
|
|
let info = null;
|
|
if (!refresh) {
|
|
info = await rgthreeApi.getModelsInfo({ type: this.modelInfoType, files: [file], light });
|
|
}
|
|
else {
|
|
info = await rgthreeApi.refreshModelsInfo({ type: this.modelInfoType, files: [file] });
|
|
}
|
|
info = (_a = info === null || info === void 0 ? void 0 : info[0]) !== null && _a !== void 0 ? _a : null;
|
|
if (!light) {
|
|
this.fileToInfo.set(file, info);
|
|
}
|
|
return info;
|
|
}
|
|
setFreshInfo(file, info) {
|
|
this.fileToInfo.set(file, info);
|
|
}
|
|
}
|
|
class LoraInfoService extends BaseModelInfoService {
|
|
constructor() {
|
|
super(...arguments);
|
|
this.apiRefreshEventString = "rgthree-refreshed-loras-info";
|
|
this.modelInfoType = 'loras';
|
|
}
|
|
}
|
|
class CheckpointInfoService extends BaseModelInfoService {
|
|
constructor() {
|
|
super(...arguments);
|
|
this.apiRefreshEventString = "rgthree-refreshed-checkpoints-info";
|
|
this.modelInfoType = 'checkpoints';
|
|
}
|
|
}
|
|
export const LORA_INFO_SERVICE = new LoraInfoService();
|
|
export const CHECKPOINT_INFO_SERVICE = new CheckpointInfoService();
|