UNPKG

tprompter

Version:
97 lines (96 loc) 4.36 kB
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) { var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d; if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc); else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r; return c > 3 && r && Object.defineProperty(target, key, r), r; }; var __metadata = (this && this.__metadata) || function (k, v) { if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v); }; var __param = (this && this.__param) || function (paramIndex, decorator) { return function (target, key) { decorator(target, key, paramIndex); } }; var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) { function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); } return new (P || (P = Promise))(function (resolve, reject) { function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } } function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } } function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); } step((generator = generator.apply(thisArg, _arguments || [])).next()); }); }; import { Service } from 'typedi'; import { PersistentConfigRepository } from './PersistentConfigRepository.js'; import { AppConfigData, AppConfigDataValuesTransformers, } from './AppConfigData.js'; import { CLIArgumentsToAppConfigMapper } from './CLIArgumentsToConfigMapper.js'; import { InjectLogger } from '../logger/logger.decorator.js'; export const DEFAULT_APP_CONFIG = new AppConfigData({ agentDefaultModel: 'gpt-4o-mini', agentMaxTokens: 15000, appName: 'tprompter', }); let AppConfig = class AppConfig { constructor(repo, mapper, logger) { this.repo = repo; this.mapper = mapper; this.logger = logger; this.defaultData = DEFAULT_APP_CONFIG.clone(); this.persistentData = AppConfigData.empty(); this.argumentsData = AppConfigData.empty(); } loadPersistent() { return __awaiter(this, void 0, void 0, function* () { try { const raw = yield this.repo.getRawConfig(); this.persistentData = AppConfigData.deserialize(raw); } catch (err) { if (err.code === 'ENOENT') { this.logger.debug('No persistent config found'); } else { this.logger.error('Failed to load persistent config: ', err); } } }); } applyCLIArguments(data) { return __awaiter(this, void 0, void 0, function* () { this.logger.debug('Applying CLI arguments to config', data); this.argumentsData = new AppConfigData(this.mapper.map(data)); }); } resetConfig() { this.argumentsData = AppConfigData.empty(); this.persistentData = AppConfigData.empty(); } setPersistentValue(key, value) { return __awaiter(this, void 0, void 0, function* () { const val = AppConfigDataValuesTransformers[key](value, true); this.persistentData[key] = val; yield this.savePersistentConfig(); }); } deletePersistentValue(key) { return __awaiter(this, void 0, void 0, function* () { delete this.persistentData[key]; yield this.savePersistentConfig(); }); } savePersistentConfig() { return this.repo.saveRawConfig(this.persistentData.serialize()); } getConfig() { return this.defaultData.merge(this.persistentData).merge(this.argumentsData); } getAvailableConfigKeys() { return AppConfigData.getAvailableKeys(); } }; AppConfig = __decorate([ Service(), __param(2, InjectLogger()), __metadata("design:paramtypes", [PersistentConfigRepository, CLIArgumentsToAppConfigMapper, Object]) ], AppConfig); export { AppConfig };