/*
 * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
 */

import * as z from "zod/v4-mini";
import { remap as remap$ } from "../lib/primitives.js";
import { ClosedEnum } from "../types/enums.js";

export const Mode = {
  Compact: "compact",
  Balanced: "balanced",
  Recall: "recall",
} as const;
export type Mode = ClosedEnum<typeof Mode>;

export type AgentMemoryBudget = {
  maxCharacters?: number | undefined;
  maxChars?: number | undefined;
  maxContextCharacters?: number | undefined;
  maxContextChars?: number | undefined;
  mode?: Mode | undefined;
};

/** @internal */
export const Mode$outboundSchema: z.ZodMiniEnum<typeof Mode> = z.enum(Mode);

/** @internal */
export type AgentMemoryBudget$Outbound = {
  maxCharacters?: number | undefined;
  max_chars?: number | undefined;
  maxContextCharacters?: number | undefined;
  max_context_chars?: number | undefined;
  mode?: string | undefined;
};

/** @internal */
export const AgentMemoryBudget$outboundSchema: z.ZodMiniType<
  AgentMemoryBudget$Outbound,
  AgentMemoryBudget
> = z.pipe(
  z.object({
    maxCharacters: z.optional(z.int()),
    maxChars: z.optional(z.int()),
    maxContextCharacters: z.optional(z.int()),
    maxContextChars: z.optional(z.int()),
    mode: z.optional(Mode$outboundSchema),
  }),
  z.transform((v) => {
    return remap$(v, {
      maxChars: "max_chars",
      maxContextChars: "max_context_chars",
    });
  }),
);

export function agentMemoryBudgetToJSON(
  agentMemoryBudget: AgentMemoryBudget,
): string {
  return JSON.stringify(
    AgentMemoryBudget$outboundSchema.parse(agentMemoryBudget),
  );
}