Documentation
¶
Index ¶
- type Entity
- type EntityCollection
- func (ec *EntityCollection) AddEntities(entities ...*Entity) error
- func (ec *EntityCollection) CountEntities() (int, error)
- func (ec *EntityCollection) DeleteEntity(entityID string) error
- func (ec *EntityCollection) GetEntity(entityID string) (*Entity, error)
- func (ec *EntityCollection) ListAllEntities() ([]*Entity, error)
- func (ec *EntityCollection) SearchEntities(query string, limit int) ([]*Entity, error)
- func (ec *EntityCollection) SearchEntitiesByType(entityType EntityType, query string, limit int) ([]*Entity, error)
- func (ec *EntityCollection) UpdateEntity(entity *Entity) error
- type EntityType
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Entity ¶
type Entity struct {
ID string `json:"id"` // 实体唯一标识
Name string `json:"name"` // 实体名称
Type EntityType `json:"type"` // 实体类型
Description string `json:"description"` // 详细描述
Aliases []string `json:"aliases"` // 别名
Properties map[string]interface{} `json:"properties"` // 扩展属性
Tags []string `json:"tags"` // 标签
CreatedAt time.Time `json:"created_at"` // 创建时间
UpdatedAt time.Time `json:"updated_at"` // 更新时间
}
Entity 知识图谱实体
func GenerateRandomEntities ¶
GenerateRandomEntities 生成随机实体数据用于测试
func NewEntity ¶
func NewEntity(id, name string, entityType EntityType, description string) *Entity
NewEntity 创建新实体
func (*Entity) SetProperty ¶
SetProperty 设置属性
func (*Entity) ToRAGDocument ¶
func (e *Entity) ToRAGDocument() *vectorstore.Document
ToRAGDocument 转换为RAG文档
type EntityCollection ¶
type EntityCollection struct {
// contains filtered or unexported fields
}
EntityCollection RAG实体集合管理器
func NewEntityCollection ¶
func NewEntityCollection(db *gorm.DB, collectionName string) (*EntityCollection, error)
NewEntityCollection 创建实体集合管理器
func (*EntityCollection) AddEntities ¶
func (ec *EntityCollection) AddEntities(entities ...*Entity) error
AddEntities 批量导入实体到RAG集合
func (*EntityCollection) CountEntities ¶
func (ec *EntityCollection) CountEntities() (int, error)
CountEntities 获取实体总数
func (*EntityCollection) DeleteEntity ¶
func (ec *EntityCollection) DeleteEntity(entityID string) error
DeleteEntity 删除实体
func (*EntityCollection) GetEntity ¶
func (ec *EntityCollection) GetEntity(entityID string) (*Entity, error)
GetEntity 根据ID获取实体
func (*EntityCollection) ListAllEntities ¶
func (ec *EntityCollection) ListAllEntities() ([]*Entity, error)
ListAllEntities 列出所有实体
func (*EntityCollection) SearchEntities ¶
func (ec *EntityCollection) SearchEntities(query string, limit int) ([]*Entity, error)
SearchEntities 在RAG集合中搜索实体
func (*EntityCollection) SearchEntitiesByType ¶
func (ec *EntityCollection) SearchEntitiesByType(entityType EntityType, query string, limit int) ([]*Entity, error)
SearchEntitiesByType 按类型搜索实体
func (*EntityCollection) UpdateEntity ¶
func (ec *EntityCollection) UpdateEntity(entity *Entity) error
UpdateEntity 更新实体
type EntityType ¶
type EntityType string
EntityType 实体类型
const ( EntityTypePerson EntityType = "person" // 人物 EntityTypeOrganization EntityType = "organization" // 组织/公司 EntityTypeTechnology EntityType = "technology" // 技术/工具 EntityTypeVulnerability EntityType = "vulnerability" // 漏洞 EntityTypeConcept EntityType = "concept" // 概念 EntityTypeProduct EntityType = "product" // 产品 EntityTypeLocation EntityType = "location" // 地点 EntityTypeEvent EntityType = "event" // 事件 )
Source Files
Click to show internal directories.
Click to hide internal directories.