big_model.go 984 B

123456789101112131415161718192021222324252627282930313233343536
  1. package dto
  2. type AskAiModel struct {
  3. Content string `form:"content" json:"content"`
  4. }
  5. type BigModelParamsMessage struct {
  6. Role string `json:"role"`
  7. Content string `json:"content"`
  8. }
  9. type BigModelParams struct {
  10. Model string `json:"model"`
  11. Messages []BigModelParamsMessage `json:"messages"`
  12. }
  13. type BigModelChoice struct {
  14. FinishReason string `json:"finish_reason"`
  15. Index int64 `json:"index"`
  16. Message BigModelParamsMessage `json:"message"`
  17. }
  18. type ModelUsage struct {
  19. CompletionTokens int64 `json:"completion_tokens"`
  20. PromptTokens int64 `json:"prompt_tokens"`
  21. TotalTokens int64 `json:"total_tokens"`
  22. }
  23. type BigModelResp struct {
  24. Choices []BigModelChoice `json:"choices"`
  25. Created int64 `json:"created"`
  26. Id string `json:"id"`
  27. Model string `json:"model"`
  28. RequestId string `json:"request_id"`
  29. Usage ModelUsage `json:"usage"`
  30. }