|
@@ -43390,6 +43390,8 @@ type UsageTotalMutation struct {
|
|
|
bot_id *string
|
|
|
total_tokens *uint64
|
|
|
addtotal_tokens *int64
|
|
|
+ credits *float64
|
|
|
+ addcredits *float64
|
|
|
start_index *uint64
|
|
|
addstart_index *int64
|
|
|
end_index *uint64
|
|
@@ -43824,6 +43826,76 @@ func (m *UsageTotalMutation) ResetTotalTokens() {
|
|
|
delete(m.clearedFields, usagetotal.FieldTotalTokens)
|
|
|
}
|
|
|
|
|
|
+// SetCredits sets the "credits" field.
|
|
|
+func (m *UsageTotalMutation) SetCredits(f float64) {
|
|
|
+ m.credits = &f
|
|
|
+ m.addcredits = nil
|
|
|
+}
|
|
|
+
|
|
|
+// Credits returns the value of the "credits" field in the mutation.
|
|
|
+func (m *UsageTotalMutation) Credits() (r float64, exists bool) {
|
|
|
+ v := m.credits
|
|
|
+ if v == nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ return *v, true
|
|
|
+}
|
|
|
+
|
|
|
+// OldCredits returns the old "credits" field's value of the UsageTotal entity.
|
|
|
+// If the UsageTotal object wasn't provided to the builder, the object is fetched from the database.
|
|
|
+// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
|
|
|
+func (m *UsageTotalMutation) OldCredits(ctx context.Context) (v float64, err error) {
|
|
|
+ if !m.op.Is(OpUpdateOne) {
|
|
|
+ return v, errors.New("OldCredits is only allowed on UpdateOne operations")
|
|
|
+ }
|
|
|
+ if m.id == nil || m.oldValue == nil {
|
|
|
+ return v, errors.New("OldCredits requires an ID field in the mutation")
|
|
|
+ }
|
|
|
+ oldValue, err := m.oldValue(ctx)
|
|
|
+ if err != nil {
|
|
|
+ return v, fmt.Errorf("querying old value for OldCredits: %w", err)
|
|
|
+ }
|
|
|
+ return oldValue.Credits, nil
|
|
|
+}
|
|
|
+
|
|
|
+// AddCredits adds f to the "credits" field.
|
|
|
+func (m *UsageTotalMutation) AddCredits(f float64) {
|
|
|
+ if m.addcredits != nil {
|
|
|
+ *m.addcredits += f
|
|
|
+ } else {
|
|
|
+ m.addcredits = &f
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+// AddedCredits returns the value that was added to the "credits" field in this mutation.
|
|
|
+func (m *UsageTotalMutation) AddedCredits() (r float64, exists bool) {
|
|
|
+ v := m.addcredits
|
|
|
+ if v == nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ return *v, true
|
|
|
+}
|
|
|
+
|
|
|
+// ClearCredits clears the value of the "credits" field.
|
|
|
+func (m *UsageTotalMutation) ClearCredits() {
|
|
|
+ m.credits = nil
|
|
|
+ m.addcredits = nil
|
|
|
+ m.clearedFields[usagetotal.FieldCredits] = struct{}{}
|
|
|
+}
|
|
|
+
|
|
|
+// CreditsCleared returns if the "credits" field was cleared in this mutation.
|
|
|
+func (m *UsageTotalMutation) CreditsCleared() bool {
|
|
|
+ _, ok := m.clearedFields[usagetotal.FieldCredits]
|
|
|
+ return ok
|
|
|
+}
|
|
|
+
|
|
|
+// ResetCredits resets all changes to the "credits" field.
|
|
|
+func (m *UsageTotalMutation) ResetCredits() {
|
|
|
+ m.credits = nil
|
|
|
+ m.addcredits = nil
|
|
|
+ delete(m.clearedFields, usagetotal.FieldCredits)
|
|
|
+}
|
|
|
+
|
|
|
// SetStartIndex sets the "start_index" field.
|
|
|
func (m *UsageTotalMutation) SetStartIndex(u uint64) {
|
|
|
m.start_index = &u
|
|
@@ -44068,7 +44140,7 @@ func (m *UsageTotalMutation) Type() string {
|
|
|
// order to get all numeric fields that were incremented/decremented, call
|
|
|
// AddedFields().
|
|
|
func (m *UsageTotalMutation) Fields() []string {
|
|
|
- fields := make([]string, 0, 9)
|
|
|
+ fields := make([]string, 0, 10)
|
|
|
if m.created_at != nil {
|
|
|
fields = append(fields, usagetotal.FieldCreatedAt)
|
|
|
}
|
|
@@ -44087,6 +44159,9 @@ func (m *UsageTotalMutation) Fields() []string {
|
|
|
if m.total_tokens != nil {
|
|
|
fields = append(fields, usagetotal.FieldTotalTokens)
|
|
|
}
|
|
|
+ if m.credits != nil {
|
|
|
+ fields = append(fields, usagetotal.FieldCredits)
|
|
|
+ }
|
|
|
if m.start_index != nil {
|
|
|
fields = append(fields, usagetotal.FieldStartIndex)
|
|
|
}
|
|
@@ -44116,6 +44191,8 @@ func (m *UsageTotalMutation) Field(name string) (ent.Value, bool) {
|
|
|
return m.BotID()
|
|
|
case usagetotal.FieldTotalTokens:
|
|
|
return m.TotalTokens()
|
|
|
+ case usagetotal.FieldCredits:
|
|
|
+ return m.Credits()
|
|
|
case usagetotal.FieldStartIndex:
|
|
|
return m.StartIndex()
|
|
|
case usagetotal.FieldEndIndex:
|
|
@@ -44143,6 +44220,8 @@ func (m *UsageTotalMutation) OldField(ctx context.Context, name string) (ent.Val
|
|
|
return m.OldBotID(ctx)
|
|
|
case usagetotal.FieldTotalTokens:
|
|
|
return m.OldTotalTokens(ctx)
|
|
|
+ case usagetotal.FieldCredits:
|
|
|
+ return m.OldCredits(ctx)
|
|
|
case usagetotal.FieldStartIndex:
|
|
|
return m.OldStartIndex(ctx)
|
|
|
case usagetotal.FieldEndIndex:
|
|
@@ -44200,6 +44279,13 @@ func (m *UsageTotalMutation) SetField(name string, value ent.Value) error {
|
|
|
}
|
|
|
m.SetTotalTokens(v)
|
|
|
return nil
|
|
|
+ case usagetotal.FieldCredits:
|
|
|
+ v, ok := value.(float64)
|
|
|
+ if !ok {
|
|
|
+ return fmt.Errorf("unexpected type %T for field %s", value, name)
|
|
|
+ }
|
|
|
+ m.SetCredits(v)
|
|
|
+ return nil
|
|
|
case usagetotal.FieldStartIndex:
|
|
|
v, ok := value.(uint64)
|
|
|
if !ok {
|
|
@@ -44238,6 +44324,9 @@ func (m *UsageTotalMutation) AddedFields() []string {
|
|
|
if m.addtotal_tokens != nil {
|
|
|
fields = append(fields, usagetotal.FieldTotalTokens)
|
|
|
}
|
|
|
+ if m.addcredits != nil {
|
|
|
+ fields = append(fields, usagetotal.FieldCredits)
|
|
|
+ }
|
|
|
if m.addstart_index != nil {
|
|
|
fields = append(fields, usagetotal.FieldStartIndex)
|
|
|
}
|
|
@@ -44261,6 +44350,8 @@ func (m *UsageTotalMutation) AddedField(name string) (ent.Value, bool) {
|
|
|
return m.AddedType()
|
|
|
case usagetotal.FieldTotalTokens:
|
|
|
return m.AddedTotalTokens()
|
|
|
+ case usagetotal.FieldCredits:
|
|
|
+ return m.AddedCredits()
|
|
|
case usagetotal.FieldStartIndex:
|
|
|
return m.AddedStartIndex()
|
|
|
case usagetotal.FieldEndIndex:
|
|
@@ -44297,6 +44388,13 @@ func (m *UsageTotalMutation) AddField(name string, value ent.Value) error {
|
|
|
}
|
|
|
m.AddTotalTokens(v)
|
|
|
return nil
|
|
|
+ case usagetotal.FieldCredits:
|
|
|
+ v, ok := value.(float64)
|
|
|
+ if !ok {
|
|
|
+ return fmt.Errorf("unexpected type %T for field %s", value, name)
|
|
|
+ }
|
|
|
+ m.AddCredits(v)
|
|
|
+ return nil
|
|
|
case usagetotal.FieldStartIndex:
|
|
|
v, ok := value.(int64)
|
|
|
if !ok {
|
|
@@ -44335,6 +44433,9 @@ func (m *UsageTotalMutation) ClearedFields() []string {
|
|
|
if m.FieldCleared(usagetotal.FieldTotalTokens) {
|
|
|
fields = append(fields, usagetotal.FieldTotalTokens)
|
|
|
}
|
|
|
+ if m.FieldCleared(usagetotal.FieldCredits) {
|
|
|
+ fields = append(fields, usagetotal.FieldCredits)
|
|
|
+ }
|
|
|
if m.FieldCleared(usagetotal.FieldStartIndex) {
|
|
|
fields = append(fields, usagetotal.FieldStartIndex)
|
|
|
}
|
|
@@ -44367,6 +44468,9 @@ func (m *UsageTotalMutation) ClearField(name string) error {
|
|
|
case usagetotal.FieldTotalTokens:
|
|
|
m.ClearTotalTokens()
|
|
|
return nil
|
|
|
+ case usagetotal.FieldCredits:
|
|
|
+ m.ClearCredits()
|
|
|
+ return nil
|
|
|
case usagetotal.FieldStartIndex:
|
|
|
m.ClearStartIndex()
|
|
|
return nil
|
|
@@ -44402,6 +44506,9 @@ func (m *UsageTotalMutation) ResetField(name string) error {
|
|
|
case usagetotal.FieldTotalTokens:
|
|
|
m.ResetTotalTokens()
|
|
|
return nil
|
|
|
+ case usagetotal.FieldCredits:
|
|
|
+ m.ResetCredits()
|
|
|
+ return nil
|
|
|
case usagetotal.FieldStartIndex:
|
|
|
m.ResetStartIndex()
|
|
|
return nil
|