alexion
6081e3e325
Imported as a linear vendor commit so forge rebases do not replay the upstream root tree. git-subtree-dir: third_party/pi git-subtree-split: 71023aec1546d6d677a65669c5665c0927ce0dd0
69 lines
2.3 KiB
TypeScript
69 lines
2.3 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { stream as streamOpenAIResponses } from "../src/api/openai-responses.ts";
|
|
import type { Model } from "../src/types.ts";
|
|
import { normalizeContext } from "../src/utils/transcript.ts";
|
|
|
|
const usageLimitError = {
|
|
code: "subscription_sharing_usage_limit_exceeded",
|
|
message: "Usage limit reached.",
|
|
};
|
|
|
|
const model: Model<"openai-responses"> = {
|
|
id: "gpt-5-mini",
|
|
name: "GPT-5 Mini",
|
|
api: "openai-responses",
|
|
provider: "openai",
|
|
baseUrl: "https://api.openai.com/v1",
|
|
reasoning: true,
|
|
input: ["text"],
|
|
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
|
|
contextWindow: 400000,
|
|
maxTokens: 128000,
|
|
};
|
|
|
|
const context = normalizeContext({
|
|
systemPrompt: "",
|
|
messages: [{ role: "user", content: [{ type: "text", text: "hi" }], timestamp: 0 }],
|
|
tools: [],
|
|
});
|
|
|
|
async function getErrorMessage(response: Response): Promise<string | undefined> {
|
|
const result = await streamOpenAIResponses(model, context, {
|
|
apiKey: "test",
|
|
fetch: async () => response,
|
|
}).result();
|
|
expect(result.stopReason).toBe("error");
|
|
return result.errorMessage;
|
|
}
|
|
|
|
describe("OpenAI Responses ChatGPT usage limit", () => {
|
|
it("links to ChatGPT usage when the request is rejected", async () => {
|
|
const response = new Response(JSON.stringify({ error: { ...usageLimitError, type: "rate_limit_error" } }), {
|
|
status: 429,
|
|
headers: { "content-type": "application/json" },
|
|
});
|
|
|
|
const errorMessage = await getErrorMessage(response);
|
|
|
|
expect(errorMessage).toContain("subscription_sharing_usage_limit_exceeded");
|
|
expect(errorMessage).toContain("Check your ChatGPT usage: https://chatgpt.com/settings/usage");
|
|
});
|
|
|
|
it("links to ChatGPT usage when the stream fails", async () => {
|
|
const event = {
|
|
type: "response.failed",
|
|
sequence_number: 0,
|
|
response: { id: "resp_failed", status: "failed", error: usageLimitError },
|
|
};
|
|
const response = new Response(`event: response.failed\ndata: ${JSON.stringify(event)}\n\n`, {
|
|
status: 200,
|
|
headers: { "content-type": "text/event-stream" },
|
|
});
|
|
|
|
const errorMessage = await getErrorMessage(response);
|
|
|
|
expect(errorMessage).toContain("subscription_sharing_usage_limit_exceeded: Usage limit reached.");
|
|
expect(errorMessage).toContain("Check your ChatGPT usage: https://chatgpt.com/settings/usage");
|
|
});
|
|
});
|