mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-05-26 10:47:21 +08:00
f0122179dd
### What problem does this PR solve? align time units with Python and centralize timestamp injection in BaseModel ### Type of change - [x] Refactoring
90 lines
2.7 KiB
Go
90 lines
2.7 KiB
Go
//
|
|
// Copyright 2026 The InfiniFlow Authors. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
package entity
|
|
|
|
import "time"
|
|
|
|
// Space status constants
|
|
const (
|
|
SpaceStatusActive = "1" // Normal active space
|
|
SpaceStatusDeleted = "0" // Soft-deleted space
|
|
SpaceStatusDeleting = "2" // Space is being asynchronously deleted
|
|
)
|
|
|
|
// SkillSpace represents a skills space (library) that contains skills
|
|
type SkillSpace struct {
|
|
ID string `gorm:"column:id;primaryKey;size:32" json:"id"`
|
|
TenantID string `gorm:"column:tenant_id;size:32;not null;index" json:"tenant_id"`
|
|
Name string `gorm:"column:name;size:128;not null" json:"name"`
|
|
FolderID string `gorm:"column:folder_id;size:32;not null" json:"folder_id"`
|
|
Description string `gorm:"column:description;type:text" json:"description"`
|
|
EmbdID string `gorm:"column:embd_id;size:128" json:"embd_id"`
|
|
RerankID string `gorm:"column:rerank_id;size:128" json:"rerank_id"`
|
|
TopK int `gorm:"column:top_k;default:10" json:"top_k"`
|
|
Status string `gorm:"column:status;size:1;default:1" json:"status"`
|
|
BaseModel
|
|
}
|
|
|
|
// TableName returns the table name for SkillSpace model
|
|
func (SkillSpace) TableName() string {
|
|
return "skill_spaces"
|
|
}
|
|
|
|
// StatusDescription returns a human-readable status string
|
|
func (s *SkillSpace) StatusDescription() string {
|
|
switch s.Status {
|
|
case SpaceStatusActive:
|
|
return "active"
|
|
case SpaceStatusDeleted:
|
|
return "deleted"
|
|
case SpaceStatusDeleting:
|
|
return "deleting"
|
|
default:
|
|
return "unknown"
|
|
}
|
|
}
|
|
|
|
// ToMap converts SkillSpace to a map for JSON response
|
|
func (s *SkillSpace) ToMap() map[string]interface{} {
|
|
result := map[string]interface{}{
|
|
"id": s.ID,
|
|
"tenant_id": s.TenantID,
|
|
"name": s.Name,
|
|
"folder_id": s.FolderID,
|
|
"top_k": s.TopK,
|
|
"status": s.StatusDescription(),
|
|
}
|
|
|
|
if s.Description != "" {
|
|
result["description"] = s.Description
|
|
}
|
|
if s.EmbdID != "" {
|
|
result["embd_id"] = s.EmbdID
|
|
}
|
|
if s.RerankID != "" {
|
|
result["rerank_id"] = s.RerankID
|
|
}
|
|
if s.CreateTime != nil {
|
|
result["create_time"] = s.CreateTime
|
|
}
|
|
if s.UpdateTime != nil {
|
|
result["update_time"] = time.UnixMilli(*s.UpdateTime).Format("2006-01-02 15:04:05")
|
|
}
|
|
|
|
return result
|
|
}
|