usagedetail.go 8.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249
  1. // Code generated by ent, DO NOT EDIT.
  2. package ent
  3. import (
  4. "fmt"
  5. "strings"
  6. "time"
  7. "wechat-api/ent/usagedetail"
  8. "entgo.io/ent"
  9. "entgo.io/ent/dialect/sql"
  10. )
  11. // UsageDetail is the model entity for the UsageDetail schema.
  12. type UsageDetail struct {
  13. config `json:"-"`
  14. // ID of the ent.
  15. ID uint64 `json:"id,omitempty"`
  16. // Create Time | 创建日期
  17. CreatedAt time.Time `json:"created_at,omitempty"`
  18. // Update Time | 修改日期
  19. UpdatedAt time.Time `json:"updated_at,omitempty"`
  20. // Status 1: normal 2: ban | 状态 1 正常 2 禁用
  21. Status uint8 `json:"status,omitempty"`
  22. // 1 微信 2 名片
  23. Type int `json:"type,omitempty"`
  24. // 微信或名片id
  25. BotID string `json:"bot_id,omitempty"`
  26. // 微信id或open_id
  27. ReceiverID string `json:"receiver_id,omitempty"`
  28. // 1 cow-basic 2 cow-agent 3 cow-sop 4 mp-card 5 mp-employee
  29. App int `json:"app,omitempty"`
  30. // 名片会话id
  31. SessionID uint64 `json:"session_id,omitempty"`
  32. // 请求内容
  33. Request string `json:"request,omitempty"`
  34. // 响应内容
  35. Response string `json:"response,omitempty"`
  36. // 使用token总数
  37. TotalTokens uint64 `json:"total_tokens,omitempty"`
  38. // 请求token数
  39. PromptTokens uint64 `json:"prompt_tokens,omitempty"`
  40. // 响应token数
  41. CompletionTokens uint64 `json:"completion_tokens,omitempty"`
  42. // 机构 ID
  43. OrganizationID uint64 `json:"organization_id,omitempty"`
  44. selectValues sql.SelectValues
  45. }
  46. // scanValues returns the types for scanning values from sql.Rows.
  47. func (*UsageDetail) scanValues(columns []string) ([]any, error) {
  48. values := make([]any, len(columns))
  49. for i := range columns {
  50. switch columns[i] {
  51. case usagedetail.FieldID, usagedetail.FieldStatus, usagedetail.FieldType, usagedetail.FieldApp, usagedetail.FieldSessionID, usagedetail.FieldTotalTokens, usagedetail.FieldPromptTokens, usagedetail.FieldCompletionTokens, usagedetail.FieldOrganizationID:
  52. values[i] = new(sql.NullInt64)
  53. case usagedetail.FieldBotID, usagedetail.FieldReceiverID, usagedetail.FieldRequest, usagedetail.FieldResponse:
  54. values[i] = new(sql.NullString)
  55. case usagedetail.FieldCreatedAt, usagedetail.FieldUpdatedAt:
  56. values[i] = new(sql.NullTime)
  57. default:
  58. values[i] = new(sql.UnknownType)
  59. }
  60. }
  61. return values, nil
  62. }
  63. // assignValues assigns the values that were returned from sql.Rows (after scanning)
  64. // to the UsageDetail fields.
  65. func (ud *UsageDetail) assignValues(columns []string, values []any) error {
  66. if m, n := len(values), len(columns); m < n {
  67. return fmt.Errorf("mismatch number of scan values: %d != %d", m, n)
  68. }
  69. for i := range columns {
  70. switch columns[i] {
  71. case usagedetail.FieldID:
  72. value, ok := values[i].(*sql.NullInt64)
  73. if !ok {
  74. return fmt.Errorf("unexpected type %T for field id", value)
  75. }
  76. ud.ID = uint64(value.Int64)
  77. case usagedetail.FieldCreatedAt:
  78. if value, ok := values[i].(*sql.NullTime); !ok {
  79. return fmt.Errorf("unexpected type %T for field created_at", values[i])
  80. } else if value.Valid {
  81. ud.CreatedAt = value.Time
  82. }
  83. case usagedetail.FieldUpdatedAt:
  84. if value, ok := values[i].(*sql.NullTime); !ok {
  85. return fmt.Errorf("unexpected type %T for field updated_at", values[i])
  86. } else if value.Valid {
  87. ud.UpdatedAt = value.Time
  88. }
  89. case usagedetail.FieldStatus:
  90. if value, ok := values[i].(*sql.NullInt64); !ok {
  91. return fmt.Errorf("unexpected type %T for field status", values[i])
  92. } else if value.Valid {
  93. ud.Status = uint8(value.Int64)
  94. }
  95. case usagedetail.FieldType:
  96. if value, ok := values[i].(*sql.NullInt64); !ok {
  97. return fmt.Errorf("unexpected type %T for field type", values[i])
  98. } else if value.Valid {
  99. ud.Type = int(value.Int64)
  100. }
  101. case usagedetail.FieldBotID:
  102. if value, ok := values[i].(*sql.NullString); !ok {
  103. return fmt.Errorf("unexpected type %T for field bot_id", values[i])
  104. } else if value.Valid {
  105. ud.BotID = value.String
  106. }
  107. case usagedetail.FieldReceiverID:
  108. if value, ok := values[i].(*sql.NullString); !ok {
  109. return fmt.Errorf("unexpected type %T for field receiver_id", values[i])
  110. } else if value.Valid {
  111. ud.ReceiverID = value.String
  112. }
  113. case usagedetail.FieldApp:
  114. if value, ok := values[i].(*sql.NullInt64); !ok {
  115. return fmt.Errorf("unexpected type %T for field app", values[i])
  116. } else if value.Valid {
  117. ud.App = int(value.Int64)
  118. }
  119. case usagedetail.FieldSessionID:
  120. if value, ok := values[i].(*sql.NullInt64); !ok {
  121. return fmt.Errorf("unexpected type %T for field session_id", values[i])
  122. } else if value.Valid {
  123. ud.SessionID = uint64(value.Int64)
  124. }
  125. case usagedetail.FieldRequest:
  126. if value, ok := values[i].(*sql.NullString); !ok {
  127. return fmt.Errorf("unexpected type %T for field request", values[i])
  128. } else if value.Valid {
  129. ud.Request = value.String
  130. }
  131. case usagedetail.FieldResponse:
  132. if value, ok := values[i].(*sql.NullString); !ok {
  133. return fmt.Errorf("unexpected type %T for field response", values[i])
  134. } else if value.Valid {
  135. ud.Response = value.String
  136. }
  137. case usagedetail.FieldTotalTokens:
  138. if value, ok := values[i].(*sql.NullInt64); !ok {
  139. return fmt.Errorf("unexpected type %T for field total_tokens", values[i])
  140. } else if value.Valid {
  141. ud.TotalTokens = uint64(value.Int64)
  142. }
  143. case usagedetail.FieldPromptTokens:
  144. if value, ok := values[i].(*sql.NullInt64); !ok {
  145. return fmt.Errorf("unexpected type %T for field prompt_tokens", values[i])
  146. } else if value.Valid {
  147. ud.PromptTokens = uint64(value.Int64)
  148. }
  149. case usagedetail.FieldCompletionTokens:
  150. if value, ok := values[i].(*sql.NullInt64); !ok {
  151. return fmt.Errorf("unexpected type %T for field completion_tokens", values[i])
  152. } else if value.Valid {
  153. ud.CompletionTokens = uint64(value.Int64)
  154. }
  155. case usagedetail.FieldOrganizationID:
  156. if value, ok := values[i].(*sql.NullInt64); !ok {
  157. return fmt.Errorf("unexpected type %T for field organization_id", values[i])
  158. } else if value.Valid {
  159. ud.OrganizationID = uint64(value.Int64)
  160. }
  161. default:
  162. ud.selectValues.Set(columns[i], values[i])
  163. }
  164. }
  165. return nil
  166. }
  167. // Value returns the ent.Value that was dynamically selected and assigned to the UsageDetail.
  168. // This includes values selected through modifiers, order, etc.
  169. func (ud *UsageDetail) Value(name string) (ent.Value, error) {
  170. return ud.selectValues.Get(name)
  171. }
  172. // Update returns a builder for updating this UsageDetail.
  173. // Note that you need to call UsageDetail.Unwrap() before calling this method if this UsageDetail
  174. // was returned from a transaction, and the transaction was committed or rolled back.
  175. func (ud *UsageDetail) Update() *UsageDetailUpdateOne {
  176. return NewUsageDetailClient(ud.config).UpdateOne(ud)
  177. }
  178. // Unwrap unwraps the UsageDetail entity that was returned from a transaction after it was closed,
  179. // so that all future queries will be executed through the driver which created the transaction.
  180. func (ud *UsageDetail) Unwrap() *UsageDetail {
  181. _tx, ok := ud.config.driver.(*txDriver)
  182. if !ok {
  183. panic("ent: UsageDetail is not a transactional entity")
  184. }
  185. ud.config.driver = _tx.drv
  186. return ud
  187. }
  188. // String implements the fmt.Stringer.
  189. func (ud *UsageDetail) String() string {
  190. var builder strings.Builder
  191. builder.WriteString("UsageDetail(")
  192. builder.WriteString(fmt.Sprintf("id=%v, ", ud.ID))
  193. builder.WriteString("created_at=")
  194. builder.WriteString(ud.CreatedAt.Format(time.ANSIC))
  195. builder.WriteString(", ")
  196. builder.WriteString("updated_at=")
  197. builder.WriteString(ud.UpdatedAt.Format(time.ANSIC))
  198. builder.WriteString(", ")
  199. builder.WriteString("status=")
  200. builder.WriteString(fmt.Sprintf("%v", ud.Status))
  201. builder.WriteString(", ")
  202. builder.WriteString("type=")
  203. builder.WriteString(fmt.Sprintf("%v", ud.Type))
  204. builder.WriteString(", ")
  205. builder.WriteString("bot_id=")
  206. builder.WriteString(ud.BotID)
  207. builder.WriteString(", ")
  208. builder.WriteString("receiver_id=")
  209. builder.WriteString(ud.ReceiverID)
  210. builder.WriteString(", ")
  211. builder.WriteString("app=")
  212. builder.WriteString(fmt.Sprintf("%v", ud.App))
  213. builder.WriteString(", ")
  214. builder.WriteString("session_id=")
  215. builder.WriteString(fmt.Sprintf("%v", ud.SessionID))
  216. builder.WriteString(", ")
  217. builder.WriteString("request=")
  218. builder.WriteString(ud.Request)
  219. builder.WriteString(", ")
  220. builder.WriteString("response=")
  221. builder.WriteString(ud.Response)
  222. builder.WriteString(", ")
  223. builder.WriteString("total_tokens=")
  224. builder.WriteString(fmt.Sprintf("%v", ud.TotalTokens))
  225. builder.WriteString(", ")
  226. builder.WriteString("prompt_tokens=")
  227. builder.WriteString(fmt.Sprintf("%v", ud.PromptTokens))
  228. builder.WriteString(", ")
  229. builder.WriteString("completion_tokens=")
  230. builder.WriteString(fmt.Sprintf("%v", ud.CompletionTokens))
  231. builder.WriteString(", ")
  232. builder.WriteString("organization_id=")
  233. builder.WriteString(fmt.Sprintf("%v", ud.OrganizationID))
  234. builder.WriteByte(')')
  235. return builder.String()
  236. }
  237. // UsageDetails is a parsable slice of UsageDetail.
  238. type UsageDetails []*UsageDetail