fix: adapt usage annotation to upstream agent->public_agent rename + lint

Author: adityasingh2400

src/agents/run_internal/run_loop.py

@@ -1640,7 +1640,7 @@ async def rewind_model_request() -> None:
     if final_response is not None:
         context_wrapper.usage.add(
             final_response.usage,
-            agent_name=agent.name,
+            agent_name=public_agent.name,
             model_name=_get_model_name(model),
         )
         await asyncio.gather(
@@ -1922,7 +1922,7 @@ async def rewind_model_request() -> None:
 
     context_wrapper.usage.add(
         new_response.usage,
-        agent_name=agent.name,
+        agent_name=public_agent.name,
         model_name=_get_model_name(model),
     )
 

src/agents/usage.py

@@ -248,8 +248,12 @@ def add(
                     total_tokens=entry.total_tokens,
                     input_tokens_details=entry.input_tokens_details,
                     output_tokens_details=entry.output_tokens_details,
-                    agent_name=agent_name if (agent_name is not None and entry.agent_name is None) else entry.agent_name,
-                    model_name=model_name if (model_name is not None and entry.model_name is None) else entry.model_name,
+                    agent_name=agent_name
+                    if (agent_name is not None and entry.agent_name is None)
+                    else entry.agent_name,
+                    model_name=model_name
+                    if (model_name is not None and entry.model_name is None)
+                    else entry.model_name,
                 )
                 self.request_usage_entries.append(annotated_entry)