RomanNum3ral
/
cai
mirror of https://github.com/aliasrobotics/cai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix history error

This commit is contained in:
Mery-Sanz 2025-04-25 14:12:57 +02:00 committed by Víctor Mayoral Vilches
parent e561a5b59a
commit 6973e23ece
1 changed files with 24 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
src/cai/repl/commands/history.py
View File

@ -51,30 +51,34 @@ class HistoryCommand(Command):
# Add messages to the table
for idx, msg in enumerate(message_history, 1):
role = msg.get("role", "unknown")
content = msg.get("content", "")
try:
role = msg.get("role", "unknown")
content = msg.get("content", "")
# Truncate long content for better display
if len(content) > 100:
content = content[:97] + "..."
# Truncate long content for better display
if len(content) > 100:
content = content[:97] + "..."
# Color the role based on type
if role == "user":
role_style = "cyan"
elif role == "assistant":
role_style = "yellow"
else:
role_style = "red"
# Color the role based on type
if role == "user":
role_style = "cyan"
elif role == "assistant":
role_style = "yellow"
else:
role_style = "red"
# Add a newline between each role for better readability
if idx > 1:
table.add_row("", "", "")
# Add a newline between each role for better readability
if idx > 1:
table.add_row("", "", "")
table.add_row(
str(idx),
f"[{role_style}]{role}[/{role_style}]",
content
)
table.add_row(
str(idx),
f"[{role_style}]{role}[/{role_style}]",
content
)
except Exception:
# Si hay un error, evita esto (no agregues la fila)
continue
console.print(table)
return True