Merge pull request #58 from luis5tb/agent-name

Fix typo on accessing agent name for logging

Author: luis5tb

src/lightspeed_agent/api/a2a/logging_plugin.py

@@ -50,7 +50,7 @@ async def before_run_callback(
         logger.info(
             "Agent run started (invocation_id=%s, agent=%s)",
             invocation_context.invocation_id,
-            invocation_context.agent_name,  # type: ignore[attr-defined]
+            invocation_context.agent.name,
         )
         return None
 

tests/test_logging_plugin.py

@@ -44,7 +44,7 @@ def plugin(self):
     async def test_before_run_logs_info(self, plugin, caplog):
         ctx = MagicMock()
         ctx.invocation_id = "inv-123"
-        ctx.agent_name = "test_agent"
+        ctx.agent = MagicMock(name="test_agent")
 
         with caplog.at_level(logging.INFO):
             result = await plugin.before_run_callback(invocation_context=ctx)
@@ -270,7 +270,7 @@ def plugin(self):
     async def test_all_callbacks_return_none(self, plugin):
         ctx = MagicMock()
         ctx.invocation_id = "inv-1"
-        ctx.agent_name = "agent"
+        ctx.agent = MagicMock(name="agent")
 
         tool = MagicMock()
         tool.name = "test_tool"