diff --git a/apps/open-swe/src/graphs/programmer/nodes/summarize-task-steps.ts b/apps/open-swe/src/graphs/programmer/nodes/summarize-task-steps.ts index 3e54e21d..c3678996 100644 --- a/apps/open-swe/src/graphs/programmer/nodes/summarize-task-steps.ts +++ b/apps/open-swe/src/graphs/programmer/nodes/summarize-task-steps.ts @@ -14,7 +14,6 @@ import { getMessageString } from "../../../utils/message/content.js"; import { removeLastTaskMessages } from "../../../utils/message/modify-array.js"; import { Command } from "@langchain/langgraph"; import { ConfigurableModel } from "langchain/chat_models/universal"; -import { traceable } from "langsmith/traceable"; import { completePlanItem, getActivePlanItems, @@ -90,7 +89,7 @@ const formatUserMessage = ( ); }; -async function generateTaskSummaryFunc( +async function generateTaskSummary( state: GraphState, model: ConfigurableModel, ): Promise<{ planItemIndex: number; summary: string }> { @@ -101,16 +100,18 @@ async function generateTaskSummaryFunc( } logger.info(`Summarizing task steps...`); - const response = await model.withConfig({ tags: ["nostream"] }).invoke([ - { - role: "system", - content: formatPrompt(activePlanItems), - }, - { - role: "user", - content: formatUserMessage(state.internalMessages, activePlanItems), - }, - ]); + const response = await model + .withConfig({ tags: ["nostream"], runName: "generate-task-summary" }) + .invoke([ + { + role: "system", + content: formatPrompt(activePlanItems), + }, + { + role: "user", + content: formatUserMessage(state.internalMessages, activePlanItems), + }, + ]); return { planItemIndex: lastCompletedTask.index, @@ -118,10 +119,6 @@ async function generateTaskSummaryFunc( }; } -const generateTaskSummary = traceable(generateTaskSummaryFunc, { - name: "generate-task-summary", -}); - export async function summarizeTaskSteps( state: GraphState, config: GraphConfig,