update generateProjectRecommendationQuestions input

wren-ui/src/apollo/server/resolvers/askingResolver.ts

@@ -75,11 +75,10 @@ export class AskingResolver {
 
   public async generateProjectRecommendationQuestions(
     _root: any,
-    args: { projectId: number },
+    _args: any,
     ctx: IContext,
   ): Promise<boolean> {
-    const { projectId } = args;
-    await ctx.projectService.generateProjectRecommendationQuestions(projectId);
+    await ctx.projectService.generateProjectRecommendationQuestions();
     return true;
   }
 

wren-ui/src/apollo/server/resolvers/projectResolver.ts

@@ -107,7 +107,7 @@ export class ProjectResolver {
     });
 
     // update project recommendation questions
-    await ctx.projectService.generateProjectRecommendationQuestions(project.id);
+    await ctx.projectService.generateProjectRecommendationQuestions();
     return true;
   }
 
@@ -623,7 +623,7 @@ export class ProjectResolver {
     const { manifest } = await ctx.mdlService.makeCurrentModelMDL();
     const deployRes = await ctx.deployService.deploy(manifest, project.id);
 
-    await ctx.projectService.generateProjectRecommendationQuestions(project.id);
+    await ctx.projectService.generateProjectRecommendationQuestions();
     return deployRes;
   }
 

wren-ui/src/apollo/server/schema.ts

@@ -863,6 +863,6 @@ export const typeDefs = gql`
 
     # Recommendation questions
     generateThreadRecommendationQuestions(threadId: Int!): Boolean!
-    generateProjectRecommendationQuestions(projectId: Int!): Boolean!
+    generateProjectRecommendationQuestions: Boolean!
   }
 `;

wren-ui/src/apollo/server/services/projectService.ts

@@ -58,7 +58,7 @@ export interface IProjectService {
   getProjectRecommendationQuestions: () => Promise<ProjectRecommendationQuestionsResult>;
 
   // recommend questions
-  generateProjectRecommendationQuestions: (projectId: number) => Promise<void>;
+  generateProjectRecommendationQuestions: () => Promise<void>;
 }
 
 export class ProjectService implements IProjectService {
@@ -92,12 +92,10 @@ export class ProjectService implements IProjectService {
       });
   }
 
-  public async generateProjectRecommendationQuestions(
-    projectId: number,
-  ): Promise<void> {
-    const project = await this.getProjectById(projectId);
+  public async generateProjectRecommendationQuestions(): Promise<void> {
+    const project = await this.getCurrentProject();
     if (!project) {
-      throw new Error(`Project ${projectId} not found`);
+      throw new Error(`Project not found`);
     }
     const { manifest } = await this.mdlService.makeCurrentModelMDL();
 
@@ -117,6 +115,10 @@ export class ProjectService implements IProjectService {
       this.projectRecommendQuestionBackgroundTracker.taskKey(updatedProject);
     if (!tasks[taskKey]) {
       this.projectRecommendQuestionBackgroundTracker.addTask(updatedProject);
+    } else {
+      logger.debug(
+        `Generate Project Recommendation Questions Task ${taskKey} already exists, skip adding`,
+      );
     }
   }