import { Entity } from "../../../common";
import { PolymorphicDiscriminatorData } from "../../../common/interfaces/entity.schema.interface";
import type { Company } from "../../company/entities/company";
import type { Assistant } from "../../assistant/entities/assistant";
import type { Chunk } from "../../chunk/entities/chunk.entity";
import type { AssistantAction } from "../../assistant-action/entities/assistant-action";
export type AssistantMessageRole = "user" | "assistant" | "system";
export type AssistantMessage = Entity & {
    role: AssistantMessageRole;
    content: string;
    position: number;
    suggestedQuestions?: string[];
    inputTokens?: number;
    outputTokens?: number;
    trace?: string;
    messageType?: string;
    company: Company;
    assistant: Assistant;
    references?: unknown[];
    citations?: Chunk[];
    action?: AssistantAction;
};
export declare const AssistantMessageDescriptor: import("../../../common").EntityDescriptor<AssistantMessage, {
    assistant: {
        model: import("../../../common").DataMeta;
        direction: "in";
        relationship: string;
        cardinality: "one";
        dtoKey: string;
        required: true;
        immutable: true;
    };
    references: {
        model: import("../../../common").DataMeta;
        direction: "out";
        relationship: string;
        cardinality: "many";
        dtoKey: string;
        fields: ({
            name: string;
            type: "number";
            required: false;
        } | {
            name: string;
            type: "string";
            required: false;
        })[];
        polymorphic: {
            candidates: any[];
            discriminator: (data: PolymorphicDiscriminatorData) => import("../../../common").DataModelInterface<any>;
        };
    };
    citations: {
        model: import("../../../common").DataMeta;
        direction: "out";
        relationship: string;
        cardinality: "many";
        required: false;
        dtoKey: string;
        fields: ({
            name: string;
            type: "number";
            required: true;
        } | {
            name: string;
            type: "string";
            required: false;
        })[];
    };
    action: {
        model: import("../../../common").DataMeta;
        direction: "out";
        relationship: string;
        cardinality: "one";
        required: false;
        dtoKey: string;
        immutable: true;
    };
}>;
export type AssistantMessageDescriptorType = typeof AssistantMessageDescriptor;
//# sourceMappingURL=assistant-message.d.ts.map