@autobe/agent 0.0.0 → 0.3.6

package/package.json

@@ -1,10 +1,17 @@
 {
   "name": "@autobe/agent",
-  "version": "0.0.0",
+  "version": "0.3.6",
   "description": "AI backend server code generator",
   "main": "lib/index.js",
   "author": "Wrtn Technologies",
   "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/wrtnlabs/autobe"
+  },
+  "bugs": {
+    "url": "https://github.com/wrtnlabs/autobe/issues"
+  },
   "files": [
     "lib",
     "src",
@@ -13,19 +20,31 @@
     "README.md"
   ],
   "dependencies": {
-    "@agentica/core": "^0.25.0",
-    "@samchon/openapi": "^4.3.0",
+    "@agentica/core": "^0.27.2",
+    "@rollup/plugin-json": "^6.1.0",
+    "@prisma/internals": "^6.7.0",
+    "@samchon/openapi": "^4.3.3",
+    "@types/node": "^22.15.18",
+    "chalk": "4.1.2",
     "openai": "^4.80.0",
-    "typia": "^9.3.0",
-    "@autobe/interface": "^0.0.0"
+    "tstl": "^3.0.0",
+    "typia": "^9.3.1",
+    "uuid": "^11.1.0",
+    "@autobe/interface": "^0.3.6"
   },
   "devDependencies": {
     "@rollup/plugin-terser": "^0.4.4",
     "@rollup/plugin-typescript": "^12.1.2",
+    "@types/node": "^22.15.19",
+    "@types/uuid": "^10.0.0",
+    "chalk": "4.1.2",
     "rimraf": "^6.0.1",
     "rollup": "^4.40.1",
+    "ts-node": "^10.9.2",
     "ts-patch": "^3.3.0",
-    "typescript": "~5.8.3"
+    "typescript": "~5.8.3",
+    "@autobe/compiler": "^0.3.6",
+    "@autobe/filesystem": "^0.3.6"
   },
   "keywords": [
     "ai",
@@ -60,8 +79,10 @@
     "access": "public"
   },
   "scripts": {
-    "build": "rimraf lib && rollup -c && tsc",
-    "dev": "rimraf lib && tsc --watch"
+    "build": "rimraf lib && pnpm run build:prompt && rollup -c && tsc",
+    "build:examples": "ts-node build/examples.ts",
+    "build:prompt": "ts-node build/prompt.ts",
+    "dev": "rimraf lib && pnpm run build:prompt && tsc --watch"
   },
   "typings": "lib/index.d.ts"
 }

package/src/AutoBeAgent.ts

@@ -1,101 +1,227 @@
-import { MicroAgentica } from "@agentica/core";
-import { AutoBeHistory, AutoBeUserMessageContent } from "@autobe/interface";
-import { ILlmSchema } from "@samchon/openapi";
-
-import { AutoBeContext } from "./context/AutoBeContext";
-import { AutoBeState } from "./context/AutoBeState";
-import { AutoBeTokenUsage } from "./context/AutoBeTokenUsage";
-import { createAutoBeController } from "./internal/createAutoBeApplication";
-import { createAutoBeState } from "./internal/createAutoBeState";
-import { transformAgenticaHistory } from "./internal/transformAgenticaHistory";
-import { IAutoBeProps } from "./structures/IAutoBeProps";
-
-export class AutoBeAgent<Model extends ILlmSchema.Model> {
-  private readonly agentica_: MicroAgentica<Model>;
-  private readonly histories_: AutoBeHistory[];
-  private readonly context_: AutoBeContext<Model>;
-  private readonly state_: AutoBeState;
-
-  /* -----------------------------------------------------------
-    CONSTRUCTOR
-  ----------------------------------------------------------- */
-  /**
-   * Initializer constructor.
-   *
-   * @param props Properties to construct the agent
-   */
-  public constructor(private readonly props: IAutoBeProps<Model>) {
-    this.histories_ = props.histories?.slice() ?? [];
-    this.state_ = createAutoBeState(this.histories_);
-    this.context_ = {
-      model: props.model,
-      vendor: props.vendor,
-      compiler: props.compiler,
-      histories: this.histories_,
-      usage: () => this.agentica_.getTokenUsage(),
-      state: () => this.state_,
-    };
-
-    this.agentica_ = new MicroAgentica({
-      model: props.model,
-      vendor: props.vendor,
-      config: {
-        ...(props.config ?? {}),
-        executor: {
-          describe: null,
-        },
-      },
-      controllers: [
-        createAutoBeController({
-          model: props.model,
-          context: this.context_,
-        }),
-      ],
-    });
-    this.agentica_.getHistories().push(
-      ...this.histories_
-        .map((history) =>
-          transformAgenticaHistory({
-            operations: this.agentica_.getOperations(),
-            history,
-          }),
-        )
-        .filter((h) => h !== null),
-    );
-  }
-
-  /**
-   * @internal
-   */
-  public clone(): AutoBeAgent<Model> {
-    return new AutoBeAgent<Model>({
-      ...this.props,
-      histories: this.histories_.slice(),
-    });
-  }
-
-  /* -----------------------------------------------------------
-    ACCESSORS
-  ----------------------------------------------------------- */
-  public async conversate(
-    content: string | AutoBeUserMessageContent | AutoBeUserMessageContent[],
-  ): Promise<AutoBeHistory[]> {
-    content;
-    return [];
-  }
-
-  public getHistories(): AutoBeHistory[] {
-    return this.histories_;
-  }
-
-  public getTokenUsage(): AutoBeTokenUsage {
-    return this.agentica_.getTokenUsage();
-  }
-
-  /**
-   * @internal
-   */
-  public getContext(): AutoBeContext<Model> {
-    return this.context_;
-  }
-}
+import { MicroAgentica } from "@agentica/core";
+import {
+  AutoBeAssistantMessageHistory,
+  AutoBeEvent,
+  AutoBeHistory,
+  AutoBeUserMessageContent,
+  AutoBeUserMessageHistory,
+} from "@autobe/interface";
+import { ILlmSchema } from "@samchon/openapi";
+import { v4 } from "uuid";
+
+import { AutoBeContext } from "./context/AutoBeContext";
+import { AutoBeState } from "./context/AutoBeState";
+import { AutoBeTokenUsage } from "./context/AutoBeTokenUsage";
+import { createAgenticaHistory } from "./factory/createAgenticaHistory";
+import { createAutoBeController } from "./factory/createAutoBeApplication";
+import { createAutoBeState } from "./factory/createAutoBeState";
+import { transformFacadeStateMessage } from "./orchestrate/facade/transformFacadeStateMessage";
+import { IAutoBeProps } from "./structures/IAutoBeProps";
+import { emplaceMap } from "./utils/emplaceMap";
+
+export class AutoBeAgent<Model extends ILlmSchema.Model> {
+  private readonly agentica_: MicroAgentica<Model>;
+  private readonly histories_: AutoBeHistory[];
+  private readonly context_: AutoBeContext<Model>;
+
+  private readonly state_: AutoBeState;
+  private readonly listeners_: Map<
+    string,
+    Set<(event: AutoBeEvent) => Promise<void> | void>
+  >;
+
+  /* -----------------------------------------------------------
+    CONSTRUCTOR
+  ----------------------------------------------------------- */
+  /**
+   * Initializer constructor.
+   *
+   * @param props Properties to construct the agent
+   */
+  public constructor(private readonly props: IAutoBeProps<Model>) {
+    this.histories_ = props.histories?.slice() ?? [];
+    this.state_ = createAutoBeState(this.histories_);
+    this.context_ = {
+      model: props.model,
+      vendor: props.vendor,
+      config: props.config,
+      compiler: props.compiler,
+      histories: () => this.histories_,
+      state: () => this.state_,
+      usage: () => this.agentica_.getTokenUsage(),
+      files: () => this.getFiles(),
+      dispatch: (event) => {
+        this.dispatch(event).catch(() => {});
+      },
+    };
+    this.listeners_ = new Map();
+
+    this.agentica_ = new MicroAgentica({
+      model: props.model,
+      vendor: props.vendor,
+      config: {
+        ...(props.config ?? {}),
+        executor: {
+          describe: null,
+        },
+        systemPrompt: {
+          execute: () => transformFacadeStateMessage(this.state_),
+        },
+      },
+      controllers: [
+        createAutoBeController({
+          model: props.model,
+          context: this.context_,
+        }),
+      ],
+    });
+    this.agentica_.getHistories().push(
+      ...this.histories_
+        .map((history) =>
+          createAgenticaHistory({
+            operations: this.agentica_.getOperations(),
+            history,
+          }),
+        )
+        .filter((h) => h !== null),
+    );
+    this.agentica_.on("assistantMessage", async (message) => {
+      const start = new Date();
+      const history: AutoBeAssistantMessageHistory = {
+        id: v4(),
+        type: "assistantMessage",
+        text: await message.join(),
+        created_at: start.toISOString(),
+        completed_at: new Date().toISOString(),
+      };
+      this.histories_.push(history);
+      this.dispatch({
+        type: "assistantMessage",
+        text: history.text,
+        created_at: history.created_at,
+      }).catch(() => {});
+    });
+  }
+
+  /** @internal */
+  public clone(): AutoBeAgent<Model> {
+    return new AutoBeAgent<Model>({
+      ...this.props,
+      histories: this.histories_.slice(),
+    });
+  }
+
+  public on<Type extends AutoBeEvent.Type>(
+    type: Type,
+    listener: (event: AutoBeEvent.Mapper[Type]) => Promise<void> | void,
+  ): this {
+    emplaceMap(this.listeners_, type, () => new Set()).add(
+      listener as (event: AutoBeEvent) => any,
+    );
+    return this;
+  }
+
+  public off<Type extends AutoBeEvent.Type>(
+    type: Type,
+    listener: (event: AutoBeEvent.Mapper[Type]) => Promise<void> | void,
+  ): this {
+    const set = this.listeners_.get(type);
+    if (set === undefined) return this;
+
+    set.delete(listener as (event: AutoBeEvent) => any);
+    if (set.size === 0) this.listeners_.delete(type);
+    return this;
+  }
+
+  /* -----------------------------------------------------------
+    ACCESSORS
+  ----------------------------------------------------------- */
+  public async conversate(
+    content: string | AutoBeUserMessageContent | AutoBeUserMessageContent[],
+  ): Promise<AutoBeHistory[]> {
+    const index: number = this.histories_.length;
+    const userMessageHistory: AutoBeUserMessageHistory = {
+      id: v4(),
+      type: "userMessage",
+      contents:
+        typeof content === "string"
+          ? [
+              {
+                type: "text",
+                text: content,
+              },
+            ]
+          : Array.isArray(content)
+            ? content
+            : [content],
+      created_at: new Date().toISOString(),
+    };
+    this.histories_.push(userMessageHistory);
+    this.dispatch(userMessageHistory).catch(() => {});
+
+    await this.agentica_.conversate(content);
+    return this.histories_.slice(index);
+  }
+
+  public getFiles(): Record<string, string> {
+    return {
+      ...Object.fromEntries(
+        this.state_.analyze
+          ? Object.entries(this.state_.analyze.files).map(([key, value]) => [
+              `docs/analysis/${key.split("/").at(-1)}`,
+              value,
+            ])
+          : [],
+      ),
+      ...Object.fromEntries(
+        this.state_.prisma?.result.type === "success"
+          ? [
+              ...Object.entries(this.state_.prisma.result.schemas).map(
+                ([key, value]) => [
+                  `prisma/schema/${key.split("/").at(-1)}`,
+                  value,
+                ],
+              ),
+              ["docs/ERD.md", this.state_.prisma.result.document],
+            ]
+          : [],
+      ),
+      ...(this.state_.interface ? this.state_.interface.files : {}),
+      ...(this.state_.test?.result.type === "success"
+        ? this.state_.test.files
+        : {}),
+      ...(this.state_.realize?.result.type === "success"
+        ? this.state_.realize.files
+        : {}),
+    };
+  }
+
+  public getHistories(): AutoBeHistory[] {
+    return this.histories_;
+  }
+
+  public getTokenUsage(): AutoBeTokenUsage {
+    return this.agentica_.getTokenUsage();
+  }
+
+  /* -----------------------------------------------------------
+    CONTEXTS
+  ----------------------------------------------------------- */
+  /** @internal */
+  public getContext(): AutoBeContext<Model> {
+    return this.context_;
+  }
+
+  /** @internal */
+  private async dispatch(event: AutoBeEvent): Promise<void> {
+    const set = this.listeners_.get(event.type);
+    if (set === undefined) return;
+    await Promise.all(
+      Array.from(set).map(async (listener) => {
+        try {
+          await listener(event);
+        } catch {}
+      }),
+    );
+  }
+}

package/src/analyze/AnalyzeAgent.ts

@@ -0,0 +1,167 @@
+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 { createReviewerAgent } from "./CreateReviewerAgent";
+import { IPlanning, Planning } from "./Planning";
+
+type Filename = string;
+type FileContent = string;
+
+export class AnalyzeAgent<Model extends ILlmSchema.Model> {
+  private readonly agent: MicroAgentica<Model>;
+  private readonly fileMap: Record<Filename, FileContent> = {};
+
+  constructor(
+    private readonly createReviewerAgentFn: typeof createReviewerAgent,
+    private readonly ctx: AutoBeContext<Model>,
+    private readonly pointer: IPointer<{
+      files: Record<Filename, FileContent>;
+    } | null>,
+  ) {
+    assertSchemaModel(ctx.model);
+
+    const controller = createController<Model>({
+      model: ctx.model,
+      execute: new Planning(this.fileMap),
+      build: async (files: Record<Filename, FileContent>) => {
+        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,
+              )
+              .replace("{% User Locale %}", ctx.config?.locale ?? "en-US");
+          },
+        },
+      },
+      tokenUsage: ctx.usage(),
+    });
+  }
+
+  /**
+   * Conversate with planner agent
+   *
+   * @param content Conversation from user in this time.
+   * @returns
+   */
+  async conversate(content: string): Promise<string> {
+    const response = await this.agent.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 currentFiles = this.fileMap;
+
+      const reviewer = this.createReviewerAgentFn(this.ctx, {
+        query: content,
+        currentFiles,
+      });
+
+      const [review] = await reviewer.conversate(lastMessage.text);
+
+      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,
+            }),
+          );
+        }
+      }
+    }
+
+    return "COMPLETE";
+  }
+}
+
+function createController<Model extends ILlmSchema.Model>(props: {
+  model: Model;
+  execute: Planning;
+  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;
+      },
+      createOrUpdateFile: (input) => {
+        const response = props.execute.createOrUpdateFile(input);
+        props.build(props.execute.allFiles());
+        return response;
+      },
+    } satisfies IPlanning,
+  };
+}
+
+const claude = typia.llm.application<Planning, "claude", { reference: true }>();
+const collection = {
+  chatgpt: typia.llm.application<Planning, "chatgpt", { reference: true }>(),
+  claude,
+  llama: claude,
+  deepseek: claude,
+  "3.1": claude,
+  "3.0": typia.llm.application<Planning, "3.0">(),
+};