fix: 使用 ResultMessage.result 获取最终响应

- 改用 ResultMessage.result 替代手动收集 AssistantMessage 的 TextBlock
- 符合 SDK 设计意图，ResultMessage.result 包含完整聚合响应
- 移除不再需要的 AssistantMessage 和 TextBlock 导入
- 添加 SDK 响应结构测试用于验证消息类型

Author: BukeLy

agent-sdk-server/agent_session.py

@@ -12,9 +12,7 @@
     query,
     ClaudeAgentOptions,
     AgentDefinition,
-    AssistantMessage,
     ResultMessage,
-    TextBlock,
 )
 
 # Config source (in Docker image) and destination (Lambda writable)
@@ -178,8 +176,8 @@ async def process_message(
         agents=agents if agents else None,
     )
 
-    response_texts: list[str] = []
     result_session_id = session_id or ''
+    result_text = ''
     cost_usd = 0.0
     num_turns = 0
     is_error = False
@@ -190,19 +188,14 @@ async def process_message(
         os.environ['AWS_PROFILE'] = 'bedrock'
 
         async for message in query(prompt=user_message, options=options):
-            # Handle AssistantMessage - extract text from content blocks
-            if isinstance(message, AssistantMessage):
-                for block in message.content:
-                    if isinstance(block, TextBlock):
-                        response_texts.append(block.text)
-
-            # Handle ResultMessage - extract session_id and metadata
-            elif isinstance(message, ResultMessage):
+            # Handle ResultMessage - contains final response and metadata
+            if isinstance(message, ResultMessage):
                 result_session_id = message.session_id
+                result_text = message.result or ''
                 cost_usd = message.total_cost_usd or 0.0
                 num_turns = message.num_turns
                 is_error = message.is_error
-                if is_error and message.result:
+                if is_error:
                     error_message = message.result
 
     except Exception as e:
@@ -214,7 +207,7 @@ async def process_message(
             del os.environ['AWS_PROFILE']
 
     return {
-        'response': '\n'.join(response_texts) if response_texts else '',
+        'response': result_text,
         'session_id': result_session_id,
         'cost_usd': cost_usd,
         'num_turns': num_turns,

tests/test_sdk_response.py

@@ -0,0 +1,45 @@
+"""测试 Claude Agent SDK 返回的消息结构."""
+import asyncio
+import os
+
+from claude_agent_sdk import (
+    query,
+    ClaudeAgentOptions,
+    AssistantMessage,
+    ResultMessage,
+    TextBlock,
+)
+
+
+async def test_sdk_response():
+    """测试 SDK 返回内容."""
+    options = ClaudeAgentOptions(
+        cwd='/tmp',
+        model='haiku',
+        permission_mode='bypassPermissions',
+        max_turns=5,
+    )
+
+    print("=" * 60)
+    print("测试 SDK 返回的消息结构")
+    print("=" * 60)
+
+    async for message in query(prompt="说 Hello World", options=options):
+        print(f"\n--- Message Type: {type(message).__name__} ---")
+
+        if isinstance(message, AssistantMessage):
+            print(f"  model: {message.model}")
+            print(f"  content blocks: {len(message.content)}")
+            for i, block in enumerate(message.content):
+                print(f"    [{i}] {type(block).__name__}: {repr(block)[:200]}")
+
+        elif isinstance(message, ResultMessage):
+            print(f"  session_id: {message.session_id}")
+            print(f"  is_error: {message.is_error}")
+            print(f"  result: {message.result}")
+            print(f"  num_turns: {message.num_turns}")
+            print(f"  total_cost_usd: {message.total_cost_usd}")
+
+
+if __name__ == '__main__':
+    asyncio.run(test_sdk_response())