update

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

Author: namchuai

cortex-js/src/infrastructure/controllers/models.controller.ts

@@ -196,15 +196,21 @@ export class ModelsController {
   @ApiOperation({
     summary: 'Update model',
     description: "Updates a model instance defined by a model's `id`.",
+    parameters: [
+      {
+        in: 'path',
+        name: 'model',
+        required: true,
+        description: 'The unique identifier of the model.',
+      },
+    ],
   })
-  @ApiParam({
-    name: 'id',
-    required: true,
-    description: 'The unique identifier of the model.',
-  })
-  @Patch(':id(*)')
-  update(@Param('id') id: string, @Body() updateModelDto: UpdateModelDto) {
-    return this.modelsUsecases.update(id, updateModelDto);
+  @Post(':model(*)/config')
+  async update(
+    @Param('model') model: string,
+    @Body() updateModelDto: UpdateModelDto,
+  ) {
+    return this.modelsUsecases.update(model, updateModelDto);
   }
 
   @ApiResponse({

cortex-js/src/infrastructure/exception/duplicate-assistant.exception.ts

@@ -0,0 +1,10 @@
+import { HttpException, HttpStatus } from '@nestjs/common';
+
+export class DuplicateAssistantException extends HttpException {
+  constructor(assistantId: string) {
+    super(
+      `Assistant with the id ${assistantId} is already exists.`,
+      HttpStatus.CONFLICT,
+    );
+  }
+}

cortex-js/src/usecases/assistants/assistants.usecases.ts

@@ -1,11 +1,12 @@
 import { Inject, Injectable } from '@nestjs/common';
 import { AssistantEntity } from '@/infrastructure/entities/assistant.entity';
-import { Repository } from 'typeorm';
+import { QueryFailedError, Repository } from 'typeorm';
 import { CreateAssistantDto } from '@/infrastructure/dtos/assistants/create-assistant.dto';
 import { Assistant } from '@/domain/models/assistant.interface';
 import { PageDto } from '@/infrastructure/dtos/page.dto';
 import { ModelRepository } from '@/domain/repositories/model.interface';
 import { ModelNotFoundException } from '@/infrastructure/exception/model-not-found.exception';
+import { DuplicateAssistantException } from '@/infrastructure/exception/duplicate-assistant.exception';
 
 @Injectable()
 export class AssistantsUsecases {
@@ -16,7 +17,7 @@ export class AssistantsUsecases {
   ) {}
 
   async create(createAssistantDto: CreateAssistantDto) {
-    const { top_p, temperature, model } = createAssistantDto;
+    const { top_p, temperature, model, id } = createAssistantDto;
     if (model !== '*') {
       const modelEntity = await this.modelRepository.findOne(model);
       if (!modelEntity) {
@@ -33,7 +34,18 @@ export class AssistantsUsecases {
       top_p: top_p ?? null,
       temperature: temperature ?? null,
     };
-    await this.assistantRepository.insert(assistant);
+
+    try {
+      await this.assistantRepository.insert(assistant);
+    } catch (err) {
+      if (err instanceof QueryFailedError) {
+        if (err.driverError.code === 'SQLITE_CONSTRAINT')
+          throw new DuplicateAssistantException(id);
+      }
+
+      throw err;
+    }
+
     return this.findOne(assistant.id);
   }