knowledge

type Entity struct {
	// ID 实体唯一标识
	ID string `json:"id"`

	// Name 实体名称
	Name string `json:"name"`

	// Type 实体类型（如 Person, Company, Language）
	Type string `json:"type"`

	// Properties 属性
	Properties map[string]any `json:"properties,omitempty"`

	// CreatedAt 创建时间
	CreatedAt time.Time `json:"created_at,omitempty"`
}

type GraphQueryResult struct {
	// Entities 匹配的实体
	Entities []Entity `json:"entities,omitempty"`

	// Relations 匹配的关系
	Relations []Relation `json:"relations,omitempty"`

	// Paths 匹配的路径
	Paths [][]string `json:"paths,omitempty"`

	// RawResult 原始查询结果
	RawResult any `json:"raw_result,omitempty"`
}

type GraphRetriever struct {
	// contains filtered or unexported fields
}

type GraphRetrieverOption func(*GraphRetriever)

type GraphStore interface {
	// AddEntity 添加实体
	AddEntity(ctx context.Context, entity Entity) error

	// AddRelation 添加关系
	AddRelation(ctx context.Context, relation Relation) error

	// GetEntity 获取实体
	GetEntity(ctx context.Context, nameOrID string) (*Entity, error)

	// GetRelations 获取实体的关系
	GetRelations(ctx context.Context, entityNameOrID string, direction string) ([]Relation, error)

	// SearchEntities 搜索实体（按名称或类型）
	SearchEntities(ctx context.Context, query string, entityType string, limit int) ([]Entity, error)

	// ExecuteCypher 执行 Cypher 查询
	ExecuteCypher(ctx context.Context, cypher string, params map[string]any) (*GraphQueryResult, error)

	// GetNeighbors 获取 N 跳邻居
	GetNeighbors(ctx context.Context, entityNameOrID string, hops int) ([]Entity, []Relation, error)

	// DeleteEntity 删除实体
	DeleteEntity(ctx context.Context, nameOrID string) error

	// DeleteRelation 删除关系
	DeleteRelation(ctx context.Context, id string) error

	// Clear 清空所有数据
	Clear(ctx context.Context) error

	// Stats 统计信息
	Stats(ctx context.Context) (entityCount, relationCount int, err error)
}

type MemoryGraphStore struct {
	// contains filtered or unexported fields
}

type PropertyGraph struct {
	// contains filtered or unexported fields
}

type Relation struct {
	// ID 关系唯一标识
	ID string `json:"id"`

	// From 源实体名称或 ID
	From string `json:"from"`

	// To 目标实体名称或 ID
	To string `json:"to"`

	// Type 关系类型（如 created_by, works_at, has_skill）
	Type string `json:"type"`

	// Properties 关系属性
	Properties map[string]any `json:"properties,omitempty"`

	// Weight 关系权重
	Weight float64 `json:"weight,omitempty"`

	// CreatedAt 创建时间
	CreatedAt time.Time `json:"created_at,omitempty"`
}

type Text2Cypher struct {
	// contains filtered or unexported fields
}

type Text2CypherOption func(*Text2Cypher)