import type { AssistantMessage } from "@mariozechner/pi-ai";

import { describe, expect, it } from "vitest";

import { MALFORMED_STREAMING_FRAGMENT_ERROR_MESSAGE } from "../shared/assistant-error-format.js";

import {

BILLING_ERROR_USER_MESSAGE,

formatBillingErrorMessage,

formatAssistantErrorText,

getApiErrorPayloadFingerprint,

formatRawAssistantErrorForUi,

isRawApiErrorPayload,

sanitizeUserFacingText,

} from "./pi-embedded-helpers.js";

import { makeAssistantMessageFixture } from "./test-helpers/assistant-message-fixtures.js";

"LLM request failed: provider returned an invalid streaming response. Please try again.",

);

});

it("sanitizes transport-classified malformed streaming fragments (#59076)", () => {

const msg = makeAssistantError(MALFORMED_STREAMING_FRAGMENT_ERROR_MESSAGE);

expect(formatAssistantErrorText(msg)).toBe(

"LLM streaming response contained a malformed fragment. Please try again.",

);

});

it("does not broadly rewrite non-streaming 'Unexpected token' JSON parse errors", () => {

const msg = makeAssistantError("Unexpected token < in JSON at position 0");

expect(formatAssistantErrorText(msg)).toBe("Unexpected token < in JSON at position 0");

});

it("does not rewrite non-streaming provider JSON request-validation diagnostics", () => {

const msg = makeAssistantError("Expected value in JSON at position 12 for messages.0.content");

expect(formatAssistantErrorText(msg)).toBe(

"Expected value in JSON at position 12 for messages.0.content",

);

});

it("keeps provider request-validation JSON diagnostics actionable", () => {

const msg = makeAssistantError(

'{"type":"error","error":{"type":"invalid_request_error","message":"Expected value in JSON at position 12 for messages.0.content"}}',

);

expect(formatAssistantErrorText(msg)).toBe(

"LLM request rejected: Expected value in JSON at position 12 for messages.0.content",

);

});

});

describe("formatRawAssistantErrorForUi", () => {

);

});

});

describe("sanitizeUserFacingText — streaming JSON parse error (#59076)", () => {

it("rewrites transport-classified malformed streaming fragments in error context", () => {

const result = sanitizeUserFacingText(MALFORMED_STREAMING_FRAGMENT_ERROR_MESSAGE, {

errorContext: true,

});

expect(result).toBe("LLM streaming response contained a malformed fragment. Please try again.");

});

it("does not rewrite JSON parse error when not in error context", () => {

// When not in error context, the text could be legitimate assistant content

// mentioning JSON errors. Don't rewrite.

const text =

"Expected ',' or '}' after property value in JSON at position 334 (line 1 column 335)";

const result = sanitizeUserFacingText(text, { errorContext: false });

expect(result).toBe(text);

});

});