Spaces:
Running
Running
File size: 2,992 Bytes
fb4d8fe | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 | import { describe, expect, it } from "vitest";
import {
extractContentFromMessage,
extractTextFromMessage,
extractThinkingFromMessage,
isCommandMessage,
} from "./tui-formatters.js";
describe("extractTextFromMessage", () => {
it("renders errorMessage when assistant content is empty", () => {
const text = extractTextFromMessage({
role: "assistant",
content: [],
stopReason: "error",
errorMessage:
'429 {"type":"error","error":{"type":"rate_limit_error","message":"This request would exceed your account\\u0027s rate limit. Please try again later."},"request_id":"req_123"}',
});
expect(text).toContain("HTTP 429");
expect(text).toContain("rate_limit_error");
expect(text).toContain("req_123");
});
it("falls back to a generic message when errorMessage is missing", () => {
const text = extractTextFromMessage({
role: "assistant",
content: [],
stopReason: "error",
errorMessage: "",
});
expect(text).toContain("unknown error");
});
it("joins multiple text blocks with single newlines", () => {
const text = extractTextFromMessage({
role: "assistant",
content: [
{ type: "text", text: "first" },
{ type: "text", text: "second" },
],
});
expect(text).toBe("first\nsecond");
});
it("places thinking before content when included", () => {
const text = extractTextFromMessage(
{
role: "assistant",
content: [
{ type: "text", text: "hello" },
{ type: "thinking", thinking: "ponder" },
],
},
{ includeThinking: true },
);
expect(text).toBe("[thinking]\nponder\n\nhello");
});
});
describe("extractThinkingFromMessage", () => {
it("collects only thinking blocks", () => {
const text = extractThinkingFromMessage({
role: "assistant",
content: [
{ type: "thinking", thinking: "alpha" },
{ type: "text", text: "hello" },
{ type: "thinking", thinking: "beta" },
],
});
expect(text).toBe("alpha\nbeta");
});
});
describe("extractContentFromMessage", () => {
it("collects only text blocks", () => {
const text = extractContentFromMessage({
role: "assistant",
content: [
{ type: "thinking", thinking: "alpha" },
{ type: "text", text: "hello" },
],
});
expect(text).toBe("hello");
});
it("renders error text when stopReason is error and content is not an array", () => {
const text = extractContentFromMessage({
role: "assistant",
stopReason: "error",
errorMessage: '429 {"error":{"message":"rate limit"}}',
});
expect(text).toContain("HTTP 429");
});
});
describe("isCommandMessage", () => {
it("detects command-marked messages", () => {
expect(isCommandMessage({ command: true })).toBe(true);
expect(isCommandMessage({ command: false })).toBe(false);
expect(isCommandMessage({})).toBe(false);
});
});
|