can1357
96822471c2
- Added cell-by-cell Python execution tracking with duration and exit codes.
- Added template system with {{placeholders}} for task tool context rendering.
- Enhanced patch applicator to preserve exact indentation when intended.
- Changed Python tool to use cells array instead of single code parameter.
- Updated task tool schema to require vars instead of task field.
47 lines
1.5 KiB
TypeScript
47 lines
1.5 KiB
TypeScript
import { describe, expect, it } from "bun:test";
|
|
import stripAnsi from "strip-ansi";
|
|
import { pythonToolRenderer } from "../../src/core/tools/python";
|
|
import { getThemeByName } from "../../src/modes/interactive/theme/theme";
|
|
|
|
describe("pythonToolRenderer", () => {
|
|
it("renders truncated output when collapsed and full output when expanded", () => {
|
|
const theme = getThemeByName("dark");
|
|
expect(theme).toBeDefined();
|
|
const uiTheme = theme!;
|
|
|
|
const fullOutput = ["line 1", "line 2", "line 3", "line 4"].join("\n");
|
|
|
|
const result = {
|
|
content: [{ type: "text", text: fullOutput }],
|
|
details: {
|
|
cells: [
|
|
{
|
|
index: 0,
|
|
title: "run",
|
|
code: "print('hello')",
|
|
output: fullOutput,
|
|
status: "complete" as const,
|
|
durationMs: 12,
|
|
},
|
|
],
|
|
},
|
|
};
|
|
|
|
const collapsed = pythonToolRenderer.renderResult(
|
|
result,
|
|
{ expanded: false, isPartial: false, renderContext: { previewLines: 2 } },
|
|
uiTheme,
|
|
);
|
|
const collapsedLines = stripAnsi(collapsed.render(80).join("\n"));
|
|
expect(collapsedLines).toContain("line 4");
|
|
expect(collapsedLines).not.toContain("line 1");
|
|
expect(collapsedLines).toContain("more lines");
|
|
|
|
const expanded = pythonToolRenderer.renderResult(result, { expanded: true, isPartial: false }, uiTheme);
|
|
const expandedLines = stripAnsi(expanded.render(80).join("\n"));
|
|
expect(expandedLines).toContain("line 1");
|
|
expect(expandedLines).toContain("line 4");
|
|
expect(expandedLines).not.toContain("more lines");
|
|
});
|
|
});
|