Skip to content

Commit 60e2590

Browse files
committed
chore: cleanup naming
1 parent a460a6e commit 60e2590

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

src/agent_chat_cli/components/chat_history.py

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -13,37 +13,37 @@
1313

1414
class ChatHistory(Container):
1515
def add_message(self, message: Message) -> None:
16-
widget = self._create_message_widget(message)
16+
widget = self._create_message(message)
1717
self.mount(widget)
1818

19-
def _create_message_widget(
19+
def _create_message(
2020
self, message: Message
2121
) -> SystemMessage | UserMessage | AgentMessage | ToolMessage:
2222
match message.type:
2323
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
2727

2828
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
3232

3333
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
3737

3838
case RoleType.TOOL:
39-
tool_widget = ToolMessage()
39+
tool_message = ToolMessage()
4040

4141
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"]
4343

4444
try:
45-
tool_widget.tool_input = json.loads(message.content)
45+
tool_message.tool_input = json.loads(message.content)
4646
except (json.JSONDecodeError, TypeError):
47-
tool_widget.tool_input = {"raw": message.content}
47+
tool_message.tool_input = {"raw": message.content}
4848

49-
return tool_widget
49+
return tool_message

0 commit comments

Comments
 (0)