chore: update models to extends from openai

Signed-off-by: James <namnh0122@gmail.com>

Author: namchuai

cortex-js/src/app.module.ts

@@ -10,7 +10,6 @@ import { ExtensionModule } from './infrastructure/repositories/extensions/extens
 import { CortexModule } from './usecases/cortex/cortex.module';
 import { ConfigModule } from '@nestjs/config';
 import { env } from 'node:process';
-import { SeedService } from './usecases/seed/seed.service';
 import { FileManagerModule } from './infrastructure/services/file-manager/file-manager.module';
 import { AppLoggerMiddleware } from './infrastructure/middlewares/app.logger.middleware';
 import { TelemetryModule } from './usecases/telemetry/telemetry.module';
@@ -26,6 +25,7 @@ import { ModelsController } from './infrastructure/controllers/models.controller
 import { ThreadsController } from './infrastructure/controllers/threads.controller';
 import { StatusController } from './infrastructure/controllers/status.controller';
 import { ProcessController } from './infrastructure/controllers/process.controller';
+import { DownloadManagerModule } from './download-manager/download-manager.module';
 
 @Module({
   imports: [
@@ -37,6 +37,7 @@ import { ProcessController } from './infrastructure/controllers/process.controll
       envFilePath: env.NODE_ENV !== 'production' ? '.env.development' : '.env',
     }),
     EventEmitterModule.forRoot(),
+    DownloadManagerModule,
     DatabaseModule,
     MessagesModule,
     ThreadsModule,
@@ -60,7 +61,6 @@ import { ProcessController } from './infrastructure/controllers/process.controll
     EventsController,
   ],
   providers: [
-    SeedService,
     {
       provide: APP_FILTER,
       useClass: GlobalExceptionFilter,

cortex-js/src/domain/models/assistant.interface.ts

@@ -1,39 +1,14 @@
-/**
- * Assistant type defines the shape of an assistant object.
- * @stored
- */
+import { Assistant as OpenAiAssistant } from 'openai/resources/beta/assistants';
+import { AssistantResponseFormatOption as OpenAIAssistantResponseFormatOption } from 'openai/resources/beta/threads/threads';
 
-export interface AssistantTool {
-  type: string;
-  enabled: boolean;
-  settings: any;
-}
+export interface Assistant extends OpenAiAssistant {
+  avatar?: string;
+
+  response_format?: AssistantResponseFormatOption;
 
-export interface Assistant {
-  /** Represents the unique identifier of the object. */
-  id: string;
-  /** Represents the avatar of the user. */
-  avatar: string;
-  /** Represents the location of the thread. */
-  thread_location?: string;
-  /** Represents the object. */
-  object: string;
-  /** Represents the creation timestamp of the object. */
-  created_at: number;
-  /** Represents the name of the object. */
-  name: string;
-  /** Represents the description of the object. */
-  description?: string;
-  /** Represents the model of the object. */
-  model: string;
-  /** Represents the instructions for the object. */
-  instructions?: string;
-  /** Represents the tools associated with the object. */
-  tools?: AssistantTool[];
-  /** Represents the file identifiers associated with the object. */
-  file_ids: string[];
-  /** Represents the metadata of the object. */
-  metadata?: AssistantMetadata;
+  tool_resources?: AssistantToolResources;
 }
 
-export interface AssistantMetadata {}
+export type AssistantResponseFormatOption = OpenAIAssistantResponseFormatOption;
+
+export interface AssistantToolResources extends OpenAiAssistant.ToolResources {}

cortex-js/src/domain/models/thread.interface.ts

@@ -1,46 +1,12 @@
-import { AssistantTool } from './assistant.interface';
-import { Model } from './model.interface';
+import { Thread as OpenAiThread } from 'openai/resources/beta/threads/threads';
+import { Assistant } from './assistant.interface';
 
-export interface Thread {
-  /** Unique identifier for the thread, generated by default using the ULID method. **/
-  id: string;
+export interface ThreadToolResources extends OpenAiThread.ToolResources {}
 
-  /** Object name **/
-  object: string;
-
-  /** The title of this thread. **/
+export interface Thread extends OpenAiThread {
   title: string;
 
-  /** Assistants in this thread. **/
-  assistants: ThreadAssistantInfo[];
-
-  /** The timestamp indicating when this thread was created, represented in ISO 8601 format. **/
-  createdAt: number;
-
-  /** The timestamp indicating when this thread was updated, represented in ISO 8601 format. **/
-  updatedAt?: number;
-
-  /**
-   * The additional metadata of this thread.
-   **/
-  metadata?: ThreadMetadata;
-}
-
-/**
- * The additional metadata of this thread.
- */
-export interface ThreadMetadata {
-  lastMessage?: string;
-}
+  assistants: Assistant[];
 
-/**
- * Represents the information about an assistant in a thread.
- * @stored
- */
-export interface ThreadAssistantInfo {
-  assistant_id: string;
-  assistant_name: string;
-  model: Partial<Model>;
-  instructions?: string;
-  tools?: AssistantTool[];
+  tool_resources: ThreadToolResources | null;
 }

cortex-js/src/infrastructure/commanders/usecases/chat.cli.usecases.ts

@@ -265,9 +265,16 @@ export class ChatCliUsecases {
     if (!assistant) throw new Error('No assistant available');
 
     const assistantDto: CreateThreadAssistantDto = {
-      assistant_id: assistant.id,
-      assistant_name: assistant.name,
-      model: model,
+      avatar: '',
+      id: 'jan',
+      object: 'assistant',
+      created_at: Date.now(),
+      name: 'Jan',
+      description: 'A default assistant that can use all downloaded models',
+      model: modelId,
+      instructions: '',
+      tools: [],
+      metadata: {},
     };
 
     const createThreadDto: CreateThreadDto = {

cortex-js/src/infrastructure/dtos/assistants/assistant-tool.dto.ts

@@ -1,33 +1,37 @@
-import { Type } from 'class-transformer';
-import { IsArray, IsOptional, IsString, ValidateNested } from 'class-validator';
-import {
-  Assistant,
-  AssistantMetadata,
-} from '@/domain/models/assistant.interface';
-import { AssistantToolDto } from './assistant-tool.dto';
+import { IsArray, IsOptional, IsString } from 'class-validator';
+import { Assistant } from '@/domain/models/assistant.interface';
 import { ApiProperty } from '@nestjs/swagger';
 
 export class CreateAssistantDto implements Partial<Assistant> {
   @ApiProperty({
     description: 'The unique identifier of the assistant.',
+    example: 'jan',
+    default: 'jan',
   })
   @IsString()
   id: string;
 
   @ApiProperty({
     description: 'The avatar of the assistant.',
+    example: '',
+    default: '',
   })
+  @IsOptional()
   @IsString()
-  avatar: string;
+  avatar?: string;
 
   @ApiProperty({
     description: 'The name of the assistant.',
+    example: 'Jan',
+    default: 'Jan',
   })
   @IsString()
   name: string;
 
   @ApiProperty({
     description: 'The description of the assistant.',
+    example: 'A default assistant that can use all downloaded models',
+    default: 'A default assistant that can use all downloaded models',
   })
   @IsString()
   description: string;
@@ -40,29 +44,23 @@ export class CreateAssistantDto implements Partial<Assistant> {
 
   @ApiProperty({
     description: 'The instructions for the assistant.',
+    example: '',
+    default: '',
   })
   @IsString()
   instructions: string;
 
   @ApiProperty({
     description: 'The tools associated with the assistant.',
+    example: [],
+    default: [],
   })
   @IsArray()
-  @ValidateNested({ each: true })
-  @Type(() => AssistantToolDto)
-  tools: AssistantToolDto[];
-
-  @ApiProperty({
-    description:
-      'The identifiers of the files that have been uploaded to the thread.',
-  })
-  @IsArray()
-  @IsOptional()
-  file_ids: string[];
+  tools: any[];
 
   @ApiProperty({
     description: 'The metadata of the assistant.',
   })
   @IsOptional()
-  metadata?: AssistantMetadata;
+  metadata: unknown | null;
 }

cortex-js/src/infrastructure/dtos/assistants/create-assistant.dto.ts

@@ -1,34 +1,73 @@
-import { IsArray, IsOptional, IsString, ValidateNested } from 'class-validator';
-import { ThreadAssistantInfo } from '@/domain/models/thread.interface';
-import { AssistantToolDto } from '@/infrastructure/dtos/assistants/assistant-tool.dto';
-import { Type } from 'class-transformer';
+import { IsArray, IsNumber, IsOptional, IsString } from 'class-validator';
 import { ApiProperty } from '@nestjs/swagger';
-import { CreateModelDto } from '../models/create-model.dto';
+import {
+  Assistant,
+  AssistantToolResources,
+} from '@/domain/models/assistant.interface';
+import { AssistantResponseFormatOption } from 'openai/resources/beta/threads/threads';
 
-export class CreateThreadAssistantDto implements ThreadAssistantInfo {
-  @ApiProperty({ description: 'The unique identifier of the assistant.' })
+export class CreateThreadAssistantDto implements Assistant {
+  @ApiProperty({
+    description: 'The unique identifier of the assistant.',
+    type: 'string',
+  })
+  @IsString()
+  id: string;
+
+  @ApiProperty()
+  @IsOptional()
   @IsString()
-  assistant_id: string;
+  avatar?: string;
 
   @ApiProperty({ description: 'The name of the assistant.' })
   @IsString()
-  assistant_name: string;
+  name: string;
 
   @ApiProperty({ description: "The model's unique identifier and settings." })
-  @ValidateNested()
-  model: CreateModelDto;
+  @IsString()
+  model: string;
 
   @ApiProperty({ description: "The assistant's specific instructions." })
-  @IsOptional()
   @IsString()
-  instructions?: string;
+  instructions: string;
 
   @ApiProperty({
     description: "The thread's tool(Knowledge Retrieval) configurations.",
   })
   @IsOptional()
   @IsArray()
-  @ValidateNested({ each: true })
-  @Type(() => AssistantToolDto)
-  tools?: AssistantToolDto[];
+  tools: any;
+
+  @ApiProperty()
+  @IsString()
+  @IsOptional()
+  description: string | null;
+
+  @ApiProperty()
+  @IsOptional()
+  metadata: Record<string, unknown> | null;
+
+  @ApiProperty()
+  object: 'assistant';
+
+  @ApiProperty()
+  @IsNumber()
+  @IsOptional()
+  temperature?: number | null;
+
+  @ApiProperty()
+  @IsNumber()
+  @IsOptional()
+  top_p?: number | null;
+
+  @ApiProperty()
+  created_at: number;
+
+  @ApiProperty()
+  @IsOptional()
+  response_format?: AssistantResponseFormatOption;
+
+  @ApiProperty()
+  @IsOptional()
+  tool_resources?: AssistantToolResources;
 }

cortex-js/src/infrastructure/dtos/threads/create-thread-assistant.dto.ts

@@ -1,13 +1,10 @@
-import { IsArray, ValidateNested } from 'class-validator';
+import { IsArray } from 'class-validator';
 import { Thread } from '@/domain/models/thread.interface';
 import { CreateThreadAssistantDto } from './create-thread-assistant.dto';
-import { Type } from 'class-transformer';
 import { ApiProperty } from '@nestjs/swagger';
 
 export class CreateThreadDto implements Partial<Thread> {
   @ApiProperty({ description: "The details of the thread's settings." })
   @IsArray()
-  @ValidateNested({ each: true })
-  @Type(() => CreateThreadAssistantDto)
   assistants: CreateThreadAssistantDto[];
 }