feat: Adds session id to Agent context metadata (#3559)

Co-authored-by: Brandon Wisnicki <bwisnicki@palantir.com>
This commit is contained in:
Brandon
2025-10-30 13:15:21 -04:00
committed by GitHub
parent e944ff0286
commit c68607fb2b

View File

@@ -33,6 +33,14 @@ export const TaskTool = Tool.define("task", async () => {
}) })
const msg = await Session.getMessage({ sessionID: ctx.sessionID, messageID: ctx.messageID }) const msg = await Session.getMessage({ sessionID: ctx.sessionID, messageID: ctx.messageID })
if (msg.info.role !== "assistant") throw new Error("Not an assistant message") if (msg.info.role !== "assistant") throw new Error("Not an assistant message")
ctx.metadata({
title: params.description,
metadata: {
sessionId: session.id,
},
})
const messageID = Identifier.ascending("message") const messageID = Identifier.ascending("message")
const parts: Record<string, MessageV2.ToolPart> = {} const parts: Record<string, MessageV2.ToolPart> = {}
const unsub = Bus.subscribe(MessageV2.Event.PartUpdated, async (evt) => { const unsub = Bus.subscribe(MessageV2.Event.PartUpdated, async (evt) => {
@@ -44,6 +52,7 @@ export const TaskTool = Tool.define("task", async () => {
title: params.description, title: params.description,
metadata: { metadata: {
summary: Object.values(parts).sort((a, b) => a.id?.localeCompare(b.id)), summary: Object.values(parts).sort((a, b) => a.id?.localeCompare(b.id)),
sessionId: session.id,
}, },
}) })
}) })
@@ -87,6 +96,7 @@ export const TaskTool = Tool.define("task", async () => {
title: params.description, title: params.description,
metadata: { metadata: {
summary: all, summary: all,
sessionId: session.id,
}, },
output: (result.parts.findLast((x: any) => x.type === "text") as any)?.text ?? "", output: (result.parts.findLast((x: any) => x.type === "text") as any)?.text ?? "",
} }