refactor: domain repositories (#4403)

server/src/domain/smart-info/dto/model-config.dto.ts

@@ -2,7 +2,7 @@ import { ApiProperty } from '@nestjs/swagger';
 import { Type } from 'class-transformer';
 import { IsBoolean, IsEnum, IsNotEmpty, IsNumber, IsString, Max, Min } from 'class-validator';
 import { Optional } from '../../domain.util';
-import { CLIPMode, ModelType } from '../machine-learning.interface';
+import { CLIPMode, ModelType } from '../../repositories';
 
 export class ModelConfig {
   @IsBoolean()

server/src/domain/smart-info/index.ts

@@ -1,4 +1,2 @@
 export * from './dto';
-export * from './machine-learning.interface';
-export * from './smart-info.repository';
 export * from './smart-info.service';

server/src/domain/smart-info/machine-learning.interface.ts

@@ -1,44 +0,0 @@
-import { ClassificationConfig, CLIPConfig, RecognitionConfig } from './dto';
-
-export const IMachineLearningRepository = 'IMachineLearningRepository';
-
-export interface VisionModelInput {
-  imagePath: string;
-}
-
-export interface TextModelInput {
-  text: string;
-}
-
-export interface BoundingBox {
-  x1: number;
-  y1: number;
-  x2: number;
-  y2: number;
-}
-
-export interface DetectFaceResult {
-  imageWidth: number;
-  imageHeight: number;
-  boundingBox: BoundingBox;
-  score: number;
-  embedding: number[];
-}
-
-export enum ModelType {
-  IMAGE_CLASSIFICATION = 'image-classification',
-  FACIAL_RECOGNITION = 'facial-recognition',
-  CLIP = 'clip',
-}
-
-export enum CLIPMode {
-  VISION = 'vision',
-  TEXT = 'text',
-}
-
-export interface IMachineLearningRepository {
-  classifyImage(url: string, input: VisionModelInput, config: ClassificationConfig): Promise<string[]>;
-  encodeImage(url: string, input: VisionModelInput, config: CLIPConfig): Promise<number[]>;
-  encodeText(url: string, input: TextModelInput, config: CLIPConfig): Promise<number[]>;
-  detectFaces(url: string, input: VisionModelInput, config: RecognitionConfig): Promise<DetectFaceResult[]>;
-}

server/src/domain/smart-info/smart-info.repository.ts

@@ -1,7 +0,0 @@
-import { SmartInfoEntity } from '@app/infra/entities';
-
-export const ISmartInfoRepository = 'ISmartInfoRepository';
-
-export interface ISmartInfoRepository {
-  upsert(info: Partial<SmartInfoEntity>): Promise<void>;
-}

server/src/domain/smart-info/smart-info.service.spec.ts

@@ -7,11 +7,15 @@ import {
   newSmartInfoRepositoryMock,
   newSystemConfigRepositoryMock,
 } from '@test';
-import { IAssetRepository, WithoutProperty } from '../asset';
-import { IJobRepository, JobName } from '../job';
-import { ISystemConfigRepository } from '../system-config';
-import { IMachineLearningRepository } from './machine-learning.interface';
-import { ISmartInfoRepository } from './smart-info.repository';
+import { JobName } from '../job';
+import {
+  IAssetRepository,
+  IJobRepository,
+  IMachineLearningRepository,
+  ISmartInfoRepository,
+  ISystemConfigRepository,
+  WithoutProperty,
+} from '../repositories';
 import { SmartInfoService } from './smart-info.service';
 
 const asset = {

server/src/domain/smart-info/smart-info.service.ts

@@ -1,10 +1,15 @@
 import { Inject, Injectable } from '@nestjs/common';
-import { IAssetRepository, WithoutProperty } from '../asset';
 import { usePagination } from '../domain.util';
-import { IBaseJob, IEntityJob, IJobRepository, JOBS_ASSET_PAGINATION_SIZE, JobName } from '../job';
-import { ISystemConfigRepository, SystemConfigCore } from '../system-config';
-import { IMachineLearningRepository } from './machine-learning.interface';
-import { ISmartInfoRepository } from './smart-info.repository';
+import { IBaseJob, IEntityJob, JOBS_ASSET_PAGINATION_SIZE, JobName } from '../job';
+import {
+  IAssetRepository,
+  IJobRepository,
+  IMachineLearningRepository,
+  ISmartInfoRepository,
+  ISystemConfigRepository,
+  WithoutProperty,
+} from '../repositories';
+import { SystemConfigCore } from '../system-config';
 
 @Injectable()
 export class SmartInfoService {