mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 08:03:15 +00:00
chore: Fix formatting (#557)
This commit is contained in:
parent
d89b37c568
commit
263a3010b0
1 changed files with 28 additions and 7 deletions
|
|
@ -11,7 +11,11 @@ import { ThreadMetadata } from "./types";
|
|||
import { useStream } from "@langchain/langgraph-sdk/react";
|
||||
import { ManagerGraphState } from "@open-swe/shared/open-swe/manager/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 { ThemeToggle } from "../theme-toggle";
|
||||
import { HumanMessage } from "@langchain/core/messages";
|
||||
|
|
@ -47,24 +51,38 @@ interface ThreadViewProps {
|
|||
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) {
|
||||
return [];
|
||||
}
|
||||
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) {
|
||||
return Array.isArray(plannerTokenData) ? plannerTokenData : [plannerTokenData];
|
||||
return Array.isArray(plannerTokenData)
|
||||
? plannerTokenData
|
||||
: [plannerTokenData];
|
||||
}
|
||||
|
||||
if (!plannerTokenData && programmerTokenData) {
|
||||
return Array.isArray(programmerTokenData) ? programmerTokenData : [programmerTokenData];
|
||||
return Array.isArray(programmerTokenData)
|
||||
? programmerTokenData
|
||||
: [programmerTokenData];
|
||||
}
|
||||
|
||||
return [];
|
||||
}
|
||||
};
|
||||
|
||||
export function ThreadView({
|
||||
stream,
|
||||
|
|
@ -360,7 +378,10 @@ export function ThreadView({
|
|||
/>
|
||||
)}
|
||||
<TokenUsage
|
||||
tokenData={joinTokenData(plannerStream.values.tokenData, programmerStream.values.tokenData)}
|
||||
tokenData={joinTokenData(
|
||||
plannerStream.values.tokenData,
|
||||
programmerStream.values.tokenData,
|
||||
)}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue