adam/open-swe
1
0
Fork 0
mirror of https://github.com/Sea-Haven-Industries/open-swe.git synced 2026-09-30 09:13:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: Pass proper config when retrying run (#808)

Browse source 
* fix: Pass proper config when retrying run

* fix
This commit is contained in:
Brace Sproul 2025-08-27 12:49:44 -07:00 • committed by GitHub
parent 504be7c563
commit d81080e71e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 30 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
apps/web/src/app/api/restart-run/route.ts
View file

@ -20,7 +20,11 @@ import { RestartRunRequest } from "./types";
import { Client, StreamMode, ThreadState } from "@langchain/langgraph-sdk";
import { ManagerGraphState } from "@openswe/shared/open-swe/manager/types";
import { PlannerGraphState } from "@openswe/shared/open-swe/planner/types";
import { AgentSession, GraphState } from "@openswe/shared/open-swe/types";
import {
AgentSession,
GraphConfig,
GraphState,
} from "@openswe/shared/open-swe/types";
import { END } from "@langchain/langgraph/web";
import { getCustomConfigurableFields } from "@openswe/shared/open-swe/utils/config";
@ -60,10 +64,12 @@ async function createNewSession(
threadState: ThreadState<
ManagerGraphState | PlannerGraphState | GraphState
>;
threadConfig: GraphConfig;
},
): Promise<AgentSession> {
const newThreadId = uuidv4();
const hasNext = inputs.threadState.next.length > 0;
const run = await client.runs.create(newThreadId, inputs.graphId, {
command: {
update: inputs.threadState.values,
@ -74,9 +80,7 @@ async function createNewSession(
streamResumable: true,
config: {
recursion_limit: 400,
configurable: getCustomConfigurableFields(
inputs.threadState.metadata as Record<string, any>,
),
configurable: getCustomConfigurableFields(inputs.threadConfig),
},
});
return {
@ -101,14 +105,25 @@ export async function POST(request: NextRequest): Promise<NextResponse> {
defaultHeaders: await getRequestHeaders(request),
});
const [managerThreadState, plannerThreadState, programmerThreadState] =
await Promise.all([
langGraphClient.threads.getState<ManagerGraphState>(managerThreadId),
langGraphClient.threads.getState<PlannerGraphState>(plannerThreadId),
programmerThreadId
? langGraphClient.threads.getState<GraphState>(programmerThreadId)
: null,
]);
const [
managerThread,
managerThreadState,
plannerThread,
plannerThreadState,
programmerThread,
programmerThreadState,
] = await Promise.all([
langGraphClient.threads.get<ManagerGraphState>(managerThreadId),
langGraphClient.threads.getState<ManagerGraphState>(managerThreadId),
langGraphClient.threads.get<PlannerGraphState>(plannerThreadId),
langGraphClient.threads.getState<PlannerGraphState>(plannerThreadId),
programmerThreadId
? langGraphClient.threads.get<GraphState>(programmerThreadId)
: null,
programmerThreadId
? langGraphClient.threads.getState<GraphState>(programmerThreadId)
: null,
]);
if (!managerThreadState || !plannerThreadState) {
return NextResponse.json(
{
@ -123,6 +138,7 @@ export async function POST(request: NextRequest): Promise<NextResponse> {
? await createNewSession(langGraphClient, {
graphId: PROGRAMMER_GRAPH_ID,
threadState: programmerThreadState,
threadConfig: (programmerThread as Record<string, any>)?.config,
})
: undefined;
@ -140,6 +156,7 @@ export async function POST(request: NextRequest): Promise<NextResponse> {
...plannerThreadState,
values: newPlannerState,
},
threadConfig: (plannerThread as Record<string, any>)?.config,
});
const newManagerState: ManagerGraphState = {
@ -152,6 +169,7 @@ export async function POST(request: NextRequest): Promise<NextResponse> {
...managerThreadState,
values: newManagerState,
},
threadConfig: (managerThread as Record<string, any>)?.config,
});
return NextResponse.json({