chore: Fix formatting (#557)

This commit is contained in:
Brace Sproul 2025-07-27 17:25:57 -07:00 • committed by GitHub
parent d89b37c568
commit 263a3010b0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -11,7 +11,11 @@ import { ThreadMetadata } from "./types";
import { useStream } from "@langchain/langgraph-sdk/react"; import { useStream } from "@langchain/langgraph-sdk/react";
import { ManagerGraphState } from "@open-swe/shared/open-swe/manager/types"; import { ManagerGraphState } from "@open-swe/shared/open-swe/manager/types";
import { PlannerGraphState } from "@open-swe/shared/open-swe/planner/types"; import { PlannerGraphState } from "@open-swe/shared/open-swe/planner/types";
import { GraphState, CacheMetrics, ModelTokenData } from "@open-swe/shared/open-swe/types"; import {
GraphState,
CacheMetrics,
ModelTokenData,
} from "@open-swe/shared/open-swe/types";
import { ActionsRenderer } from "./actions-renderer"; import { ActionsRenderer } from "./actions-renderer";
import { ThemeToggle } from "../theme-toggle"; import { ThemeToggle } from "../theme-toggle";
import { HumanMessage } from "@langchain/core/messages"; import { HumanMessage } from "@langchain/core/messages";
@ -47,24 +51,38 @@ interface ThreadViewProps {
onBackToHome: () => void; onBackToHome: () => void;
} }
const joinTokenData = (plannerTokenData?: CacheMetrics | ModelTokenData[], programmerTokenData?: CacheMetrics | ModelTokenData[]): ModelTokenData[] | CacheMetrics[] => { const joinTokenData = (
plannerTokenData?: CacheMetrics | ModelTokenData[],
programmerTokenData?: CacheMetrics | ModelTokenData[],
): ModelTokenData[] | CacheMetrics[] => {
if (!plannerTokenData && !programmerTokenData) { if (!plannerTokenData && !programmerTokenData) {
return []; return [];
} }
if (plannerTokenData && programmerTokenData) { if (plannerTokenData && programmerTokenData) {
return [...(Array.isArray(plannerTokenData) ? plannerTokenData : [plannerTokenData]), ...(Array.isArray(programmerTokenData) ? programmerTokenData : [programmerTokenData])]; return [
...(Array.isArray(plannerTokenData)
? plannerTokenData
: [plannerTokenData]),
...(Array.isArray(programmerTokenData)
? programmerTokenData
: [programmerTokenData]),
];
} }
if (plannerTokenData && !programmerTokenData) { if (plannerTokenData && !programmerTokenData) {
return Array.isArray(plannerTokenData) ? plannerTokenData : [plannerTokenData]; return Array.isArray(plannerTokenData)
? plannerTokenData
: [plannerTokenData];
} }
if (!plannerTokenData && programmerTokenData) { if (!plannerTokenData && programmerTokenData) {
return Array.isArray(programmerTokenData) ? programmerTokenData : [programmerTokenData]; return Array.isArray(programmerTokenData)
? programmerTokenData
: [programmerTokenData];
} }
return []; return [];
} };
export function ThreadView({ export function ThreadView({
stream, stream,
@ -360,7 +378,10 @@ export function ThreadView({
/> />
)} )}
<TokenUsage <TokenUsage
tokenData={joinTokenData(plannerStream.values.tokenData, programmerStream.values.tokenData)} tokenData={joinTokenData(
plannerStream.values.tokenData,
programmerStream.values.tokenData,
)}
/> />
</div> </div>
</div> </div>