indexer

type ColumnInfo struct {
	// Name 列名
	Name string `json:"name"`

	// Type 数据类型
	Type string `json:"type"`

	// Nullable 是否可空
	Nullable bool `json:"nullable"`

	// Default 默认值
	Default *string `json:"default,omitempty"`

	// Description 列描述
	Description string `json:"description,omitempty"`

	// IsPrimaryKey 是否主键
	IsPrimaryKey bool `json:"is_primary_key,omitempty"`

	// ForeignKey 外键引用
	ForeignKey *ForeignKeyInfo `json:"foreign_key,omitempty"`
}

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

type ConcurrentIndexerOption func(*ConcurrentIndexer)

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

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

	// Type 实体类型
	Type EntityType `json:"type"`

	// Description 实体描述
	Description string `json:"description,omitempty"`

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

	// SourceDocIDs 来源文档ID列表
	SourceDocIDs []string `json:"source_doc_ids,omitempty"`

	// Embedding 实体向量 (可选)
	Embedding []float32 `json:"embedding,omitempty"`

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

	// UpdatedAt 更新时间
	UpdatedAt time.Time `json:"updated_at"`
}

type EntityExtractor interface {
	// Extract 从文本中抽取实体和关系
	Extract(ctx context.Context, text string) ([]*Entity, []*Relation, error)
}

type EntityQuery struct {
	// Types 实体类型过滤
	Types []EntityType

	// NamePattern 名称模式 (支持通配符)
	NamePattern string

	// Properties 属性过滤
	Properties map[string]any

	// Limit 返回数量限制
	Limit int

	// Offset 偏移量
	Offset int
}

type EntityType string

type ForeignKeyInfo struct {
	// Table 引用表
	Table string `json:"table"`

	// Column 引用列
	Column string `json:"column"`
}

type GraphContext struct {
	// Entities 相关实体
	Entities []*Entity

	// Relations 相关关系
	Relations []*Relation

	// Triples 三元组列表
	Triples []Triple
}

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

type GraphIndexerOption func(*GraphIndexer)

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

type GraphRetrieverOption func(*GraphRetriever)

type GraphStats struct {
	// EntityCount 实体数量
	EntityCount int

	// RelationCount 关系数量
	RelationCount int

	// EntityTypeCounts 各类型实体数量
	EntityTypeCounts map[EntityType]int

	// RelationTypeCounts 各类型关系数量
	RelationTypeCounts map[RelationType]int
}

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

	// AddEntities 批量添加实体
	AddEntities(ctx context.Context, entities []*Entity) error

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

	// GetEntityByName 按名称获取实体
	GetEntityByName(ctx context.Context, name string) (*Entity, error)

	// SearchEntities 搜索实体
	SearchEntities(ctx context.Context, query EntityQuery) ([]*Entity, error)

	// UpdateEntity 更新实体
	UpdateEntity(ctx context.Context, entity *Entity) error

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

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

	// AddRelations 批量添加关系
	AddRelations(ctx context.Context, relations []*Relation) error

	// GetRelation 获取关系
	GetRelation(ctx context.Context, id string) (*Relation, error)

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

	// SearchRelations 搜索关系
	SearchRelations(ctx context.Context, query RelationQuery) ([]*Relation, error)

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

	// GetNeighbors 获取邻居实体
	GetNeighbors(ctx context.Context, entityID string, depth int) ([]*Entity, error)

	// GetSubgraph 获取子图
	GetSubgraph(ctx context.Context, entityIDs []string, depth int) (*Subgraph, error)

	// Clear 清空图
	Clear(ctx context.Context) error

	// Stats 获取统计信息
	Stats(ctx context.Context) (*GraphStats, error)
}

type ImportOptions struct {
	// BatchSize 批量大小
	BatchSize int

	// CreateTable 自动创建表
	CreateTable bool

	// TruncateFirst 先清空表
	TruncateFirst bool

	// OnConflict 冲突处理
	OnConflict string // "ignore", "update", "error"
}

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

type IncrementalIndexerOption func(*IncrementalIndexer)

type IndexInfo struct {
	// Name 索引名
	Name string `json:"name"`

	// Columns 索引列
	Columns []string `json:"columns"`

	// Unique 是否唯一
	Unique bool `json:"unique"`
}

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

type LLMExtractorOption func(*LLMEntityExtractor)

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

type NLQuery interface {
	// ToSQL 转换为 SQL
	ToSQL(ctx context.Context, question string, tables []*TableInfo) (string, error)
}

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

type QueryResult struct {
	// Columns 列名
	Columns []string `json:"columns"`

	// Rows 行数据
	Rows []map[string]any `json:"rows"`

	// RowCount 行数
	RowCount int `json:"row_count"`

	// Query 执行的查询
	Query string `json:"query,omitempty"`
}

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

	// SourceID 源实体ID
	SourceID string `json:"source_id"`

	// TargetID 目标实体ID
	TargetID string `json:"target_id"`

	// Type 关系类型
	Type RelationType `json:"type"`

	// Description 关系描述
	Description string `json:"description,omitempty"`

	// Weight 关系权重 (0-1)
	Weight float64 `json:"weight"`

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

	// SourceDocIDs 来源文档ID列表
	SourceDocIDs []string `json:"source_doc_ids,omitempty"`

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

type RelationDirection string

type RelationQuery struct {
	// SourceID 源实体ID
	SourceID string

	// TargetID 目标实体ID
	TargetID string

	// Types 关系类型过滤
	Types []RelationType

	// MinWeight 最小权重
	MinWeight float64

	// Limit 返回数量限制
	Limit int
}

type RelationType string

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

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

type SQLIndexConfig struct {
	// Driver 数据库驱动
	Driver string

	// DSN 数据源名称
	DSN string

	// MaxOpenConns 最大连接数
	MaxOpenConns int

	// MaxIdleConns 最大空闲连接数
	MaxIdleConns int

	// TablePrefix 表前缀
	TablePrefix string

	// AutoMigrate 自动迁移
	AutoMigrate bool
}

type Subgraph struct {
	// Entities 实体列表
	Entities []*Entity

	// Relations 关系列表
	Relations []*Relation
}

type TableInfo struct {
	// Name 表名
	Name string `json:"name"`

	// Columns 列信息
	Columns []*ColumnInfo `json:"columns"`

	// PrimaryKey 主键
	PrimaryKey string `json:"primary_key,omitempty"`

	// Indexes 索引
	Indexes []*IndexInfo `json:"indexes,omitempty"`

	// Description 表描述
	Description string `json:"description,omitempty"`

	// RowCount 行数（估算）
	RowCount int64 `json:"row_count,omitempty"`
}

type Triple struct {
	Subject   string       `json:"subject"`
	Predicate RelationType `json:"predicate"`
	Object    string       `json:"object"`
}

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

type VectorIndexerOption func(*VectorIndexer)