package org.xyzh.ai.client.dto;

import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.Data;
import java.util.List;
import java.util.Map;

/**
 * @description 对话响应（阻塞模式）
 * @filename ChatResponse.java
 * @author AI Assistant
 * @copyright xyzh
 * @since 2025-11-04
 */
@Data
public class ChatResponse {

    /**
     * 消息ID
     */
    @JsonProperty("message_id")
    private String messageId;

    /**
     * 对话ID
     */
    @JsonProperty("conversation_id")
    private String conversationId;

    /**
     * 模式
     */
    private String mode;

    /**
     * 回答内容
     */
    private String answer;

    /**
     * 元数据
     */
    private Map<String, Object> metadata;

    /**
     * 创建时间
     */
    @JsonProperty("created_at")
    private Long createdAt;

    /**
     * Token使用情况
     */
    private Usage usage;

    /**
     * 检索信息
     */
    @JsonProperty("retrieval_info")
    private List<RetrievalInfo> retrievalInfo;

    @Data
    public static class Usage {
        @JsonProperty("prompt_tokens")
        private Integer promptTokens;

        @JsonProperty("prompt_unit_price")
        private String promptUnitPrice;

        @JsonProperty("prompt_price_unit")
        private String promptPriceUnit;

        @JsonProperty("prompt_price")
        private String promptPrice;

        @JsonProperty("completion_tokens")
        private Integer completionTokens;

        @JsonProperty("completion_unit_price")
        private String completionUnitPrice;

        @JsonProperty("completion_price_unit")
        private String completionPriceUnit;

        @JsonProperty("completion_price")
        private String completionPrice;

        @JsonProperty("total_tokens")
        private Integer totalTokens;

        @JsonProperty("total_price")
        private String totalPrice;

        private String currency;

        private Double latency;
    }

    @Data
    public static class RetrievalInfo {
        @JsonProperty("dataset_id")
        private String datasetId;

        @JsonProperty("dataset_name")
        private String datasetName;

        @JsonProperty("document_id")
        private String documentId;

        @JsonProperty("document_name")
        private String documentName;

        @JsonProperty("segment_id")
        private String segmentId;

        private Double score;

        private String content;
    }
}

