api_key.go 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. package schema
  2. import (
  3. "entgo.io/ent/schema/edge"
  4. "wechat-api/ent/schema/localmixin"
  5. "entgo.io/ent"
  6. "entgo.io/ent/dialect/entsql"
  7. "entgo.io/ent/schema"
  8. "entgo.io/ent/schema/field"
  9. "entgo.io/ent/schema/index"
  10. "github.com/suyuan32/simple-admin-common/orm/ent/mixins"
  11. )
  12. type ApiKey struct {
  13. ent.Schema
  14. }
  15. func (ApiKey) Fields() []ent.Field {
  16. return []ent.Field{
  17. field.String("title").Optional().Default("").Comment("Title"),
  18. field.String("key").Optional().Default("").Comment("Key"),
  19. field.Uint64("organization_id").Default(0).Comment("租户ID"),
  20. field.Uint64("agent_id").Default(0).Comment("智能体ID"),
  21. field.String("custom_agent_base").Optional().Default("").Comment("定制agent服务地址"),
  22. field.String("custom_agent_key").Optional().Default("").Comment("定制agent服务密钥"),
  23. field.String("openai_base").Optional().Default("").Comment("大模型服务地址"),
  24. field.String("openai_key").Optional().Default("").Comment("大模型服务密钥"),
  25. }
  26. }
  27. func (ApiKey) Mixin() []ent.Mixin {
  28. return []ent.Mixin{
  29. mixins.IDMixin{},
  30. localmixin.SoftDeleteMixin{},
  31. }
  32. }
  33. func (ApiKey) Indexes() []ent.Index {
  34. return []ent.Index{
  35. index.Fields("key").Unique(),
  36. }
  37. }
  38. func (ApiKey) Edges() []ent.Edge {
  39. return []ent.Edge{
  40. edge.From("agent", Agent.Type).
  41. Ref("key_agent").
  42. Unique().
  43. Field("agent_id").
  44. Required(),
  45. }
  46. }
  47. func (ApiKey) Annotations() []schema.Annotation {
  48. return []schema.Annotation{
  49. entsql.WithComments(true),
  50. entsql.Annotation{Table: "api_key"},
  51. }
  52. }