Skip to content

Commit fc95b0e

Browse files
committed
fix: Remove "Rendering diagram..." when rendering the diagram
Blinking cursor should be enough of an indication that the model is working.
1 parent 779524c commit fc95b0e

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

packages/navie/src/agents/diagram-agent.ts

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ export default class DiagramAgent implements Agent {
3333
return new MermaidFilter(this.history, this.mermaidFixerService);
3434
}
3535

36-
async perform(options: AgentOptions, tokensAvailable: () => number): Promise<AgentResponse> {
36+
async perform(options: AgentOptions, tokensAvailable: () => number): Promise<void> {
3737
const agentPrompt = [DIAGRAM_AGENT_PROMPT];
3838
// With the /noformat option, the user will explain the desired output format in their message.
3939
if (options.userOptions.isEnabled('format', true)) {
@@ -50,8 +50,6 @@ export default class DiagramAgent implements Agent {
5050
);
5151

5252
await this.contextService.searchContext(options, tokensAvailable);
53-
54-
return { response: 'Rendering diagram...\n', abort: false };
5553
}
5654

5755
applyQuestionPrompt(question: string): void {

packages/navie/test/agents/diagram-agent.spec.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,6 @@ describe('@diagram agent', () => {
7272
);
7373

7474
// Verify that the result returned by the perform method is as expected
75-
expect(result).toEqual({ response: 'Rendering diagram...\n', abort: false });
75+
expect(result).toBeUndefined();
7676
});
7777
});

0 commit comments

Comments
 (0)