works

nina-kollman · nina-kollman · commit aa91b3efb25f · 2025-11-11T12:11:15.000+02:00
diff --git a/traceloop-sdk/agent.go b/traceloop-sdk/agent.go
@@ -22,7 +22,7 @@ func (agent *Agent) End() {
 
 func (agent *Agent) LogPrompt(prompt Prompt) LLMSpan {
 	if agent.workflow != nil {
-		return agent.workflow.LogPrompt(prompt)
+		return agent.sdk.LogPrompt(agent.ctx, prompt, agent.workflow.Attributes)
 	}
 	return agent.sdk.LogPrompt(agent.ctx, prompt, WorkflowAttributes{})
 }
diff --git a/traceloop-sdk/tool.go b/traceloop-sdk/tool.go
@@ -19,5 +19,5 @@ func (tool *Tool) End() {
 }
 
 func (tool *Tool) LogPrompt(prompt Prompt) LLMSpan {
-	return tool.agent.LogPrompt(prompt)
+	return tool.agent.sdk.LogPrompt(tool.ctx, prompt, tool.agent.workflow.Attributes)
 }

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ func (agent *Agent) End() {`
`22`	`22`
`23`	`23`	`func (agent *Agent) LogPrompt(prompt Prompt) LLMSpan {`
`24`	`24`	`if agent.workflow != nil {`
`25`		`- return agent.workflow.LogPrompt(prompt)`
	`25`	`+ return agent.sdk.LogPrompt(agent.ctx, prompt, agent.workflow.Attributes)`
`26`	`26`	`}`
`27`	`27`	`return agent.sdk.LogPrompt(agent.ctx, prompt, WorkflowAttributes{})`
`28`	`28`	`}`
Original file line number	Diff line number	Diff line change
`@@ -19,5 +19,5 @@ func (tool *Tool) End() {`
`19`	`19`	`}`
`20`	`20`
`21`	`21`	`func (tool *Tool) LogPrompt(prompt Prompt) LLMSpan {`
`22`		`- return tool.agent.LogPrompt(prompt)`
	`22`	`+ return tool.agent.sdk.LogPrompt(tool.ctx, prompt, tool.agent.workflow.Attributes)`
`23`	`23`	`}`