refactor: rename ConfigLoader and ModelLoader files, update imports, …

…and adjust ChatsConfig structure
Sma1lboy · Dec 15, 2024 · 9c34f08 · 9c34f08
1 parent eb55281
commit 9c34f08
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 10 deletions.
diff --git a/backend/src/config/ConfigLoader.ts → backend/src/config/config-loader.ts b/backend/src/config/ConfigLoader.ts → backend/src/config/config-loader.ts
@@ -1,18 +1,16 @@
 import * as fs from 'fs';
 import * as path from 'path';
 import _ from 'lodash';
-export interface ChatsConfig {
-  chats: ChatConfig[];
-}
 export interface ChatConfig {
   model: string;
   endpoint?: string;
   token?: string;
   default?: boolean;
   task?: string;
 }
+
 export class ConfigLoader {
-  private chatsConfig: ChatsConfig;
+  private chatsConfig: ChatConfig[];
 
   private readonly configPath: string;
 

diff --git a/backend/src/main.ts b/backend/src/main.ts
@@ -1,7 +1,7 @@
 import { NestFactory } from '@nestjs/core';
 import { AppModule } from './app.module';
 import 'reflect-metadata';
-import { downloadAllModels } from './model/ModelLoader';
+import { downloadAllModels } from './model/utils';
 
 async function bootstrap() {
   const app = await NestFactory.create(AppModule);

diff --git a/backend/src/model/__tests__/loadAllChatsModels.spec.ts b/backend/src/model/__tests__/loadAllChatsModels.spec.ts
@@ -1,6 +1,6 @@
-import { ConfigLoader } from '../../config/ConfigLoader';
-import { ModelDownloader } from '../ModelDownloader';
-import { downloadAllModels } from '../ModelLoader';
+import { ConfigLoader } from '../../config/config-loader';
+import { ModelDownloader } from '../model-downloader';
+import { downloadAllModels } from '../utils';
 
 const originalIsArray = Array.isArray;
 

diff --git a/backend/src/model/ModelDownloader.ts → backend/src/model/model-downloader.ts b/backend/src/model/ModelDownloader.ts → backend/src/model/model-downloader.ts
diff --git a/backend/src/model/ModelLoader.ts → backend/src/model/utils.ts b/backend/src/model/ModelLoader.ts → backend/src/model/utils.ts
@@ -1,5 +1,5 @@
-import { ChatConfig, ConfigLoader } from 'src/config/ConfigLoader';
-import { ModelDownloader } from './ModelDownloader';
+import { ChatConfig, ConfigLoader } from 'src/config/config-loader';
+import { ModelDownloader } from './model-downloader';
 
 export async function downloadAllModels(): Promise<void> {
   const configLoader = new ConfigLoader();