From fd0d04ba21816a5b8802733d7de33ae0f5b2ae49 Mon Sep 17 00:00:00 2001 From: Jeff Haynie Date: Thu, 28 Aug 2025 17:10:42 -0500 Subject: [PATCH] fix issue with session_id --- agentuity/server/__init__.py | 5 +++-- tests/server/test_request_handlers.py | 12 +++++++++++- 2 files changed, 14 insertions(+), 3 deletions(-) diff --git a/agentuity/server/__init__.py b/agentuity/server/__init__.py index 9fe1206..2d7acd5 100644 --- a/agentuity/server/__init__.py +++ b/agentuity/server/__init__.py @@ -11,6 +11,7 @@ import traceback from opentelemetry import trace +from opentelemetry.trace import format_trace_id from opentelemetry.propagate import extract, inject from agentuity.otel import init @@ -343,7 +344,7 @@ async def handle_agent_request(request: web.Request): metadata = {} scope = "local" if span.is_recording(): - run_id = span.get_span_context().trace_id + run_id = format_trace_id(span.get_span_context().trace_id) else: run_id = None for key, value in headers.items(): @@ -446,7 +447,7 @@ async def handle_agent_request(request: web.Request): agent=agent, agents_by_id=agents_by_id, port=port, - session_id=run_id, + session_id=str(run_id), scope=scope, ) agent_response = AgentResponse( diff --git a/tests/server/test_request_handlers.py b/tests/server/test_request_handlers.py index 451b213..5e911e6 100644 --- a/tests/server/test_request_handlers.py +++ b/tests/server/test_request_handlers.py @@ -88,11 +88,21 @@ async def test_handle_agent_request_success(self, mock_request): mock_response.status = 200 mock_response.content_type = "application/json" + # Create a mock span with proper trace_id formatting + mock_span_context = MagicMock() + mock_span_context.trace_id = 12345678901234567890123456789012345678 + mock_span = MagicMock() + mock_span.get_span_context.return_value = mock_span_context + mock_span.is_recording.return_value = True + mock_tracer = MagicMock() + mock_tracer.start_span.return_value.__enter__.return_value = mock_span + with ( patch( - "agentuity.server.trace.get_tracer", return_value=MagicMock() + "agentuity.server.trace.get_tracer", return_value=mock_tracer ) as mock_get_tracer, patch("agentuity.server.extract", return_value={}), + patch("agentuity.server.format_trace_id", return_value="test-trace-id"), patch( "agentuity.server.run_agent", new_callable=AsyncMock ) as mock_run_agent,