diff --git a/main.py b/main.py index 939bf97..7051bde 100644 --- a/main.py +++ b/main.py @@ -26,6 +26,7 @@ while True: tools=tools, stream=True) ran_tool = False + saw_content = False for chunk in stream: msg = chunk.message @@ -42,9 +43,14 @@ while True: }) ran_tool = True - # 2️⃣ normal user-visible text - elif msg.content: - print(msg.content, end="", flush=True) + if msg.thinking: + print(f"\033[90m{msg.thinking}", end="", flush=True) + + if msg.content: + if not saw_content: + print("") + saw_content = True + print(f"\033[97m{msg.content}", end="", flush=True) # loop again if we just satisfied a tool call if not ran_tool: