Skip to content

Commit 606fc5e

Browse files
Fix typing: safely access TaskMessageContent attributes in streaming tutorial
1 parent e6d6f57 commit 606fc5e

File tree

1 file changed

+6
-6
lines changed
  • examples/tutorials/00_sync/020_streaming/project

1 file changed

+6
-6
lines changed

examples/tutorials/00_sync/020_streaming/project/acp.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -41,11 +41,11 @@ async def handle_message_send(
4141
if not params.content:
4242
return
4343

44-
if params.content.type != "text":
45-
raise ValueError(f"Expected text message, got {params.content.type}")
44+
if not hasattr(params.content, 'type') or params.content.type != "text":
45+
raise ValueError(f"Expected text message, got {getattr(params.content, 'type', 'unknown')}")
4646

47-
if params.content.author != "user":
48-
raise ValueError(f"Expected user message, got {params.content.author}")
47+
if not hasattr(params.content, 'author') or params.content.author != "user":
48+
raise ValueError(f"Expected user message, got {getattr(params.content, 'author', 'unknown')}")
4949

5050
if not os.environ.get("OPENAI_API_KEY"):
5151
yield StreamTaskMessageFull(
@@ -72,9 +72,9 @@ async def handle_message_send(
7272
llm_messages = [
7373
SystemMessage(content=state.system_prompt),
7474
*[
75-
UserMessage(content=message.content.content) if message.content.author == "user" else AssistantMessage(content=message.content.content)
75+
UserMessage(content=getattr(message.content, 'content', '')) if getattr(message.content, 'author', None) == "user" else AssistantMessage(content=getattr(message.content, 'content', ''))
7676
for message in task_messages
77-
if message.content and message.content.type == "text"
77+
if message.content and getattr(message.content, 'type', None) == "text"
7878
]
7979
]
8080

0 commit comments

Comments
 (0)