package com.zy.ai.entity;

import lombok.Data;

import java.util.List;

@Data
public class ChatCompletionRequest {

    private String model;
    private List<Message> messages;
    // 可选参数
    private Double temperature;
    private Integer max_tokens;
    private Boolean stream;
    private List<Object> tools;
    private Object tool_choice;
    private Thinking thinking;

    @Data
    public static class Message {
        private String role;    // "user" / "assistant" / "system"
        private String content;
        private String name;
        private String tool_call_id;
        private List<ToolCall> tool_calls;
    }

    @Data
    public static class ToolCall {
        private String id;
        private String type;
        private Function function;
    }

    @Data
    public static class Function {
        private String name;
        private String arguments;
    }

    @Data
    public static class Thinking {
        private String type;
    }

}