@autobe/agent 0.4.0 → 0.4.2

package/src/analyze/AnalyzeAgent.ts

@@ -6,8 +6,11 @@ import typia from "typia";
 import { AutoBeSystemPromptConstant } from "../constants/AutoBeSystemPromptConstant";
 import { AutoBeContext } from "../context/AutoBeContext";
 import { assertSchemaModel } from "../context/assertSchemaModel";
-import { createReviewerAgent } from "./CreateReviewerAgent";
-import { IPlanning, Planning } from "./Planning";
+import {
+  AutoBeAnalyzeFileSystem,
+  IAutoBeAnalyzeFileSystem,
+} from "./AutoBeAnalyzeFileSystem";
+import { AutoBeAnalyzeReviewer } from "./AutoBeAnalyzeReviewer";
 
 type Filename = string;
 type FileContent = string;
@@ -17,7 +20,7 @@ export class AnalyzeAgent<Model extends ILlmSchema.Model> {
   private readonly fileMap: Record<Filename, FileContent> = {};
 
   constructor(
-    private readonly createReviewerAgentFn: typeof createReviewerAgent,
+    private readonly createReviewerAgentFn: typeof AutoBeAnalyzeReviewer,
     private readonly ctx: AutoBeContext<Model>,
     private readonly pointer: IPointer<{
       files: Record<Filename, FileContent>;
@@ -27,7 +30,7 @@ export class AnalyzeAgent<Model extends ILlmSchema.Model> {
 
     const controller = createController<Model>({
       model: ctx.model,
-      execute: new Planning(this.fileMap),
+      execute: new AutoBeAnalyzeFileSystem(this.fileMap),
       build: async (files: Record<Filename, FileContent>) => {
         this.pointer.value = { files };
       },
@@ -44,12 +47,7 @@ export class AnalyzeAgent<Model extends ILlmSchema.Model> {
               return AutoBeSystemPromptConstant.ANALYZE.replace(
                 "{% Guidelines %}",
                 AutoBeSystemPromptConstant.ANALYZE_GUIDELINE,
-              )
-                .replace(
-                  "{% Example Documentation %}",
-                  AutoBeSystemPromptConstant.ANALYZE_EXAMPLE,
-                )
-                .replace("{% User Locale %}", ctx.config?.locale ?? "en-US");
+              ).replace("{% User Locale %}", ctx.config?.locale ?? "en-US");
             },
             describe: () => {
               return "Answer only 'completion' or 'failure'.";
@@ -98,11 +96,10 @@ export class AnalyzeAgent<Model extends ILlmSchema.Model> {
         return lastMessage.text;
       }
 
-      const currentFiles = this.fileMap;
-
+      const currentFiles = JSON.stringify(this.fileMap);
       const reviewer = this.createReviewerAgentFn(this.ctx, {
         query: content,
-        currentFiles,
+        files: currentFiles,
       });
 
       const [review] = await reviewer.conversate(lastMessage.text);
@@ -135,7 +132,7 @@ export class AnalyzeAgent<Model extends ILlmSchema.Model> {
 
 function createController<Model extends ILlmSchema.Model>(props: {
   model: Model;
-  execute: Planning;
+  execute: AutoBeAnalyzeFileSystem;
   build: (input: Record<Filename, FileContent>) => void;
 }): IAgenticaController.IClass<Model> {
   assertSchemaModel(props.model);
@@ -144,7 +141,7 @@ function createController<Model extends ILlmSchema.Model>(props: {
   ] as unknown as ILlmApplication<Model>;
   return {
     protocol: "class",
-    name: "Planning",
+    name: "AutoBeAnalyzeFileSystem",
     application,
     // execute: props.execute,
     execute: {
@@ -163,16 +160,24 @@ function createController<Model extends ILlmSchema.Model>(props: {
         props.build(props.execute.allFiles());
         return response;
       },
-    } satisfies IPlanning,
+    } satisfies IAutoBeAnalyzeFileSystem,
   };
 }
 
-const claude = typia.llm.application<Planning, "claude", { reference: true }>();
+const claude = typia.llm.application<
+  AutoBeAnalyzeFileSystem,
+  "claude",
+  { reference: true }
+>();
 const collection = {
-  chatgpt: typia.llm.application<Planning, "chatgpt", { reference: true }>(),
+  chatgpt: typia.llm.application<
+    AutoBeAnalyzeFileSystem,
+    "chatgpt",
+    { reference: true }
+  >(),
   claude,
   llama: claude,
   deepseek: claude,
   "3.1": claude,
-  "3.0": typia.llm.application<Planning, "3.0">(),
+  "3.0": typia.llm.application<AutoBeAnalyzeFileSystem, "3.0">(),
 };

package/src/analyze/AutoBeAnalyzeAgent.ts

@@ -0,0 +1,217 @@
+import { IAgenticaController, MicroAgentica } from "@agentica/core";
+import { ILlmApplication, ILlmSchema } from "@samchon/openapi";
+import { IPointer } from "tstl";
+import typia from "typia";
+
+import { AutoBeSystemPromptConstant } from "../constants/AutoBeSystemPromptConstant";
+import { AutoBeContext } from "../context/AutoBeContext";
+import { assertSchemaModel } from "../context/assertSchemaModel";
+import {
+  AutoBeAnalyzeFileSystem,
+  IAutoBeAnalyzeFileSystem,
+} from "./AutoBeAnalyzeFileSystem";
+import { AutoBeAnalyzeReviewer } from "./AutoBeAnalyzeReviewer";
+
+type Filename = string;
+type FileContent = string;
+
+export class AutoBeAnalyzeAgent<Model extends ILlmSchema.Model> {
+  private readonly createAnalyzeAgent: () => MicroAgentica<Model>;
+  private readonly fileMap: Record<Filename, FileContent> = {};
+
+  constructor(
+    private readonly createReviewerAgentFn: typeof AutoBeAnalyzeReviewer,
+    private readonly ctx: AutoBeContext<Model>,
+    private readonly pointer: IPointer<{
+      files: Record<Filename, FileContent>;
+    } | null>,
+    filenames: string[],
+  ) {
+    assertSchemaModel(ctx.model);
+
+    const controller = createController<Model>({
+      model: ctx.model,
+      execute: new AutoBeAnalyzeFileSystem(this.fileMap),
+      build: async (files: Record<Filename, FileContent>) => {
+        this.pointer.value = { files };
+      },
+    });
+
+    this.createAnalyzeAgent = (): MicroAgentica<Model> => {
+      const agent = new MicroAgentica({
+        controllers: [controller],
+        model: ctx.model,
+        vendor: ctx.vendor,
+        config: {
+          locale: ctx.config?.locale,
+          systemPrompt: {
+            describe: () => {
+              return "Answer only 'completion' or 'failure'.";
+            },
+          },
+        },
+        tokenUsage: ctx.usage(),
+        histories: [
+          {
+            type: "systemMessage",
+            text: AutoBeSystemPromptConstant.ANALYZE.replace(
+              "{% User Locale %}",
+              ctx.config?.locale ?? "en-US",
+            ),
+          },
+          {
+            type: "systemMessage",
+            text: [
+              "# Guidelines",
+              "If the user specifies the exact number of pages, please follow it precisely.",
+              AutoBeSystemPromptConstant.ANALYZE_GUIDELINE,
+            ].join("\n"),
+          },
+          {
+            type: "systemMessage",
+            text: [
+              "The following is the name of the entire file.",
+              "Use it to build a table of contents.",
+              filenames.map((filename) => `- ${filename}`),
+              "",
+              "However, do not touch other than the file you have to create.",
+            ].join("\n"),
+          },
+        ],
+      });
+
+      agent.on("request", (event) => {
+        if (event.body.tools) {
+          event.body.tool_choice = "required";
+        }
+      });
+
+      return agent;
+    };
+  }
+
+  /**
+   * Conversate with planner agent
+   *
+   * @param content Conversation from user in this time.
+   * @returns
+   */
+  async conversate(content: string, retry = 3): Promise<string> {
+    if (retry === 0) {
+      return "Abort due to excess retry count";
+    }
+
+    const response = await this.createAnalyzeAgent().conversate(content);
+    const lastMessage = response[response.length - 1]!;
+
+    if ("text" in lastMessage) {
+      this.ctx.dispatch({
+        type: "analyzeWriteDocument",
+        files: this.fileMap,
+        created_at: new Date().toISOString(),
+        step: this.ctx.state().analyze?.step ?? 0,
+      });
+
+      const aborted =
+        lastMessage.type === "describe" &&
+        lastMessage.executes.some((el) => {
+          if (
+            el.protocol === "class" &&
+            el.operation.function.name === "abort"
+          ) {
+            el.arguments;
+            return true;
+          }
+        });
+
+      if (aborted === true) {
+        return lastMessage.text;
+      }
+
+      const reviewer = this.createReviewerAgentFn(this.ctx, {
+        query: content,
+        files: JSON.stringify(this.fileMap),
+      });
+
+      const filenames = Object.keys(this.fileMap).join(",");
+      const command = `Request for review of these files.: ${filenames}`;
+      const [review] = await reviewer.conversate(command);
+
+      if (review) {
+        if (review.type === "assistantMessage") {
+          this.ctx.dispatch({
+            type: "analyzeReview",
+            review: review.text,
+            created_at: new Date().toISOString(),
+            step: this.ctx.state().analyze?.step ?? 0,
+          });
+
+          return this.conversate(
+            JSON.stringify({
+              user_query: content,
+              message: `THIS IS ANSWER OF REVIEW AGENT. FOLLOW THIS INSTRUCTIONS. AND DON\'T REQUEST ANYTHING.`,
+              review: review.text,
+            }),
+            retry--,
+          );
+        }
+      }
+
+      return `COMPLETE WITHOUT REVIEW`;
+    }
+
+    return "COMPLETE";
+  }
+}
+
+function createController<Model extends ILlmSchema.Model>(props: {
+  model: Model;
+  execute: AutoBeAnalyzeFileSystem;
+  build: (input: Record<Filename, FileContent>) => void;
+}): IAgenticaController.IClass<Model> {
+  assertSchemaModel(props.model);
+  const application: ILlmApplication<Model> = collection[
+    props.model
+  ] as unknown as ILlmApplication<Model>;
+  return {
+    protocol: "class",
+    name: "Planning",
+    application,
+    // execute: props.execute,
+    execute: {
+      removeFile: (input) => {
+        const response = props.execute.removeFile(input);
+        props.build(props.execute.allFiles());
+        return response;
+      },
+      abort: (input) => {
+        const response = props.execute.abort(input);
+        props.build(props.execute.allFiles());
+        return response;
+      },
+      createOrUpdateFiles: (input) => {
+        const response = props.execute.createOrUpdateFiles(input);
+        props.build(props.execute.allFiles());
+        return response;
+      },
+    } satisfies IAutoBeAnalyzeFileSystem,
+  };
+}
+
+const claude = typia.llm.application<
+  AutoBeAnalyzeFileSystem,
+  "claude",
+  { reference: true }
+>();
+const collection = {
+  chatgpt: typia.llm.application<
+    AutoBeAnalyzeFileSystem,
+    "chatgpt",
+    { reference: true }
+  >(),
+  claude,
+  llama: claude,
+  deepseek: claude,
+  "3.1": claude,
+  "3.0": typia.llm.application<AutoBeAnalyzeFileSystem, "3.0">(),
+};

package/src/analyze/AutoBeAnalyzeFileSystem.ts

@@ -0,0 +1,81 @@
+import { tags } from "typia";
+
+export interface IFile {
+  /**
+   * Describe briefly why you made this document, and if you have any plans for
+   * the next one.
+   */
+  reason: string;
+
+  /** Filename to generate or overwrite. */
+  filename: `${string}.md`;
+
+  /**
+   * Markdown file content. Only write the content of the file. Do not include
+   * any questions. This should contain only the contents of the file. Do not
+   * write down any questions or appreciation. For example, remove a sentence
+   * such as "Is it okay if we proceed with the table of contents? Please let me
+   * know if there is anything to add or exclude from the table of contents!"
+   */
+  markdown: string;
+}
+
+export interface ICreateOrUpdateInput {
+  /**
+   * Fill in the elements of the array as many files as you want to create.
+   * Overwrite if you point to the name of the file that already exists.
+   *
+   * @title files to create or update
+   */
+  files: Array<IFile> & tags.MinItems<1>;
+}
+
+type Filename = string;
+type FileContent = string;
+
+export interface IAutoBeAnalyzeFileSystem {
+  /**
+   * Generate multiple markdown files. if there is already created files,
+   * overwrite it. Generate several markdown files at once. It is recommended
+   * that you create multiple files at a time.
+   */
+  createOrUpdateFiles(input: ICreateOrUpdateInput): Promise<void>;
+
+  /** Remove markdown file. */
+  removeFile(input: Pick<IFile, "filename">): Promise<void>;
+
+  /**
+   * If you decide that you no longer need any reviews, or if the reviewer
+   * refuses to do so, call abort. This is a function to end document creation
+   * and review, and to respond to users.
+   *
+   * When there is content you are unsure about and need to ask the user a
+   * question, abort the process and ask the user directly. The reason for
+   * aborting should be included as the content of the question.
+   */
+  abort(input: { reason: string }): "OK";
+}
+
+export class AutoBeAnalyzeFileSystem implements IAutoBeAnalyzeFileSystem {
+  constructor(private readonly fileMap: Record<Filename, FileContent> = {}) {}
+  async createOrUpdateFiles(input: {
+    files: Array<IFile> & tags.MinItems<1>;
+  }): Promise<void> {
+    input.files.forEach((file) => {
+      this.fileMap[file.filename] = file.markdown;
+    });
+  }
+
+  async removeFile(input: Pick<IFile, "filename">): Promise<void> {
+    delete this.fileMap[input.filename];
+  }
+
+  abort(_input: { reason: string }): "OK" {
+    return "OK";
+  }
+
+  /** @ignore */
+  allFiles(): Record<string, string> {
+    return this.fileMap;
+  }
+}

package/src/analyze/AutoBeAnalyzeReviewer.ts

@@ -0,0 +1,60 @@
+import { MicroAgentica } from "@agentica/core";
+import { ILlmSchema } from "@samchon/openapi";
+
+import { AutoBeSystemPromptConstant } from "../constants/AutoBeSystemPromptConstant";
+import { AutoBeContext } from "../context/AutoBeContext";
+
+export const AutoBeAnalyzeReviewer = <Model extends ILlmSchema.Model>(
+  ctx: AutoBeContext<Model>,
+  input: AutoBeAnalyzeReviewer.ICreateReviewerAgentInput,
+) => {
+  const agent = new MicroAgentica({
+    model: ctx.model,
+    vendor: ctx.vendor,
+    controllers: [],
+    config: {
+      systemPrompt: {
+        describe: () => {
+          return "Answer only 'completion' or 'failure'.";
+        },
+      },
+      locale: ctx.config?.locale,
+    },
+    histories: [
+      ...ctx
+        .histories()
+        .filter(
+          (el) => el.type === "assistantMessage" || el.type === "userMessage",
+        ),
+      {
+        type: "systemMessage",
+        text: AutoBeSystemPromptConstant.ANALYZE_REVIEWER,
+      },
+      {
+        type: "systemMessage",
+        text: [
+          "Below are all of the files.",
+          "```json",
+          input.files,
+          "```",
+        ].join("\n"),
+      },
+    ],
+    tokenUsage: ctx.usage(),
+  });
+
+  return agent;
+};
+
+export namespace AutoBeAnalyzeReviewer {
+  export interface ICreateReviewerAgentInput {
+    /**
+     * Indicates the initial utterance of the user. Identify the purpose of your
+     * documentation for better review.
+     */
+    query: string;
+
+    /** Total file names */
+    files: string;
+  }
+}