@zaip-ai/cli 1.0.8 → 1.3.1

package/templates/.agents/dashboard/packages/local-runtime/src/sqlite-read-model-multi-provider-fixture.ts

@@ -0,0 +1,227 @@
+import { mkdir, writeFile } from "node:fs/promises";
+import { join } from "node:path";
+
+export async function prepareMultiProviderFixture(root: string) {
+  await mkdir(join(root, ".agents", "records", "runs", "2026", "08"), { recursive: true });
+  await mkdir(join(root, ".agents", "config"), { recursive: true });
+  await mkdir(join(root, ".agents", "config", "agents"), { recursive: true });
+  await mkdir(join(root, ".agents", "config", "pricing"), { recursive: true });
+  await writeFile(join(root, ".agents", "config", "project.yaml"), "zaip:\n  enabled: true\n");
+  await writeFile(
+    join(root, ".agents", "config", "agents", "catalog.yaml"),
+    "agents:\n  - id: agent-catalog\n    codex_name: agent\n    persona: { nome: ana, cargo: quality_engineer }\n    role: reviewer\n",
+  );
+  await writeFile(
+    join(root, ".agents", "config", "pricing", "models.json"),
+    JSON.stringify({
+      schema_version: 1,
+      updated_at: "2026-08-04T00:00:00.000Z",
+      source: "fixture",
+      disclaimer: "fixture",
+      models: [
+        {
+          id: "gpt-a",
+          provider: "openai",
+          aliases: [],
+          input_per_million_usd: 2,
+          cached_input_per_million_usd: 0.2,
+          output_per_million_usd: 8,
+          reasoning_per_million_usd: 8,
+          reasoning_billed_separately: false,
+          long_context_threshold_tokens: null,
+          long_context_input_multiplier: 1,
+          long_context_output_multiplier: 1,
+          source_label: "fixture-openai",
+          source_url: "https://example.test/openai",
+          notes: [],
+        },
+        {
+          id: "claude-b",
+          provider: "anthropic",
+          aliases: [],
+          input_per_million_usd: 3,
+          cached_input_per_million_usd: 0.3,
+          output_per_million_usd: 12,
+          reasoning_per_million_usd: null,
+          reasoning_billed_separately: false,
+          long_context_threshold_tokens: null,
+          long_context_input_multiplier: 1,
+          long_context_output_multiplier: 1,
+          source_label: "fixture-anthropic",
+          source_url: "https://example.test/anthropic",
+          notes: [],
+        },
+      ],
+    }),
+  );
+  const started = {
+    schema_version: 2,
+    event_id: "RUN-MULTI:1",
+    run_id: "RUN-MULTI",
+    sequence: 1,
+    occurred_at: "2026-08-04T12:00:00.000Z",
+    type: "run.started",
+    attribution_status: "attributed",
+    phase: "execution",
+    run_role: "orchestrator",
+    run_purpose: "execution",
+    requires_finalizer: false,
+    execution_kind: "delivery",
+    execution_eligibility: ["delivery"],
+    entrypoint: "task",
+    work_type: "new",
+    storage_mode: "tracked_sanitized",
+    client: "codex",
+    agent: "agent",
+    agent_name: "agent",
+    provider: "mixed",
+    model: "mixed",
+    reasoning_effort: null,
+    client_thread_id: "multi",
+    client_turn_id: "multi",
+    conversation_id: "CONV-MULTI",
+    execution_id: "EXEC-MULTI",
+    task_id: "TASK-MULTI",
+    spec_id: null,
+    prd_id: null,
+    change_id: null,
+    parent_run_id: null,
+    continuation_of_run_id: null,
+    retry_of_run_id: null,
+    agent_invocation_id: "AINV-MULTI",
+    root_agent_invocation_id: "AINV-MULTI",
+    parent_agent_invocation_id: null,
+    invocation_kind: "orchestrator",
+    delegation_depth: 0,
+  };
+  const usage = (
+    sequence: number,
+    provider: string,
+    model: string,
+    tokens: number,
+    brl: number,
+  ) => ({
+    schema_version: 2,
+    event_id: `RUN-MULTI:${sequence}`,
+    run_id: "RUN-MULTI",
+    sequence,
+    occurred_at: "2026-08-04T12:00:01.000Z",
+    type: "usage.recorded",
+    phase: "execution",
+    activity_kind: "model_usage",
+    classification_source: "explicit",
+    tokens: {
+      input: tokens,
+      cached_input: 0,
+      output: 0,
+      reasoning: 0,
+      total: tokens,
+      source: "provider_reported",
+    },
+    cost: { usd: brl / 5, brl, fx_rate: 5 },
+    provider,
+    model,
+    source_segment_key: `${provider}-${model}`,
+    model_call_id: `MCALL-${sequence}`,
+    runtime: { reasoning_effort: null, service_tier: null },
+    supersedes_event_id: null,
+    source_started_at: "2026-08-04T12:00:00.000Z",
+    source_finished_at: "2026-08-04T12:01:00.000Z",
+    duration_ms: null,
+    duration_provenance: null,
+  });
+  const finished = {
+    schema_version: 2,
+    event_id: "RUN-MULTI:5",
+    run_id: "RUN-MULTI",
+    sequence: 5,
+    occurred_at: "2026-08-04T12:01:00.000Z",
+    type: "run.finished",
+    phase: "execution",
+    status: "completed",
+    duration_ms: 60_000,
+    duration_provenance: "observed",
+  };
+  const mixedNullSegment = {
+    ...usage(4, "openai", "gpt-a", 0, 0),
+    event_id: "RUN-MULTI:4",
+    provider: null,
+    model: null,
+    source_segment_key: "mixed-null",
+    model_call_id: "MCALL-MIXED-NULL",
+  };
+  await writeFile(
+    join(root, ".agents", "records", "runs", "2026", "08", "RUN-MULTI.jsonl"),
+    `${[started, usage(2, "openai", "gpt-a", 10, 5), usage(3, "anthropic", "claude-b", 20, 10), mixedNullSegment, finished].map((row) => JSON.stringify(row)).join("\n")}\n`,
+  );
+  const unavailableStarted = {
+    ...started,
+    event_id: "RUN-UNAVAILABLE:1",
+    run_id: "RUN-UNAVAILABLE",
+    occurred_at: "2026-08-04T14:00:00.000Z",
+    client_turn_id: "unavailable",
+    conversation_id: "CONV-UNAVAILABLE",
+    execution_id: "EXEC-UNAVAILABLE",
+    task_id: "TASK-UNAVAILABLE",
+    agent_invocation_id: "AINV-UNAVAILABLE",
+    root_agent_invocation_id: "AINV-UNAVAILABLE",
+    provider: "unavailable",
+    model: "unavailable",
+  };
+  const unavailableFinished = {
+    ...finished,
+    event_id: "RUN-UNAVAILABLE:2",
+    run_id: "RUN-UNAVAILABLE",
+    sequence: 3,
+    occurred_at: "2026-08-04T14:01:00.000Z",
+  };
+  const unavailableUsage = {
+    ...usage(2, "openai", "gpt-a", 0, 0),
+    event_id: "RUN-UNAVAILABLE:usage",
+    run_id: "RUN-UNAVAILABLE",
+    sequence: 2,
+    occurred_at: "2026-08-04T14:00:01.000Z",
+    provider: null,
+    model: null,
+    source_started_at: "2026-08-04T14:00:00.000Z",
+    source_finished_at: "2026-08-04T14:01:00.000Z",
+    source_segment_key: "unknown-segment",
+    model_call_id: "MCALL-UNAVAILABLE",
+  };
+  await writeFile(
+    join(root, ".agents", "records", "runs", "2026", "08", "RUN-UNAVAILABLE.jsonl"),
+    `${[unavailableStarted, unavailableUsage, unavailableFinished].map((row) => JSON.stringify(row)).join("\n")}\n`,
+  );
+  // It is a canonical activity but not an Agent Invocation. Cost Monitor
+  // must not blend this population into its token/cost/session rate.
+  const {
+    agent_invocation_id: _invocation,
+    root_agent_invocation_id: _rootInvocation,
+    parent_agent_invocation_id: _parentInvocation,
+    ...unassistedBase
+  } = started;
+  const unassistedStarted = {
+    ...unassistedBase,
+    event_id: "RUN-UNASSISTED:1",
+    run_id: "RUN-UNASSISTED",
+    occurred_at: "2026-08-04T16:00:00.000Z",
+    client_turn_id: "unassisted",
+    conversation_id: "CONV-UNASSISTED",
+    execution_id: "EXEC-MULTI",
+    task_id: "TASK-MULTI",
+  };
+  const unassistedUsage = {
+    ...usage(2, "openai", "gpt-a", 1_000, 500),
+    event_id: "RUN-UNASSISTED:2",
+    run_id: "RUN-UNASSISTED",
+    occurred_at: "2026-08-04T16:00:01.000Z",
+    source_started_at: "2026-08-04T16:00:00.000Z",
+    source_finished_at: "2026-08-04T16:01:00.000Z",
+    source_segment_key: "unassisted-segment",
+  };
+  await writeFile(
+    join(root, ".agents", "records", "runs", "2026", "08", "RUN-UNASSISTED.jsonl"),
+    `${[unassistedStarted, unassistedUsage, { ...finished, event_id: "RUN-UNASSISTED:3", run_id: "RUN-UNASSISTED", sequence: 3 }].map((row) => JSON.stringify(row)).join("\n")}\n`,
+  );
+  return { started, usage, finished };
+}