stagent 0.1.11 → 0.1.12

package/src/lib/settings/runtime-setup.ts

@@ -0,0 +1,71 @@
+import {
+  getRuntimeCatalogEntry,
+  SUPPORTED_AGENT_RUNTIMES,
+  type AgentRuntimeId,
+} from "@/lib/agents/runtime/catalog";
+import { getAuthSettings } from "./auth";
+import { getOpenAIAuthSettings } from "./openai-auth";
+import type { ApiKeySource, AuthMethod } from "@/lib/constants/settings";
+
+export type RuntimeBillingMode = "usage" | "subscription";
+export type RuntimeSetupMethod = AuthMethod | "none";
+
+export interface RuntimeSetupState {
+  runtimeId: AgentRuntimeId;
+  label: string;
+  providerId: "anthropic" | "openai";
+  configured: boolean;
+  authMethod: RuntimeSetupMethod;
+  apiKeySource: ApiKeySource;
+  billingMode: RuntimeBillingMode;
+}
+
+export async function getRuntimeSetupStates(): Promise<
+  Record<AgentRuntimeId, RuntimeSetupState>
+> {
+  const [claudeAuth, openAIAuth] = await Promise.all([
+    getAuthSettings(),
+    getOpenAIAuthSettings(),
+  ]);
+
+  const claudeRuntime = getRuntimeCatalogEntry("claude-code");
+  const openAIRuntime = getRuntimeCatalogEntry("openai-codex-app-server");
+
+  const claudeAuthMethod: RuntimeSetupMethod =
+    claudeAuth.method === "oauth" || claudeAuth.apiKeySource === "oauth"
+      ? "oauth"
+      : claudeAuth.hasKey
+        ? "api_key"
+        : "none";
+  const claudeConfigured =
+    claudeAuth.hasKey || claudeAuth.apiKeySource === "oauth";
+
+  const states = {
+    "claude-code": {
+      runtimeId: "claude-code",
+      label: claudeRuntime.label,
+      providerId: claudeRuntime.providerId,
+      configured: claudeConfigured,
+      authMethod: claudeAuthMethod,
+      apiKeySource: claudeAuth.apiKeySource,
+      billingMode: claudeAuthMethod === "oauth" ? "subscription" : "usage",
+    },
+    "openai-codex-app-server": {
+      runtimeId: "openai-codex-app-server",
+      label: openAIRuntime.label,
+      providerId: openAIRuntime.providerId,
+      configured: openAIAuth.hasKey,
+      authMethod: openAIAuth.hasKey ? "api_key" : "none",
+      apiKeySource: openAIAuth.apiKeySource,
+      billingMode: "usage",
+    },
+  } satisfies Record<AgentRuntimeId, RuntimeSetupState>;
+
+  return states;
+}
+
+export function listConfiguredRuntimeIds(
+  states: Record<AgentRuntimeId, RuntimeSetupState>
+) {
+  return SUPPORTED_AGENT_RUNTIMES.filter((runtimeId) => states[runtimeId].configured);
+}

package/src/lib/usage/__tests__/ledger.test.ts

@@ -85,13 +85,13 @@ describe("usage ledger", () => {
     const priced = rows.find((row) => row.modelId === "claude-sonnet-4-20250514");
     expect(priced?.costMicros).toBe(10_500);
     expect(priced?.status).toBe("completed");
-    expect(priced?.pricingVersion).toBe("registry-2026-03-15");
+    expect(priced?.pricingVersion).toBe("anthropic-claude-sonnet");
 
     // Unknown model: fallback pricing (conservative Opus-tier for OpenAI: $10/$30)
     const fallback = rows.find((row) => row.modelId === "codex-unknown");
     expect(fallback?.costMicros).toBeGreaterThan(0);
     expect(fallback?.status).toBe("completed");
-    expect(fallback?.pricingVersion).toBe("registry-2026-03-15-fallback");
+    expect(fallback?.pricingVersion).toBe("openai-fallback");
 
     // Null modelId: truly unknown
     const unknown = rows.find((row) => row.modelId === null);

package/src/lib/usage/__tests__/pricing-registry.test.ts

@@ -0,0 +1,78 @@
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { mkdtempSync, rmSync } from "fs";
+import { join } from "path";
+import { tmpdir } from "os";
+
+let tempDir: string;
+
+beforeEach(() => {
+  tempDir = mkdtempSync(join(tmpdir(), "stagent-pricing-registry-"));
+  vi.resetModules();
+  vi.stubEnv("STAGENT_DATA_DIR", tempDir);
+});
+
+afterEach(() => {
+  vi.unstubAllEnvs();
+  vi.unstubAllGlobals();
+  rmSync(tempDir, { recursive: true, force: true });
+});
+
+describe("pricing registry", () => {
+  it("refreshes pricing from official provider pages and updates visible rows", async () => {
+    const fetch = vi.fn(async (input: string) => {
+      if (input.includes("anthropic.com")) {
+        return new Response(`
+          <html>
+            <body>
+              Claude Sonnet 4 $3 / 1M input tokens $15 / 1M output tokens
+              Claude Opus 4 $15 / 1M input tokens $75 / 1M output tokens
+              Claude Haiku 3.5 $0.80 / 1M input tokens $4 / 1M output tokens
+              Claude Pro $20
+              Max 5x $100
+              Max 20x $200
+            </body>
+          </html>
+        `);
+      }
+
+      return new Response(`
+        <html>
+          <body>
+            GPT-5 $10 / 1M input tokens $30 / 1M output tokens
+            GPT-4o $2.50 / 1M input tokens $10 / 1M output tokens
+          </body>
+        </html>
+      `);
+    });
+    vi.stubGlobal("fetch", fetch);
+
+    const { refreshPricingRegistry } = await import("../pricing-registry");
+    const snapshot = await refreshPricingRegistry();
+
+    expect(snapshot.providers.anthropic.rows.find((row) => row.key === "anthropic-plan-pro")?.monthlyPriceUsd).toBe(20);
+    expect(snapshot.providers.anthropic.rows.find((row) => row.key === "anthropic-claude-sonnet")?.inputCostPerMillionMicros).toBe(3_000_000);
+    expect(snapshot.providers.openai.rows.find((row) => row.key === "openai-gpt-5")?.outputCostPerMillionMicros).toBe(30_000_000);
+  });
+
+  it("keeps last-known-good pricing when a refresh fails", async () => {
+    const fetch = vi.fn(async (input: string) => {
+      if (input.includes("anthropic.com")) {
+        return new Response(`<html><body>Claude Pro $20</body></html>`);
+      }
+
+      throw new Error("network down");
+    });
+    vi.stubGlobal("fetch", fetch);
+
+    const { refreshPricingRegistry, getPricingRegistrySnapshot } = await import(
+      "../pricing-registry"
+    );
+
+    await refreshPricingRegistry();
+    const snapshot = await getPricingRegistrySnapshot();
+
+    expect(snapshot.providers.anthropic.refreshError).toBeNull();
+    expect(snapshot.providers.openai.refreshError).toContain("network down");
+    expect(snapshot.providers.openai.rows.find((row) => row.key === "openai-gpt-5")?.inputCostPerMillionMicros).toBe(10_000_000);
+  });
+});

package/src/lib/usage/ledger.ts

@@ -207,7 +207,7 @@ export async function recordUsageLedgerEntry(input: UsageLedgerWriteInput) {
     (normalizedInputTokens != null && normalizedOutputTokens != null
       ? normalizedInputTokens + normalizedOutputTokens
       : null);
-  const { costMicros, pricingVersion } = deriveUsageCostMicros({
+  const { costMicros, pricingVersion } = await deriveUsageCostMicros({
     providerId: input.providerId,
     modelId: input.modelId,
     inputTokens: normalizedInputTokens,