few small fixes

Author: sokoliva

src/a2a/client/text_client.py

@@ -94,7 +94,7 @@ async def send_text_message(
             elif event.HasField('message'):
                 response_parts.append(get_message_text(event.message))
             elif event.HasField('status_update'):
-                if event.status_update.task_id:
+                if not self._task_id and event.status_update.task_id:
                     self._task_id = event.status_update.task_id
                 if event.status_update.status.state in _TERMINAL_STATES:
                     self._task_id = None

tests/client/test_text_client.py

@@ -55,9 +55,9 @@ async def create_stream(*args, **kwargs):
         resp0.task.id = 'task-1'
         yield resp0
 
-        # Event 1: direct message
+        # Event 1: status update
         resp1 = StreamResponse()
-        resp1.message.parts.append(Part(text='Hello'))
+        resp1.status_update.status.message.parts.append(Part(text='Hello'))
         yield resp1
 
         # Event 2: status update without message
@@ -87,20 +87,17 @@ async def create_stream(*args, **kwargs):
 
 
 @pytest.mark.asyncio
-async def test_send_text_message_custom_delimiter(
+async def test_send_text_message(
     text_client: TextClient, mock_client: AsyncMock
 ) -> None:
     async def create_stream(*args, **kwargs):
         resp1 = StreamResponse()
         resp1.message.parts.append(Part(text='Hello'))
         yield resp1
-        resp2 = StreamResponse()
-        resp2.artifact_update.artifact.parts.append(Part(text='World'))
-        yield resp2
 
     mock_client.send_message.return_value = create_stream()
-    response = await text_client.send_text_message('Hi', delimiter='\n')
-    assert response == 'Hello\nWorld'
+    response = await text_client.send_text_message('Hi')
+    assert response == 'Hello'
 
 
 @pytest.mark.asyncio