WeKnora deployment source with the reviewed wiki identity, summary and recovery fixes

internal/workbench/operations_models.go

@@ -0,0 +1,121 @@
+package workbench
+
+import "time"
+
+type Balance struct {
+	TenantID  uint64    `json:"tenant_id" gorm:"primaryKey"`
+	UserID    string    `json:"user_id" gorm:"primaryKey"`
+	Balance   int64     `json:"balance"`
+	Reserved  int64     `json:"reserved"`
+	UpdatedAt time.Time `json:"updated_at"`
+}
+
+func (Balance) TableName() string { return "workbench_balances" }
+
+type Transaction struct {
+	ID             string    `json:"id" gorm:"primaryKey"`
+	TenantID       uint64    `json:"tenant_id" gorm:"uniqueIndex:wb_transaction_key"`
+	UserID         string    `json:"user_id"`
+	Delta          int64     `json:"delta"`
+	Kind           string    `json:"kind"`
+	Reason         string    `json:"reason"`
+	ActorID        string    `json:"actor_id"`
+	IdempotencyKey string    `json:"-" gorm:"uniqueIndex:wb_transaction_key"`
+	Metadata       JSON      `json:"metadata" gorm:"type:jsonb"`
+	CreatedAt      time.Time `json:"created_at"`
+}
+
+func (Transaction) TableName() string { return "workbench_transactions" }
+
+type Usage struct {
+	ID           string    `json:"id" gorm:"primaryKey"`
+	TenantID     uint64    `json:"tenant_id" gorm:"uniqueIndex:wb_usage_key"`
+	OwnerID      string    `json:"owner_id"`
+	JobID        string    `json:"job_id"`
+	CallID       string    `json:"call_id" gorm:"uniqueIndex:wb_usage_key"`
+	AgentID      string    `json:"agent_id"`
+	ModelID      string    `json:"model_id"`
+	Action       string    `json:"action"`
+	InputTokens  int64     `json:"input_tokens"`
+	OutputTokens int64     `json:"output_tokens"`
+	Points       int64     `json:"points"`
+	Status       string    `json:"status"`
+	Metadata     JSON      `json:"metadata" gorm:"type:jsonb"`
+	CreatedAt    time.Time `json:"created_at"`
+}
+
+func (Usage) TableName() string { return "workbench_usage" }
+
+type BalanceHold struct {
+	ID        string `gorm:"primaryKey"`
+	TenantID  uint64
+	UserID    string
+	Amount    int64
+	Status    string
+	ExpiresAt time.Time
+	CreatedAt time.Time
+}
+
+func (BalanceHold) TableName() string { return "workbench_balance_holds" }
+
+type Consumption struct {
+	ID          string     `json:"id" gorm:"primaryKey"`
+	TenantID    uint64     `json:"tenant_id" gorm:"uniqueIndex:wb_consumption_key"`
+	OwnerID     string     `json:"owner_id" gorm:"uniqueIndex:wb_consumption_key"`
+	AgentID     string     `json:"agent_id" gorm:"uniqueIndex:wb_consumption_key"`
+	KBID        string     `json:"kb_id" gorm:"uniqueIndex:wb_consumption_key"`
+	ChunkID     string     `json:"chunk_id" gorm:"uniqueIndex:wb_consumption_key"`
+	KnowledgeID string     `json:"knowledge_id"`
+	FileName    string     `json:"file_name"`
+	Category    string     `json:"category"`
+	Excerpt     string     `json:"excerpt"`
+	ResourceID  string     `json:"resource_id"`
+	SegmentID   string     `json:"segment_id"`
+	JobID       string     `json:"job_id"`
+	Status      string     `json:"status"`
+	Banned      bool       `json:"banned"`
+	LeaseUntil  *time.Time `json:"lease_until"`
+	CreatedAt   time.Time  `json:"created_at"`
+	UpdatedAt   time.Time  `json:"updated_at"`
+}
+
+func (Consumption) TableName() string { return "workbench_consumption" }
+
+type OperationLog struct {
+	ID         string    `json:"id" gorm:"primaryKey"`
+	TenantID   uint64    `json:"tenant_id"`
+	OwnerID    string    `json:"owner_id"`
+	OwnerName  string    `json:"owner_name"`
+	Kind       string    `json:"kind"`
+	Level      string    `json:"level"`
+	Action     string    `json:"action"`
+	Message    string    `json:"message"`
+	ResourceID string    `json:"resource_id"`
+	TraceID    string    `json:"trace_id"`
+	Status     int       `json:"status"`
+	Metadata   JSON      `json:"metadata" gorm:"type:jsonb"`
+	CreatedAt  time.Time `json:"created_at"`
+}
+
+func (OperationLog) TableName() string { return "workbench_operation_logs" }
+
+type PendingMemory struct {
+	ID                              string     `json:"id" gorm:"primaryKey"`
+	TenantID                        uint64     `json:"tenant_id"`
+	OwnerID                         string     `json:"owner_id"`
+	AgentID                         string     `json:"agent_id"`
+	Content                         string     `json:"content"`
+	SourceRef                       string     `json:"source_ref"`
+	Status                          string     `json:"status"`
+	ReviewedBy                      string     `json:"reviewed_by"`
+	CreatedAt                       time.Time  `json:"created_at"`
+	UpdatedAt                       time.Time  `json:"updated_at"`
+	LegacyHistoryID                 string     `json:"legacy_history_id,omitempty"`
+	NativeMemoryID                  string     `json:"native_memory_id,omitempty"`
+	ReconfirmedContent              string     `json:"reconfirmed_content,omitempty"`
+	ReconfirmedAt                   *time.Time `json:"reconfirmed_at,omitempty"`
+	CanReconfirm                    bool       `json:"can_reconfirm" gorm:"-"`
+	ReconfirmationUnavailableReason string     `json:"reconfirmation_unavailable_reason,omitempty" gorm:"-"`
+}
+
+func (PendingMemory) TableName() string { return "workbench_pending_memories" }