@autobe/utils 0.29.1 → 0.30.0-dev.20260315

package/src/aggregate/AutoBeProcessAggregateFactory.ts

@@ -1,152 +1,161 @@
-import {
-  AutoBeAggregateEventBase,
-  AutoBePhase,
-  AutoBeProcessAggregate,
-  AutoBeProcessAggregateCollection,
-} from "@autobe/interface";
-
-import { AutoBeFunctionCallingMetricFactory } from "./AutoBeFunctionCallingMetricFactory";
-import { TokenUsageComputer } from "./TokenUsageComputer";
-
-export namespace AutoBeProcessAggregateFactory {
-  export const createAggregate = (): AutoBeProcessAggregate => ({
-    metric: AutoBeFunctionCallingMetricFactory.create(),
-    tokenUsage: {
-      total: 0,
-      input: {
-        total: 0,
-        cached: 0,
-      },
-      output: {
-        total: 0,
-        reasoning: 0,
-        accepted_prediction: 0,
-        rejected_prediction: 0,
-      },
-    },
-  });
-
-  export const createCollection = <
-    Phase extends AutoBePhase | "all",
-  >(): AutoBeProcessAggregateCollection<Phase> =>
-    ({
-      total: createAggregate(),
-    }) satisfies AutoBeProcessAggregateCollection as AutoBeProcessAggregateCollection<Phase>;
-
-  export const computeTotal = <Phase extends AutoBePhase | "all">(
-    collection: AutoBeProcessAggregateCollection<Phase>,
-  ): AutoBeProcessAggregate => {
-    const total: AutoBeProcessAggregate = createAggregate();
-    for (const [key, value] of Object.entries(collection)) {
-      if (key === "total") continue;
-      AutoBeFunctionCallingMetricFactory.increment(total.metric, value.metric);
-      TokenUsageComputer.increment(total.tokenUsage, value.tokenUsage);
-    }
-    return total;
-  };
-
-  export const emplaceEvent = <
-    Event extends AutoBeAggregateEventBase & {
-      type: string;
-    },
-  >(
-    collection: AutoBeProcessAggregateCollection,
-    event: Event,
-  ): void => {
-    (collection as any)[event.type] ??= createAggregate();
-    collection.total ??= computeTotal(collection);
-
-    const local: AutoBeProcessAggregate = (collection as any)[
-      event.type
-    ] as AutoBeProcessAggregate;
-    const total: AutoBeProcessAggregate = collection.total;
-
-    AutoBeFunctionCallingMetricFactory.increment(local.metric, event.metric);
-    AutoBeFunctionCallingMetricFactory.increment(total.metric, event.metric);
-    TokenUsageComputer.increment(local.tokenUsage, event.tokenUsage);
-    TokenUsageComputer.increment(total.tokenUsage, event.tokenUsage);
-  };
-
-  export const filterPhase = <Phase extends AutoBePhase>(
-    collection: AutoBeProcessAggregateCollection,
-    phase: Phase,
-  ): AutoBeProcessAggregateCollection<Phase> => {
-    const result: AutoBeProcessAggregateCollection<Phase> = createCollection();
-    for (const [key, value] of Object.entries(collection)) {
-      if (key === "total") continue;
-      else if (key.startsWith(phase) === false) continue;
-
-      (result as any)[key] = value;
-      AutoBeFunctionCallingMetricFactory.increment(
-        result.total.metric,
-        value.metric,
-      );
-      TokenUsageComputer.increment(result.total.tokenUsage, value.tokenUsage);
-    }
-    return result;
-  };
-
-  export const reduce = (
-    collections: AutoBeProcessAggregateCollection[],
-  ): AutoBeProcessAggregateCollection => {
-    const result: AutoBeProcessAggregateCollection = createCollection();
-    for (const collection of collections) {
-      for (const [key, value] of Object.entries(collection)) {
-        if (key === "total") continue;
-        (result as any)[key] ??= createAggregate();
-        const local: AutoBeProcessAggregate = (result as any)[
-          key
-        ] as AutoBeProcessAggregate;
-        AutoBeFunctionCallingMetricFactory.increment(
-          local.metric,
-          value.metric,
-        );
-        TokenUsageComputer.increment(local.tokenUsage, value.tokenUsage);
-        AutoBeFunctionCallingMetricFactory.increment(
-          result.total.metric,
-          value.metric,
-        );
-        TokenUsageComputer.increment(result.total.tokenUsage, value.tokenUsage);
-      }
-    }
-    result.total ??= createAggregate();
-    Object.assign(result.total, computeTotal(result));
-    return result;
-  };
-
-  export const increment = (
-    x: AutoBeProcessAggregateCollection,
-    y: AutoBeProcessAggregateCollection,
-  ): void => {
-    for (const [key, value] of Object.entries(y)) {
-      if (key === "total") continue;
-      (x as any)[key] ??= createAggregate();
-      const local: AutoBeProcessAggregate = (x as any)[
-        key
-      ] as AutoBeProcessAggregate;
-      AutoBeFunctionCallingMetricFactory.increment(local.metric, value.metric);
-    }
-    x.total ??= createAggregate();
-    Object.assign(x.total, computeTotal(x));
-  };
-
-  export const minus = (
-    x: AutoBeProcessAggregateCollection,
-    y: AutoBeProcessAggregateCollection,
-  ): AutoBeProcessAggregateCollection => {
-    const result = JSON.parse(
-      JSON.stringify(x),
-    ) as AutoBeProcessAggregateCollection;
-    for (const [key, value] of Object.entries(y)) {
-      if (key === "total") continue;
-      (result as any)[key] ??= createAggregate();
-      const local: AutoBeProcessAggregate = (result as any)[
-        key
-      ] as AutoBeProcessAggregate;
-      AutoBeFunctionCallingMetricFactory.minus(local.metric, value.metric);
-    }
-    result.total ??= createAggregate();
-    Object.assign(result.total, computeTotal(result));
-    return result;
-  };
-}
+import {
+  AutoBeAggregateEventBase,
+  AutoBePhase,
+  AutoBeProcessAggregate,
+  AutoBeProcessAggregateCollection,
+} from "@autobe/interface";
+
+import { AutoBeFunctionCallingMetricFactory } from "./AutoBeFunctionCallingMetricFactory";
+import { TokenUsageComputer } from "./TokenUsageComputer";
+
+export namespace AutoBeProcessAggregateFactory {
+  export const createAggregate = (): AutoBeProcessAggregate => ({
+    metric: AutoBeFunctionCallingMetricFactory.create(),
+    tokenUsage: {
+      total: 0,
+      input: {
+        total: 0,
+        cached: 0,
+      },
+      output: {
+        total: 0,
+        reasoning: 0,
+        accepted_prediction: 0,
+        rejected_prediction: 0,
+      },
+    },
+  });
+
+  export const createCollection = <
+    Phase extends AutoBePhase | "all",
+  >(): AutoBeProcessAggregateCollection<Phase> =>
+    ({
+      total: createAggregate(),
+    }) satisfies AutoBeProcessAggregateCollection as AutoBeProcessAggregateCollection<Phase>;
+
+  export const computeTotal = <Phase extends AutoBePhase | "all">(
+    collection: AutoBeProcessAggregateCollection<Phase>,
+  ): AutoBeProcessAggregate => {
+    const total: AutoBeProcessAggregate = createAggregate();
+    for (const [key, value] of Object.entries(collection)) {
+      if (key === "total") continue;
+      AutoBeFunctionCallingMetricFactory.increment(total.metric, value.metric);
+      TokenUsageComputer.increment(total.tokenUsage, value.tokenUsage);
+    }
+    return total;
+  };
+
+  export const emplaceEvent = <
+    Event extends AutoBeAggregateEventBase & {
+      type: string;
+    },
+  >(
+    collection: AutoBeProcessAggregateCollection,
+    event: Event,
+  ): void => {
+    // biome-ignore lint: intended
+    (collection as any)[event.type] ??= createAggregate();
+    collection.total ??= computeTotal(collection);
+
+    // biome-ignore lint: intended
+    const local: AutoBeProcessAggregate = (collection as any)[
+      event.type
+    ] as AutoBeProcessAggregate;
+    const total: AutoBeProcessAggregate = collection.total;
+
+    AutoBeFunctionCallingMetricFactory.increment(local.metric, event.metric);
+    AutoBeFunctionCallingMetricFactory.increment(total.metric, event.metric);
+    TokenUsageComputer.increment(local.tokenUsage, event.tokenUsage);
+    TokenUsageComputer.increment(total.tokenUsage, event.tokenUsage);
+  };
+
+  export const filterPhase = <Phase extends AutoBePhase>(
+    collection: AutoBeProcessAggregateCollection,
+    phase: Phase,
+  ): AutoBeProcessAggregateCollection<Phase> => {
+    const result: AutoBeProcessAggregateCollection<Phase> = createCollection();
+    for (const [key, value] of Object.entries(collection)) {
+      if (key === "total") continue;
+      else if (key.startsWith(phase) === false) continue;
+
+      // biome-ignore lint: intended
+      (result as any)[key] = value;
+      AutoBeFunctionCallingMetricFactory.increment(
+        result.total.metric,
+        value.metric,
+      );
+      TokenUsageComputer.increment(result.total.tokenUsage, value.tokenUsage);
+    }
+    return result;
+  };
+
+  export const reduce = (
+    collections: AutoBeProcessAggregateCollection[],
+  ): AutoBeProcessAggregateCollection => {
+    const result: AutoBeProcessAggregateCollection = createCollection();
+    for (const collection of collections) {
+      for (const [key, value] of Object.entries(collection)) {
+        if (key === "total") continue;
+        // biome-ignore lint: intended
+        (result as any)[key] ??= createAggregate();
+        // biome-ignore lint: intended
+        const local: AutoBeProcessAggregate = (result as any)[
+          key
+        ] as AutoBeProcessAggregate;
+        AutoBeFunctionCallingMetricFactory.increment(
+          local.metric,
+          value.metric,
+        );
+        TokenUsageComputer.increment(local.tokenUsage, value.tokenUsage);
+        AutoBeFunctionCallingMetricFactory.increment(
+          result.total.metric,
+          value.metric,
+        );
+        TokenUsageComputer.increment(result.total.tokenUsage, value.tokenUsage);
+      }
+    }
+    result.total ??= createAggregate();
+    Object.assign(result.total, computeTotal(result));
+    return result;
+  };
+
+  export const increment = (
+    x: AutoBeProcessAggregateCollection,
+    y: AutoBeProcessAggregateCollection,
+  ): void => {
+    for (const [key, value] of Object.entries(y)) {
+      if (key === "total") continue;
+      // biome-ignore lint: intended
+      (x as any)[key] ??= createAggregate();
+      // biome-ignore lint: intended
+      const local: AutoBeProcessAggregate = (x as any)[
+        key
+      ] as AutoBeProcessAggregate;
+      AutoBeFunctionCallingMetricFactory.increment(local.metric, value.metric);
+    }
+    x.total ??= createAggregate();
+    Object.assign(x.total, computeTotal(x));
+  };
+
+  export const minus = (
+    x: AutoBeProcessAggregateCollection,
+    y: AutoBeProcessAggregateCollection,
+  ): AutoBeProcessAggregateCollection => {
+    const result = JSON.parse(
+      JSON.stringify(x),
+    ) as AutoBeProcessAggregateCollection;
+    for (const [key, value] of Object.entries(y)) {
+      if (key === "total") continue;
+      // biome-ignore lint: intended
+      (result as any)[key] ??= createAggregate();
+      // biome-ignore lint: intended
+      const local: AutoBeProcessAggregate = (result as any)[
+        key
+      ] as AutoBeProcessAggregate;
+      AutoBeFunctionCallingMetricFactory.minus(local.metric, value.metric);
+    }
+    result.total ??= createAggregate();
+    Object.assign(result.total, computeTotal(result));
+    return result;
+  };
+}

package/src/aggregate/TokenUsageComputer.ts

@@ -1,49 +1,49 @@
-import { IAutoBeTokenUsageJson } from "@autobe/interface";
-
-export namespace TokenUsageComputer {
-  export const zero = (): IAutoBeTokenUsageJson.IComponent => ({
-    total: 0,
-    input: {
-      total: 0,
-      cached: 0,
-    },
-    output: {
-      total: 0,
-      reasoning: 0,
-      accepted_prediction: 0,
-      rejected_prediction: 0,
-    },
-  });
-
-  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;
-  };
-}
+import { IAutoBeTokenUsageJson } from "@autobe/interface";
+
+export namespace TokenUsageComputer {
+  export const zero = (): IAutoBeTokenUsageJson.IComponent => ({
+    total: 0,
+    input: {
+      total: 0,
+      cached: 0,
+    },
+    output: {
+      total: 0,
+      reasoning: 0,
+      accepted_prediction: 0,
+      rejected_prediction: 0,
+    },
+  });
+
+  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/src/aggregate/index.ts

@@ -1,3 +1,3 @@
-export * from "./AutoBeFunctionCallingMetricFactory";
-export * from "./AutoBeProcessAggregateFactory";
-export * from "./TokenUsageComputer";
+export * from "./AutoBeFunctionCallingMetricFactory";
+export * from "./AutoBeProcessAggregateFactory";
+export * from "./TokenUsageComputer";

package/src/index.ts

@@ -1,9 +1,10 @@
-export * from "./aggregate";
-
-export * from "./prisma";
-export * from "./interface";
-export * from "./test";
-
-export * from "./ArrayUtil";
-export * from "./MapUtil";
-export * from "./StringUtil";
+export * from "./aggregate";
+
+export * from "./prisma";
+export * from "./interface";
+export * from "./test";
+
+export * from "./ArrayUtil";
+export * from "./AutoBeEscaper";
+export * from "./MapUtil";
+export * from "./StringUtil";

package/src/interface/AutoBeOpenApiEndpointComparator.ts

@@ -1,20 +1,27 @@
-import { AutoBeOpenApi } from "@autobe/interface";
-import { hash } from "tstl";
-
-export namespace AutoBeOpenApiEndpointComparator {
-  export function compare(
-    x: AutoBeOpenApi.IEndpoint,
-    y: AutoBeOpenApi.IEndpoint,
-  ): number {
-    if (x.path !== y.path) return x.path.localeCompare(y.path);
-    return x.method.localeCompare(y.method);
-  }
-
-  export const equals = (
-    x: AutoBeOpenApi.IEndpoint,
-    y: AutoBeOpenApi.IEndpoint,
-  ): boolean => x.method === y.method && x.path === y.path;
-
-  export const hashCode = (endpoint: AutoBeOpenApi.IEndpoint): number =>
-    hash(endpoint.path, endpoint.method);
-}
+import { AutoBeOpenApi } from "@autobe/interface";
+import { hash } from "tstl";
+
+export namespace AutoBeOpenApiEndpointComparator {
+  export function compare(
+    x: AutoBeOpenApi.IEndpoint,
+    y: AutoBeOpenApi.IEndpoint,
+  ): number {
+    if (x.path !== y.path) return x.path.localeCompare(y.path);
+    return x.method.localeCompare(y.method);
+  }
+
+  export const equals = (
+    x: AutoBeOpenApi.IEndpoint,
+    y: AutoBeOpenApi.IEndpoint,
+  ): boolean => x.method === y.method && x.path === y.path;
+
+  export const hashCode = (endpoint: AutoBeOpenApi.IEndpoint): number =>
+    hash(endpoint.path, endpoint.method);
+
+  export const clone = (
+    endpoint: AutoBeOpenApi.IEndpoint,
+  ): AutoBeOpenApi.IEndpoint => ({
+    path: endpoint.path,
+    method: endpoint.method,
+  });
+}