Add tool exec error to trace

Signed-off-by: Casper Guldbech Nielsen <scni@novonordisk.com>
This commit is contained in:
Casper Guldbech Nielsen 2025-05-03 23:34:46 +02:00
parent fc383a1c5d
commit 5be92d9021
No known key found for this signature in database
GPG Key ID: B004583B52B9A446
1 changed files with 7 additions and 2 deletions

View File

@ -37,7 +37,7 @@ from dapr_agents.agent.telemetry import (
)
from opentelemetry import trace
from opentelemetry.trace import Tracer, set_tracer_provider
from opentelemetry.trace import Tracer, set_tracer_provider, Status, StatusCode
logger = logging.getLogger(__name__)
@ -399,7 +399,12 @@ class AssistantAgent(AgentWorkflowBase):
)
except (ToolError, AgentToolExecutorError) as e:
logger.info(f"####### '{e}'")
logger.info(e)
span = trace.get_current_span()
span.set_status(Status(StatusCode.ERROR))
span.record_exception(e)
workflow_tool_message = {
"tool_call_id": tool_call.get("id"),
"function_name": function_name,