feat: add assistant skills

marian2js · marian2js · commit 35653469f6e0 · 2023-12-23T17:30:20.000-03:00
diff --git a/apps/api/src/chat/chat.module.ts b/apps/api/src/chat/chat.module.ts
@@ -17,16 +17,19 @@ import { WorkflowTriggersModule } from '../workflow-triggers/workflow-triggers.m
 import { WorkflowsModule } from '../workflows/workflows.module'
 import { ContactSubscriptionController } from './controllers/contact-subscription.controller'
 import { Assistant, AssistantAuthorizer } from './entities/assistant'
+import { AssistantSkill, AssistantSkillAuthorizer } from './entities/assistant-skill'
 import { Campaign, CampaignAuthorizer } from './entities/campaign'
 import { CampaignMessage } from './entities/campaign-message'
 import { Contact, ContactAuthorizer } from './entities/contact'
 import { Menu, MenuAuthorizer } from './entities/menu'
 import { Order, OrderAuthorizer } from './entities/order'
+import { AssistantSkillResolver } from './resolvers/assistant-skill.resolver'
 import { AssistantResolver } from './resolvers/assistant.resolver'
 import { CampaignResolver } from './resolvers/campaign.resolver'
 import { ContactResolver } from './resolvers/contact.resolver'
 import { MenuResolver } from './resolvers/menu.resolver'
 import { OrderResolver } from './resolvers/order.resolver'
+import { AssistantSkillService } from './services/assistant-skill.service'
 import { AssistantService } from './services/assistant.service'
 import { BroadcastConsumer } from './services/broadcast.consumer'
 import { CampaignMessageService } from './services/campaign-message.service'
@@ -62,6 +65,10 @@ import { OrderService } from './services/order.service'
       imports: [NestjsQueryTypegooseModule.forFeature([Assistant])],
       dtos: [{ DTOClass: Assistant }],
     }),
+    NestjsQueryGraphQLModule.forFeature({
+      imports: [NestjsQueryTypegooseModule.forFeature([AssistantSkill])],
+      dtos: [{ DTOClass: AssistantSkill }],
+    }),
     BullModule.registerQueue({
       name: 'chatbotMessage',
       settings: {
@@ -122,6 +129,7 @@ import { OrderService } from './services/order.service'
     MenuResolver,
     OrderResolver,
     AssistantResolver,
+    AssistantSkillResolver,
 
     // Services
     ContactService,
@@ -130,13 +138,15 @@ import { OrderService } from './services/order.service'
     MenuService,
     OrderService,
     AssistantService,
+    AssistantSkillService,
 
     // Authorizers
     ContactAuthorizer,
     CampaignAuthorizer,
     MenuAuthorizer,
     OrderAuthorizer,
     AssistantAuthorizer,
+    AssistantSkillAuthorizer,
 
     // Consumers
     ChatbotConsumer,
diff --git a/apps/api/src/chat/entities/assistant-skill.ts b/apps/api/src/chat/entities/assistant-skill.ts
@@ -0,0 +1,54 @@
+import { BaseEntity } from '@app/common/base/base-entity'
+import { OwnedAuthorizer } from '@app/common/base/owned.authorizer'
+import { OwnedEntity } from '@app/common/decorators/owned-entity.decorator'
+import { Reference } from '@app/common/typings/mongodb'
+import { Injectable } from '@nestjs/common'
+import { Field, ID, InputType, ObjectType } from '@nestjs/graphql'
+import { Authorize, FilterableField } from '@ptc-org/nestjs-query-graphql'
+import { prop } from '@typegoose/typegoose'
+import { GraphQLJSONObject } from 'graphql-type-json'
+import { User } from '../../users/entities/user'
+import { Assistant } from './assistant'
+
+@Injectable()
+export class AssistantSkillAuthorizer extends OwnedAuthorizer<AssistantSkill> {}
+
+@ObjectType()
+@OwnedEntity()
+@Authorize<AssistantSkill>(AssistantSkillAuthorizer)
+export class AssistantSkill extends BaseEntity {
+  @prop({ ref: User, required: true })
+  readonly owner!: Reference<User>
+
+  @FilterableField(() => ID)
+  @prop({ ref: 'Assistant', required: true })
+  readonly assistant!: Reference<Assistant>
+
+  @Field()
+  @prop({ required: true })
+  name: string
+
+  @Field(() => GraphQLJSONObject)
+  inputs: object
+}
+
+@InputType()
+export class CreateAssistantSkillInput {
+  @Field(() => ID)
+  assistant: Reference<Assistant>
+
+  @Field()
+  name: string
+
+  @Field(() => GraphQLJSONObject)
+  inputs: object
+}
+
+@InputType()
+export class UpdateAssistantSkillInput {
+  @Field({ nullable: true })
+  name: string
+
+  @Field(() => GraphQLJSONObject, { nullable: true })
+  inputs?: object
+}
diff --git a/apps/api/src/chat/resolvers/assistant-skill.resolver.ts b/apps/api/src/chat/resolvers/assistant-skill.resolver.ts
@@ -0,0 +1,23 @@
+import { BaseResolver } from '@app/common/base/base.resolver'
+import { UseGuards, UseInterceptors } from '@nestjs/common'
+import { Resolver } from '@nestjs/graphql'
+import { Authorizer, AuthorizerInterceptor, InjectAuthorizer } from '@ptc-org/nestjs-query-graphql'
+import { GraphqlGuard } from '../../auth/guards/graphql.guard'
+import { AssistantSkill, CreateAssistantSkillInput, UpdateAssistantSkillInput } from '../entities/assistant-skill'
+import { AssistantSkillService } from '../services/assistant-skill.service'
+
+@Resolver(() => AssistantSkill)
+@UseGuards(GraphqlGuard)
+@UseInterceptors(AuthorizerInterceptor(AssistantSkill))
+export class AssistantSkillResolver extends BaseResolver(AssistantSkill, {
+  CreateDTOClass: CreateAssistantSkillInput,
+  UpdateDTOClass: UpdateAssistantSkillInput,
+  guards: [GraphqlGuard],
+}) {
+  constructor(
+    protected assistantSkillService: AssistantSkillService,
+    @InjectAuthorizer(AssistantSkill) readonly authorizer: Authorizer<AssistantSkill>,
+  ) {
+    super(assistantSkillService)
+  }
+}
diff --git a/apps/api/src/chat/services/assistant-skill.service.ts b/apps/api/src/chat/services/assistant-skill.service.ts
@@ -0,0 +1,82 @@
+import { BaseService } from '@app/common/base/base.service'
+import { InjectQueue } from '@nestjs/bull'
+import { BadRequestException, Injectable, Logger, NotFoundException } from '@nestjs/common'
+import { DeleteOneOptions, UpdateOneOptions } from '@ptc-org/nestjs-query-core'
+import { ReturnModelType } from '@typegoose/typegoose'
+import { Queue } from 'bull'
+import { UpdateResult } from 'mongodb'
+import { FilterQuery, UpdateQuery } from 'mongoose'
+import { InjectModel } from 'nestjs-typegoose'
+import { AssistantSkill } from '../entities/assistant-skill'
+import { AssistantService } from './assistant.service'
+
+@Injectable()
+export class AssistantSkillService extends BaseService<AssistantSkill> {
+  protected readonly logger = new Logger(AssistantSkillService.name)
+  static instance: AssistantSkillService
+
+  constructor(
+    @InjectModel(AssistantSkill) protected readonly model: ReturnModelType<typeof AssistantSkill>,
+    @InjectQueue('assistants') private assistantsQueue: Queue,
+    private assistantService: AssistantService,
+  ) {
+    super(model)
+    AssistantSkillService.instance = this
+  }
+
+  async createOne(record: Partial<AssistantSkill>): Promise<AssistantSkill> {
+    if (!record.assistant || !record.owner) {
+      throw new BadRequestException()
+    }
+
+    // Verify assistant exist and the user has access to it
+    const assistant = await this.assistantService.findById(record.assistant?.toString())
+    if (!assistant?.owner || assistant.owner.toString() !== record.owner.toString()) {
+      throw new NotFoundException(`Assistant ${record.assistant} not found`)
+    }
+
+    return super.createOne(record)
+  }
+
+  async afterCreateOne(assistantSkill: AssistantSkill) {
+    this.assistantsQueue.add({
+      type: 'skill-created',
+      id: assistantSkill._id,
+    })
+  }
+
+  async updateOne(
+    id: string,
+    update: Partial<AssistantSkill>,
+    opts?: UpdateOneOptions<AssistantSkill> | undefined,
+  ): Promise<AssistantSkill> {
+    this.assistantsQueue.add({
+      type: 'skill-updated',
+      id,
+    })
+    return super.updateOne(id, update, opts)
+  }
+
+  updateOneNative(
+    conditions: FilterQuery<new () => AssistantSkill>,
+    query: UpdateQuery<new () => AssistantSkill>,
+  ): Promise<UpdateResult> {
+    this.assistantsQueue.add({
+      type: 'skill-updated',
+      id: conditions._id,
+    })
+    return super.updateOneNative(conditions, query)
+  }
+
+  async deleteOne(id: string, opts?: DeleteOneOptions<AssistantSkill> | undefined): Promise<AssistantSkill> {
+    const assistantSkill = await this.findById(id)
+    if (assistantSkill) {
+      this.assistantsQueue.add({
+        type: 'skill-deleted',
+        id,
+        assistantId: assistantSkill.assistant,
+      })
+    }
+    return super.deleteOne(id, opts)
+  }
+}
diff --git a/generated/graphql.ts b/generated/graphql.ts
@@ -173,6 +173,12 @@ export enum AssistantSortFields {
     createdAt = "createdAt"
 }
 
+export enum AssistantSkillSortFields {
+    id = "id",
+    createdAt = "createdAt",
+    assistant = "assistant"
+}
+
 export enum UserDatabaseSortFields {
     id = "id",
     createdAt = "createdAt"
@@ -592,6 +598,20 @@ export interface AssistantSort {
     nulls?: Nullable<SortNulls>;
 }
 
+export interface AssistantSkillFilter {
+    and?: Nullable<AssistantSkillFilter[]>;
+    or?: Nullable<AssistantSkillFilter[]>;
+    id?: Nullable<IDFilterComparison>;
+    createdAt?: Nullable<DateFieldComparison>;
+    assistant?: Nullable<IDFilterComparison>;
+}
+
+export interface AssistantSkillSort {
+    field: AssistantSkillSortFields;
+    direction: SortDirection;
+    nulls?: Nullable<SortNulls>;
+}
+
 export interface UserDatabaseFilter {
     and?: Nullable<UserDatabaseFilter[]>;
     or?: Nullable<UserDatabaseFilter[]>;
@@ -921,6 +941,30 @@ export interface DeleteOneAssistantInput {
     id: string;
 }
 
+export interface CreateOneAssistantSkillInput {
+    assistantSkill: CreateAssistantSkillInput;
+}
+
+export interface CreateAssistantSkillInput {
+    assistant: string;
+    name: string;
+    inputs: JSONObject;
+}
+
+export interface UpdateOneAssistantSkillInput {
+    id: string;
+    update: UpdateAssistantSkillInput;
+}
+
+export interface UpdateAssistantSkillInput {
+    name?: Nullable<string>;
+    inputs?: Nullable<JSONObject>;
+}
+
+export interface DeleteOneAssistantSkillInput {
+    id: string;
+}
+
 export interface CreateOneUserDatabaseInput {
     userDatabase: CreateUserDatabase;
 }
@@ -1425,6 +1469,14 @@ export interface Assistant {
     enabled?: Nullable<boolean>;
 }
 
+export interface AssistantSkill {
+    id: string;
+    createdAt: DateTime;
+    assistant: string;
+    name: string;
+    inputs: JSONObject;
+}
+
 export interface Campaign {
     id: string;
     createdAt: DateTime;
@@ -1440,6 +1492,24 @@ export interface Campaign {
     error?: Nullable<string>;
 }
 
+export interface AssistantSkillDeleteResponse {
+    id?: Nullable<string>;
+    createdAt?: Nullable<DateTime>;
+    assistant?: Nullable<string>;
+    name?: Nullable<string>;
+    inputs?: Nullable<JSONObject>;
+}
+
+export interface AssistantSkillEdge {
+    node: AssistantSkill;
+    cursor: ConnectionCursor;
+}
+
+export interface AssistantSkillConnection {
+    pageInfo: PageInfo;
+    edges: AssistantSkillEdge[];
+}
+
 export interface AssistantDeleteResponse {
     id?: Nullable<string>;
     createdAt?: Nullable<DateTime>;
@@ -1752,6 +1822,8 @@ export interface IQuery {
     orderSummary(from: DateTime, to: DateTime): OrderSummary | Promise<OrderSummary>;
     assistant(id: string): Assistant | Promise<Assistant>;
     assistants(paging?: Nullable<CursorPaging>, filter?: Nullable<AssistantFilter>, sorting?: Nullable<AssistantSort[]>): AssistantConnection | Promise<AssistantConnection>;
+    assistantSkill(id: string): AssistantSkill | Promise<AssistantSkill>;
+    assistantSkills(paging?: Nullable<CursorPaging>, filter?: Nullable<AssistantSkillFilter>, sorting?: Nullable<AssistantSkillSort[]>): AssistantSkillConnection | Promise<AssistantSkillConnection>;
     contractSchema(chainId: number, address: string, type: string): ContractSchema | Promise<ContractSchema>;
     asyncSchemas(integrationId: string, accountCredentialId: string, names: string[], inputs?: Nullable<JSONObject>, integrationTriggerId?: Nullable<string>, integrationActionId?: Nullable<string>): AsyncSchema | Promise<AsyncSchema>;
     manyAsyncSchemas(asyncSchemaInputs: JSONObject[]): AsyncSchema | Promise<AsyncSchema>;
@@ -1814,6 +1886,9 @@ export interface IMutation {
     createOneAssistant(input: CreateOneAssistantInput): Assistant | Promise<Assistant>;
     updateOneAssistant(input: UpdateOneAssistantInput): Assistant | Promise<Assistant>;
     deleteOneAssistant(input: DeleteOneAssistantInput): AssistantDeleteResponse | Promise<AssistantDeleteResponse>;
+    createOneAssistantSkill(input: CreateOneAssistantSkillInput): AssistantSkill | Promise<AssistantSkill>;
+    updateOneAssistantSkill(input: UpdateOneAssistantSkillInput): AssistantSkill | Promise<AssistantSkill>;
+    deleteOneAssistantSkill(input: DeleteOneAssistantSkillInput): AssistantSkillDeleteResponse | Promise<AssistantSkillDeleteResponse>;
     createOneUserDatabase(input: CreateOneUserDatabaseInput): UserDatabase | Promise<UserDatabase>;
     updateOneUserDatabase(input: UpdateOneUserDatabaseInput): UserDatabase | Promise<UserDatabase>;
     deleteOneUserDatabase(input: DeleteOneUserDatabaseInput): UserDatabaseDeleteResponse | Promise<UserDatabaseDeleteResponse>;
diff --git a/generated/schema.graphql b/generated/schema.graphql
