Removed list_messages method from thread (#564)

python/packages/main/tests/main/test_agents.py

@@ -227,8 +227,9 @@ async def test_chat_client_agent_update_thread_messages(chat_client: ChatClient)
     assert result.text == "test response"
 
     assert thread.service_thread_id is None
+    assert thread.message_store is not None
 
-    chat_messages: list[ChatMessage] | None = await thread.list_messages()
+    chat_messages: list[ChatMessage] = await thread.message_store.list_messages()
 
     assert chat_messages is not None
     assert len(chat_messages) == 2

python/packages/main/tests/main/test_threads.py

@@ -122,7 +122,9 @@ class TestAgentThread:
         store = ChatMessageList(sample_messages)
         thread = AgentThread(message_store=store)
 
-        messages: list[ChatMessage] | None = await thread.list_messages()
+        assert thread.message_store is not None
+
+        messages: list[ChatMessage] = await thread.message_store.list_messages()
 
         assert messages is not None
         assert len(messages) == 3
@@ -134,9 +136,7 @@ class TestAgentThread:
         """Test get_messages when no message_store is set."""
         thread = AgentThread()
 
-        messages: list[ChatMessage] | None = await thread.list_messages()
-
-        assert messages is None
+        assert thread.message_store is None
 
     async def test_on_new_messages_with_service_thread_id(self, sample_message: ChatMessage) -> None:
         """Test _on_new_messages when service_thread_id is set (should do nothing)."""