Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
64 changes: 64 additions & 0 deletions console-ui/__tests__/api.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ import {
fetchModels,
fetchPricing,
healthCheck,
streamWeaverChat,
} from "@/lib/api";

// ---------------------------------------------------------------------------
Expand All @@ -24,6 +25,21 @@ function jsonResponse(body: unknown, status = 200): Response {
} as unknown as Response;
}

function sseResponse(events: string): Response {
const encoder = new TextEncoder();
return {
ok: true,
status: 200,
headers: new Headers({ "content-type": "text/event-stream" }),
body: new ReadableStream({
start(controller) {
controller.enqueue(encoder.encode(events));
controller.close();
},
}),
} as unknown as Response;
}

// ---------------------------------------------------------------------------
// Setup
// ---------------------------------------------------------------------------
Expand Down Expand Up @@ -262,6 +278,54 @@ describe("healthCheck", () => {
});
});

describe("streamWeaverChat", () => {
it("sends weaver options and routes orchestration SSE events", async () => {
fetchMock.mockResolvedValueOnce(sseResponse([
'event: weaver_init\ndata: {"mode":"deep","status":"generating","candidateCount":4,"verifierCount":3,"candidates":[],"verifierScores":[]}\n\n',
'event: candidate_delta\ndata: {"candidate_id":"candidate-2","delta":"hello"}\n\n',
'event: verifier_score\ndata: {"candidate_id":"candidate-2","verifier_model":"verifier-a","score":8,"rationale":"clear"}\n\n',
'event: weaver_final\ndata: {"selected_candidate":"candidate-2","confidence":0.8,"content":"final answer","usage":{"tokenCount":12,"tps":0,"ttft":0}}\n\n',
'event: done\ndata: {"status":"complete"}\n\n',
].join("")));

const seen: string[] = [];
await streamWeaverChat(
[{ role: "user", content: "hi" }],
"base-model",
{ mode: "deep", candidateCount: 4, verifierCount: 3 },
{
onWeaverInit: (trace) => seen.push(`init:${trace.candidateCount}`),
onWeaverStatus: (status) => seen.push(`status:${status}`),
onCandidateStart: () => {},
onCandidateDelta: (id, delta) => seen.push(`delta:${id}:${delta}`),
onCandidateDone: () => {},
onCandidateError: () => {},
onVerifierScore: (score) => seen.push(`score:${score.candidateId}:${score.score}`),
onFinal: (content, selected, confidence, usage) => seen.push(`final:${selected}:${confidence}:${usage?.tokenCount}:${content}`),
onDone: () => seen.push("done"),
onError: (error) => seen.push(`error:${error}`),
}
);

const [, opts] = fetchMock.mock.calls[0];
const body = JSON.parse(opts.body);
expect(body.weaver).toEqual({
mode: "deep",
candidate_count: 4,
verifier_count: 3,
verifier_selection: "auto",
trace: true,
});
expect(seen).toEqual([
"init:4",
"delta:candidate-2:hello",
"score:candidate-2:8",
"final:candidate-2:0.8:12:final answer",
"done",
]);
});
});

// ---------------------------------------------------------------------------
// proxyHeaders
// ---------------------------------------------------------------------------
Expand Down
86 changes: 85 additions & 1 deletion console-ui/__tests__/components.test.tsx
Original file line number Diff line number Diff line change
@@ -1,7 +1,12 @@
import { describe, it, expect } from "vitest";
import { render, screen } from "@testing-library/react";
import { fireEvent, render, screen } from "@testing-library/react";
import { ChatInput } from "@/components/ChatInput";
import { WeaverTraceStrip } from "@/components/WeaverTraceStrip";
import { WeaverTraceModal } from "@/components/WeaverTraceModal";
import { TrustBadge } from "@/components/TrustBadge";
import { VerificationModeProvider } from "@/lib/verification-mode";
import { useStore } from "@/lib/store";
import type { WeaverTrace } from "@/lib/api";
import type { TrustMetadata } from "@/lib/api";

// ---------------------------------------------------------------------------
Expand Down Expand Up @@ -93,3 +98,82 @@ describe("TrustBadge (normal mode)", () => {
expect(span!.getAttribute("title")).toBe("Apple-verified hardware");
});
});

describe("WeaverTraceStrip", () => {
it("renders one dynamic diamond per candidate", () => {
const weaver: WeaverTrace = {
mode: "deep_plus",
status: "generating",
candidateCount: 8,
verifierCount: 5,
candidates: Array.from({ length: 8 }, (_, index) => ({
id: `candidate-${index + 1}`,
index,
label: `Agent ${index + 1}`,
model: "base",
status: index === 0 ? "streaming" : "queued",
content: "",
})),
verifierScores: [],
};

render(<WeaverTraceStrip weaver={weaver} onOpen={() => {}} />);

expect(screen.getByText("Deep+ · generating")).toBeInTheDocument();
expect(screen.getAllByLabelText(/Agent \d/)).toHaveLength(8);
});

it("renders dynamic agent tabs in the modal", () => {
const weaver: WeaverTrace = {
mode: "deep_plus",
status: "complete",
candidateCount: 8,
verifierCount: 5,
selectedCandidate: "candidate-3",
confidence: 0.7,
candidates: Array.from({ length: 8 }, (_, index) => ({
id: `candidate-${index + 1}`,
index,
label: `Agent ${index + 1}`,
model: "base",
status: index === 2 ? "selected" : "done",
content: `answer ${index + 1}`,
})),
verifierScores: [],
};

render(
<WeaverTraceModal
weaver={weaver}
finalContent="final"
usage={{ tokenCount: 12 }}
onClose={() => {}}
onSelectTab={() => {}}
/>,
);

expect(screen.getAllByRole("button", { name: /Agent \d/ })).toHaveLength(8);
expect(screen.getByRole("button", { name: "Verifiers" })).toBeInTheDocument();
expect(screen.getByRole("button", { name: "Final" })).toBeInTheDocument();
expect(screen.getByRole("dialog", { name: "Weaver trace" })).toHaveClass("h-[min(760px,86vh)]");
});
});

describe("ChatInput", () => {
it("opens the model selector even when models have not loaded", () => {
useStore.setState({ models: [], selectedModel: "" });

render(
<ChatInput
onSend={() => {}}
onStop={() => {}}
isStreaming={false}
/>,
);

fireEvent.click(screen.getByRole("button", { name: "Select model" }));

expect(screen.getByText("No models loaded")).toBeInTheDocument();
expect(screen.getByText("Check your API key or coordinator connection.")).toBeInTheDocument();
});
});
36 changes: 36 additions & 0 deletions console-ui/src/app/api/device/approve/route.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,36 @@
import { NextRequest, NextResponse } from "next/server";

const DEFAULT_COORD = process.env.NEXT_PUBLIC_COORDINATOR_URL || "https://api.darkbloom.dev";

export async function POST(req: NextRequest) {
let authHeader = req.headers.get("authorization") || "";
if (!authHeader) {
const privyToken = req.cookies.get("privy-token")?.value;
if (privyToken) {
authHeader = `Bearer ${privyToken}`;
}
}
if (!authHeader) {
return NextResponse.json({ error: "missing privy token" }, { status: 401 });
}

const res = await fetch(`${DEFAULT_COORD}/v1/device/approve`, {
method: "POST",
headers: {
"Content-Type": "application/json",
Authorization: authHeader,
},
body: JSON.stringify(await req.json()),
});
const text = await res.text();
if (!res.ok) {
return new Response(text, {
status: res.status,
headers: { "Content-Type": res.headers.get("content-type") || "application/json" },
});
}
return new Response(text, {
status: res.status,
headers: { "Content-Type": res.headers.get("content-type") || "application/json" },
});
}
30 changes: 30 additions & 0 deletions console-ui/src/app/api/provider/account-earnings/route.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
import { NextRequest, NextResponse } from "next/server";

const DEFAULT_COORD = process.env.NEXT_PUBLIC_COORDINATOR_URL || "https://api.darkbloom.dev";

export async function GET(req: NextRequest) {
let authHeader = req.headers.get("authorization") || "";
if (!authHeader) {
const privyToken = req.cookies.get("privy-token")?.value;
if (privyToken) {
authHeader = `Bearer ${privyToken}`;
}
}
if (!authHeader) {
return NextResponse.json({ error: "missing auth token" }, { status: 401 });
}

const limit = req.nextUrl.searchParams.get("limit") || "100";
const upstream = new URL(`${DEFAULT_COORD}/v1/provider/account-earnings`);
upstream.searchParams.set("limit", limit);

const res = await fetch(upstream, {
headers: { Authorization: authHeader },
cache: "no-store",
});
if (!res.ok) {
const text = await res.text();
return NextResponse.json({ error: text || `Upstream ${res.status}` }, { status: res.status });
}
return NextResponse.json(await res.json());
}
15 changes: 15 additions & 0 deletions console-ui/src/app/api/providers/attestation/route.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
import { NextResponse } from "next/server";

const DEFAULT_COORD =
process.env.NEXT_PUBLIC_COORDINATOR_URL || "https://api.darkbloom.dev";

export async function GET() {
const res = await fetch(`${DEFAULT_COORD}/v1/providers/attestation`);
if (!res.ok) {
return NextResponse.json(
{ error: `Upstream ${res.status}` },
{ status: res.status }
);
}
return NextResponse.json(await res.json());
}
12 changes: 12 additions & 0 deletions console-ui/src/app/api/stats/route.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
import { NextResponse } from "next/server";

const DEFAULT_COORD = process.env.NEXT_PUBLIC_COORDINATOR_URL || "https://api.darkbloom.dev";

export async function GET() {
const res = await fetch(`${DEFAULT_COORD}/v1/stats`, { cache: "no-store" });
if (!res.ok) {
const text = await res.text();
return NextResponse.json({ error: text || `Upstream ${res.status}` }, { status: res.status });
}
return NextResponse.json(await res.json());
}
6 changes: 1 addition & 5 deletions console-ui/src/app/link/DeviceLinkForm.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,6 @@ import { useState, useCallback } from "react";
import { useAuthContext } from "@/components/providers/PrivyClientProvider";
import { trackEvent } from "@/lib/google-analytics";

const COORDINATOR_URL =
process.env.NEXT_PUBLIC_COORDINATOR_URL ||
"https://api.darkbloom.dev";

type LinkStatus = "idle" | "submitting" | "success" | "error";

export function DeviceLinkForm() {
Expand Down Expand Up @@ -39,7 +35,7 @@ export function DeviceLinkForm() {
return;
}

const res = await fetch(`${COORDINATOR_URL}/v1/device/approve`, {
const res = await fetch("/api/device/approve", {
method: "POST",
headers: {
"Content-Type": "application/json",
Expand Down
Loading