mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-10-07 00:02:12 +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 { 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>
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue