@autobe/agent 0.26.0 → 0.28.0

package/src/orchestrate/test/orchestrateTestWrite.ts

@@ -77,7 +77,7 @@ async function process<Model extends ILlmSchema.Model>(
   const pointer: IPointer<IAutoBeTestWriteApplication.IProps | null> = {
     value: null,
   };
-  const { tokenUsage } = await ctx.conversate({
+  const { metric, tokenUsage } = await ctx.conversate({
     source: "testWrite",
     histories: await transformTestWriteHistories(ctx, {
       scenario,
@@ -122,6 +122,7 @@ async function process<Model extends ILlmSchema.Model>(
     draft: pointer.value.draft,
     review: pointer.value.revise?.review,
     final: pointer.value.revise?.final ?? undefined,
+    metric,
     tokenUsage,
     completed: ++progress.completed,
     total: progress.total,
@@ -154,7 +155,11 @@ function createController<Model extends ILlmSchema.Model>(props: {
       : result;
   };
   const application: ILlmApplication<Model> = collection[
-    props.model === "chatgpt" ? "chatgpt" : "claude"
+    props.model === "chatgpt"
+      ? "chatgpt"
+      : props.model === "gemini"
+        ? "gemini"
+        : "claude"
   ](
     validate,
   ) satisfies ILlmApplication<any> as unknown as ILlmApplication<Model>;
@@ -183,6 +188,12 @@ const collection = {
         write: validate,
       },
     }),
+  gemini: (validate: Validator) =>
+    typia.llm.application<IAutoBeTestWriteApplication, "gemini">({
+      validate: {
+        write: validate,
+      },
+    }),
 };
 
 type Validator = (

package/src/utils/TokenUsageComputer.ts

@@ -0,0 +1,35 @@
+import { IAutoBeTokenUsageJson } from "@autobe/interface";
+
+export namespace TokenUsageComputer {
+  export const plus = (
+    x: IAutoBeTokenUsageJson.IComponent,
+    y: IAutoBeTokenUsageJson.IComponent,
+  ): IAutoBeTokenUsageJson.IComponent => ({
+    total: x.total + y.total,
+    input: {
+      total: x.input.total + y.input.total,
+      cached: x.input.cached + y.input.cached,
+    },
+    output: {
+      total: x.output.total + y.output.total,
+      reasoning: x.output.reasoning + y.output.reasoning,
+      accepted_prediction:
+        x.output.accepted_prediction + y.output.accepted_prediction,
+      rejected_prediction:
+        x.output.rejected_prediction + y.output.rejected_prediction,
+    },
+  });
+
+  export const increment = (
+    x: IAutoBeTokenUsageJson.IComponent,
+    y: IAutoBeTokenUsageJson.IComponent,
+  ): void => {
+    x.total += y.total;
+    x.input.total += y.input.total;
+    x.input.cached += y.input.cached;
+    x.output.total += y.output.total;
+    x.output.reasoning += y.output.reasoning;
+    x.output.accepted_prediction += y.output.accepted_prediction;
+    x.output.rejected_prediction += y.output.rejected_prediction;
+  };
+}

package/lib/context/IAutoBeFacadeApplication.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"IAutoBeFacadeApplication.js","sourceRoot":"","sources":["../../src/context/IAutoBeFacadeApplication.ts"],"names":[],"mappings":""}

package/lib/context/IAutoBeFacadeApplicationProps.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"IAutoBeFacadeApplicationProps.js","sourceRoot":"","sources":["../../src/context/IAutoBeFacadeApplicationProps.ts"],"names":[],"mappings":""}

package/lib/context/IAutoBeFacadeApplicationResult.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"IAutoBeFacadeApplicationResult.js","sourceRoot":"","sources":["../../src/context/IAutoBeFacadeApplicationResult.ts"],"names":[],"mappings":""}