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

fix: Tracing issue where LLM calls were being traced outside of the main run (#375)

Browse source
This commit is contained in:
Brace Sproul 2025-07-09 15:49:46 -07:00 • committed by GitHub
parent 1756575b4c
commit 417c2e6650
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 13 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
apps/open-swe/src/graphs/programmer/nodes/summarize-task-steps.ts
View file

@ -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,