usagedetail_create.go 54 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844
  1. // Code generated by ent, DO NOT EDIT.
  2. package ent
  3. import (
  4. "context"
  5. "errors"
  6. "fmt"
  7. "time"
  8. "wechat-api/ent/custom_types"
  9. "wechat-api/ent/usagedetail"
  10. "entgo.io/ent/dialect/sql"
  11. "entgo.io/ent/dialect/sql/sqlgraph"
  12. "entgo.io/ent/schema/field"
  13. )
  14. // UsageDetailCreate is the builder for creating a UsageDetail entity.
  15. type UsageDetailCreate struct {
  16. config
  17. mutation *UsageDetailMutation
  18. hooks []Hook
  19. conflict []sql.ConflictOption
  20. }
  21. // SetCreatedAt sets the "created_at" field.
  22. func (udc *UsageDetailCreate) SetCreatedAt(t time.Time) *UsageDetailCreate {
  23. udc.mutation.SetCreatedAt(t)
  24. return udc
  25. }
  26. // SetNillableCreatedAt sets the "created_at" field if the given value is not nil.
  27. func (udc *UsageDetailCreate) SetNillableCreatedAt(t *time.Time) *UsageDetailCreate {
  28. if t != nil {
  29. udc.SetCreatedAt(*t)
  30. }
  31. return udc
  32. }
  33. // SetUpdatedAt sets the "updated_at" field.
  34. func (udc *UsageDetailCreate) SetUpdatedAt(t time.Time) *UsageDetailCreate {
  35. udc.mutation.SetUpdatedAt(t)
  36. return udc
  37. }
  38. // SetNillableUpdatedAt sets the "updated_at" field if the given value is not nil.
  39. func (udc *UsageDetailCreate) SetNillableUpdatedAt(t *time.Time) *UsageDetailCreate {
  40. if t != nil {
  41. udc.SetUpdatedAt(*t)
  42. }
  43. return udc
  44. }
  45. // SetStatus sets the "status" field.
  46. func (udc *UsageDetailCreate) SetStatus(u uint8) *UsageDetailCreate {
  47. udc.mutation.SetStatus(u)
  48. return udc
  49. }
  50. // SetNillableStatus sets the "status" field if the given value is not nil.
  51. func (udc *UsageDetailCreate) SetNillableStatus(u *uint8) *UsageDetailCreate {
  52. if u != nil {
  53. udc.SetStatus(*u)
  54. }
  55. return udc
  56. }
  57. // SetType sets the "type" field.
  58. func (udc *UsageDetailCreate) SetType(i int) *UsageDetailCreate {
  59. udc.mutation.SetType(i)
  60. return udc
  61. }
  62. // SetNillableType sets the "type" field if the given value is not nil.
  63. func (udc *UsageDetailCreate) SetNillableType(i *int) *UsageDetailCreate {
  64. if i != nil {
  65. udc.SetType(*i)
  66. }
  67. return udc
  68. }
  69. // SetBotID sets the "bot_id" field.
  70. func (udc *UsageDetailCreate) SetBotID(s string) *UsageDetailCreate {
  71. udc.mutation.SetBotID(s)
  72. return udc
  73. }
  74. // SetNillableBotID sets the "bot_id" field if the given value is not nil.
  75. func (udc *UsageDetailCreate) SetNillableBotID(s *string) *UsageDetailCreate {
  76. if s != nil {
  77. udc.SetBotID(*s)
  78. }
  79. return udc
  80. }
  81. // SetReceiverID sets the "receiver_id" field.
  82. func (udc *UsageDetailCreate) SetReceiverID(s string) *UsageDetailCreate {
  83. udc.mutation.SetReceiverID(s)
  84. return udc
  85. }
  86. // SetNillableReceiverID sets the "receiver_id" field if the given value is not nil.
  87. func (udc *UsageDetailCreate) SetNillableReceiverID(s *string) *UsageDetailCreate {
  88. if s != nil {
  89. udc.SetReceiverID(*s)
  90. }
  91. return udc
  92. }
  93. // SetApp sets the "app" field.
  94. func (udc *UsageDetailCreate) SetApp(i int) *UsageDetailCreate {
  95. udc.mutation.SetApp(i)
  96. return udc
  97. }
  98. // SetNillableApp sets the "app" field if the given value is not nil.
  99. func (udc *UsageDetailCreate) SetNillableApp(i *int) *UsageDetailCreate {
  100. if i != nil {
  101. udc.SetApp(*i)
  102. }
  103. return udc
  104. }
  105. // SetSessionID sets the "session_id" field.
  106. func (udc *UsageDetailCreate) SetSessionID(u uint64) *UsageDetailCreate {
  107. udc.mutation.SetSessionID(u)
  108. return udc
  109. }
  110. // SetNillableSessionID sets the "session_id" field if the given value is not nil.
  111. func (udc *UsageDetailCreate) SetNillableSessionID(u *uint64) *UsageDetailCreate {
  112. if u != nil {
  113. udc.SetSessionID(*u)
  114. }
  115. return udc
  116. }
  117. // SetRequest sets the "request" field.
  118. func (udc *UsageDetailCreate) SetRequest(s string) *UsageDetailCreate {
  119. udc.mutation.SetRequest(s)
  120. return udc
  121. }
  122. // SetNillableRequest sets the "request" field if the given value is not nil.
  123. func (udc *UsageDetailCreate) SetNillableRequest(s *string) *UsageDetailCreate {
  124. if s != nil {
  125. udc.SetRequest(*s)
  126. }
  127. return udc
  128. }
  129. // SetResponse sets the "response" field.
  130. func (udc *UsageDetailCreate) SetResponse(s string) *UsageDetailCreate {
  131. udc.mutation.SetResponse(s)
  132. return udc
  133. }
  134. // SetNillableResponse sets the "response" field if the given value is not nil.
  135. func (udc *UsageDetailCreate) SetNillableResponse(s *string) *UsageDetailCreate {
  136. if s != nil {
  137. udc.SetResponse(*s)
  138. }
  139. return udc
  140. }
  141. // SetOriginalData sets the "original_data" field.
  142. func (udc *UsageDetailCreate) SetOriginalData(ctd custom_types.OriginalData) *UsageDetailCreate {
  143. udc.mutation.SetOriginalData(ctd)
  144. return udc
  145. }
  146. // SetTotalTokens sets the "total_tokens" field.
  147. func (udc *UsageDetailCreate) SetTotalTokens(u uint64) *UsageDetailCreate {
  148. udc.mutation.SetTotalTokens(u)
  149. return udc
  150. }
  151. // SetNillableTotalTokens sets the "total_tokens" field if the given value is not nil.
  152. func (udc *UsageDetailCreate) SetNillableTotalTokens(u *uint64) *UsageDetailCreate {
  153. if u != nil {
  154. udc.SetTotalTokens(*u)
  155. }
  156. return udc
  157. }
  158. // SetPromptTokens sets the "prompt_tokens" field.
  159. func (udc *UsageDetailCreate) SetPromptTokens(u uint64) *UsageDetailCreate {
  160. udc.mutation.SetPromptTokens(u)
  161. return udc
  162. }
  163. // SetNillablePromptTokens sets the "prompt_tokens" field if the given value is not nil.
  164. func (udc *UsageDetailCreate) SetNillablePromptTokens(u *uint64) *UsageDetailCreate {
  165. if u != nil {
  166. udc.SetPromptTokens(*u)
  167. }
  168. return udc
  169. }
  170. // SetCompletionTokens sets the "completion_tokens" field.
  171. func (udc *UsageDetailCreate) SetCompletionTokens(u uint64) *UsageDetailCreate {
  172. udc.mutation.SetCompletionTokens(u)
  173. return udc
  174. }
  175. // SetNillableCompletionTokens sets the "completion_tokens" field if the given value is not nil.
  176. func (udc *UsageDetailCreate) SetNillableCompletionTokens(u *uint64) *UsageDetailCreate {
  177. if u != nil {
  178. udc.SetCompletionTokens(*u)
  179. }
  180. return udc
  181. }
  182. // SetOrganizationID sets the "organization_id" field.
  183. func (udc *UsageDetailCreate) SetOrganizationID(u uint64) *UsageDetailCreate {
  184. udc.mutation.SetOrganizationID(u)
  185. return udc
  186. }
  187. // SetNillableOrganizationID sets the "organization_id" field if the given value is not nil.
  188. func (udc *UsageDetailCreate) SetNillableOrganizationID(u *uint64) *UsageDetailCreate {
  189. if u != nil {
  190. udc.SetOrganizationID(*u)
  191. }
  192. return udc
  193. }
  194. // SetModel sets the "model" field.
  195. func (udc *UsageDetailCreate) SetModel(s string) *UsageDetailCreate {
  196. udc.mutation.SetModel(s)
  197. return udc
  198. }
  199. // SetNillableModel sets the "model" field if the given value is not nil.
  200. func (udc *UsageDetailCreate) SetNillableModel(s *string) *UsageDetailCreate {
  201. if s != nil {
  202. udc.SetModel(*s)
  203. }
  204. return udc
  205. }
  206. // SetCredits sets the "credits" field.
  207. func (udc *UsageDetailCreate) SetCredits(f float64) *UsageDetailCreate {
  208. udc.mutation.SetCredits(f)
  209. return udc
  210. }
  211. // SetNillableCredits sets the "credits" field if the given value is not nil.
  212. func (udc *UsageDetailCreate) SetNillableCredits(f *float64) *UsageDetailCreate {
  213. if f != nil {
  214. udc.SetCredits(*f)
  215. }
  216. return udc
  217. }
  218. // SetID sets the "id" field.
  219. func (udc *UsageDetailCreate) SetID(u uint64) *UsageDetailCreate {
  220. udc.mutation.SetID(u)
  221. return udc
  222. }
  223. // Mutation returns the UsageDetailMutation object of the builder.
  224. func (udc *UsageDetailCreate) Mutation() *UsageDetailMutation {
  225. return udc.mutation
  226. }
  227. // Save creates the UsageDetail in the database.
  228. func (udc *UsageDetailCreate) Save(ctx context.Context) (*UsageDetail, error) {
  229. udc.defaults()
  230. return withHooks(ctx, udc.sqlSave, udc.mutation, udc.hooks)
  231. }
  232. // SaveX calls Save and panics if Save returns an error.
  233. func (udc *UsageDetailCreate) SaveX(ctx context.Context) *UsageDetail {
  234. v, err := udc.Save(ctx)
  235. if err != nil {
  236. panic(err)
  237. }
  238. return v
  239. }
  240. // Exec executes the query.
  241. func (udc *UsageDetailCreate) Exec(ctx context.Context) error {
  242. _, err := udc.Save(ctx)
  243. return err
  244. }
  245. // ExecX is like Exec, but panics if an error occurs.
  246. func (udc *UsageDetailCreate) ExecX(ctx context.Context) {
  247. if err := udc.Exec(ctx); err != nil {
  248. panic(err)
  249. }
  250. }
  251. // defaults sets the default values of the builder before save.
  252. func (udc *UsageDetailCreate) defaults() {
  253. if _, ok := udc.mutation.CreatedAt(); !ok {
  254. v := usagedetail.DefaultCreatedAt()
  255. udc.mutation.SetCreatedAt(v)
  256. }
  257. if _, ok := udc.mutation.UpdatedAt(); !ok {
  258. v := usagedetail.DefaultUpdatedAt()
  259. udc.mutation.SetUpdatedAt(v)
  260. }
  261. if _, ok := udc.mutation.Status(); !ok {
  262. v := usagedetail.DefaultStatus
  263. udc.mutation.SetStatus(v)
  264. }
  265. if _, ok := udc.mutation.GetType(); !ok {
  266. v := usagedetail.DefaultType
  267. udc.mutation.SetType(v)
  268. }
  269. if _, ok := udc.mutation.BotID(); !ok {
  270. v := usagedetail.DefaultBotID
  271. udc.mutation.SetBotID(v)
  272. }
  273. if _, ok := udc.mutation.ReceiverID(); !ok {
  274. v := usagedetail.DefaultReceiverID
  275. udc.mutation.SetReceiverID(v)
  276. }
  277. if _, ok := udc.mutation.App(); !ok {
  278. v := usagedetail.DefaultApp
  279. udc.mutation.SetApp(v)
  280. }
  281. if _, ok := udc.mutation.SessionID(); !ok {
  282. v := usagedetail.DefaultSessionID
  283. udc.mutation.SetSessionID(v)
  284. }
  285. if _, ok := udc.mutation.Request(); !ok {
  286. v := usagedetail.DefaultRequest
  287. udc.mutation.SetRequest(v)
  288. }
  289. if _, ok := udc.mutation.Response(); !ok {
  290. v := usagedetail.DefaultResponse
  291. udc.mutation.SetResponse(v)
  292. }
  293. if _, ok := udc.mutation.TotalTokens(); !ok {
  294. v := usagedetail.DefaultTotalTokens
  295. udc.mutation.SetTotalTokens(v)
  296. }
  297. if _, ok := udc.mutation.PromptTokens(); !ok {
  298. v := usagedetail.DefaultPromptTokens
  299. udc.mutation.SetPromptTokens(v)
  300. }
  301. if _, ok := udc.mutation.CompletionTokens(); !ok {
  302. v := usagedetail.DefaultCompletionTokens
  303. udc.mutation.SetCompletionTokens(v)
  304. }
  305. if _, ok := udc.mutation.OrganizationID(); !ok {
  306. v := usagedetail.DefaultOrganizationID
  307. udc.mutation.SetOrganizationID(v)
  308. }
  309. if _, ok := udc.mutation.Model(); !ok {
  310. v := usagedetail.DefaultModel
  311. udc.mutation.SetModel(v)
  312. }
  313. }
  314. // check runs all checks and user-defined validators on the builder.
  315. func (udc *UsageDetailCreate) check() error {
  316. if _, ok := udc.mutation.CreatedAt(); !ok {
  317. return &ValidationError{Name: "created_at", err: errors.New(`ent: missing required field "UsageDetail.created_at"`)}
  318. }
  319. if _, ok := udc.mutation.UpdatedAt(); !ok {
  320. return &ValidationError{Name: "updated_at", err: errors.New(`ent: missing required field "UsageDetail.updated_at"`)}
  321. }
  322. if _, ok := udc.mutation.BotID(); !ok {
  323. return &ValidationError{Name: "bot_id", err: errors.New(`ent: missing required field "UsageDetail.bot_id"`)}
  324. }
  325. if _, ok := udc.mutation.ReceiverID(); !ok {
  326. return &ValidationError{Name: "receiver_id", err: errors.New(`ent: missing required field "UsageDetail.receiver_id"`)}
  327. }
  328. if _, ok := udc.mutation.Request(); !ok {
  329. return &ValidationError{Name: "request", err: errors.New(`ent: missing required field "UsageDetail.request"`)}
  330. }
  331. if _, ok := udc.mutation.Response(); !ok {
  332. return &ValidationError{Name: "response", err: errors.New(`ent: missing required field "UsageDetail.response"`)}
  333. }
  334. if _, ok := udc.mutation.OriginalData(); !ok {
  335. return &ValidationError{Name: "original_data", err: errors.New(`ent: missing required field "UsageDetail.original_data"`)}
  336. }
  337. return nil
  338. }
  339. func (udc *UsageDetailCreate) sqlSave(ctx context.Context) (*UsageDetail, error) {
  340. if err := udc.check(); err != nil {
  341. return nil, err
  342. }
  343. _node, _spec := udc.createSpec()
  344. if err := sqlgraph.CreateNode(ctx, udc.driver, _spec); err != nil {
  345. if sqlgraph.IsConstraintError(err) {
  346. err = &ConstraintError{msg: err.Error(), wrap: err}
  347. }
  348. return nil, err
  349. }
  350. if _spec.ID.Value != _node.ID {
  351. id := _spec.ID.Value.(int64)
  352. _node.ID = uint64(id)
  353. }
  354. udc.mutation.id = &_node.ID
  355. udc.mutation.done = true
  356. return _node, nil
  357. }
  358. func (udc *UsageDetailCreate) createSpec() (*UsageDetail, *sqlgraph.CreateSpec) {
  359. var (
  360. _node = &UsageDetail{config: udc.config}
  361. _spec = sqlgraph.NewCreateSpec(usagedetail.Table, sqlgraph.NewFieldSpec(usagedetail.FieldID, field.TypeUint64))
  362. )
  363. _spec.OnConflict = udc.conflict
  364. if id, ok := udc.mutation.ID(); ok {
  365. _node.ID = id
  366. _spec.ID.Value = id
  367. }
  368. if value, ok := udc.mutation.CreatedAt(); ok {
  369. _spec.SetField(usagedetail.FieldCreatedAt, field.TypeTime, value)
  370. _node.CreatedAt = value
  371. }
  372. if value, ok := udc.mutation.UpdatedAt(); ok {
  373. _spec.SetField(usagedetail.FieldUpdatedAt, field.TypeTime, value)
  374. _node.UpdatedAt = value
  375. }
  376. if value, ok := udc.mutation.Status(); ok {
  377. _spec.SetField(usagedetail.FieldStatus, field.TypeUint8, value)
  378. _node.Status = value
  379. }
  380. if value, ok := udc.mutation.GetType(); ok {
  381. _spec.SetField(usagedetail.FieldType, field.TypeInt, value)
  382. _node.Type = value
  383. }
  384. if value, ok := udc.mutation.BotID(); ok {
  385. _spec.SetField(usagedetail.FieldBotID, field.TypeString, value)
  386. _node.BotID = value
  387. }
  388. if value, ok := udc.mutation.ReceiverID(); ok {
  389. _spec.SetField(usagedetail.FieldReceiverID, field.TypeString, value)
  390. _node.ReceiverID = value
  391. }
  392. if value, ok := udc.mutation.App(); ok {
  393. _spec.SetField(usagedetail.FieldApp, field.TypeInt, value)
  394. _node.App = value
  395. }
  396. if value, ok := udc.mutation.SessionID(); ok {
  397. _spec.SetField(usagedetail.FieldSessionID, field.TypeUint64, value)
  398. _node.SessionID = value
  399. }
  400. if value, ok := udc.mutation.Request(); ok {
  401. _spec.SetField(usagedetail.FieldRequest, field.TypeString, value)
  402. _node.Request = value
  403. }
  404. if value, ok := udc.mutation.Response(); ok {
  405. _spec.SetField(usagedetail.FieldResponse, field.TypeString, value)
  406. _node.Response = value
  407. }
  408. if value, ok := udc.mutation.OriginalData(); ok {
  409. _spec.SetField(usagedetail.FieldOriginalData, field.TypeJSON, value)
  410. _node.OriginalData = value
  411. }
  412. if value, ok := udc.mutation.TotalTokens(); ok {
  413. _spec.SetField(usagedetail.FieldTotalTokens, field.TypeUint64, value)
  414. _node.TotalTokens = value
  415. }
  416. if value, ok := udc.mutation.PromptTokens(); ok {
  417. _spec.SetField(usagedetail.FieldPromptTokens, field.TypeUint64, value)
  418. _node.PromptTokens = value
  419. }
  420. if value, ok := udc.mutation.CompletionTokens(); ok {
  421. _spec.SetField(usagedetail.FieldCompletionTokens, field.TypeUint64, value)
  422. _node.CompletionTokens = value
  423. }
  424. if value, ok := udc.mutation.OrganizationID(); ok {
  425. _spec.SetField(usagedetail.FieldOrganizationID, field.TypeUint64, value)
  426. _node.OrganizationID = value
  427. }
  428. if value, ok := udc.mutation.Model(); ok {
  429. _spec.SetField(usagedetail.FieldModel, field.TypeString, value)
  430. _node.Model = value
  431. }
  432. if value, ok := udc.mutation.Credits(); ok {
  433. _spec.SetField(usagedetail.FieldCredits, field.TypeFloat64, value)
  434. _node.Credits = value
  435. }
  436. return _node, _spec
  437. }
  438. // OnConflict allows configuring the `ON CONFLICT` / `ON DUPLICATE KEY` clause
  439. // of the `INSERT` statement. For example:
  440. //
  441. // client.UsageDetail.Create().
  442. // SetCreatedAt(v).
  443. // OnConflict(
  444. // // Update the row with the new values
  445. // // the was proposed for insertion.
  446. // sql.ResolveWithNewValues(),
  447. // ).
  448. // // Override some of the fields with custom
  449. // // update values.
  450. // Update(func(u *ent.UsageDetailUpsert) {
  451. // SetCreatedAt(v+v).
  452. // }).
  453. // Exec(ctx)
  454. func (udc *UsageDetailCreate) OnConflict(opts ...sql.ConflictOption) *UsageDetailUpsertOne {
  455. udc.conflict = opts
  456. return &UsageDetailUpsertOne{
  457. create: udc,
  458. }
  459. }
  460. // OnConflictColumns calls `OnConflict` and configures the columns
  461. // as conflict target. Using this option is equivalent to using:
  462. //
  463. // client.UsageDetail.Create().
  464. // OnConflict(sql.ConflictColumns(columns...)).
  465. // Exec(ctx)
  466. func (udc *UsageDetailCreate) OnConflictColumns(columns ...string) *UsageDetailUpsertOne {
  467. udc.conflict = append(udc.conflict, sql.ConflictColumns(columns...))
  468. return &UsageDetailUpsertOne{
  469. create: udc,
  470. }
  471. }
  472. type (
  473. // UsageDetailUpsertOne is the builder for "upsert"-ing
  474. // one UsageDetail node.
  475. UsageDetailUpsertOne struct {
  476. create *UsageDetailCreate
  477. }
  478. // UsageDetailUpsert is the "OnConflict" setter.
  479. UsageDetailUpsert struct {
  480. *sql.UpdateSet
  481. }
  482. )
  483. // SetUpdatedAt sets the "updated_at" field.
  484. func (u *UsageDetailUpsert) SetUpdatedAt(v time.Time) *UsageDetailUpsert {
  485. u.Set(usagedetail.FieldUpdatedAt, v)
  486. return u
  487. }
  488. // UpdateUpdatedAt sets the "updated_at" field to the value that was provided on create.
  489. func (u *UsageDetailUpsert) UpdateUpdatedAt() *UsageDetailUpsert {
  490. u.SetExcluded(usagedetail.FieldUpdatedAt)
  491. return u
  492. }
  493. // SetStatus sets the "status" field.
  494. func (u *UsageDetailUpsert) SetStatus(v uint8) *UsageDetailUpsert {
  495. u.Set(usagedetail.FieldStatus, v)
  496. return u
  497. }
  498. // UpdateStatus sets the "status" field to the value that was provided on create.
  499. func (u *UsageDetailUpsert) UpdateStatus() *UsageDetailUpsert {
  500. u.SetExcluded(usagedetail.FieldStatus)
  501. return u
  502. }
  503. // AddStatus adds v to the "status" field.
  504. func (u *UsageDetailUpsert) AddStatus(v uint8) *UsageDetailUpsert {
  505. u.Add(usagedetail.FieldStatus, v)
  506. return u
  507. }
  508. // ClearStatus clears the value of the "status" field.
  509. func (u *UsageDetailUpsert) ClearStatus() *UsageDetailUpsert {
  510. u.SetNull(usagedetail.FieldStatus)
  511. return u
  512. }
  513. // SetType sets the "type" field.
  514. func (u *UsageDetailUpsert) SetType(v int) *UsageDetailUpsert {
  515. u.Set(usagedetail.FieldType, v)
  516. return u
  517. }
  518. // UpdateType sets the "type" field to the value that was provided on create.
  519. func (u *UsageDetailUpsert) UpdateType() *UsageDetailUpsert {
  520. u.SetExcluded(usagedetail.FieldType)
  521. return u
  522. }
  523. // AddType adds v to the "type" field.
  524. func (u *UsageDetailUpsert) AddType(v int) *UsageDetailUpsert {
  525. u.Add(usagedetail.FieldType, v)
  526. return u
  527. }
  528. // ClearType clears the value of the "type" field.
  529. func (u *UsageDetailUpsert) ClearType() *UsageDetailUpsert {
  530. u.SetNull(usagedetail.FieldType)
  531. return u
  532. }
  533. // SetBotID sets the "bot_id" field.
  534. func (u *UsageDetailUpsert) SetBotID(v string) *UsageDetailUpsert {
  535. u.Set(usagedetail.FieldBotID, v)
  536. return u
  537. }
  538. // UpdateBotID sets the "bot_id" field to the value that was provided on create.
  539. func (u *UsageDetailUpsert) UpdateBotID() *UsageDetailUpsert {
  540. u.SetExcluded(usagedetail.FieldBotID)
  541. return u
  542. }
  543. // SetReceiverID sets the "receiver_id" field.
  544. func (u *UsageDetailUpsert) SetReceiverID(v string) *UsageDetailUpsert {
  545. u.Set(usagedetail.FieldReceiverID, v)
  546. return u
  547. }
  548. // UpdateReceiverID sets the "receiver_id" field to the value that was provided on create.
  549. func (u *UsageDetailUpsert) UpdateReceiverID() *UsageDetailUpsert {
  550. u.SetExcluded(usagedetail.FieldReceiverID)
  551. return u
  552. }
  553. // SetApp sets the "app" field.
  554. func (u *UsageDetailUpsert) SetApp(v int) *UsageDetailUpsert {
  555. u.Set(usagedetail.FieldApp, v)
  556. return u
  557. }
  558. // UpdateApp sets the "app" field to the value that was provided on create.
  559. func (u *UsageDetailUpsert) UpdateApp() *UsageDetailUpsert {
  560. u.SetExcluded(usagedetail.FieldApp)
  561. return u
  562. }
  563. // AddApp adds v to the "app" field.
  564. func (u *UsageDetailUpsert) AddApp(v int) *UsageDetailUpsert {
  565. u.Add(usagedetail.FieldApp, v)
  566. return u
  567. }
  568. // ClearApp clears the value of the "app" field.
  569. func (u *UsageDetailUpsert) ClearApp() *UsageDetailUpsert {
  570. u.SetNull(usagedetail.FieldApp)
  571. return u
  572. }
  573. // SetSessionID sets the "session_id" field.
  574. func (u *UsageDetailUpsert) SetSessionID(v uint64) *UsageDetailUpsert {
  575. u.Set(usagedetail.FieldSessionID, v)
  576. return u
  577. }
  578. // UpdateSessionID sets the "session_id" field to the value that was provided on create.
  579. func (u *UsageDetailUpsert) UpdateSessionID() *UsageDetailUpsert {
  580. u.SetExcluded(usagedetail.FieldSessionID)
  581. return u
  582. }
  583. // AddSessionID adds v to the "session_id" field.
  584. func (u *UsageDetailUpsert) AddSessionID(v uint64) *UsageDetailUpsert {
  585. u.Add(usagedetail.FieldSessionID, v)
  586. return u
  587. }
  588. // ClearSessionID clears the value of the "session_id" field.
  589. func (u *UsageDetailUpsert) ClearSessionID() *UsageDetailUpsert {
  590. u.SetNull(usagedetail.FieldSessionID)
  591. return u
  592. }
  593. // SetRequest sets the "request" field.
  594. func (u *UsageDetailUpsert) SetRequest(v string) *UsageDetailUpsert {
  595. u.Set(usagedetail.FieldRequest, v)
  596. return u
  597. }
  598. // UpdateRequest sets the "request" field to the value that was provided on create.
  599. func (u *UsageDetailUpsert) UpdateRequest() *UsageDetailUpsert {
  600. u.SetExcluded(usagedetail.FieldRequest)
  601. return u
  602. }
  603. // SetResponse sets the "response" field.
  604. func (u *UsageDetailUpsert) SetResponse(v string) *UsageDetailUpsert {
  605. u.Set(usagedetail.FieldResponse, v)
  606. return u
  607. }
  608. // UpdateResponse sets the "response" field to the value that was provided on create.
  609. func (u *UsageDetailUpsert) UpdateResponse() *UsageDetailUpsert {
  610. u.SetExcluded(usagedetail.FieldResponse)
  611. return u
  612. }
  613. // SetOriginalData sets the "original_data" field.
  614. func (u *UsageDetailUpsert) SetOriginalData(v custom_types.OriginalData) *UsageDetailUpsert {
  615. u.Set(usagedetail.FieldOriginalData, v)
  616. return u
  617. }
  618. // UpdateOriginalData sets the "original_data" field to the value that was provided on create.
  619. func (u *UsageDetailUpsert) UpdateOriginalData() *UsageDetailUpsert {
  620. u.SetExcluded(usagedetail.FieldOriginalData)
  621. return u
  622. }
  623. // SetTotalTokens sets the "total_tokens" field.
  624. func (u *UsageDetailUpsert) SetTotalTokens(v uint64) *UsageDetailUpsert {
  625. u.Set(usagedetail.FieldTotalTokens, v)
  626. return u
  627. }
  628. // UpdateTotalTokens sets the "total_tokens" field to the value that was provided on create.
  629. func (u *UsageDetailUpsert) UpdateTotalTokens() *UsageDetailUpsert {
  630. u.SetExcluded(usagedetail.FieldTotalTokens)
  631. return u
  632. }
  633. // AddTotalTokens adds v to the "total_tokens" field.
  634. func (u *UsageDetailUpsert) AddTotalTokens(v uint64) *UsageDetailUpsert {
  635. u.Add(usagedetail.FieldTotalTokens, v)
  636. return u
  637. }
  638. // ClearTotalTokens clears the value of the "total_tokens" field.
  639. func (u *UsageDetailUpsert) ClearTotalTokens() *UsageDetailUpsert {
  640. u.SetNull(usagedetail.FieldTotalTokens)
  641. return u
  642. }
  643. // SetPromptTokens sets the "prompt_tokens" field.
  644. func (u *UsageDetailUpsert) SetPromptTokens(v uint64) *UsageDetailUpsert {
  645. u.Set(usagedetail.FieldPromptTokens, v)
  646. return u
  647. }
  648. // UpdatePromptTokens sets the "prompt_tokens" field to the value that was provided on create.
  649. func (u *UsageDetailUpsert) UpdatePromptTokens() *UsageDetailUpsert {
  650. u.SetExcluded(usagedetail.FieldPromptTokens)
  651. return u
  652. }
  653. // AddPromptTokens adds v to the "prompt_tokens" field.
  654. func (u *UsageDetailUpsert) AddPromptTokens(v uint64) *UsageDetailUpsert {
  655. u.Add(usagedetail.FieldPromptTokens, v)
  656. return u
  657. }
  658. // ClearPromptTokens clears the value of the "prompt_tokens" field.
  659. func (u *UsageDetailUpsert) ClearPromptTokens() *UsageDetailUpsert {
  660. u.SetNull(usagedetail.FieldPromptTokens)
  661. return u
  662. }
  663. // SetCompletionTokens sets the "completion_tokens" field.
  664. func (u *UsageDetailUpsert) SetCompletionTokens(v uint64) *UsageDetailUpsert {
  665. u.Set(usagedetail.FieldCompletionTokens, v)
  666. return u
  667. }
  668. // UpdateCompletionTokens sets the "completion_tokens" field to the value that was provided on create.
  669. func (u *UsageDetailUpsert) UpdateCompletionTokens() *UsageDetailUpsert {
  670. u.SetExcluded(usagedetail.FieldCompletionTokens)
  671. return u
  672. }
  673. // AddCompletionTokens adds v to the "completion_tokens" field.
  674. func (u *UsageDetailUpsert) AddCompletionTokens(v uint64) *UsageDetailUpsert {
  675. u.Add(usagedetail.FieldCompletionTokens, v)
  676. return u
  677. }
  678. // ClearCompletionTokens clears the value of the "completion_tokens" field.
  679. func (u *UsageDetailUpsert) ClearCompletionTokens() *UsageDetailUpsert {
  680. u.SetNull(usagedetail.FieldCompletionTokens)
  681. return u
  682. }
  683. // SetOrganizationID sets the "organization_id" field.
  684. func (u *UsageDetailUpsert) SetOrganizationID(v uint64) *UsageDetailUpsert {
  685. u.Set(usagedetail.FieldOrganizationID, v)
  686. return u
  687. }
  688. // UpdateOrganizationID sets the "organization_id" field to the value that was provided on create.
  689. func (u *UsageDetailUpsert) UpdateOrganizationID() *UsageDetailUpsert {
  690. u.SetExcluded(usagedetail.FieldOrganizationID)
  691. return u
  692. }
  693. // AddOrganizationID adds v to the "organization_id" field.
  694. func (u *UsageDetailUpsert) AddOrganizationID(v uint64) *UsageDetailUpsert {
  695. u.Add(usagedetail.FieldOrganizationID, v)
  696. return u
  697. }
  698. // ClearOrganizationID clears the value of the "organization_id" field.
  699. func (u *UsageDetailUpsert) ClearOrganizationID() *UsageDetailUpsert {
  700. u.SetNull(usagedetail.FieldOrganizationID)
  701. return u
  702. }
  703. // SetModel sets the "model" field.
  704. func (u *UsageDetailUpsert) SetModel(v string) *UsageDetailUpsert {
  705. u.Set(usagedetail.FieldModel, v)
  706. return u
  707. }
  708. // UpdateModel sets the "model" field to the value that was provided on create.
  709. func (u *UsageDetailUpsert) UpdateModel() *UsageDetailUpsert {
  710. u.SetExcluded(usagedetail.FieldModel)
  711. return u
  712. }
  713. // ClearModel clears the value of the "model" field.
  714. func (u *UsageDetailUpsert) ClearModel() *UsageDetailUpsert {
  715. u.SetNull(usagedetail.FieldModel)
  716. return u
  717. }
  718. // SetCredits sets the "credits" field.
  719. func (u *UsageDetailUpsert) SetCredits(v float64) *UsageDetailUpsert {
  720. u.Set(usagedetail.FieldCredits, v)
  721. return u
  722. }
  723. // UpdateCredits sets the "credits" field to the value that was provided on create.
  724. func (u *UsageDetailUpsert) UpdateCredits() *UsageDetailUpsert {
  725. u.SetExcluded(usagedetail.FieldCredits)
  726. return u
  727. }
  728. // AddCredits adds v to the "credits" field.
  729. func (u *UsageDetailUpsert) AddCredits(v float64) *UsageDetailUpsert {
  730. u.Add(usagedetail.FieldCredits, v)
  731. return u
  732. }
  733. // ClearCredits clears the value of the "credits" field.
  734. func (u *UsageDetailUpsert) ClearCredits() *UsageDetailUpsert {
  735. u.SetNull(usagedetail.FieldCredits)
  736. return u
  737. }
  738. // UpdateNewValues updates the mutable fields using the new values that were set on create except the ID field.
  739. // Using this option is equivalent to using:
  740. //
  741. // client.UsageDetail.Create().
  742. // OnConflict(
  743. // sql.ResolveWithNewValues(),
  744. // sql.ResolveWith(func(u *sql.UpdateSet) {
  745. // u.SetIgnore(usagedetail.FieldID)
  746. // }),
  747. // ).
  748. // Exec(ctx)
  749. func (u *UsageDetailUpsertOne) UpdateNewValues() *UsageDetailUpsertOne {
  750. u.create.conflict = append(u.create.conflict, sql.ResolveWithNewValues())
  751. u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(s *sql.UpdateSet) {
  752. if _, exists := u.create.mutation.ID(); exists {
  753. s.SetIgnore(usagedetail.FieldID)
  754. }
  755. if _, exists := u.create.mutation.CreatedAt(); exists {
  756. s.SetIgnore(usagedetail.FieldCreatedAt)
  757. }
  758. }))
  759. return u
  760. }
  761. // Ignore sets each column to itself in case of conflict.
  762. // Using this option is equivalent to using:
  763. //
  764. // client.UsageDetail.Create().
  765. // OnConflict(sql.ResolveWithIgnore()).
  766. // Exec(ctx)
  767. func (u *UsageDetailUpsertOne) Ignore() *UsageDetailUpsertOne {
  768. u.create.conflict = append(u.create.conflict, sql.ResolveWithIgnore())
  769. return u
  770. }
  771. // DoNothing configures the conflict_action to `DO NOTHING`.
  772. // Supported only by SQLite and PostgreSQL.
  773. func (u *UsageDetailUpsertOne) DoNothing() *UsageDetailUpsertOne {
  774. u.create.conflict = append(u.create.conflict, sql.DoNothing())
  775. return u
  776. }
  777. // Update allows overriding fields `UPDATE` values. See the UsageDetailCreate.OnConflict
  778. // documentation for more info.
  779. func (u *UsageDetailUpsertOne) Update(set func(*UsageDetailUpsert)) *UsageDetailUpsertOne {
  780. u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(update *sql.UpdateSet) {
  781. set(&UsageDetailUpsert{UpdateSet: update})
  782. }))
  783. return u
  784. }
  785. // SetUpdatedAt sets the "updated_at" field.
  786. func (u *UsageDetailUpsertOne) SetUpdatedAt(v time.Time) *UsageDetailUpsertOne {
  787. return u.Update(func(s *UsageDetailUpsert) {
  788. s.SetUpdatedAt(v)
  789. })
  790. }
  791. // UpdateUpdatedAt sets the "updated_at" field to the value that was provided on create.
  792. func (u *UsageDetailUpsertOne) UpdateUpdatedAt() *UsageDetailUpsertOne {
  793. return u.Update(func(s *UsageDetailUpsert) {
  794. s.UpdateUpdatedAt()
  795. })
  796. }
  797. // SetStatus sets the "status" field.
  798. func (u *UsageDetailUpsertOne) SetStatus(v uint8) *UsageDetailUpsertOne {
  799. return u.Update(func(s *UsageDetailUpsert) {
  800. s.SetStatus(v)
  801. })
  802. }
  803. // AddStatus adds v to the "status" field.
  804. func (u *UsageDetailUpsertOne) AddStatus(v uint8) *UsageDetailUpsertOne {
  805. return u.Update(func(s *UsageDetailUpsert) {
  806. s.AddStatus(v)
  807. })
  808. }
  809. // UpdateStatus sets the "status" field to the value that was provided on create.
  810. func (u *UsageDetailUpsertOne) UpdateStatus() *UsageDetailUpsertOne {
  811. return u.Update(func(s *UsageDetailUpsert) {
  812. s.UpdateStatus()
  813. })
  814. }
  815. // ClearStatus clears the value of the "status" field.
  816. func (u *UsageDetailUpsertOne) ClearStatus() *UsageDetailUpsertOne {
  817. return u.Update(func(s *UsageDetailUpsert) {
  818. s.ClearStatus()
  819. })
  820. }
  821. // SetType sets the "type" field.
  822. func (u *UsageDetailUpsertOne) SetType(v int) *UsageDetailUpsertOne {
  823. return u.Update(func(s *UsageDetailUpsert) {
  824. s.SetType(v)
  825. })
  826. }
  827. // AddType adds v to the "type" field.
  828. func (u *UsageDetailUpsertOne) AddType(v int) *UsageDetailUpsertOne {
  829. return u.Update(func(s *UsageDetailUpsert) {
  830. s.AddType(v)
  831. })
  832. }
  833. // UpdateType sets the "type" field to the value that was provided on create.
  834. func (u *UsageDetailUpsertOne) UpdateType() *UsageDetailUpsertOne {
  835. return u.Update(func(s *UsageDetailUpsert) {
  836. s.UpdateType()
  837. })
  838. }
  839. // ClearType clears the value of the "type" field.
  840. func (u *UsageDetailUpsertOne) ClearType() *UsageDetailUpsertOne {
  841. return u.Update(func(s *UsageDetailUpsert) {
  842. s.ClearType()
  843. })
  844. }
  845. // SetBotID sets the "bot_id" field.
  846. func (u *UsageDetailUpsertOne) SetBotID(v string) *UsageDetailUpsertOne {
  847. return u.Update(func(s *UsageDetailUpsert) {
  848. s.SetBotID(v)
  849. })
  850. }
  851. // UpdateBotID sets the "bot_id" field to the value that was provided on create.
  852. func (u *UsageDetailUpsertOne) UpdateBotID() *UsageDetailUpsertOne {
  853. return u.Update(func(s *UsageDetailUpsert) {
  854. s.UpdateBotID()
  855. })
  856. }
  857. // SetReceiverID sets the "receiver_id" field.
  858. func (u *UsageDetailUpsertOne) SetReceiverID(v string) *UsageDetailUpsertOne {
  859. return u.Update(func(s *UsageDetailUpsert) {
  860. s.SetReceiverID(v)
  861. })
  862. }
  863. // UpdateReceiverID sets the "receiver_id" field to the value that was provided on create.
  864. func (u *UsageDetailUpsertOne) UpdateReceiverID() *UsageDetailUpsertOne {
  865. return u.Update(func(s *UsageDetailUpsert) {
  866. s.UpdateReceiverID()
  867. })
  868. }
  869. // SetApp sets the "app" field.
  870. func (u *UsageDetailUpsertOne) SetApp(v int) *UsageDetailUpsertOne {
  871. return u.Update(func(s *UsageDetailUpsert) {
  872. s.SetApp(v)
  873. })
  874. }
  875. // AddApp adds v to the "app" field.
  876. func (u *UsageDetailUpsertOne) AddApp(v int) *UsageDetailUpsertOne {
  877. return u.Update(func(s *UsageDetailUpsert) {
  878. s.AddApp(v)
  879. })
  880. }
  881. // UpdateApp sets the "app" field to the value that was provided on create.
  882. func (u *UsageDetailUpsertOne) UpdateApp() *UsageDetailUpsertOne {
  883. return u.Update(func(s *UsageDetailUpsert) {
  884. s.UpdateApp()
  885. })
  886. }
  887. // ClearApp clears the value of the "app" field.
  888. func (u *UsageDetailUpsertOne) ClearApp() *UsageDetailUpsertOne {
  889. return u.Update(func(s *UsageDetailUpsert) {
  890. s.ClearApp()
  891. })
  892. }
  893. // SetSessionID sets the "session_id" field.
  894. func (u *UsageDetailUpsertOne) SetSessionID(v uint64) *UsageDetailUpsertOne {
  895. return u.Update(func(s *UsageDetailUpsert) {
  896. s.SetSessionID(v)
  897. })
  898. }
  899. // AddSessionID adds v to the "session_id" field.
  900. func (u *UsageDetailUpsertOne) AddSessionID(v uint64) *UsageDetailUpsertOne {
  901. return u.Update(func(s *UsageDetailUpsert) {
  902. s.AddSessionID(v)
  903. })
  904. }
  905. // UpdateSessionID sets the "session_id" field to the value that was provided on create.
  906. func (u *UsageDetailUpsertOne) UpdateSessionID() *UsageDetailUpsertOne {
  907. return u.Update(func(s *UsageDetailUpsert) {
  908. s.UpdateSessionID()
  909. })
  910. }
  911. // ClearSessionID clears the value of the "session_id" field.
  912. func (u *UsageDetailUpsertOne) ClearSessionID() *UsageDetailUpsertOne {
  913. return u.Update(func(s *UsageDetailUpsert) {
  914. s.ClearSessionID()
  915. })
  916. }
  917. // SetRequest sets the "request" field.
  918. func (u *UsageDetailUpsertOne) SetRequest(v string) *UsageDetailUpsertOne {
  919. return u.Update(func(s *UsageDetailUpsert) {
  920. s.SetRequest(v)
  921. })
  922. }
  923. // UpdateRequest sets the "request" field to the value that was provided on create.
  924. func (u *UsageDetailUpsertOne) UpdateRequest() *UsageDetailUpsertOne {
  925. return u.Update(func(s *UsageDetailUpsert) {
  926. s.UpdateRequest()
  927. })
  928. }
  929. // SetResponse sets the "response" field.
  930. func (u *UsageDetailUpsertOne) SetResponse(v string) *UsageDetailUpsertOne {
  931. return u.Update(func(s *UsageDetailUpsert) {
  932. s.SetResponse(v)
  933. })
  934. }
  935. // UpdateResponse sets the "response" field to the value that was provided on create.
  936. func (u *UsageDetailUpsertOne) UpdateResponse() *UsageDetailUpsertOne {
  937. return u.Update(func(s *UsageDetailUpsert) {
  938. s.UpdateResponse()
  939. })
  940. }
  941. // SetOriginalData sets the "original_data" field.
  942. func (u *UsageDetailUpsertOne) SetOriginalData(v custom_types.OriginalData) *UsageDetailUpsertOne {
  943. return u.Update(func(s *UsageDetailUpsert) {
  944. s.SetOriginalData(v)
  945. })
  946. }
  947. // UpdateOriginalData sets the "original_data" field to the value that was provided on create.
  948. func (u *UsageDetailUpsertOne) UpdateOriginalData() *UsageDetailUpsertOne {
  949. return u.Update(func(s *UsageDetailUpsert) {
  950. s.UpdateOriginalData()
  951. })
  952. }
  953. // SetTotalTokens sets the "total_tokens" field.
  954. func (u *UsageDetailUpsertOne) SetTotalTokens(v uint64) *UsageDetailUpsertOne {
  955. return u.Update(func(s *UsageDetailUpsert) {
  956. s.SetTotalTokens(v)
  957. })
  958. }
  959. // AddTotalTokens adds v to the "total_tokens" field.
  960. func (u *UsageDetailUpsertOne) AddTotalTokens(v uint64) *UsageDetailUpsertOne {
  961. return u.Update(func(s *UsageDetailUpsert) {
  962. s.AddTotalTokens(v)
  963. })
  964. }
  965. // UpdateTotalTokens sets the "total_tokens" field to the value that was provided on create.
  966. func (u *UsageDetailUpsertOne) UpdateTotalTokens() *UsageDetailUpsertOne {
  967. return u.Update(func(s *UsageDetailUpsert) {
  968. s.UpdateTotalTokens()
  969. })
  970. }
  971. // ClearTotalTokens clears the value of the "total_tokens" field.
  972. func (u *UsageDetailUpsertOne) ClearTotalTokens() *UsageDetailUpsertOne {
  973. return u.Update(func(s *UsageDetailUpsert) {
  974. s.ClearTotalTokens()
  975. })
  976. }
  977. // SetPromptTokens sets the "prompt_tokens" field.
  978. func (u *UsageDetailUpsertOne) SetPromptTokens(v uint64) *UsageDetailUpsertOne {
  979. return u.Update(func(s *UsageDetailUpsert) {
  980. s.SetPromptTokens(v)
  981. })
  982. }
  983. // AddPromptTokens adds v to the "prompt_tokens" field.
  984. func (u *UsageDetailUpsertOne) AddPromptTokens(v uint64) *UsageDetailUpsertOne {
  985. return u.Update(func(s *UsageDetailUpsert) {
  986. s.AddPromptTokens(v)
  987. })
  988. }
  989. // UpdatePromptTokens sets the "prompt_tokens" field to the value that was provided on create.
  990. func (u *UsageDetailUpsertOne) UpdatePromptTokens() *UsageDetailUpsertOne {
  991. return u.Update(func(s *UsageDetailUpsert) {
  992. s.UpdatePromptTokens()
  993. })
  994. }
  995. // ClearPromptTokens clears the value of the "prompt_tokens" field.
  996. func (u *UsageDetailUpsertOne) ClearPromptTokens() *UsageDetailUpsertOne {
  997. return u.Update(func(s *UsageDetailUpsert) {
  998. s.ClearPromptTokens()
  999. })
  1000. }
  1001. // SetCompletionTokens sets the "completion_tokens" field.
  1002. func (u *UsageDetailUpsertOne) SetCompletionTokens(v uint64) *UsageDetailUpsertOne {
  1003. return u.Update(func(s *UsageDetailUpsert) {
  1004. s.SetCompletionTokens(v)
  1005. })
  1006. }
  1007. // AddCompletionTokens adds v to the "completion_tokens" field.
  1008. func (u *UsageDetailUpsertOne) AddCompletionTokens(v uint64) *UsageDetailUpsertOne {
  1009. return u.Update(func(s *UsageDetailUpsert) {
  1010. s.AddCompletionTokens(v)
  1011. })
  1012. }
  1013. // UpdateCompletionTokens sets the "completion_tokens" field to the value that was provided on create.
  1014. func (u *UsageDetailUpsertOne) UpdateCompletionTokens() *UsageDetailUpsertOne {
  1015. return u.Update(func(s *UsageDetailUpsert) {
  1016. s.UpdateCompletionTokens()
  1017. })
  1018. }
  1019. // ClearCompletionTokens clears the value of the "completion_tokens" field.
  1020. func (u *UsageDetailUpsertOne) ClearCompletionTokens() *UsageDetailUpsertOne {
  1021. return u.Update(func(s *UsageDetailUpsert) {
  1022. s.ClearCompletionTokens()
  1023. })
  1024. }
  1025. // SetOrganizationID sets the "organization_id" field.
  1026. func (u *UsageDetailUpsertOne) SetOrganizationID(v uint64) *UsageDetailUpsertOne {
  1027. return u.Update(func(s *UsageDetailUpsert) {
  1028. s.SetOrganizationID(v)
  1029. })
  1030. }
  1031. // AddOrganizationID adds v to the "organization_id" field.
  1032. func (u *UsageDetailUpsertOne) AddOrganizationID(v uint64) *UsageDetailUpsertOne {
  1033. return u.Update(func(s *UsageDetailUpsert) {
  1034. s.AddOrganizationID(v)
  1035. })
  1036. }
  1037. // UpdateOrganizationID sets the "organization_id" field to the value that was provided on create.
  1038. func (u *UsageDetailUpsertOne) UpdateOrganizationID() *UsageDetailUpsertOne {
  1039. return u.Update(func(s *UsageDetailUpsert) {
  1040. s.UpdateOrganizationID()
  1041. })
  1042. }
  1043. // ClearOrganizationID clears the value of the "organization_id" field.
  1044. func (u *UsageDetailUpsertOne) ClearOrganizationID() *UsageDetailUpsertOne {
  1045. return u.Update(func(s *UsageDetailUpsert) {
  1046. s.ClearOrganizationID()
  1047. })
  1048. }
  1049. // SetModel sets the "model" field.
  1050. func (u *UsageDetailUpsertOne) SetModel(v string) *UsageDetailUpsertOne {
  1051. return u.Update(func(s *UsageDetailUpsert) {
  1052. s.SetModel(v)
  1053. })
  1054. }
  1055. // UpdateModel sets the "model" field to the value that was provided on create.
  1056. func (u *UsageDetailUpsertOne) UpdateModel() *UsageDetailUpsertOne {
  1057. return u.Update(func(s *UsageDetailUpsert) {
  1058. s.UpdateModel()
  1059. })
  1060. }
  1061. // ClearModel clears the value of the "model" field.
  1062. func (u *UsageDetailUpsertOne) ClearModel() *UsageDetailUpsertOne {
  1063. return u.Update(func(s *UsageDetailUpsert) {
  1064. s.ClearModel()
  1065. })
  1066. }
  1067. // SetCredits sets the "credits" field.
  1068. func (u *UsageDetailUpsertOne) SetCredits(v float64) *UsageDetailUpsertOne {
  1069. return u.Update(func(s *UsageDetailUpsert) {
  1070. s.SetCredits(v)
  1071. })
  1072. }
  1073. // AddCredits adds v to the "credits" field.
  1074. func (u *UsageDetailUpsertOne) AddCredits(v float64) *UsageDetailUpsertOne {
  1075. return u.Update(func(s *UsageDetailUpsert) {
  1076. s.AddCredits(v)
  1077. })
  1078. }
  1079. // UpdateCredits sets the "credits" field to the value that was provided on create.
  1080. func (u *UsageDetailUpsertOne) UpdateCredits() *UsageDetailUpsertOne {
  1081. return u.Update(func(s *UsageDetailUpsert) {
  1082. s.UpdateCredits()
  1083. })
  1084. }
  1085. // ClearCredits clears the value of the "credits" field.
  1086. func (u *UsageDetailUpsertOne) ClearCredits() *UsageDetailUpsertOne {
  1087. return u.Update(func(s *UsageDetailUpsert) {
  1088. s.ClearCredits()
  1089. })
  1090. }
  1091. // Exec executes the query.
  1092. func (u *UsageDetailUpsertOne) Exec(ctx context.Context) error {
  1093. if len(u.create.conflict) == 0 {
  1094. return errors.New("ent: missing options for UsageDetailCreate.OnConflict")
  1095. }
  1096. return u.create.Exec(ctx)
  1097. }
  1098. // ExecX is like Exec, but panics if an error occurs.
  1099. func (u *UsageDetailUpsertOne) ExecX(ctx context.Context) {
  1100. if err := u.create.Exec(ctx); err != nil {
  1101. panic(err)
  1102. }
  1103. }
  1104. // Exec executes the UPSERT query and returns the inserted/updated ID.
  1105. func (u *UsageDetailUpsertOne) ID(ctx context.Context) (id uint64, err error) {
  1106. node, err := u.create.Save(ctx)
  1107. if err != nil {
  1108. return id, err
  1109. }
  1110. return node.ID, nil
  1111. }
  1112. // IDX is like ID, but panics if an error occurs.
  1113. func (u *UsageDetailUpsertOne) IDX(ctx context.Context) uint64 {
  1114. id, err := u.ID(ctx)
  1115. if err != nil {
  1116. panic(err)
  1117. }
  1118. return id
  1119. }
  1120. // UsageDetailCreateBulk is the builder for creating many UsageDetail entities in bulk.
  1121. type UsageDetailCreateBulk struct {
  1122. config
  1123. err error
  1124. builders []*UsageDetailCreate
  1125. conflict []sql.ConflictOption
  1126. }
  1127. // Save creates the UsageDetail entities in the database.
  1128. func (udcb *UsageDetailCreateBulk) Save(ctx context.Context) ([]*UsageDetail, error) {
  1129. if udcb.err != nil {
  1130. return nil, udcb.err
  1131. }
  1132. specs := make([]*sqlgraph.CreateSpec, len(udcb.builders))
  1133. nodes := make([]*UsageDetail, len(udcb.builders))
  1134. mutators := make([]Mutator, len(udcb.builders))
  1135. for i := range udcb.builders {
  1136. func(i int, root context.Context) {
  1137. builder := udcb.builders[i]
  1138. builder.defaults()
  1139. var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) {
  1140. mutation, ok := m.(*UsageDetailMutation)
  1141. if !ok {
  1142. return nil, fmt.Errorf("unexpected mutation type %T", m)
  1143. }
  1144. if err := builder.check(); err != nil {
  1145. return nil, err
  1146. }
  1147. builder.mutation = mutation
  1148. var err error
  1149. nodes[i], specs[i] = builder.createSpec()
  1150. if i < len(mutators)-1 {
  1151. _, err = mutators[i+1].Mutate(root, udcb.builders[i+1].mutation)
  1152. } else {
  1153. spec := &sqlgraph.BatchCreateSpec{Nodes: specs}
  1154. spec.OnConflict = udcb.conflict
  1155. // Invoke the actual operation on the latest mutation in the chain.
  1156. if err = sqlgraph.BatchCreate(ctx, udcb.driver, spec); err != nil {
  1157. if sqlgraph.IsConstraintError(err) {
  1158. err = &ConstraintError{msg: err.Error(), wrap: err}
  1159. }
  1160. }
  1161. }
  1162. if err != nil {
  1163. return nil, err
  1164. }
  1165. mutation.id = &nodes[i].ID
  1166. if specs[i].ID.Value != nil && nodes[i].ID == 0 {
  1167. id := specs[i].ID.Value.(int64)
  1168. nodes[i].ID = uint64(id)
  1169. }
  1170. mutation.done = true
  1171. return nodes[i], nil
  1172. })
  1173. for i := len(builder.hooks) - 1; i >= 0; i-- {
  1174. mut = builder.hooks[i](mut)
  1175. }
  1176. mutators[i] = mut
  1177. }(i, ctx)
  1178. }
  1179. if len(mutators) > 0 {
  1180. if _, err := mutators[0].Mutate(ctx, udcb.builders[0].mutation); err != nil {
  1181. return nil, err
  1182. }
  1183. }
  1184. return nodes, nil
  1185. }
  1186. // SaveX is like Save, but panics if an error occurs.
  1187. func (udcb *UsageDetailCreateBulk) SaveX(ctx context.Context) []*UsageDetail {
  1188. v, err := udcb.Save(ctx)
  1189. if err != nil {
  1190. panic(err)
  1191. }
  1192. return v
  1193. }
  1194. // Exec executes the query.
  1195. func (udcb *UsageDetailCreateBulk) Exec(ctx context.Context) error {
  1196. _, err := udcb.Save(ctx)
  1197. return err
  1198. }
  1199. // ExecX is like Exec, but panics if an error occurs.
  1200. func (udcb *UsageDetailCreateBulk) ExecX(ctx context.Context) {
  1201. if err := udcb.Exec(ctx); err != nil {
  1202. panic(err)
  1203. }
  1204. }
  1205. // OnConflict allows configuring the `ON CONFLICT` / `ON DUPLICATE KEY` clause
  1206. // of the `INSERT` statement. For example:
  1207. //
  1208. // client.UsageDetail.CreateBulk(builders...).
  1209. // OnConflict(
  1210. // // Update the row with the new values
  1211. // // the was proposed for insertion.
  1212. // sql.ResolveWithNewValues(),
  1213. // ).
  1214. // // Override some of the fields with custom
  1215. // // update values.
  1216. // Update(func(u *ent.UsageDetailUpsert) {
  1217. // SetCreatedAt(v+v).
  1218. // }).
  1219. // Exec(ctx)
  1220. func (udcb *UsageDetailCreateBulk) OnConflict(opts ...sql.ConflictOption) *UsageDetailUpsertBulk {
  1221. udcb.conflict = opts
  1222. return &UsageDetailUpsertBulk{
  1223. create: udcb,
  1224. }
  1225. }
  1226. // OnConflictColumns calls `OnConflict` and configures the columns
  1227. // as conflict target. Using this option is equivalent to using:
  1228. //
  1229. // client.UsageDetail.Create().
  1230. // OnConflict(sql.ConflictColumns(columns...)).
  1231. // Exec(ctx)
  1232. func (udcb *UsageDetailCreateBulk) OnConflictColumns(columns ...string) *UsageDetailUpsertBulk {
  1233. udcb.conflict = append(udcb.conflict, sql.ConflictColumns(columns...))
  1234. return &UsageDetailUpsertBulk{
  1235. create: udcb,
  1236. }
  1237. }
  1238. // UsageDetailUpsertBulk is the builder for "upsert"-ing
  1239. // a bulk of UsageDetail nodes.
  1240. type UsageDetailUpsertBulk struct {
  1241. create *UsageDetailCreateBulk
  1242. }
  1243. // UpdateNewValues updates the mutable fields using the new values that
  1244. // were set on create. Using this option is equivalent to using:
  1245. //
  1246. // client.UsageDetail.Create().
  1247. // OnConflict(
  1248. // sql.ResolveWithNewValues(),
  1249. // sql.ResolveWith(func(u *sql.UpdateSet) {
  1250. // u.SetIgnore(usagedetail.FieldID)
  1251. // }),
  1252. // ).
  1253. // Exec(ctx)
  1254. func (u *UsageDetailUpsertBulk) UpdateNewValues() *UsageDetailUpsertBulk {
  1255. u.create.conflict = append(u.create.conflict, sql.ResolveWithNewValues())
  1256. u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(s *sql.UpdateSet) {
  1257. for _, b := range u.create.builders {
  1258. if _, exists := b.mutation.ID(); exists {
  1259. s.SetIgnore(usagedetail.FieldID)
  1260. }
  1261. if _, exists := b.mutation.CreatedAt(); exists {
  1262. s.SetIgnore(usagedetail.FieldCreatedAt)
  1263. }
  1264. }
  1265. }))
  1266. return u
  1267. }
  1268. // Ignore sets each column to itself in case of conflict.
  1269. // Using this option is equivalent to using:
  1270. //
  1271. // client.UsageDetail.Create().
  1272. // OnConflict(sql.ResolveWithIgnore()).
  1273. // Exec(ctx)
  1274. func (u *UsageDetailUpsertBulk) Ignore() *UsageDetailUpsertBulk {
  1275. u.create.conflict = append(u.create.conflict, sql.ResolveWithIgnore())
  1276. return u
  1277. }
  1278. // DoNothing configures the conflict_action to `DO NOTHING`.
  1279. // Supported only by SQLite and PostgreSQL.
  1280. func (u *UsageDetailUpsertBulk) DoNothing() *UsageDetailUpsertBulk {
  1281. u.create.conflict = append(u.create.conflict, sql.DoNothing())
  1282. return u
  1283. }
  1284. // Update allows overriding fields `UPDATE` values. See the UsageDetailCreateBulk.OnConflict
  1285. // documentation for more info.
  1286. func (u *UsageDetailUpsertBulk) Update(set func(*UsageDetailUpsert)) *UsageDetailUpsertBulk {
  1287. u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(update *sql.UpdateSet) {
  1288. set(&UsageDetailUpsert{UpdateSet: update})
  1289. }))
  1290. return u
  1291. }
  1292. // SetUpdatedAt sets the "updated_at" field.
  1293. func (u *UsageDetailUpsertBulk) SetUpdatedAt(v time.Time) *UsageDetailUpsertBulk {
  1294. return u.Update(func(s *UsageDetailUpsert) {
  1295. s.SetUpdatedAt(v)
  1296. })
  1297. }
  1298. // UpdateUpdatedAt sets the "updated_at" field to the value that was provided on create.
  1299. func (u *UsageDetailUpsertBulk) UpdateUpdatedAt() *UsageDetailUpsertBulk {
  1300. return u.Update(func(s *UsageDetailUpsert) {
  1301. s.UpdateUpdatedAt()
  1302. })
  1303. }
  1304. // SetStatus sets the "status" field.
  1305. func (u *UsageDetailUpsertBulk) SetStatus(v uint8) *UsageDetailUpsertBulk {
  1306. return u.Update(func(s *UsageDetailUpsert) {
  1307. s.SetStatus(v)
  1308. })
  1309. }
  1310. // AddStatus adds v to the "status" field.
  1311. func (u *UsageDetailUpsertBulk) AddStatus(v uint8) *UsageDetailUpsertBulk {
  1312. return u.Update(func(s *UsageDetailUpsert) {
  1313. s.AddStatus(v)
  1314. })
  1315. }
  1316. // UpdateStatus sets the "status" field to the value that was provided on create.
  1317. func (u *UsageDetailUpsertBulk) UpdateStatus() *UsageDetailUpsertBulk {
  1318. return u.Update(func(s *UsageDetailUpsert) {
  1319. s.UpdateStatus()
  1320. })
  1321. }
  1322. // ClearStatus clears the value of the "status" field.
  1323. func (u *UsageDetailUpsertBulk) ClearStatus() *UsageDetailUpsertBulk {
  1324. return u.Update(func(s *UsageDetailUpsert) {
  1325. s.ClearStatus()
  1326. })
  1327. }
  1328. // SetType sets the "type" field.
  1329. func (u *UsageDetailUpsertBulk) SetType(v int) *UsageDetailUpsertBulk {
  1330. return u.Update(func(s *UsageDetailUpsert) {
  1331. s.SetType(v)
  1332. })
  1333. }
  1334. // AddType adds v to the "type" field.
  1335. func (u *UsageDetailUpsertBulk) AddType(v int) *UsageDetailUpsertBulk {
  1336. return u.Update(func(s *UsageDetailUpsert) {
  1337. s.AddType(v)
  1338. })
  1339. }
  1340. // UpdateType sets the "type" field to the value that was provided on create.
  1341. func (u *UsageDetailUpsertBulk) UpdateType() *UsageDetailUpsertBulk {
  1342. return u.Update(func(s *UsageDetailUpsert) {
  1343. s.UpdateType()
  1344. })
  1345. }
  1346. // ClearType clears the value of the "type" field.
  1347. func (u *UsageDetailUpsertBulk) ClearType() *UsageDetailUpsertBulk {
  1348. return u.Update(func(s *UsageDetailUpsert) {
  1349. s.ClearType()
  1350. })
  1351. }
  1352. // SetBotID sets the "bot_id" field.
  1353. func (u *UsageDetailUpsertBulk) SetBotID(v string) *UsageDetailUpsertBulk {
  1354. return u.Update(func(s *UsageDetailUpsert) {
  1355. s.SetBotID(v)
  1356. })
  1357. }
  1358. // UpdateBotID sets the "bot_id" field to the value that was provided on create.
  1359. func (u *UsageDetailUpsertBulk) UpdateBotID() *UsageDetailUpsertBulk {
  1360. return u.Update(func(s *UsageDetailUpsert) {
  1361. s.UpdateBotID()
  1362. })
  1363. }
  1364. // SetReceiverID sets the "receiver_id" field.
  1365. func (u *UsageDetailUpsertBulk) SetReceiverID(v string) *UsageDetailUpsertBulk {
  1366. return u.Update(func(s *UsageDetailUpsert) {
  1367. s.SetReceiverID(v)
  1368. })
  1369. }
  1370. // UpdateReceiverID sets the "receiver_id" field to the value that was provided on create.
  1371. func (u *UsageDetailUpsertBulk) UpdateReceiverID() *UsageDetailUpsertBulk {
  1372. return u.Update(func(s *UsageDetailUpsert) {
  1373. s.UpdateReceiverID()
  1374. })
  1375. }
  1376. // SetApp sets the "app" field.
  1377. func (u *UsageDetailUpsertBulk) SetApp(v int) *UsageDetailUpsertBulk {
  1378. return u.Update(func(s *UsageDetailUpsert) {
  1379. s.SetApp(v)
  1380. })
  1381. }
  1382. // AddApp adds v to the "app" field.
  1383. func (u *UsageDetailUpsertBulk) AddApp(v int) *UsageDetailUpsertBulk {
  1384. return u.Update(func(s *UsageDetailUpsert) {
  1385. s.AddApp(v)
  1386. })
  1387. }
  1388. // UpdateApp sets the "app" field to the value that was provided on create.
  1389. func (u *UsageDetailUpsertBulk) UpdateApp() *UsageDetailUpsertBulk {
  1390. return u.Update(func(s *UsageDetailUpsert) {
  1391. s.UpdateApp()
  1392. })
  1393. }
  1394. // ClearApp clears the value of the "app" field.
  1395. func (u *UsageDetailUpsertBulk) ClearApp() *UsageDetailUpsertBulk {
  1396. return u.Update(func(s *UsageDetailUpsert) {
  1397. s.ClearApp()
  1398. })
  1399. }
  1400. // SetSessionID sets the "session_id" field.
  1401. func (u *UsageDetailUpsertBulk) SetSessionID(v uint64) *UsageDetailUpsertBulk {
  1402. return u.Update(func(s *UsageDetailUpsert) {
  1403. s.SetSessionID(v)
  1404. })
  1405. }
  1406. // AddSessionID adds v to the "session_id" field.
  1407. func (u *UsageDetailUpsertBulk) AddSessionID(v uint64) *UsageDetailUpsertBulk {
  1408. return u.Update(func(s *UsageDetailUpsert) {
  1409. s.AddSessionID(v)
  1410. })
  1411. }
  1412. // UpdateSessionID sets the "session_id" field to the value that was provided on create.
  1413. func (u *UsageDetailUpsertBulk) UpdateSessionID() *UsageDetailUpsertBulk {
  1414. return u.Update(func(s *UsageDetailUpsert) {
  1415. s.UpdateSessionID()
  1416. })
  1417. }
  1418. // ClearSessionID clears the value of the "session_id" field.
  1419. func (u *UsageDetailUpsertBulk) ClearSessionID() *UsageDetailUpsertBulk {
  1420. return u.Update(func(s *UsageDetailUpsert) {
  1421. s.ClearSessionID()
  1422. })
  1423. }
  1424. // SetRequest sets the "request" field.
  1425. func (u *UsageDetailUpsertBulk) SetRequest(v string) *UsageDetailUpsertBulk {
  1426. return u.Update(func(s *UsageDetailUpsert) {
  1427. s.SetRequest(v)
  1428. })
  1429. }
  1430. // UpdateRequest sets the "request" field to the value that was provided on create.
  1431. func (u *UsageDetailUpsertBulk) UpdateRequest() *UsageDetailUpsertBulk {
  1432. return u.Update(func(s *UsageDetailUpsert) {
  1433. s.UpdateRequest()
  1434. })
  1435. }
  1436. // SetResponse sets the "response" field.
  1437. func (u *UsageDetailUpsertBulk) SetResponse(v string) *UsageDetailUpsertBulk {
  1438. return u.Update(func(s *UsageDetailUpsert) {
  1439. s.SetResponse(v)
  1440. })
  1441. }
  1442. // UpdateResponse sets the "response" field to the value that was provided on create.
  1443. func (u *UsageDetailUpsertBulk) UpdateResponse() *UsageDetailUpsertBulk {
  1444. return u.Update(func(s *UsageDetailUpsert) {
  1445. s.UpdateResponse()
  1446. })
  1447. }
  1448. // SetOriginalData sets the "original_data" field.
  1449. func (u *UsageDetailUpsertBulk) SetOriginalData(v custom_types.OriginalData) *UsageDetailUpsertBulk {
  1450. return u.Update(func(s *UsageDetailUpsert) {
  1451. s.SetOriginalData(v)
  1452. })
  1453. }
  1454. // UpdateOriginalData sets the "original_data" field to the value that was provided on create.
  1455. func (u *UsageDetailUpsertBulk) UpdateOriginalData() *UsageDetailUpsertBulk {
  1456. return u.Update(func(s *UsageDetailUpsert) {
  1457. s.UpdateOriginalData()
  1458. })
  1459. }
  1460. // SetTotalTokens sets the "total_tokens" field.
  1461. func (u *UsageDetailUpsertBulk) SetTotalTokens(v uint64) *UsageDetailUpsertBulk {
  1462. return u.Update(func(s *UsageDetailUpsert) {
  1463. s.SetTotalTokens(v)
  1464. })
  1465. }
  1466. // AddTotalTokens adds v to the "total_tokens" field.
  1467. func (u *UsageDetailUpsertBulk) AddTotalTokens(v uint64) *UsageDetailUpsertBulk {
  1468. return u.Update(func(s *UsageDetailUpsert) {
  1469. s.AddTotalTokens(v)
  1470. })
  1471. }
  1472. // UpdateTotalTokens sets the "total_tokens" field to the value that was provided on create.
  1473. func (u *UsageDetailUpsertBulk) UpdateTotalTokens() *UsageDetailUpsertBulk {
  1474. return u.Update(func(s *UsageDetailUpsert) {
  1475. s.UpdateTotalTokens()
  1476. })
  1477. }
  1478. // ClearTotalTokens clears the value of the "total_tokens" field.
  1479. func (u *UsageDetailUpsertBulk) ClearTotalTokens() *UsageDetailUpsertBulk {
  1480. return u.Update(func(s *UsageDetailUpsert) {
  1481. s.ClearTotalTokens()
  1482. })
  1483. }
  1484. // SetPromptTokens sets the "prompt_tokens" field.
  1485. func (u *UsageDetailUpsertBulk) SetPromptTokens(v uint64) *UsageDetailUpsertBulk {
  1486. return u.Update(func(s *UsageDetailUpsert) {
  1487. s.SetPromptTokens(v)
  1488. })
  1489. }
  1490. // AddPromptTokens adds v to the "prompt_tokens" field.
  1491. func (u *UsageDetailUpsertBulk) AddPromptTokens(v uint64) *UsageDetailUpsertBulk {
  1492. return u.Update(func(s *UsageDetailUpsert) {
  1493. s.AddPromptTokens(v)
  1494. })
  1495. }
  1496. // UpdatePromptTokens sets the "prompt_tokens" field to the value that was provided on create.
  1497. func (u *UsageDetailUpsertBulk) UpdatePromptTokens() *UsageDetailUpsertBulk {
  1498. return u.Update(func(s *UsageDetailUpsert) {
  1499. s.UpdatePromptTokens()
  1500. })
  1501. }
  1502. // ClearPromptTokens clears the value of the "prompt_tokens" field.
  1503. func (u *UsageDetailUpsertBulk) ClearPromptTokens() *UsageDetailUpsertBulk {
  1504. return u.Update(func(s *UsageDetailUpsert) {
  1505. s.ClearPromptTokens()
  1506. })
  1507. }
  1508. // SetCompletionTokens sets the "completion_tokens" field.
  1509. func (u *UsageDetailUpsertBulk) SetCompletionTokens(v uint64) *UsageDetailUpsertBulk {
  1510. return u.Update(func(s *UsageDetailUpsert) {
  1511. s.SetCompletionTokens(v)
  1512. })
  1513. }
  1514. // AddCompletionTokens adds v to the "completion_tokens" field.
  1515. func (u *UsageDetailUpsertBulk) AddCompletionTokens(v uint64) *UsageDetailUpsertBulk {
  1516. return u.Update(func(s *UsageDetailUpsert) {
  1517. s.AddCompletionTokens(v)
  1518. })
  1519. }
  1520. // UpdateCompletionTokens sets the "completion_tokens" field to the value that was provided on create.
  1521. func (u *UsageDetailUpsertBulk) UpdateCompletionTokens() *UsageDetailUpsertBulk {
  1522. return u.Update(func(s *UsageDetailUpsert) {
  1523. s.UpdateCompletionTokens()
  1524. })
  1525. }
  1526. // ClearCompletionTokens clears the value of the "completion_tokens" field.
  1527. func (u *UsageDetailUpsertBulk) ClearCompletionTokens() *UsageDetailUpsertBulk {
  1528. return u.Update(func(s *UsageDetailUpsert) {
  1529. s.ClearCompletionTokens()
  1530. })
  1531. }
  1532. // SetOrganizationID sets the "organization_id" field.
  1533. func (u *UsageDetailUpsertBulk) SetOrganizationID(v uint64) *UsageDetailUpsertBulk {
  1534. return u.Update(func(s *UsageDetailUpsert) {
  1535. s.SetOrganizationID(v)
  1536. })
  1537. }
  1538. // AddOrganizationID adds v to the "organization_id" field.
  1539. func (u *UsageDetailUpsertBulk) AddOrganizationID(v uint64) *UsageDetailUpsertBulk {
  1540. return u.Update(func(s *UsageDetailUpsert) {
  1541. s.AddOrganizationID(v)
  1542. })
  1543. }
  1544. // UpdateOrganizationID sets the "organization_id" field to the value that was provided on create.
  1545. func (u *UsageDetailUpsertBulk) UpdateOrganizationID() *UsageDetailUpsertBulk {
  1546. return u.Update(func(s *UsageDetailUpsert) {
  1547. s.UpdateOrganizationID()
  1548. })
  1549. }
  1550. // ClearOrganizationID clears the value of the "organization_id" field.
  1551. func (u *UsageDetailUpsertBulk) ClearOrganizationID() *UsageDetailUpsertBulk {
  1552. return u.Update(func(s *UsageDetailUpsert) {
  1553. s.ClearOrganizationID()
  1554. })
  1555. }
  1556. // SetModel sets the "model" field.
  1557. func (u *UsageDetailUpsertBulk) SetModel(v string) *UsageDetailUpsertBulk {
  1558. return u.Update(func(s *UsageDetailUpsert) {
  1559. s.SetModel(v)
  1560. })
  1561. }
  1562. // UpdateModel sets the "model" field to the value that was provided on create.
  1563. func (u *UsageDetailUpsertBulk) UpdateModel() *UsageDetailUpsertBulk {
  1564. return u.Update(func(s *UsageDetailUpsert) {
  1565. s.UpdateModel()
  1566. })
  1567. }
  1568. // ClearModel clears the value of the "model" field.
  1569. func (u *UsageDetailUpsertBulk) ClearModel() *UsageDetailUpsertBulk {
  1570. return u.Update(func(s *UsageDetailUpsert) {
  1571. s.ClearModel()
  1572. })
  1573. }
  1574. // SetCredits sets the "credits" field.
  1575. func (u *UsageDetailUpsertBulk) SetCredits(v float64) *UsageDetailUpsertBulk {
  1576. return u.Update(func(s *UsageDetailUpsert) {
  1577. s.SetCredits(v)
  1578. })
  1579. }
  1580. // AddCredits adds v to the "credits" field.
  1581. func (u *UsageDetailUpsertBulk) AddCredits(v float64) *UsageDetailUpsertBulk {
  1582. return u.Update(func(s *UsageDetailUpsert) {
  1583. s.AddCredits(v)
  1584. })
  1585. }
  1586. // UpdateCredits sets the "credits" field to the value that was provided on create.
  1587. func (u *UsageDetailUpsertBulk) UpdateCredits() *UsageDetailUpsertBulk {
  1588. return u.Update(func(s *UsageDetailUpsert) {
  1589. s.UpdateCredits()
  1590. })
  1591. }
  1592. // ClearCredits clears the value of the "credits" field.
  1593. func (u *UsageDetailUpsertBulk) ClearCredits() *UsageDetailUpsertBulk {
  1594. return u.Update(func(s *UsageDetailUpsert) {
  1595. s.ClearCredits()
  1596. })
  1597. }
  1598. // Exec executes the query.
  1599. func (u *UsageDetailUpsertBulk) Exec(ctx context.Context) error {
  1600. if u.create.err != nil {
  1601. return u.create.err
  1602. }
  1603. for i, b := range u.create.builders {
  1604. if len(b.conflict) != 0 {
  1605. return fmt.Errorf("ent: OnConflict was set for builder %d. Set it on the UsageDetailCreateBulk instead", i)
  1606. }
  1607. }
  1608. if len(u.create.conflict) == 0 {
  1609. return errors.New("ent: missing options for UsageDetailCreateBulk.OnConflict")
  1610. }
  1611. return u.create.Exec(ctx)
  1612. }
  1613. // ExecX is like Exec, but panics if an error occurs.
  1614. func (u *UsageDetailUpsertBulk) ExecX(ctx context.Context) {
  1615. if err := u.create.Exec(ctx); err != nil {
  1616. panic(err)
  1617. }
  1618. }