modaic 0.1.0

package/src/models/errors/modaic-client-error.ts

@@ -0,0 +1,35 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+/** The base class for all HTTP error responses */
+export class ModaicClientError extends Error {
+  /** HTTP status code */
+  public readonly statusCode: number;
+  /** HTTP body */
+  public readonly body: string;
+  /** HTTP headers */
+  public readonly headers: Headers;
+  /** HTTP content type */
+  public readonly contentType: string;
+  /** Raw response */
+  public readonly rawResponse: Response;
+
+  constructor(
+    message: string,
+    httpMeta: {
+      response: Response;
+      request: Request;
+      body: string;
+    },
+  ) {
+    super(message);
+    this.statusCode = httpMeta.response.status;
+    this.body = httpMeta.body;
+    this.headers = httpMeta.response.headers;
+    this.contentType = httpMeta.response.headers.get("content-type") || "";
+    this.rawResponse = httpMeta.response;
+
+    this.name = "ModaicClientError";
+  }
+}

package/src/models/errors/response-validation-error.ts

@@ -0,0 +1,50 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v4/core";
+import { ModaicClientError } from "./modaic-client-error.js";
+import { formatZodError } from "./sdk-validation-error.js";
+
+export class ResponseValidationError extends ModaicClientError {
+  /**
+   * The raw value that failed validation.
+   */
+  public readonly rawValue: unknown;
+
+  /**
+   * The raw message that failed validation.
+   */
+  public readonly rawMessage: unknown;
+
+  constructor(
+    message: string,
+    extra: {
+      response: Response;
+      request: Request;
+      body: string;
+      cause: unknown;
+      rawValue: unknown;
+      rawMessage: unknown;
+    },
+  ) {
+    super(message, extra);
+    this.name = "ResponseValidationError";
+    this.cause = extra.cause;
+    this.rawValue = extra.rawValue;
+    this.rawMessage = extra.rawMessage;
+  }
+
+  /**
+   * Return a pretty-formatted error message if the underlying validation error
+   * is a ZodError or some other recognized error type, otherwise return the
+   * default error message.
+   */
+  public pretty(): string {
+    if (this.cause instanceof z.$ZodError) {
+      return `${this.rawMessage}\n${formatZodError(this.cause)}`;
+    } else {
+      return this.toString();
+    }
+  }
+}

package/src/models/errors/sdk-validation-error.ts

@@ -0,0 +1,54 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v4/core";
+
+export class SDKValidationError extends Error {
+  /**
+   * The raw value that failed validation.
+   */
+  public readonly rawValue: unknown;
+
+  /**
+   * The raw message that failed validation.
+   */
+  public readonly rawMessage: unknown;
+
+  // Allows for backwards compatibility for `instanceof` checks of `ResponseValidationError`
+  static override [Symbol.hasInstance](
+    instance: unknown,
+  ): instance is SDKValidationError {
+    if (!(instance instanceof Error)) return false;
+    if (!("rawValue" in instance)) return false;
+    if (!("rawMessage" in instance)) return false;
+    if (!("pretty" in instance)) return false;
+    if (typeof instance.pretty !== "function") return false;
+    return true;
+  }
+
+  constructor(message: string, cause: unknown, rawValue: unknown) {
+    super(`${message}: ${cause}`);
+    this.name = "SDKValidationError";
+    this.cause = cause;
+    this.rawValue = rawValue;
+    this.rawMessage = message;
+  }
+
+  /**
+   * Return a pretty-formatted error message if the underlying validation error
+   * is a ZodError or some other recognized error type, otherwise return the
+   * default error message.
+   */
+  public pretty(): string {
+    if (this.cause instanceof z.$ZodError) {
+      return `${this.rawMessage}\n${formatZodError(this.cause)}`;
+    } else {
+      return this.toString();
+    }
+  }
+}
+
+export function formatZodError(err: z.$ZodError): string {
+  return z.prettifyError(err);
+}

package/src/models/export-examples-request.ts

@@ -0,0 +1,41 @@
+/*
+ * 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";
+
+/**
+ * Export examples and predictions for a repo.
+ */
+export type ExportExamplesRequest = {
+  arbiterRepo: string;
+};
+
+/** @internal */
+export type ExportExamplesRequest$Outbound = {
+  arbiter_repo: string;
+};
+
+/** @internal */
+export const ExportExamplesRequest$outboundSchema: z.ZodMiniType<
+  ExportExamplesRequest$Outbound,
+  ExportExamplesRequest
+> = z.pipe(
+  z.object({
+    arbiterRepo: z.string(),
+  }),
+  z.transform((v) => {
+    return remap$(v, {
+      arbiterRepo: "arbiter_repo",
+    });
+  }),
+);
+
+export function exportExamplesRequestToJSON(
+  exportExamplesRequest: ExportExamplesRequest,
+): string {
+  return JSON.stringify(
+    ExportExamplesRequest$outboundSchema.parse(exportExamplesRequest),
+  );
+}

package/src/models/export-examples-response.ts

@@ -0,0 +1,54 @@
+/*
+ * 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 { safeParse } from "../lib/schemas.js";
+import { Result as SafeParseResult } from "../types/fp.js";
+import * as types from "../types/primitives.js";
+import { SDKValidationError } from "./errors/sdk-validation-error.js";
+
+/**
+ * A presigned parquet download for exported examples.
+ */
+export type ExportExamplesResponse = {
+  downloadUrl: string;
+  fileKey: string;
+  fileName: string;
+  expiresIn: number;
+  rowCount: number;
+};
+
+/** @internal */
+export const ExportExamplesResponse$inboundSchema: z.ZodMiniType<
+  ExportExamplesResponse,
+  unknown
+> = z.pipe(
+  z.object({
+    download_url: types.string(),
+    file_key: types.string(),
+    file_name: types.string(),
+    expires_in: types.number(),
+    row_count: types.number(),
+  }),
+  z.transform((v) => {
+    return remap$(v, {
+      "download_url": "downloadUrl",
+      "file_key": "fileKey",
+      "file_name": "fileName",
+      "expires_in": "expiresIn",
+      "row_count": "rowCount",
+    });
+  }),
+);
+
+export function exportExamplesResponseFromJSON(
+  jsonString: string,
+): SafeParseResult<ExportExamplesResponse, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ExportExamplesResponse$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ExportExamplesResponse' from JSON`,
+  );
+}

package/src/models/field-schema.ts

@@ -0,0 +1,50 @@
+/*
+ * 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";
+
+export type FieldSchema = {
+  name: string;
+  type: string;
+  allowedValues?: Array<any> | null | undefined;
+  objectSchema?: { [k: string]: any } | null | undefined;
+  nullable?: boolean | undefined;
+  description?: string | null | undefined;
+};
+
+/** @internal */
+export type FieldSchema$Outbound = {
+  name: string;
+  type: string;
+  allowed_values?: Array<any> | null | undefined;
+  object_schema?: { [k: string]: any } | null | undefined;
+  nullable: boolean;
+  description?: string | null | undefined;
+};
+
+/** @internal */
+export const FieldSchema$outboundSchema: z.ZodMiniType<
+  FieldSchema$Outbound,
+  FieldSchema
+> = z.pipe(
+  z.object({
+    name: z.string(),
+    type: z.string(),
+    allowedValues: z.optional(z.nullable(z.array(z.any()))),
+    objectSchema: z.optional(z.nullable(z.record(z.string(), z.any()))),
+    nullable: z._default(z.boolean(), false),
+    description: z.optional(z.nullable(z.string())),
+  }),
+  z.transform((v) => {
+    return remap$(v, {
+      allowedValues: "allowed_values",
+      objectSchema: "object_schema",
+    });
+  }),
+);
+
+export function fieldSchemaToJSON(fieldSchema: FieldSchema): string {
+  return JSON.stringify(FieldSchema$outboundSchema.parse(fieldSchema));
+}

package/src/models/gepa-job-request.ts

@@ -0,0 +1,55 @@
+/*
+ * 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 {
+  GepaKwargs,
+  GepaKwargs$Outbound,
+  GepaKwargs$outboundSchema,
+} from "./gepa-kwargs.js";
+
+export type GepaJobRequest = {
+  arbiterRepo: string;
+  arbiterRevision?: string | undefined;
+  pushBranch?: string | undefined;
+  pushTag?: string | null | undefined;
+  gepaKwargs?: GepaKwargs | undefined;
+};
+
+/** @internal */
+export type GepaJobRequest$Outbound = {
+  arbiter_repo: string;
+  arbiter_revision: string;
+  push_branch: string;
+  push_tag?: string | null | undefined;
+  gepa_kwargs?: GepaKwargs$Outbound | undefined;
+};
+
+/** @internal */
+export const GepaJobRequest$outboundSchema: z.ZodMiniType<
+  GepaJobRequest$Outbound,
+  GepaJobRequest
+> = z.pipe(
+  z.object({
+    arbiterRepo: z.string(),
+    arbiterRevision: z._default(z.string(), "main"),
+    pushBranch: z._default(z.string(), "main"),
+    pushTag: z.optional(z.nullable(z.string())),
+    gepaKwargs: z.optional(GepaKwargs$outboundSchema),
+  }),
+  z.transform((v) => {
+    return remap$(v, {
+      arbiterRepo: "arbiter_repo",
+      arbiterRevision: "arbiter_revision",
+      pushBranch: "push_branch",
+      pushTag: "push_tag",
+      gepaKwargs: "gepa_kwargs",
+    });
+  }),
+);
+
+export function gepaJobRequestToJSON(gepaJobRequest: GepaJobRequest): string {
+  return JSON.stringify(GepaJobRequest$outboundSchema.parse(gepaJobRequest));
+}

package/src/models/gepa-job-response.ts

@@ -0,0 +1,79 @@
+/*
+ * 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 { safeParse } from "../lib/schemas.js";
+import * as openEnums from "../types/enums.js";
+import { OpenEnum } from "../types/enums.js";
+import { Result as SafeParseResult } from "../types/fp.js";
+import * as types from "../types/primitives.js";
+import { SDKValidationError } from "./errors/sdk-validation-error.js";
+
+export const GepaJobResponseStatus = {
+  InProgress: "in_progress",
+  Completed: "completed",
+  Failed: "failed",
+} as const;
+export type GepaJobResponseStatus = OpenEnum<typeof GepaJobResponseStatus>;
+
+export const Phase = {
+  Optimizing: "optimizing",
+  Finalizing: "finalizing",
+  RePredicting: "re_predicting",
+  Scoring: "scoring",
+  Tuning: "tuning",
+  Done: "done",
+} as const;
+export type Phase = OpenEnum<typeof Phase>;
+
+export type GepaJobResponse = {
+  jobId: string;
+  status: GepaJobResponseStatus;
+  message?: string | null | undefined;
+  results?: { [k: string]: any } | null | undefined;
+  phase?: Phase | null | undefined;
+  commitHash?: string | null | undefined;
+};
+
+/** @internal */
+export const GepaJobResponseStatus$inboundSchema: z.ZodMiniType<
+  GepaJobResponseStatus,
+  unknown
+> = openEnums.inboundSchema(GepaJobResponseStatus);
+
+/** @internal */
+export const Phase$inboundSchema: z.ZodMiniType<Phase, unknown> = openEnums
+  .inboundSchema(Phase);
+
+/** @internal */
+export const GepaJobResponse$inboundSchema: z.ZodMiniType<
+  GepaJobResponse,
+  unknown
+> = z.pipe(
+  z.object({
+    job_id: types.string(),
+    status: GepaJobResponseStatus$inboundSchema,
+    message: z.optional(z.nullable(types.string())),
+    results: z.optional(z.nullable(z.record(z.string(), z.any()))),
+    phase: z.optional(z.nullable(Phase$inboundSchema)),
+    commit_hash: z.optional(z.nullable(types.string())),
+  }),
+  z.transform((v) => {
+    return remap$(v, {
+      "job_id": "jobId",
+      "commit_hash": "commitHash",
+    });
+  }),
+);
+
+export function gepaJobResponseFromJSON(
+  jsonString: string,
+): SafeParseResult<GepaJobResponse, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => GepaJobResponse$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'GepaJobResponse' from JSON`,
+  );
+}

package/src/models/gepa-kwargs.ts

@@ -0,0 +1,122 @@
+/*
+ * 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";
+import { Lm, Lm$Outbound, Lm$outboundSchema } from "./lm.js";
+
+export const Auto = {
+  Light: "light",
+  Medium: "medium",
+  Heavy: "heavy",
+} as const;
+export type Auto = ClosedEnum<typeof Auto>;
+
+export const CandidateSelectionStrategy = {
+  Pareto: "pareto",
+  CurrentBest: "current_best",
+} as const;
+export type CandidateSelectionStrategy = ClosedEnum<
+  typeof CandidateSelectionStrategy
+>;
+
+export const ComponentSelector = {
+  RoundRobin: "round_robin",
+  All: "all",
+} as const;
+export type ComponentSelector = ClosedEnum<typeof ComponentSelector>;
+
+export type GepaKwargs = {
+  auto?: Auto | null | undefined;
+  maxFullEvals?: number | null | undefined;
+  maxMetricCalls?: number | null | undefined;
+  reflectionMinibatchSize?: number | undefined;
+  candidateSelectionStrategy?: CandidateSelectionStrategy | undefined;
+  reflectionLm?: Lm | null | undefined;
+  skipPerfectScore?: boolean | undefined;
+  addFormatFailureAsFeedback?: boolean | undefined;
+  componentSelector?: ComponentSelector | undefined;
+  useMerge?: boolean | undefined;
+  maxMergeInvocations?: number | null | undefined;
+  seed?: number | null | undefined;
+  gepaKwargs?: { [k: string]: any } | null | undefined;
+};
+
+/** @internal */
+export const Auto$outboundSchema: z.ZodMiniEnum<typeof Auto> = z.enum(Auto);
+
+/** @internal */
+export const CandidateSelectionStrategy$outboundSchema: z.ZodMiniEnum<
+  typeof CandidateSelectionStrategy
+> = z.enum(CandidateSelectionStrategy);
+
+/** @internal */
+export const ComponentSelector$outboundSchema: z.ZodMiniEnum<
+  typeof ComponentSelector
+> = z.enum(ComponentSelector);
+
+/** @internal */
+export type GepaKwargs$Outbound = {
+  auto?: string | null | undefined;
+  max_full_evals?: number | null | undefined;
+  max_metric_calls?: number | null | undefined;
+  reflection_minibatch_size: number;
+  candidate_selection_strategy: string;
+  reflection_lm?: Lm$Outbound | null | undefined;
+  skip_perfect_score: boolean;
+  add_format_failure_as_feedback: boolean;
+  component_selector: string;
+  use_merge: boolean;
+  max_merge_invocations?: number | null | undefined;
+  seed?: number | null | undefined;
+  gepa_kwargs?: { [k: string]: any } | null | undefined;
+};
+
+/** @internal */
+export const GepaKwargs$outboundSchema: z.ZodMiniType<
+  GepaKwargs$Outbound,
+  GepaKwargs
+> = z.pipe(
+  z.object({
+    auto: z.optional(z.nullable(Auto$outboundSchema)),
+    maxFullEvals: z.optional(z.nullable(z.int())),
+    maxMetricCalls: z.optional(z.nullable(z.int())),
+    reflectionMinibatchSize: z._default(z.int(), 3),
+    candidateSelectionStrategy: z._default(
+      CandidateSelectionStrategy$outboundSchema,
+      "pareto",
+    ),
+    reflectionLm: z.optional(z.nullable(Lm$outboundSchema)),
+    skipPerfectScore: z._default(z.boolean(), true),
+    addFormatFailureAsFeedback: z._default(z.boolean(), false),
+    componentSelector: z._default(
+      ComponentSelector$outboundSchema,
+      "round_robin",
+    ),
+    useMerge: z._default(z.boolean(), true),
+    maxMergeInvocations: z.optional(z.nullable(z.int())),
+    seed: z.optional(z.nullable(z.int())),
+    gepaKwargs: z.optional(z.nullable(z.record(z.string(), z.any()))),
+  }),
+  z.transform((v) => {
+    return remap$(v, {
+      maxFullEvals: "max_full_evals",
+      maxMetricCalls: "max_metric_calls",
+      reflectionMinibatchSize: "reflection_minibatch_size",
+      candidateSelectionStrategy: "candidate_selection_strategy",
+      reflectionLm: "reflection_lm",
+      skipPerfectScore: "skip_perfect_score",
+      addFormatFailureAsFeedback: "add_format_failure_as_feedback",
+      componentSelector: "component_selector",
+      useMerge: "use_merge",
+      maxMergeInvocations: "max_merge_invocations",
+      gepaKwargs: "gepa_kwargs",
+    });
+  }),
+);
+
+export function gepaKwargsToJSON(gepaKwargs: GepaKwargs): string {
+  return JSON.stringify(GepaKwargs$outboundSchema.parse(gepaKwargs));
+}

package/src/models/get-confidence-request.ts

@@ -0,0 +1,45 @@
+/*
+ * 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";
+
+export type GetConfidenceRequest = {
+  arbiterRepo: string;
+  predictionId?: string | null | undefined;
+  messages: Array<{ [k: string]: any }>;
+};
+
+/** @internal */
+export type GetConfidenceRequest$Outbound = {
+  arbiter_repo: string;
+  prediction_id?: string | null | undefined;
+  messages: Array<{ [k: string]: any }>;
+};
+
+/** @internal */
+export const GetConfidenceRequest$outboundSchema: z.ZodMiniType<
+  GetConfidenceRequest$Outbound,
+  GetConfidenceRequest
+> = z.pipe(
+  z.object({
+    arbiterRepo: z.string(),
+    predictionId: z.optional(z.nullable(z.string())),
+    messages: z.array(z.record(z.string(), z.any())),
+  }),
+  z.transform((v) => {
+    return remap$(v, {
+      arbiterRepo: "arbiter_repo",
+      predictionId: "prediction_id",
+    });
+  }),
+);
+
+export function getConfidenceRequestToJSON(
+  getConfidenceRequest: GetConfidenceRequest,
+): string {
+  return JSON.stringify(
+    GetConfidenceRequest$outboundSchema.parse(getConfidenceRequest),
+  );
+}

package/src/models/index.ts

@@ -0,0 +1,36 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+export * from "./annotate-example-request.js";
+export * from "./arbiter-request.js";
+export * from "./batch-arbiter.js";
+export * from "./batch-example.js";
+export * from "./batch-predictions-job-request.js";
+export * from "./chat-completion-choice.js";
+export * from "./chat-completion-request.js";
+export * from "./chat-completion-response.js";
+export * from "./chat-message.js";
+export * from "./completion-usage.js";
+export * from "./confidence-score-job-request.js";
+export * from "./confidence-status-response.js";
+export * from "./delete-examples-request.js";
+export * from "./dispatch-prediction-request.js";
+export * from "./export-examples-request.js";
+export * from "./export-examples-response.js";
+export * from "./field-schema.js";
+export * from "./gepa-job-request.js";
+export * from "./gepa-job-response.js";
+export * from "./gepa-kwargs.js";
+export * from "./get-confidence-request.js";
+export * from "./init-arbiter-request.js";
+export * from "./lm.js";
+export * from "./paginated-page.js";
+export * from "./predict-example-response.js";
+export * from "./predicted-example.js";
+export * from "./prediction-annotation.js";
+export * from "./security.js";
+export * from "./src-api-v1-arbiters-schemas-predict-example-request.js";
+export * from "./src-api-v2-arbiters-schemas-predict-example-request.js";
+export * from "./update-arbiter-metadata-request.js";
+export * from "./validation-error.js";

package/src/models/init-arbiter-request.ts

@@ -0,0 +1,58 @@
+/*
+ * 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 {
+  FieldSchema,
+  FieldSchema$Outbound,
+  FieldSchema$outboundSchema,
+} from "./field-schema.js";
+
+export type InitArbiterRequest = {
+  repo: string;
+  inputs: Array<FieldSchema>;
+  outputs: Array<FieldSchema>;
+  instructions?: string | null | undefined;
+  model?: string | undefined;
+  baseUrl?: string | null | undefined;
+};
+
+/** @internal */
+export type InitArbiterRequest$Outbound = {
+  repo: string;
+  inputs: Array<FieldSchema$Outbound>;
+  outputs: Array<FieldSchema$Outbound>;
+  instructions?: string | null | undefined;
+  model: string;
+  base_url?: string | null | undefined;
+};
+
+/** @internal */
+export const InitArbiterRequest$outboundSchema: z.ZodMiniType<
+  InitArbiterRequest$Outbound,
+  InitArbiterRequest
+> = z.pipe(
+  z.object({
+    repo: z.string(),
+    inputs: z.array(FieldSchema$outboundSchema),
+    outputs: z.array(FieldSchema$outboundSchema),
+    instructions: z.optional(z.nullable(z.string())),
+    model: z._default(z.string(), "qwen3-vl-32b-instruct"),
+    baseUrl: z.optional(z.nullable(z.string())),
+  }),
+  z.transform((v) => {
+    return remap$(v, {
+      baseUrl: "base_url",
+    });
+  }),
+);
+
+export function initArbiterRequestToJSON(
+  initArbiterRequest: InitArbiterRequest,
+): string {
+  return JSON.stringify(
+    InitArbiterRequest$outboundSchema.parse(initArbiterRequest),
+  );
+}