Skip to content

Commit 21b104a

Browse files
committed
fix build error
1 parent 3761b09 commit 21b104a

File tree

2 files changed

+14
-11
lines changed

2 files changed

+14
-11
lines changed

python-sdk/ag_ui/core/events.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -17,15 +17,15 @@ class EventType(str, Enum):
1717
TEXT_MESSAGE_CONTENT = "TEXT_MESSAGE_CONTENT"
1818
TEXT_MESSAGE_END = "TEXT_MESSAGE_END"
1919
TEXT_MESSAGE_CHUNK = "TEXT_MESSAGE_CHUNK"
20-
THINKING_TEXT_MESSAGE_START = "THINKING_TEXT_MESSAGE_START",
21-
THINKING_TEXT_MESSAGE_CONTENT = "THINKING_TEXT_MESSAGE_CONTENT",
22-
THINKING_TEXT_MESSAGE_END = "THINKING_TEXT_MESSAGE_END",
20+
THINKING_TEXT_MESSAGE_START = "THINKING_TEXT_MESSAGE_START"
21+
THINKING_TEXT_MESSAGE_CONTENT = "THINKING_TEXT_MESSAGE_CONTENT"
22+
THINKING_TEXT_MESSAGE_END = "THINKING_TEXT_MESSAGE_END"
2323
TOOL_CALL_START = "TOOL_CALL_START"
2424
TOOL_CALL_ARGS = "TOOL_CALL_ARGS"
2525
TOOL_CALL_END = "TOOL_CALL_END"
2626
TOOL_CALL_CHUNK = "TOOL_CALL_CHUNK"
27-
THINKING_START = "THINKING_START",
28-
THINKING_END = "THINKING_END",
27+
THINKING_START = "THINKING_START"
28+
THINKING_END = "THINKING_END"
2929
STATE_SNAPSHOT = "STATE_SNAPSHOT"
3030
STATE_DELTA = "STATE_DELTA"
3131
MESSAGES_SNAPSHOT = "MESSAGES_SNAPSHOT"
@@ -149,7 +149,7 @@ class ThinkingStartEvent(BaseEvent):
149149
Event indicating the start of a thinking step event.
150150
"""
151151
type: Literal[EventType.THINKING_START]
152-
title: Optional[str] = 'Thinking...'
152+
title: Optional[str] = None
153153

154154
class ThinkingEndEvent(BaseEvent):
155155
"""

typescript-sdk/packages/core/src/events.ts

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -87,15 +87,18 @@ export const TextMessageChunkEventSchema = BaseEventSchema.extend({
8787

8888
export const ThinkingTextMessageStartEventSchema = BaseEventSchema.extend({
8989
type: z.literal(EventType.THINKING_TEXT_MESSAGE_START),
90-
})
90+
});
9191

92-
export const ThinkingTextMessageContentEventSchema = TextMessageContentEventSchema.omit({ messageId: true, type: true }).extend({
92+
export const ThinkingTextMessageContentEventSchema = TextMessageContentEventSchema.omit({
93+
messageId: true,
94+
type: true,
95+
}).extend({
9396
type: z.literal(EventType.THINKING_TEXT_MESSAGE_CONTENT),
94-
})
97+
});
9598

9699
export const ThinkingTextMessageEndEventSchema = BaseEventSchema.extend({
97100
type: z.literal(EventType.THINKING_TEXT_MESSAGE_END),
98-
})
101+
});
99102

100103
export const ToolCallStartEventSchema = BaseEventSchema.extend({
101104
type: z.literal(EventType.TOOL_CALL_START),
@@ -125,7 +128,7 @@ export const ToolCallChunkEventSchema = BaseEventSchema.extend({
125128

126129
export const ThinkingStartEventSchema = BaseEventSchema.extend({
127130
type: z.literal(EventType.THINKING_START),
128-
title: z.string().optional().default("Thinking..."),
131+
title: z.string().optional(),
129132
});
130133

131134
export const ThinkingEndEventSchema = BaseEventSchema.extend({

0 commit comments

Comments
 (0)