Merge pull request #193 from import-ai/feat/open_tag

feat(tag): Add tag support when create resource via open api

Author: LucienShui

src/namespace-resources/open.resource.controller.ts

@@ -18,6 +18,7 @@ import { FileInterceptor } from '@nestjs/platform-express';
 import { UserId } from 'omniboxd/decorators/user-id.decorator';
 import { OpenCreateResourceDto } from 'omniboxd/namespace-resources/dto/open.create-resource.dto';
 import { ResourceType } from 'omniboxd/resources/entities/resource.entity';
+import { isEmpty } from 'omniboxd/utils/is-empty';
 
 @Controller('open/api/v1/resources')
 export class OpenResourcesController {
@@ -59,13 +60,23 @@ export class OpenResourcesController {
       resourceData,
     );
 
-    if (!newResource.name || newResource.name.trim() === '') {
-      await this.wizardTaskService.createGenerateTitleTask(
-        userId,
-        apiKey.namespaceId,
-        { resource_id: newResource.id },
-        { text: data.content },
-      );
+    if (!isEmpty(newResource.content?.trim())) {
+      if (isEmpty(newResource.name?.trim())) {
+        await this.wizardTaskService.createGenerateTitleTask(
+          userId,
+          apiKey.namespaceId,
+          { resource_id: newResource.id },
+          { text: data.content },
+        );
+      }
+      if (isEmpty(newResource.tagIds)) {
+        await this.wizardTaskService.createExtractTagsTask(
+          userId,
+          newResource.id,
+          apiKey.namespaceId,
+          newResource.content,
+        );
+      }
     }
 
     return { id: newResource.id, name: newResource.name };

src/tasks/wizard-task.service.ts

@@ -66,7 +66,32 @@ export class WizardTaskService {
     );
   }
 
-  async createExtractTagsTask(parentTask: Task, repo?: Repository<Task>) {
+  async createExtractTagsTask(
+    userId: string,
+    resourceId: string,
+    namespaceId: string,
+    text: string,
+    repo?: Repository<Task>,
+  ) {
+    const lang = await this.getUserLanguage(userId);
+    return this.create(
+      {
+        function: 'extract_tags',
+        input: { text, lang },
+        namespaceId,
+        payload: {
+          resource_id: resourceId,
+        },
+        userId,
+      },
+      repo,
+    );
+  }
+
+  async createExtractTagsTaskFromTask(
+    parentTask: Task,
+    repo?: Repository<Task>,
+  ) {
     const lang = await this.getUserLanguage(parentTask.userId);
     return this.create(
       {

src/wizard/wizard.service.ts

@@ -216,7 +216,7 @@ export class WizardService {
   private async triggerExtractTags(parentTask: Task): Promise<void> {
     try {
       const extractTagsTask =
-        await this.wizardTaskService.createExtractTagsTask(parentTask);
+        await this.wizardTaskService.createExtractTagsTaskFromTask(parentTask);
 
       this.logger.debug(
         `Triggered extract_tags task ${extractTagsTask.id} for parent task ${parentTask.id}`,