cleanup types in openai agents samples

Author: JasonSteving99

openai_agents/financial_research_agent/financial_research_manager.py

@@ -3,7 +3,7 @@
 import asyncio
 from collections.abc import Sequence
 
-from agents import RunConfig, Runner, RunResult, custom_span, trace
+from agents import RunConfig, Runner, RunResult, RunResultStreaming, custom_span, trace
 from temporalio import workflow
 
 from openai_agents.financial_research_agent.agents.financials_agent import (
@@ -26,7 +26,7 @@
 )
 
 
-async def _summary_extractor(run_result: RunResult) -> str:
+async def _summary_extractor(run_result: RunResult | RunResultStreaming) -> str:
     """Custom output extractor for sub-agents that return an AnalysisSummary."""
     # The financial/risk analyst agents emit an AnalysisSummary with a `summary` field.
     # We want the tool call to return just that summary text so the writer can drop it inline.

openai_agents/tools/workflows/image_generator_workflow.py

@@ -4,6 +4,7 @@
 from typing import Optional
 
 from agents import Agent, ImageGenerationTool, Runner
+from openai.types.responses.response_output_item import ImageGenerationCall
 from temporalio import workflow
 
 
@@ -34,7 +35,7 @@ async def run(self, prompt: str) -> ImageGenerationResult:
         for item in result.new_items:
             if (
                 item.type == "tool_call_item"
-                and item.raw_item.type == "image_generation_call"
+                and isinstance(item.raw_item, ImageGenerationCall)
                 and (img_result := item.raw_item.result)
             ):
                 image_data = img_result