Q22: cap every tool result and bound connector responses
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
1 parent
4d6c74a21e
commit
8e8b66fed4
5 files changed
+79
-5
No files matched your search
@@ -0,0 +1,43 @@
|
||||
import { EventEmitter } from "node:events";
|
||||
import { describe, expect, test, vi } from "vitest";
|
||||
|
||||
const chunksToSend: Buffer[] = [];
|
||||
let aborted = false;
|
||||
|
||||
vi.mock("electron", () => ({
|
||||
net: {
|
||||
request: () => {
|
||||
const request = new EventEmitter() as EventEmitter & { setHeader: () => void; write: () => void; end: () => void; abort: () => void };
|
||||
request.setHeader = () => undefined;
|
||||
request.write = () => undefined;
|
||||
request.abort = () => { aborted = true; };
|
||||
request.end = () => {
|
||||
const response = new EventEmitter() as EventEmitter & { statusCode: number };
|
||||
response.statusCode = 200;
|
||||
request.emit("response", response);
|
||||
for (const chunk of chunksToSend) response.emit("data", chunk);
|
||||
response.emit("end");
|
||||
};
|
||||
return request;
|
||||
},
|
||||
},
|
||||
}));
|
||||
|
||||
const { ComposioClient } = await import("../../src/main/services/composio/composio-client");
|
||||
|
||||
describe("composio client response limits", () => {
|
||||
test("a multi-byte character split across chunks decodes intact", async () => {
|
||||
const json = Buffer.from(JSON.stringify({ items: [{ slug: "GMAIL_SEND", description: "发送邮件" }] }), "utf8");
|
||||
const split = json.indexOf(Buffer.from("邮", "utf8")) + 1;
|
||||
chunksToSend.splice(0, chunksToSend.length, json.subarray(0, split), json.subarray(split));
|
||||
const tools = await new ComposioClient("ak_test").getTools(["gmail"]);
|
||||
expect(tools.map((tool) => [tool.name, tool.description])).toEqual([["GMAIL_SEND", "发送邮件"]]);
|
||||
});
|
||||
|
||||
test("a response over the byte cap aborts and fails loudly", async () => {
|
||||
aborted = false;
|
||||
chunksToSend.splice(0, chunksToSend.length, Buffer.alloc(1_500_000, 32), Buffer.alloc(1_500_000, 32));
|
||||
await expect(new ComposioClient("ak_test").getTools(["gmail"])).rejects.toThrow("returned more than 2000000 bytes");
|
||||
expect(aborted).toBe(true);
|
||||
});
|
||||
});
|
||||
Reference in new issue
Block a user