@autobe/agent 0.16.1 → 0.18.0

package/src/orchestrate/interface/orchestrateInterfaceComplement.ts

@@ -1,4 +1,4 @@
-import { IAgenticaController, MicroAgentica } from "@agentica/core";
+import { IAgenticaController } from "@agentica/core";
 import { AutoBeOpenApi } from "@autobe/interface";
 import {
   ILlmApplication,
@@ -11,8 +11,6 @@ import typia from "typia";
 
 import { AutoBeContext } from "../../context/AutoBeContext";
 import { assertSchemaModel } from "../../context/assertSchemaModel";
-import { enforceToolCall } from "../../utils/enforceToolCall";
-import { forceRetry } from "../../utils/forceRetry";
 import { transformInterfaceComplementHistories } from "./histories/transformInterfaceComplementHistories";
 import { IAutoBeInterfaceComplementApplication } from "./structures/IAutoBeInterfaceComplementApplication";
 
@@ -21,7 +19,7 @@ export function orchestrateInterfaceComplement<Model extends ILlmSchema.Model>(
   document: AutoBeOpenApi.IDocument,
   life: number = 8,
 ): Promise<Record<string, AutoBeOpenApi.IJsonSchemaDescriptive>> {
-  return forceRetry(() => step(ctx, document, life));
+  return step(ctx, document, life);
 }
 
 async function step<Model extends ILlmSchema.Model>(
@@ -40,54 +38,41 @@ async function step<Model extends ILlmSchema.Model>(
   > | null> = {
     value: null,
   };
-  const agentica: MicroAgentica<Model> = new MicroAgentica({
-    model: ctx.model,
-    vendor: ctx.vendor,
-    config: {
-      ...(ctx.config ?? {}),
-      executor: {
-        describe: null,
-      },
-    },
+  const { tokenUsage } = await ctx.conversate({
+    source: "interfaceComplement",
     histories: transformInterfaceComplementHistories(
       ctx.state(),
       document,
       missed,
     ),
-    controllers: [
-      createApplication({
-        model: ctx.model,
-        build: (next) => {
-          pointer.value ??= {};
-          Object.assign(
-            pointer.value,
-            (OpenApiV3_1Emender.convertComponents({
-              schemas: next,
-            }).schemas ?? {}) as Record<
-              string,
-              AutoBeOpenApi.IJsonSchemaDescriptive
-            >,
-          );
-        },
-      }),
-    ],
-  });
-  enforceToolCall(agentica);
-
-  await agentica.conversate("Fill missing schema types please").finally(() => {
-    const tokenUsage = agentica.getTokenUsage();
-    ctx.usage().record(tokenUsage, ["interface"]);
+    controller: createController({
+      model: ctx.model,
+      build: (next) => {
+        pointer.value ??= {};
+        Object.assign(
+          pointer.value,
+          (OpenApiV3_1Emender.convertComponents({
+            schemas: next,
+          }).schemas ?? {}) as Record<
+            string,
+            AutoBeOpenApi.IJsonSchemaDescriptive
+          >,
+        );
+      },
+    }),
+    enforceFunctionCall: true,
+    message: "Fill missing schema types please",
   });
-  if (pointer.value === null) {
+  if (pointer.value === null)
     // unreachable
     throw new Error(
       "Failed to fill missing schema types. No response from agentica.",
     );
-  }
   ctx.dispatch({
     type: "interfaceComplement",
     missed,
     schemas: pointer.value,
+    tokenUsage,
     step: ctx.state().analyze?.step ?? 0,
     created_at: new Date().toISOString(),
   });
@@ -130,7 +115,7 @@ const getMissed = (document: AutoBeOpenApi.IDocument): string[] => {
   return Array.from(missed);
 };
 
-function createApplication<Model extends ILlmSchema.Model>(props: {
+function createController<Model extends ILlmSchema.Model>(props: {
   model: Model;
   build: (
     schemas: Record<string, AutoBeOpenApi.IJsonSchemaDescriptive>,
@@ -139,7 +124,7 @@ function createApplication<Model extends ILlmSchema.Model>(props: {
   assertSchemaModel(props.model);
   const application: ILlmApplication<Model> = collection[
     props.model
-  ] as unknown as ILlmApplication<Model>;
+  ] satisfies ILlmApplication<any> as unknown as ILlmApplication<Model>;
   return {
     protocol: "class",
     name: "interface",
@@ -154,14 +139,12 @@ function createApplication<Model extends ILlmSchema.Model>(props: {
 
 const claude = typia.llm.application<
   IAutoBeInterfaceComplementApplication,
-  "claude",
-  { reference: true }
+  "claude"
 >();
 const collection = {
   chatgpt: typia.llm.application<
     IAutoBeInterfaceComplementApplication,
-    "chatgpt",
-    { reference: true }
+    "chatgpt"
   >(),
   claude,
   llama: claude,

package/src/orchestrate/interface/orchestrateInterfaceEndpoints.ts

@@ -1,4 +1,4 @@
-import { IAgenticaController, MicroAgentica } from "@agentica/core";
+import { IAgenticaController } from "@agentica/core";
 import {
   AutoBeInterfaceEndpointsEvent,
   AutoBeOpenApi,
@@ -10,7 +10,6 @@ import typia from "typia";
 
 import { AutoBeContext } from "../../context/AutoBeContext";
 import { assertSchemaModel } from "../../context/assertSchemaModel";
-import { enforceToolCall } from "../../utils/enforceToolCall";
 import { transformInterfaceEndpointHistories } from "./histories/transformInterfaceEndpointHistories";
 import { IAutoBeInterfaceEndpointApplication } from "./structures/IAutoBeInterfaceEndpointApplication";
 import { OpenApiEndpointComparator } from "./utils/OpenApiEndpointComparator";
@@ -20,6 +19,7 @@ export async function orchestrateInterfaceEndpoints<
 >(
   ctx: AutoBeContext<Model>,
   groups: AutoBeInterfaceGroup[],
+  authorizations: AutoBeOpenApi.IOperation[],
   content: string = `Make endpoints for the given assets`,
 ): Promise<AutoBeOpenApi.IEndpoint[]> {
   const progress: IProgress = {
@@ -27,7 +27,9 @@ export async function orchestrateInterfaceEndpoints<
     completed: 0,
   };
   const endpoints: AutoBeOpenApi.IEndpoint[] = (
-    await Promise.all(groups.map((g) => process(ctx, g, content, progress)))
+    await Promise.all(
+      groups.map((g) => process(ctx, g, content, progress, authorizations)),
+    )
   ).flat();
   return new HashSet(
     endpoints,
@@ -39,40 +41,33 @@ export async function orchestrateInterfaceEndpoints<
 async function process<Model extends ILlmSchema.Model>(
   ctx: AutoBeContext<Model>,
   group: AutoBeInterfaceGroup,
-  content: string,
+  message: string,
   progress: IProgress,
+  authorizations: AutoBeOpenApi.IOperation[],
 ): Promise<AutoBeOpenApi.IEndpoint[]> {
   const start: Date = new Date();
   const pointer: IPointer<AutoBeOpenApi.IEndpoint[] | null> = {
     value: null,
   };
-  const agentica: MicroAgentica<Model> = new MicroAgentica({
-    model: ctx.model,
-    vendor: ctx.vendor,
-    config: {
-      ...(ctx.config ?? {}),
-      executor: {
-        describe: null,
+  const { tokenUsage } = await ctx.conversate({
+    source: "interfaceEndpoints",
+    histories: transformInterfaceEndpointHistories(
+      ctx.state(),
+      group,
+      authorizations,
+    ),
+    controller: createController({
+      model: ctx.model,
+      build: (endpoints) => {
+        pointer.value ??= endpoints;
+        pointer.value.push(...endpoints);
       },
-    },
-    histories: transformInterfaceEndpointHistories(ctx.state(), group),
-    controllers: [
-      createApplication({
-        model: ctx.model,
-        build: (endpoints) => {
-          pointer.value ??= endpoints;
-          pointer.value.push(...endpoints);
-        },
-      }),
-    ],
-  });
-  enforceToolCall(agentica);
-
-  await agentica.conversate(content).finally(() => {
-    const tokenUsage = agentica.getTokenUsage();
-    ctx.usage().record(tokenUsage, ["interface"]);
+    }),
+    enforceFunctionCall: true,
+    message,
   });
   if (pointer.value === null) throw new Error("Failed to generate endpoints."); // unreachable
+
   const event: AutoBeInterfaceEndpointsEvent = {
     type: "interfaceEndpoints",
     endpoints: new HashSet(
@@ -80,6 +75,7 @@ async function process<Model extends ILlmSchema.Model>(
       OpenApiEndpointComparator.hashCode,
       OpenApiEndpointComparator.equals,
     ).toJSON(),
+    tokenUsage,
     created_at: start.toISOString(),
     step: ctx.state().analyze?.step ?? 0,
     completed: ++progress.completed,
@@ -89,7 +85,7 @@ async function process<Model extends ILlmSchema.Model>(
   return pointer.value;
 }
 
-function createApplication<Model extends ILlmSchema.Model>(props: {
+function createController<Model extends ILlmSchema.Model>(props: {
   model: Model;
   build: (endpoints: AutoBeOpenApi.IEndpoint[]) => void;
 }): IAgenticaController.IClass<Model> {
@@ -97,7 +93,7 @@ function createApplication<Model extends ILlmSchema.Model>(props: {
 
   const application: ILlmApplication<Model> = collection[
     props.model
-  ] as unknown as ILlmApplication<Model>;
+  ] satisfies ILlmApplication<any> as unknown as ILlmApplication<Model>;
   return {
     protocol: "class",
     name: "interface",
@@ -112,14 +108,12 @@ function createApplication<Model extends ILlmSchema.Model>(props: {
 
 const claude = typia.llm.application<
   IAutoBeInterfaceEndpointApplication,
-  "claude",
-  { reference: true }
+  "claude"
 >();
 const collection = {
   chatgpt: typia.llm.application<
     IAutoBeInterfaceEndpointApplication,
-    "chatgpt",
-    { reference: true }
+    "chatgpt"
   >(),
   claude,
   llama: claude,

package/src/orchestrate/interface/orchestrateInterfaceGroups.ts

@@ -1,8 +1,4 @@
-import {
-  IAgenticaController,
-  MicroAgentica,
-  MicroAgenticaHistory,
-} from "@agentica/core";
+import { IAgenticaController, MicroAgenticaHistory } from "@agentica/core";
 import {
   AutoBeAssistantMessageHistory,
   AutoBeInterfaceGroupsEvent,
@@ -21,38 +17,24 @@ export async function orchestrateInterfaceGroups<
   Model extends ILlmSchema.Model,
 >(
   ctx: AutoBeContext<Model>,
-  content: string = "Design API operations for the given assets.",
+  message: string = "Design API operations for the given assets.",
 ): Promise<AutoBeAssistantMessageHistory | AutoBeInterfaceGroupsEvent> {
   const start: Date = new Date();
   const pointer: IPointer<IAutoBeInterfaceGroupApplication.IProps | null> = {
     value: null,
   };
-  const agentica: MicroAgentica<Model> = new MicroAgentica({
-    model: ctx.model,
-    vendor: ctx.vendor,
-    config: {
-      ...(ctx.config ?? {}),
-      executor: {
-        describe: null,
-      },
-    },
+  const { histories, tokenUsage } = await ctx.conversate({
+    source: "interfaceGroups",
     histories: transformInterfaceGroupHistories(ctx.state()),
-    controllers: [
-      createApplication({
-        model: ctx.model,
-        build: (next) => {
-          pointer.value = next;
-        },
-      }),
-    ],
+    controller: createController({
+      model: ctx.model,
+      build: (next) => {
+        pointer.value = next;
+      },
+    }),
+    enforceFunctionCall: false,
+    message,
   });
-
-  const histories: MicroAgenticaHistory<Model>[] = await agentica
-    .conversate(content)
-    .finally(() => {
-      const tokenUsage = agentica.getTokenUsage();
-      ctx.usage().record(tokenUsage, ["interface"]);
-    });
   const last: MicroAgenticaHistory<Model> = histories.at(-1)!;
   if (last.type === "assistantMessage")
     return {
@@ -67,11 +49,12 @@ export async function orchestrateInterfaceGroups<
     type: "interfaceGroups",
     created_at: start.toISOString(),
     groups: pointer.value.groups,
+    tokenUsage,
     step: ctx.state().analyze?.step ?? 0,
   } satisfies AutoBeInterfaceGroupsEvent;
 }
 
-function createApplication<Model extends ILlmSchema.Model>(props: {
+function createController<Model extends ILlmSchema.Model>(props: {
   model: Model;
   build: (next: IAutoBeInterfaceGroupApplication.IProps) => void;
 }): IAgenticaController.IClass<Model> {
@@ -79,7 +62,7 @@ function createApplication<Model extends ILlmSchema.Model>(props: {
 
   const application: ILlmApplication<Model> = collection[
     props.model
-  ] as unknown as ILlmApplication<Model>;
+  ] satisfies ILlmApplication<any> as unknown as ILlmApplication<Model>;
   return {
     protocol: "class",
     name: "interface",