Chat.java 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. package com.backendsys.modules.ai.chat.entity;
  2. import com.backendsys.config.Mybatis.handler.timezone.LocalDateTimeAdapter;
  3. import com.baomidou.mybatisplus.annotation.IdType;
  4. import com.baomidou.mybatisplus.annotation.TableField;
  5. import com.baomidou.mybatisplus.annotation.TableId;
  6. import com.baomidou.mybatisplus.annotation.TableName;
  7. import com.google.gson.annotations.JsonAdapter;
  8. import jakarta.validation.constraints.NotEmpty;
  9. import jakarta.validation.constraints.NotNull;
  10. import jakarta.validation.constraints.Size;
  11. import lombok.Data;
  12. import java.time.LocalDateTime;
  13. @Data
  14. @TableName("ai_chat")
  15. public class Chat {
  16. public static interface Abort{}
  17. public static interface Detail{}
  18. public static interface Create{}
  19. public static interface Update{}
  20. public static interface Delete{}
  21. public static interface DeleteOne{}
  22. @TableId(type = IdType.AUTO)
  23. @NotNull(message = "id 不能为空", groups = { DeleteOne.class })
  24. private Long id;
  25. @Size(max = 36, message = "对话历史记录ID长度不超过 {max} 字符", groups = { Abort.class, Detail.class, Delete.class })
  26. @NotEmpty(message = "history_code 不能为空", groups = { Abort.class, Detail.class, Delete.class })
  27. private String history_code;
  28. @NotEmpty(message = "model 不能为空", groups = { Create.class })
  29. // @RangeStringArray(message="模型取值有误,范围应是(HUNYUAN, DEEPSEEK_API, DEEPSEEK_R1)", value = {"HUNYUAN", "DEEPSEEK_API", "DEEPSEEK_R1"}, groups = { Create.class, Update.class })
  30. private String model;
  31. private String model_version;
  32. @NotEmpty(message = "tag 不能为空", groups = { Create.class })
  33. @TableField(exist = false)
  34. private String tag;
  35. private Long user_id;
  36. // private String user_nickname;
  37. // private String user_avatar;
  38. private String robot_code;
  39. private String role; // 对话角色 ("user", "assistant")
  40. @Size(max = 5000, message = "对话内容长度不超过 {max} 字符", groups = { Create.class })
  41. @NotEmpty(message = "对话内容不能为空", groups = { Create.class })
  42. private String content;
  43. private String content_type; // 对话内容类型 (LOADING-加载中, SEARCH-搜索, REPLY-回复, REPLY-回复, REPLY_ABORT-回复中止, THINK-思考, THINK_ABORT-思考中止)
  44. private Long duration;
  45. @TableField(exist = false)
  46. private Boolean internet = false; // 是否联网搜索
  47. private Integer del_flag;
  48. @JsonAdapter(LocalDateTimeAdapter.class)
  49. private LocalDateTime create_time;
  50. @JsonAdapter(LocalDateTimeAdapter.class)
  51. private LocalDateTime update_time;
  52. }