Merge pull request #594 from janhq/feat/cortex-client-sdk-compatible

feat: cortex client sdk compatible

Author: louis-jan

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

@@ -1,10 +1,20 @@
-import { Body, Controller, Delete, Get, Param, Post } from '@nestjs/common';
+import {
+  Body,
+  Controller,
+  Delete,
+  Get,
+  Param,
+  Post,
+  UseInterceptors,
+} from '@nestjs/common';
 import { AssistantsUsecases } from '@/usecases/assistants/assistants.usecases';
 import { CreateAssistantDto } from '@/infrastructure/dtos/assistants/create-assistant.dto';
 import { ApiTags } from '@nestjs/swagger';
+import { TransformInterceptor } from '../interceptors/transform.interceptor';
 
 @ApiTags('Assistants')
 @Controller('assistants')
+@UseInterceptors(TransformInterceptor)
 export class AssistantsController {
   constructor(private readonly assistantsService: AssistantsUsecases) {}
 

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

@@ -6,14 +6,17 @@ import {
   Patch,
   Param,
   Delete,
+  UseInterceptors,
 } from '@nestjs/common';
 import { MessagesUsecases } from '@/usecases/messages/messages.usecases';
 import { CreateMessageDto } from '@/infrastructure/dtos/messages/create-message.dto';
 import { UpdateMessageDto } from '@/infrastructure/dtos/messages/update-message.dto';
 import { ApiTags } from '@nestjs/swagger';
+import { TransformInterceptor } from '../interceptors/transform.interceptor';
 
 @ApiTags('Messages')
 @Controller('messages')
+@UseInterceptors(TransformInterceptor)
 export class MessagesController {
   constructor(private readonly messagesUsecases: MessagesUsecases) {}
 

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

@@ -7,16 +7,19 @@ import {
   Param,
   Delete,
   HttpCode,
+  UseInterceptors,
 } from '@nestjs/common';
 import { ModelsUsecases } from '@/usecases/models/models.usecases';
 import { CreateModelDto } from '@/infrastructure/dtos/models/create-model.dto';
 import { UpdateModelDto } from '@/infrastructure/dtos/models/update-model.dto';
 import { ApiResponse, ApiTags } from '@nestjs/swagger';
 import { StartModelSuccessDto } from '@/infrastructure/dtos/models/start-model-success.dto';
 import { ModelSettingParamsDto } from '../dtos/models/model-setting-params.dto';
+import { TransformInterceptor } from '../interceptors/transform.interceptor';
 
 @ApiTags('Models')
 @Controller('models')
+@UseInterceptors(TransformInterceptor)
 export class ModelsController {
   constructor(private readonly modelsUsecases: ModelsUsecases) {}
 

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

@@ -6,14 +6,17 @@ import {
   Patch,
   Param,
   Delete,
+  UseInterceptors,
 } from '@nestjs/common';
 import { ThreadsUsecases } from '@/usecases/threads/threads.usecases';
 import { CreateThreadDto } from '@/infrastructure/dtos/threads/create-thread.dto';
 import { UpdateThreadDto } from '@/infrastructure/dtos/threads/update-thread.dto';
 import { ApiTags } from '@nestjs/swagger';
+import { TransformInterceptor } from '../interceptors/transform.interceptor';
 
 @ApiTags('Threads')
 @Controller('threads')
+@UseInterceptors(TransformInterceptor)
 export class ThreadsController {
   constructor(private readonly threadsService: ThreadsUsecases) {}
 

cortex-js/src/infrastructure/interceptors/transform.interceptor.ts

@@ -0,0 +1,33 @@
+import {
+  CallHandler,
+  ExecutionContext,
+  Injectable,
+  NestInterceptor,
+} from '@nestjs/common';
+import { Observable } from 'rxjs';
+import { map } from 'rxjs/operators';
+
+export interface Response<T> {
+  data: T;
+}
+
+@Injectable()
+export class TransformInterceptor<T>
+  implements NestInterceptor<T, Response<T>>
+{
+  intercept(
+    context: ExecutionContext,
+    next: CallHandler,
+  ): Observable<Response<T>> {
+    return next.handle().pipe(
+      map((data) => {
+        return Array.isArray(data)
+          ? {
+              object: 'list',
+              data,
+            }
+          : data;
+      }),
+    );
+  }
+}