fix mypy

Author: eavanvalkenburg

python/semantic_kernel/agents/agent.py

@@ -390,7 +390,7 @@ def _merge_arguments(self, override_args: KernelArguments | None) -> KernelArgum
     async def _ensure_thread_exists_with_messages(
         self,
         *,
-        messages: str | ChatMessageContent | Sequence[str | ChatMessageContent] | None = None,
+        messages: ChatHistory | str | ChatMessageContent | Sequence[str | ChatMessageContent] | None = None,
         thread: AgentThread | None,
         construct_thread: Callable[[], TThreadType],
         expected_type: type[TThreadType],
@@ -402,6 +402,9 @@ async def _ensure_thread_exists_with_messages(
         if isinstance(messages, (str, ChatMessageContent)):
             messages = [messages]
 
+        if isinstance(messages, ChatHistory):
+            messages = messages.messages
+
         normalized_messages = [
             ChatMessageContent(role=AuthorRole.USER, content=msg) if isinstance(msg, str) else msg for msg in messages
         ]

python/semantic_kernel/agents/open_ai/openai_responses_agent.py

@@ -606,13 +606,13 @@ async def get_response(
             ResponseMessageContent: The response from the agent.
         """
         thread = await self._ensure_thread_exists_with_messages(
-            messages=messages.messages if isinstance(messages, ChatHistory) else messages,
+            messages=messages,
             thread=thread,
             construct_thread=lambda: ResponsesAgentThread(client=self.client, enable_store=self.store_enabled),
             expected_type=ResponsesAgentThread,
         )
 
-        chat_history = self._prepare_input_message(messages.messages if isinstance(messages, ChatHistory) else messages)
+        chat_history = messages if isinstance(messages, ChatHistory) else self._prepare_input_message(messages)
 
         if arguments is None:
             arguments = KernelArguments(**kwargs)