@autobe/agent 0.3.11 → 0.3.12

package/src/analyze/AnalyzeAgent.ts

@@ -13,7 +13,7 @@ type Filename = string;
 type FileContent = string;
 
 export class AnalyzeAgent<Model extends ILlmSchema.Model> {
-  private readonly agent: MicroAgentica<Model>;
+  private readonly createInnerAgent: () => MicroAgentica<Model>;
   private readonly fileMap: Record<Filename, FileContent> = {};
 
   constructor(
@@ -32,27 +32,36 @@ export class AnalyzeAgent<Model extends ILlmSchema.Model> {
         this.pointer.value = { files };
       },
     });
-    this.agent = new MicroAgentica({
-      controllers: [controller],
-      model: ctx.model,
-      vendor: ctx.vendor,
-      config: {
-        systemPrompt: {
-          common: () => {
-            return AutoBeSystemPromptConstant.ANALYZE.replace(
-              "{% Guidelines %}",
-              AutoBeSystemPromptConstant.ANALYZE_GUIDELINE,
-            )
-              .replace(
-                "{% Example Documentation %}",
-                AutoBeSystemPromptConstant.ANALYZE_EXAMPLE,
+
+    this.createInnerAgent = (): MicroAgentica<Model> => {
+      const agent = new MicroAgentica({
+        controllers: [controller],
+        model: ctx.model,
+        vendor: ctx.vendor,
+        config: {
+          systemPrompt: {
+            common: () => {
+              return AutoBeSystemPromptConstant.ANALYZE.replace(
+                "{% Guidelines %}",
+                AutoBeSystemPromptConstant.ANALYZE_GUIDELINE,
               )
-              .replace("{% User Locale %}", ctx.config?.locale ?? "en-US");
+                .replace(
+                  "{% Example Documentation %}",
+                  AutoBeSystemPromptConstant.ANALYZE_EXAMPLE,
+                )
+                .replace("{% User Locale %}", ctx.config?.locale ?? "en-US");
+            },
+            describe: () => {
+              return "Answer only 'completion' or 'failure'.";
+            },
           },
         },
-      },
-      tokenUsage: ctx.usage(),
-    });
+        tokenUsage: ctx.usage(),
+        histories: [],
+      });
+
+      return agent;
+    };
   }
 
   /**
@@ -62,7 +71,7 @@ export class AnalyzeAgent<Model extends ILlmSchema.Model> {
    * @returns
    */
   async conversate(content: string): Promise<string> {
-    const response = await this.agent.conversate(content);
+    const response = await this.createInnerAgent().conversate(content);
     const lastMessage = response[response.length - 1]!;
 
     if ("text" in lastMessage) {
@@ -116,6 +125,8 @@ export class AnalyzeAgent<Model extends ILlmSchema.Model> {
           );
         }
       }
+
+      return `If the document is not 1,000 characters, please fill it out in more abundance, and if it exceeds 1,000 characters, please fill out the next document. If you don't have the next document, you can exit now.`;
     }
 
     return "COMPLETE";
@@ -147,8 +158,8 @@ function createController<Model extends ILlmSchema.Model>(props: {
         props.build(props.execute.allFiles());
         return response;
       },
-      createOrUpdateFile: (input) => {
-        const response = props.execute.createOrUpdateFile(input);
+      createOrUpdateFiles: (input) => {
+        const response = props.execute.createOrUpdateFiles(input);
         props.build(props.execute.allFiles());
         return response;
       },

package/src/analyze/CreateReviewerAgent.ts

@@ -27,7 +27,7 @@ export const createReviewerAgent = <Model extends ILlmSchema.Model>(
     ),
   );
 
-  return new MicroAgentica({
+  const agent = new MicroAgentica({
     model: ctx.model,
     vendor: ctx.vendor,
     controllers: [],
@@ -70,17 +70,13 @@ export const createReviewerAgent = <Model extends ILlmSchema.Model>(
             "At this time, the document should be newly created with the name attached to the link, not modified.",
             "<CurrentFiles>",
             JSON.stringify(
-              Object.entries(input.currentFiles)
-                .slice(-2)
-                .map(([filename, content]) => {
-                  return {
-                    filename,
-                    content,
-                    content_length: content.length,
-                  };
-                }),
-              null,
-              2,
+              Object.entries(input.currentFiles).map(([filename, content]) => {
+                return {
+                  filename,
+                  content,
+                  content_length: content.length,
+                };
+              }),
             ),
             "</CurrentFiles>",
             "",
@@ -95,20 +91,24 @@ export const createReviewerAgent = <Model extends ILlmSchema.Model>(
             `</Linked Files>`,
             "",
             "Write a long document, but keep your answer short.",
-            "The planner agent can only create and modify one document at a time, so do not ask to create or modify multiple documents at a time.",
             "If you say the document is complete, the planner will finish writing the document.",
             "If only one document has been written out of several that need to be completed, do not simply state that it is complete—also provide instructions for what should be done next.",
             "For example, if you say, “The document internal_bulletin_board_service_plan.md has already been written with over 1,000 characters. Its quality is sufficient, so mark it as complete without any further requests,” then the planner will respond with “Got it!” and stop writing—even if there are still remaining documents.",
             "Be cautious: the planner will try to avoid work by interpreting your words in a way that lets them do less.",
             "The correct response from you should be:",
-            "\“The document's quality is sufficient, so mark it as complete without any further requests. Now, proceed to write the next document immediately.\”",
+            "\“The document's quality is sufficient, so mark it as complete without any further requests. Now, proceed to write the next documents immediately.\”",
             "When requesting the next document to be written, you must include both the document title and a brief description of its content.",
           ].join("\n");
         },
+        describe: () => {
+          return "Answer only 'completion' or 'failure'.";
+        },
       },
     },
     tokenUsage: ctx.usage(),
   });
+
+  return agent;
 };
 
 export interface ICreateReviewerAgentInput {

package/src/analyze/Planning.ts

@@ -3,26 +3,27 @@ type FileContent = string;
 
 export interface IPlanning {
   /**
-   * Generate markdown file. if there is already created file, overwrite it.
-   *
-   * @param input.reason Describe briefly why you made this document, and if you
-   *   have any plans for the next one.
-   * @param input.filename Filename to generate or overwrite.
-   * @param input.markdown Markdown file content. Only write the content of the
-   *   file. Do not include any questions.
+   * Generate multiple markdown files. if there is already created files,
+   * overwrite it. Generate several markdown files at once.
    */
-  createOrUpdateFile(input: {
-    reason: string;
-    filename: `${string}.md`;
-    markdown: string;
-  }): Promise<void>;
+  createOrUpdateFiles(input: {
+    files: Array<{
+      /**
+       * Describe briefly why you made this document, and if you have any plans
+       * for the next one.
+       */
+      reason: string;
 
-  /**
-   * Read markdown file content.
-   *
-   * @param input.filename Filename to read.
-   */
-  // readFile(input: { filename: `${string}.md` }): Promise<string>;
+      /** Filename to generate or overwrite. */
+      filename: `${string}.md`;
+
+      /**
+       * Markdown file content. Only write the content of the file. Do not
+       * include any questions.
+       */
+      markdown: string;
+    }>;
+  }): Promise<void>;
 
   /**
    * Remove markdown file.
@@ -48,18 +49,18 @@ export interface IPlanning {
 export class Planning implements IPlanning {
   constructor(private readonly fileMap: Record<Filename, FileContent> = {}) {}
 
-  async createOrUpdateFile(input: {
-    reason: string;
-    filename: `${string}.md`;
-    markdown: string;
+  async createOrUpdateFiles(input: {
+    files: Array<{
+      reason: string;
+      filename: `${string}.md`;
+      markdown: string;
+    }>;
   }): Promise<void> {
-    this.fileMap[input.filename] = input.markdown;
+    input.files.forEach((file) => {
+      this.fileMap[file.filename] = file.markdown;
+    });
   }
 
-  // async readFile(input: { filename: `${string}.md` }): Promise<string> {
-  //   return this.fileMap[input.filename];
-  // }
-
   async removeFile(input: { filename: `${string}.md` }): Promise<void> {
     delete this.fileMap[input.filename];
   }