54 lines
1.7 KiB
Go
54 lines
1.7 KiB
Go
|
package model
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// SecurityLevel 安全等级
|
|||
|
|
type SecurityLevel string
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
SecurityLevelSafe SecurityLevel = "safe"
|
|||
|
|
SecurityLevelReview SecurityLevel = "review"
|
|||
|
|
SecurityLevelDanger SecurityLevel = "danger"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Agent 智能体
|
|||
|
|
type Agent struct {
|
|||
|
|
ID string `json:"id" gorm:"primaryKey"`
|
|||
|
|
Name string `json:"name" gorm:"size:100;not null"`
|
|||
|
|
Description string `json:"description" gorm:"type:text"`
|
|||
|
|
OwnerID string `json:"owner_id" gorm:"size:50;not null;index"`
|
|||
|
|
|
|||
|
|
// Agent能力配置
|
|||
|
|
Capabilities []string `json:"capabilities" gorm:"type:text"` // JSON数组，可用工具列表
|
|||
|
|
MemoryLimit int64 `json:"memory_limit" gorm:"default:134217728"` // 128MB
|
|||
|
|
Timeout int `json:"timeout" gorm:"default:60"` // 60秒
|
|||
|
|
|
|||
|
|
// 安全配置
|
|||
|
|
SecurityLevel SecurityLevel `json:"security_level" gorm:"size:20;default:'safe'"`
|
|||
|
|
AllowDangerousTools bool `json:"allow_dangerous_tools" gorm:"default:false"`
|
|||
|
|
|
|||
|
|
// 状态
|
|||
|
|
IsActive bool `json:"is_active" gorm:"default:true"`
|
|||
|
|
|
|||
|
|
CreatedAt time.Time `json:"created_at"`
|
|||
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AgentRequest 聊天请求
|
|||
|
|
type AgentRequest struct {
|
|||
|
|
AgentID string `json:"agent_id" binding:"required"`
|
|||
|
|
Message string `json:"message" binding:"required"`
|
|||
|
|
SessionID string `json:"session_id"`
|
|||
|
|
Context map[string]interface{} `json:"context"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AgentResponse 聊天响应
|
|||
|
|
type AgentResponse struct {
|
|||
|
|
Reply string `json:"reply"`
|
|||
|
|
SessionID string `json:"session_id"`
|
|||
|
|
ToolsUsed []string `json:"tools_used"`
|
|||
|
|
Metadata map[string]interface{} `json:"metadata"`
|
|||
|
|
}
|