fix for 3.10 mypy

Author: eavanvalkenburg

python/packages/core/agent_framework/openai/_responses_client.py

@@ -995,9 +995,8 @@ def _parse_response_from_openai(
                         # item.result contains raw base64 string
                         # so we call detect_media_type_from_base64 to get the media type and fallback to image/png
                         image_output = Content.from_uri(
-                            uri=f"data:{(detect_media_type_from_base64(data_str=item.result) or 'image/png')};base64,{
-                                item.result
-                            }",
+                            uri=f"data:{detect_media_type_from_base64(data_str=item.result) or 'image/png'}"
+                            f";base64,{item.result}",
                             raw_representation=item.result,
                         )
                     image_id = item.id
@@ -1292,9 +1291,8 @@ def _parse_chunk_from_openai(
                 image_base64 = event.partial_image_b64
                 partial_index = event.partial_image_index
                 image_output = Content.from_uri(
-                    uri=f"data:{(detect_media_type_from_base64(data_str=image_base64) or 'image/png')};base64,{
-                        image_base64
-                    }",
+                    uri=f"data:{detect_media_type_from_base64(data_str=image_base64) or 'image/png'}"
+                    f";base64,{image_base64}",
                     additional_properties={
                         "partial_image_index": partial_index,
                         "is_partial_image": True,

python/packages/lab/lightning/tests/test_lightning.py

@@ -9,7 +9,7 @@
 
 agentlightning = pytest.importorskip("agentlightning")
 
-from agent_framework import AgentExecutor, AgentRunEvent, ChatAgent, WorkflowBuilder
+from agent_framework import AgentExecutor, AgentRunEvent, ChatAgent, WorkflowBuilder, Workflow
 from agent_framework_lab_lightning import AgentFrameworkTracer
 from agent_framework.openai import OpenAIChatClient
 from agentlightning import TracerTraceToTriplet
@@ -106,7 +106,7 @@ def workflow_two_agents():
             yield workflow
 
 
-async def test_openai_workflow_two_agents(workflow_two_agents):
+async def test_openai_workflow_two_agents(workflow_two_agents: Workflow):
     events = await workflow_two_agents.run("Please analyze the quarterly sales data")
 
     # Get all AgentRunEvent data
@@ -121,7 +121,7 @@ async def test_openai_workflow_two_agents(workflow_two_agents):
     )
 
 
-async def test_observability(workflow_two_agents):
+async def test_observability(workflow_two_agents: Workflow):
     r"""Expected trace tree:
 
                     [workflow.run]