getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-16 17:01:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Simplify assistant message spacing - just add spacer to components

Browse source
This commit is contained in:
Mario Zechner 2025-11-11 21:51:12 +01:00
parent f5176bb173
commit e2649341f0
3 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
packages/coding-agent/src/tui/streaming-message.ts
View file

@ -1,18 +1,21 @@
import type { AssistantMessage, Message } from "@mariozechner/pi-ai";
import { Container, Markdown, Text } from "@mariozechner/pi-tui";
import { Container, Markdown, Spacer, Text } from "@mariozechner/pi-tui";
import chalk from "chalk";
/**
* Component that renders a streaming message with live updates
*/
export class StreamingMessageComponent extends Container {
private spacer: Spacer;
private markdown: Markdown;
private statsText: Text;
constructor() {
super();
this.spacer = new Spacer(1);
this.markdown = new Markdown("");
this.statsText = new Text("", 1, 0);
this.addChild(this.spacer);
this.addChild(this.markdown);
this.addChild(this.statsText);
}

5
packages/coding-agent/src/tui/tui-renderer.ts
View file

@ -199,7 +199,7 @@ export class TuiRenderer {
this.editor.setText("");
this.ui.requestRender();
} else if (event.message.role === "assistant") {
// Create streaming component for assistant messages
// Create streaming component for assistant messages (has its own spacer)
this.streamingComponent = new StreamingMessageComponent();
this.chatContainer.addChild(this.streamingComponent);
this.streamingComponent.updateContent(event.message);
@ -304,6 +304,9 @@ export class TuiRenderer {
} else if (message.role === "assistant") {
const assistantMsg = message as AssistantMessage;
// Add spacer before assistant message
this.chatContainer.addChild(new Spacer(1));
// Render content in order
for (const content of assistantMsg.content) {
if (content.type === "text" && content.text.trim()) {