@autobe/agent 0.18.0 → 0.20.0

package/src/orchestrate/interface/orchestrateInterfaceOperationsReview.ts

@@ -2,13 +2,14 @@ import { IAgenticaController } from "@agentica/core";
 import {
   AutoBeInterfaceOperationsReviewEvent,
   AutoBeOpenApi,
+  AutoBeProgressEventBase,
 } from "@autobe/interface";
 import { ILlmApplication, ILlmSchema } from "@samchon/openapi";
 import { IPointer } from "tstl";
 import typia from "typia";
+import { v7 } from "uuid";
 
 import { AutoBeContext } from "../../context/AutoBeContext";
-import { IProgress } from "../internal/IProgress";
 import { transformInterfaceOperationsReviewHistories } from "./histories/transformInterfaceOperationsReviewHistories";
 import { IAutoBeInterfaceOperationsReviewApplication } from "./structures/IAutoBeInterfaceOperationsReviewApplication";
 
@@ -17,39 +18,57 @@ export async function orchestrateInterfaceOperationsReview<
 >(
   ctx: AutoBeContext<Model>,
   operations: AutoBeOpenApi.IOperation[],
-  progress: IProgress,
+  progress: AutoBeProgressEventBase,
 ): Promise<AutoBeOpenApi.IOperation[]> {
-  const pointer: IPointer<IAutoBeInterfaceOperationsReviewApplication.IProps | null> =
-    {
-      value: null,
-    };
-  const { tokenUsage } = await ctx.conversate({
-    source: "interfaceOperationsReview",
-    histories: transformInterfaceOperationsReviewHistories(ctx, operations),
-    controller: createReviewController({
-      model: ctx.model,
-      build: (next: IAutoBeInterfaceOperationsReviewApplication.IProps) => {
-        pointer.value = next;
-      },
-    }),
-    enforceFunctionCall: false,
-    message: "Review the operations",
-  });
-  if (pointer.value === null) throw new Error("Failed to review operations.");
+  try {
+    const pointer: IPointer<IAutoBeInterfaceOperationsReviewApplication.IProps | null> =
+      {
+        value: null,
+      };
+    const { tokenUsage } = await ctx.conversate({
+      source: "interfaceOperationsReview",
+      histories: transformInterfaceOperationsReviewHistories(ctx, operations),
+      controller: createReviewController({
+        model: ctx.model,
+        build: (next: IAutoBeInterfaceOperationsReviewApplication.IProps) => {
+          pointer.value = next;
+        },
+      }),
+      enforceFunctionCall: false,
+      message: "Review the operations",
+    });
+    if (pointer.value === null) {
+      console.error("Failed to review operations.");
+      progress.completed += operations.length;
+      return [];
+    }
+
+    const content: AutoBeOpenApi.IOperation[] = pointer.value.content.map(
+      (op) => ({
+        ...op,
+        authorizationType: null,
+      }),
+    );
 
-  ctx.dispatch({
-    type: "interfaceOperationsReview",
-    operations: pointer.value.content,
-    review: pointer.value.review,
-    plan: pointer.value.plan,
-    content: pointer.value.content,
-    tokenUsage,
-    created_at: new Date().toISOString(),
-    step: ctx.state().analyze?.step ?? 0,
-    total: progress.total,
-    completed: ++progress.completed,
-  } satisfies AutoBeInterfaceOperationsReviewEvent);
-  return pointer.value.content;
+    ctx.dispatch({
+      type: "interfaceOperationsReview",
+      id: v7(),
+      operations: content,
+      review: pointer.value.review,
+      plan: pointer.value.plan,
+      content,
+      tokenUsage,
+      created_at: new Date().toISOString(),
+      step: ctx.state().analyze?.step ?? 0,
+      total: progress.total,
+      completed: ++progress.completed,
+    } satisfies AutoBeInterfaceOperationsReviewEvent);
+    return content;
+  } catch (error) {
+    console.error("Error occurred during interface operations review:", error);
+    ++progress.completed;
+    return [];
+  }
 }
 
 function createReviewController<Model extends ILlmSchema.Model>(props: {

package/src/orchestrate/interface/orchestrateInterfaceSchemas.ts

@@ -1,24 +1,31 @@
 import { IAgenticaController } from "@agentica/core";
-import { AutoBeInterfaceSchemasEvent, AutoBeOpenApi } from "@autobe/interface";
-import { ILlmApplication, ILlmSchema } from "@samchon/openapi";
+import {
+  AutoBeInterfaceSchemasEvent,
+  AutoBeOpenApi,
+  AutoBeProgressEventBase,
+} from "@autobe/interface";
+import { StringUtil } from "@autobe/utils";
+import { ILlmApplication, ILlmSchema, IValidation } from "@samchon/openapi";
 import { OpenApiV3_1Emender } from "@samchon/openapi/lib/converters/OpenApiV3_1Emender";
 import { IPointer } from "tstl";
-import typia from "typia";
+import typia, { tags } from "typia";
+import { v7 } from "uuid";
 
 import { AutoBeContext } from "../../context/AutoBeContext";
 import { assertSchemaModel } from "../../context/assertSchemaModel";
 import { divideArray } from "../../utils/divideArray";
-import { IProgress } from "../internal/IProgress";
+import { executeCachedBatch } from "../../utils/executeCachedBatch";
 import { transformInterfaceSchemaHistories } from "./histories/transformInterfaceSchemaHistories";
 import { orchestrateInterfaceSchemasReview } from "./orchestrateInterfaceSchemasReview";
 import { IAutoBeInterfaceSchemaApplication } from "./structures/IAutoBeInterfaceSchemaApplication";
+import { validateAuthorizationSchema } from "./utils/validateAuthorizationSchema";
 
 export async function orchestrateInterfaceSchemas<
   Model extends ILlmSchema.Model,
 >(
   ctx: AutoBeContext<Model>,
   operations: AutoBeOpenApi.IOperation[],
-  capacity: number = 12,
+  capacity: number = 8,
 ): Promise<Record<string, AutoBeOpenApi.IJsonSchemaDescriptive>> {
   const typeNames: Set<string> = new Set();
   for (const op of operations) {
@@ -29,29 +36,47 @@ export async function orchestrateInterfaceSchemas<
     array: Array.from(typeNames),
     capacity,
   });
-  const progress: IProgress = {
+  const progress: AutoBeProgressEventBase = {
     total: typeNames.size,
     completed: 0,
   };
-  const reviewProgress: IProgress = {
+  const reviewProgress: AutoBeProgressEventBase = {
     total: matrix.length,
     completed: 0,
   };
-  const x: Record<string, AutoBeOpenApi.IJsonSchemaDescriptive> = {};
-  for (const y of await Promise.all(
-    matrix.map(async (it) => {
+  const roles: string[] =
+    ctx.state().analyze?.roles.map((role) => role.name) ?? [];
+
+  const x: Record<string, AutoBeOpenApi.IJsonSchemaDescriptive> =
+    roles.length > 0
+      ? {
+          IAuthorizationToken: authTokenSchema,
+        }
+      : {};
+  for (const y of await executeCachedBatch(
+    matrix.map((it) => async (promptCacheKey) => {
       const row: Record<string, AutoBeOpenApi.IJsonSchemaDescriptive> =
-        await divideAndConquer(ctx, operations, it, 3, progress);
-      return orchestrateInterfaceSchemasReview(
-        ctx,
-        operations,
-        row,
-        reviewProgress,
-      );
+        await divideAndConquer(
+          ctx,
+          operations,
+          it,
+          3,
+          progress,
+          promptCacheKey,
+        );
+      const newbie: Record<string, AutoBeOpenApi.IJsonSchemaDescriptive> =
+        await orchestrateInterfaceSchemasReview(
+          ctx,
+          operations,
+          row,
+          reviewProgress,
+        );
+      return { ...row, ...newbie };
     }),
   )) {
     Object.assign(x, y);
   }
+  if (x.IAuthorizationToken) x.IAuthorizationToken = authTokenSchema;
   return x;
 }
 
@@ -60,14 +85,22 @@ async function divideAndConquer<Model extends ILlmSchema.Model>(
   operations: AutoBeOpenApi.IOperation[],
   typeNames: string[],
   retry: number,
-  progress: IProgress,
+  progress: AutoBeProgressEventBase,
+  promptCacheKey: string,
 ): Promise<Record<string, AutoBeOpenApi.IJsonSchemaDescriptive>> {
   const remained: Set<string> = new Set(typeNames);
   const schemas: Record<string, AutoBeOpenApi.IJsonSchemaDescriptive> = {};
   for (let i: number = 0; i < retry; ++i) {
     if (remained.size === 0) break;
     const newbie: Record<string, AutoBeOpenApi.IJsonSchemaDescriptive> =
-      await process(ctx, operations, schemas, remained, progress);
+      await process(
+        ctx,
+        operations,
+        schemas,
+        remained,
+        progress,
+        promptCacheKey,
+      );
     for (const key of Object.keys(newbie)) {
       schemas[key] = newbie[key];
       remained.delete(key);
@@ -81,7 +114,8 @@ async function process<Model extends ILlmSchema.Model>(
   operations: AutoBeOpenApi.IOperation[],
   oldbie: Record<string, AutoBeOpenApi.IJsonSchemaDescriptive>,
   remained: Set<string>,
-  progress: IProgress,
+  progress: AutoBeProgressEventBase,
+  promptCacheKey: string,
 ): Promise<Record<string, AutoBeOpenApi.IJsonSchemaDescriptive>> {
   const already: string[] = Object.keys(oldbie);
   const pointer: IPointer<Record<
@@ -102,25 +136,29 @@ async function process<Model extends ILlmSchema.Model>(
       pointer,
     }),
     enforceFunctionCall: true,
-    message: [
-      "Make type components please.",
-      "",
-      "Here is the list of request/response bodies' type names from",
-      "OpenAPI operations. Make type components of them. If more object",
-      "types are required during making the components, please make them",
-      "too.",
-      "",
-      ...Array.from(remained).map((k) => `- \`${k}\``),
-      ...(already.length !== 0
-        ? [
-            "",
-            "> By the way, here is the list of components schemas what you've",
-            "> already made. So, you don't need to make them again.",
-            ">",
-            ...already.map((k) => `> - \`${k}\``),
-          ]
-        : []),
-    ].join("\n"),
+    promptCacheKey,
+    message: StringUtil.trim`
+      Make type components please.
+
+      Here is the list of request/response bodies' type names from
+      OpenAPI operations. Make type components of them. If more object
+      types are required during making the components, please make them
+      too.
+
+      ${Array.from(remained)
+        .map((k) => `      - \`${k}\``)
+        .join("\n")}${
+        already.length !== 0
+          ? StringUtil.trim`
+
+            > By the way, here is the list of components schemas what you've
+            > already made. So, you don't need to make them again.
+            >
+            ${already.map((k) => `> - \`${k}\``).join("\n")}
+          `
+          : ""
+      }
+    `,
   });
   if (pointer.value === null) throw new Error("Failed to create components.");
 
@@ -132,12 +170,13 @@ async function process<Model extends ILlmSchema.Model>(
     ).schemas ?? {};
   ctx.dispatch({
     type: "interfaceSchemas",
+    id: v7(),
     schemas,
     tokenUsage,
-    completed: (progress.completed += Object.keys(schemas).filter(
-      (k) => oldbie[k] === undefined,
+    completed: (progress.completed += Object.keys(schemas).length),
+    total: (progress.total += Object.keys(schemas).filter(
+      (k) => remained.has(k) === false,
     ).length),
-    total: progress.total,
     step: ctx.state().prisma?.step ?? 0,
     created_at: new Date().toISOString(),
   } satisfies AutoBeInterfaceSchemasEvent);
@@ -156,9 +195,34 @@ function createController<Model extends ILlmSchema.Model>(props: {
 }): IAgenticaController.IClass<Model> {
   assertSchemaModel(props.model);
 
+  const validate = (
+    next: unknown,
+  ): IValidation<IAutoBeInterfaceSchemaApplication.IProps> => {
+    const result: IValidation<IAutoBeInterfaceSchemaApplication.IProps> =
+      typia.validate<IAutoBeInterfaceSchemaApplication.IProps>(next);
+    if (result.success === false) return result;
+
+    // Check all IAuthorized types
+    const errors: IValidation.IError[] = [];
+    validateAuthorizationSchema({
+      errors,
+      schemas: result.data.schemas,
+      path: "$input.schemas",
+    });
+    if (errors.length !== 0)
+      return {
+        success: false,
+        errors,
+        data: next,
+      };
+    return result;
+  };
+
   const application: ILlmApplication<Model> = collection[
-    props.model
-  ] satisfies ILlmApplication<any> as unknown as ILlmApplication<Model>;
+    props.model === "chatgpt" ? "chatgpt" : "claude"
+  ](
+    validate,
+  ) satisfies ILlmApplication<any> as unknown as ILlmApplication<Model>;
   return {
     protocol: "class",
     name: "interface",
@@ -171,17 +235,71 @@ function createController<Model extends ILlmSchema.Model>(props: {
   };
 }
 
-const claude = typia.llm.application<
-  IAutoBeInterfaceSchemaApplication,
-  "claude"
->();
 const collection = {
-  chatgpt: typia.llm.application<
-    IAutoBeInterfaceSchemaApplication,
-    "chatgpt"
-  >(),
-  claude,
-  llama: claude,
-  deepseek: claude,
-  "3.1": claude,
+  chatgpt: (validate: Validator) =>
+    typia.llm.application<IAutoBeInterfaceSchemaApplication, "chatgpt">({
+      validate: {
+        makeComponents: validate,
+      },
+    }),
+  claude: (validate: Validator) =>
+    typia.llm.application<IAutoBeInterfaceSchemaApplication, "claude">({
+      validate: {
+        makeComponents: validate,
+      },
+    }),
 };
+
+type Validator = (
+  input: unknown,
+) => IValidation<IAutoBeInterfaceSchemaApplication.IProps>;
+
+/**
+ * Authorization token response structure.
+ *
+ * This interface defines the structure of the authorization token response
+ * returned after successful user authentication. It contains both access and
+ * refresh tokens along with their expiration information.
+ *
+ * This token structure is automatically included in API schemas when the system
+ * detects authorization roles in the requirements analysis phase. It provides a
+ * standard format for JWT-based authentication across the generated backend
+ * applications.
+ */
+interface IAuthorizationToken {
+  /**
+   * JWT access token for authenticated requests.
+   *
+   * This token should be included in the Authorization header for subsequent
+   * authenticated API requests as `Bearer {token}`.
+   */
+  access: string;
+
+  /**
+   * Refresh token for obtaining new access tokens.
+   *
+   * This token can be used to request new access tokens when the current access
+   * token expires, extending the user's session.
+   */
+  refresh: string;
+
+  /**
+   * Access token expiration timestamp.
+   *
+   * ISO 8601 date-time string indicating when the access token will expire and
+   * can no longer be used for authentication.
+   */
+  expired_at: string & tags.Format<"date-time">;
+
+  /**
+   * Refresh token expiration timestamp.
+   *
+   * ISO 8601 date-time string indicating the latest time until which the
+   * refresh token can be used to obtain new access tokens.
+   */
+  refreshable_until: string & tags.Format<"date-time">;
+}
+
+const authTokenSchema: AutoBeOpenApi.IJsonSchemaDescriptive =
+  typia.json.schema<IAuthorizationToken>().components.schemas!
+    .IAuthorizationToken as AutoBeOpenApi.IJsonSchemaDescriptive;