i18n-ai-translate/src/chats/chat_interface.ts

AI-powered localization CLI, Node library, and GitHub Action. Translate i18next JSON, Gettext PO, Java .properties, and iOS .strings with ChatGPT, Claude, Gemini, or local Ollama models.

import type { ChatParams } from "../types";
import type { ZodType, ZodTypeDef } from "zod";

export type InvalidKind = "translation" | "styling";

export default abstract class ChatInterface {
    abstract startChat(params: ChatParams): void;
    abstract sendMessage(
        message: string,
        format?: ZodType<any, ZodTypeDef, any>,
    ): Promise<string>;

    abstract resetChatHistory(): void;
    abstract rollbackLastMessage(): void;

    /**
     * Record that the last translation was rejected, so the next
     * sendMessage call picks up the nag message in history.
     * Implementations choose whether to tag that message as System or
     * User — Anthropic, for example, only accepts alternating User /
     * Assistant messages, so it reuses User.
     */
    abstract signalInvalid(kind: InvalidKind): void;

    protected invalidMessage(kind: InvalidKind): string {
        if (kind === "translation") {
            return "The provided translation is incorrect. Re-attempt the translation and conform to the same rules as the original prompt.";
        }

        return "The provided translation was correct, but the styling was not maintained. Re-attempt the translation and ensure that the output text maintains the same style as the original prompt.";
    }
}