|
13 | 13 |
|
14 | 14 | class ChatHistory(Container): |
15 | 15 | def add_message(self, message: Message) -> None: |
16 | | - widget = self._create_message_widget(message) |
| 16 | + widget = self._create_message(message) |
17 | 17 | self.mount(widget) |
18 | 18 |
|
19 | | - def _create_message_widget( |
| 19 | + def _create_message( |
20 | 20 | self, message: Message |
21 | 21 | ) -> SystemMessage | UserMessage | AgentMessage | ToolMessage: |
22 | 22 | match message.type: |
23 | 23 | case RoleType.SYSTEM: |
24 | | - system_widget = SystemMessage() |
25 | | - system_widget.message = message.content |
26 | | - return system_widget |
| 24 | + system_message = SystemMessage() |
| 25 | + system_message.message = message.content |
| 26 | + return system_message |
27 | 27 |
|
28 | 28 | case RoleType.USER: |
29 | | - user_widget = UserMessage() |
30 | | - user_widget.message = message.content |
31 | | - return user_widget |
| 29 | + user_message = UserMessage() |
| 30 | + user_message.message = message.content |
| 31 | + return user_message |
32 | 32 |
|
33 | 33 | case RoleType.AGENT: |
34 | | - agent_widget = AgentMessage() |
35 | | - agent_widget.message = message.content |
36 | | - return agent_widget |
| 34 | + agent_message = AgentMessage() |
| 35 | + agent_message.message = message.content |
| 36 | + return agent_message |
37 | 37 |
|
38 | 38 | case RoleType.TOOL: |
39 | | - tool_widget = ToolMessage() |
| 39 | + tool_message = ToolMessage() |
40 | 40 |
|
41 | 41 | if message.metadata and "tool_name" in message.metadata: |
42 | | - tool_widget.tool_name = message.metadata["tool_name"] |
| 42 | + tool_message.tool_name = message.metadata["tool_name"] |
43 | 43 |
|
44 | 44 | try: |
45 | | - tool_widget.tool_input = json.loads(message.content) |
| 45 | + tool_message.tool_input = json.loads(message.content) |
46 | 46 | except (json.JSONDecodeError, TypeError): |
47 | | - tool_widget.tool_input = {"raw": message.content} |
| 47 | + tool_message.tool_input = {"raw": message.content} |
48 | 48 |
|
49 | | - return tool_widget |
| 49 | + return tool_message |
0 commit comments