wechat-api/ent/agent.go

// Code generated by ent, DO NOT EDIT.

package ent

import (
	"fmt"
	"strings"
	"time"
	"wechat-api/ent/agent"

	"entgo.io/ent"
	"entgo.io/ent/dialect/sql"
)

// Agent is the model entity for the Agent schema.
type Agent struct {
	config `json:"-"`
	// ID of the ent.
	ID uint64 `json:"id,omitempty"`
	// Create Time | 创建日期
	CreatedAt time.Time `json:"created_at,omitempty"`
	// Update Time | 修改日期
	UpdatedAt time.Time `json:"updated_at,omitempty"`
	// Delete Time | 删除日期
	DeletedAt time.Time `json:"deleted_at,omitempty"`
	// name | 角色名称
	Name string `json:"name,omitempty"`
	// role | 角色设定
	Role string `json:"role,omitempty"`
	// status | 状态 1-正常 2-禁用
	Status int `json:"status,omitempty"`
	// background | 背景介绍
	Background string `json:"background,omitempty"`
	// examples | 对话案例
	Examples string `json:"examples,omitempty"`
	// organization_id | 租户ID
	OrganizationID uint64 `json:"organization_id,omitempty"`
	// dataset_id | 知识库ID
	DatasetID string `json:"dataset_id,omitempty"`
	// collection_id | 集合ID
	CollectionID string `json:"collection_id,omitempty"`
	// model | 模型
	Model string `json:"model,omitempty"`
	// api_base | api_base
	APIBase string `json:"api_base,omitempty"`
	// api_key | api_key
	APIKey string `json:"api_key,omitempty"`
	// type | 类型 1. 内置 2. 接入
	Type int `json:"type,omitempty"`
	// Edges holds the relations/edges for other nodes in the graph.
	// The values are being populated by the AgentQuery when eager-loading is set.
	Edges        AgentEdges `json:"edges"`
	selectValues sql.SelectValues
}

// AgentEdges holds the relations/edges for other nodes in the graph.
type AgentEdges struct {
	// WxAgent holds the value of the wx_agent edge.
	WxAgent []*Wx `json:"wx_agent,omitempty"`
	// TokenAgent holds the value of the token_agent edge.
	TokenAgent []*Token `json:"token_agent,omitempty"`
	// WaAgent holds the value of the wa_agent edge.
	WaAgent []*Whatsapp `json:"wa_agent,omitempty"`
	// KeyAgent holds the value of the key_agent edge.
	KeyAgent []*ApiKey `json:"key_agent,omitempty"`
	// loadedTypes holds the information for reporting if a
	// type was loaded (or requested) in eager-loading or not.
	loadedTypes [4]bool
}

// WxAgentOrErr returns the WxAgent value or an error if the edge
// was not loaded in eager-loading.
func (e AgentEdges) WxAgentOrErr() ([]*Wx, error) {
	if e.loadedTypes[0] {
		return e.WxAgent, nil
	}
	return nil, &NotLoadedError{edge: "wx_agent"}
}

// TokenAgentOrErr returns the TokenAgent value or an error if the edge
// was not loaded in eager-loading.
func (e AgentEdges) TokenAgentOrErr() ([]*Token, error) {
	if e.loadedTypes[1] {
		return e.TokenAgent, nil
	}
	return nil, &NotLoadedError{edge: "token_agent"}
}

// WaAgentOrErr returns the WaAgent value or an error if the edge
// was not loaded in eager-loading.
func (e AgentEdges) WaAgentOrErr() ([]*Whatsapp, error) {
	if e.loadedTypes[2] {
		return e.WaAgent, nil
	}
	return nil, &NotLoadedError{edge: "wa_agent"}
}

// KeyAgentOrErr returns the KeyAgent value or an error if the edge
// was not loaded in eager-loading.
func (e AgentEdges) KeyAgentOrErr() ([]*ApiKey, error) {
	if e.loadedTypes[3] {
		return e.KeyAgent, nil
	}
	return nil, &NotLoadedError{edge: "key_agent"}
}

// scanValues returns the types for scanning values from sql.Rows.
func (*Agent) scanValues(columns []string) ([]any, error) {
	values := make([]any, len(columns))
	for i := range columns {
		switch columns[i] {
		case agent.FieldID, agent.FieldStatus, agent.FieldOrganizationID, agent.FieldType:
			values[i] = new(sql.NullInt64)
		case agent.FieldName, agent.FieldRole, agent.FieldBackground, agent.FieldExamples, agent.FieldDatasetID, agent.FieldCollectionID, agent.FieldModel, agent.FieldAPIBase, agent.FieldAPIKey:
			values[i] = new(sql.NullString)
		case agent.FieldCreatedAt, agent.FieldUpdatedAt, agent.FieldDeletedAt:
			values[i] = new(sql.NullTime)
		default:
			values[i] = new(sql.UnknownType)
		}
	}
	return values, nil
}

// assignValues assigns the values that were returned from sql.Rows (after scanning)
// to the Agent fields.
func (a *Agent) assignValues(columns []string, values []any) error {
	if m, n := len(values), len(columns); m < n {
		return fmt.Errorf("mismatch number of scan values: %d != %d", m, n)
	}
	for i := range columns {
		switch columns[i] {
		case agent.FieldID:
			value, ok := values[i].(*sql.NullInt64)
			if !ok {
				return fmt.Errorf("unexpected type %T for field id", value)
			}
			a.ID = uint64(value.Int64)
		case agent.FieldCreatedAt:
			if value, ok := values[i].(*sql.NullTime); !ok {
				return fmt.Errorf("unexpected type %T for field created_at", values[i])
			} else if value.Valid {
				a.CreatedAt = value.Time
			}
		case agent.FieldUpdatedAt:
			if value, ok := values[i].(*sql.NullTime); !ok {
				return fmt.Errorf("unexpected type %T for field updated_at", values[i])
			} else if value.Valid {
				a.UpdatedAt = value.Time
			}
		case agent.FieldDeletedAt:
			if value, ok := values[i].(*sql.NullTime); !ok {
				return fmt.Errorf("unexpected type %T for field deleted_at", values[i])
			} else if value.Valid {
				a.DeletedAt = value.Time
			}
		case agent.FieldName:
			if value, ok := values[i].(*sql.NullString); !ok {
				return fmt.Errorf("unexpected type %T for field name", values[i])
			} else if value.Valid {
				a.Name = value.String
			}
		case agent.FieldRole:
			if value, ok := values[i].(*sql.NullString); !ok {
				return fmt.Errorf("unexpected type %T for field role", values[i])
			} else if value.Valid {
				a.Role = value.String
			}
		case agent.FieldStatus:
			if value, ok := values[i].(*sql.NullInt64); !ok {
				return fmt.Errorf("unexpected type %T for field status", values[i])
			} else if value.Valid {
				a.Status = int(value.Int64)
			}
		case agent.FieldBackground:
			if value, ok := values[i].(*sql.NullString); !ok {
				return fmt.Errorf("unexpected type %T for field background", values[i])
			} else if value.Valid {
				a.Background = value.String
			}
		case agent.FieldExamples:
			if value, ok := values[i].(*sql.NullString); !ok {
				return fmt.Errorf("unexpected type %T for field examples", values[i])
			} else if value.Valid {
				a.Examples = value.String
			}
		case agent.FieldOrganizationID:
			if value, ok := values[i].(*sql.NullInt64); !ok {
				return fmt.Errorf("unexpected type %T for field organization_id", values[i])
			} else if value.Valid {
				a.OrganizationID = uint64(value.Int64)
			}
		case agent.FieldDatasetID:
			if value, ok := values[i].(*sql.NullString); !ok {
				return fmt.Errorf("unexpected type %T for field dataset_id", values[i])
			} else if value.Valid {
				a.DatasetID = value.String
			}
		case agent.FieldCollectionID:
			if value, ok := values[i].(*sql.NullString); !ok {
				return fmt.Errorf("unexpected type %T for field collection_id", values[i])
			} else if value.Valid {
				a.CollectionID = value.String
			}
		case agent.FieldModel:
			if value, ok := values[i].(*sql.NullString); !ok {
				return fmt.Errorf("unexpected type %T for field model", values[i])
			} else if value.Valid {
				a.Model = value.String
			}
		case agent.FieldAPIBase:
			if value, ok := values[i].(*sql.NullString); !ok {
				return fmt.Errorf("unexpected type %T for field api_base", values[i])
			} else if value.Valid {
				a.APIBase = value.String
			}
		case agent.FieldAPIKey:
			if value, ok := values[i].(*sql.NullString); !ok {
				return fmt.Errorf("unexpected type %T for field api_key", values[i])
			} else if value.Valid {
				a.APIKey = value.String
			}
		case agent.FieldType:
			if value, ok := values[i].(*sql.NullInt64); !ok {
				return fmt.Errorf("unexpected type %T for field type", values[i])
			} else if value.Valid {
				a.Type = int(value.Int64)
			}
		default:
			a.selectValues.Set(columns[i], values[i])
		}
	}
	return nil
}

// Value returns the ent.Value that was dynamically selected and assigned to the Agent.
// This includes values selected through modifiers, order, etc.
func (a *Agent) Value(name string) (ent.Value, error) {
	return a.selectValues.Get(name)
}

// QueryWxAgent queries the "wx_agent" edge of the Agent entity.
func (a *Agent) QueryWxAgent() *WxQuery {
	return NewAgentClient(a.config).QueryWxAgent(a)
}

// QueryTokenAgent queries the "token_agent" edge of the Agent entity.
func (a *Agent) QueryTokenAgent() *TokenQuery {
	return NewAgentClient(a.config).QueryTokenAgent(a)
}

// QueryWaAgent queries the "wa_agent" edge of the Agent entity.
func (a *Agent) QueryWaAgent() *WhatsappQuery {
	return NewAgentClient(a.config).QueryWaAgent(a)
}

// QueryKeyAgent queries the "key_agent" edge of the Agent entity.
func (a *Agent) QueryKeyAgent() *ApiKeyQuery {
	return NewAgentClient(a.config).QueryKeyAgent(a)
}

// Update returns a builder for updating this Agent.
// Note that you need to call Agent.Unwrap() before calling this method if this Agent
// was returned from a transaction, and the transaction was committed or rolled back.
func (a *Agent) Update() *AgentUpdateOne {
	return NewAgentClient(a.config).UpdateOne(a)
}

// Unwrap unwraps the Agent entity that was returned from a transaction after it was closed,
// so that all future queries will be executed through the driver which created the transaction.
func (a *Agent) Unwrap() *Agent {
	_tx, ok := a.config.driver.(*txDriver)
	if !ok {
		panic("ent: Agent is not a transactional entity")
	}
	a.config.driver = _tx.drv
	return a
}

// String implements the fmt.Stringer.
func (a *Agent) String() string {
	var builder strings.Builder
	builder.WriteString("Agent(")
	builder.WriteString(fmt.Sprintf("id=%v, ", a.ID))
	builder.WriteString("created_at=")
	builder.WriteString(a.CreatedAt.Format(time.ANSIC))
	builder.WriteString(", ")
	builder.WriteString("updated_at=")
	builder.WriteString(a.UpdatedAt.Format(time.ANSIC))
	builder.WriteString(", ")
	builder.WriteString("deleted_at=")
	builder.WriteString(a.DeletedAt.Format(time.ANSIC))
	builder.WriteString(", ")
	builder.WriteString("name=")
	builder.WriteString(a.Name)
	builder.WriteString(", ")
	builder.WriteString("role=")
	builder.WriteString(a.Role)
	builder.WriteString(", ")
	builder.WriteString("status=")
	builder.WriteString(fmt.Sprintf("%v", a.Status))
	builder.WriteString(", ")
	builder.WriteString("background=")
	builder.WriteString(a.Background)
	builder.WriteString(", ")
	builder.WriteString("examples=")
	builder.WriteString(a.Examples)
	builder.WriteString(", ")
	builder.WriteString("organization_id=")
	builder.WriteString(fmt.Sprintf("%v", a.OrganizationID))
	builder.WriteString(", ")
	builder.WriteString("dataset_id=")
	builder.WriteString(a.DatasetID)
	builder.WriteString(", ")
	builder.WriteString("collection_id=")
	builder.WriteString(a.CollectionID)
	builder.WriteString(", ")
	builder.WriteString("model=")
	builder.WriteString(a.Model)
	builder.WriteString(", ")
	builder.WriteString("api_base=")
	builder.WriteString(a.APIBase)
	builder.WriteString(", ")
	builder.WriteString("api_key=")
	builder.WriteString(a.APIKey)
	builder.WriteString(", ")
	builder.WriteString("type=")
	builder.WriteString(fmt.Sprintf("%v", a.Type))
	builder.WriteByte(')')
	return builder.String()
}

// Agents is a parsable slice of Agent.
type Agents []*Agent