knowledge

type AliyunHandler struct {
	WorkspaceID string
	IndexID     string
	// contains filtered or unexported fields
}

type AliyunOptions struct {
	AccessKeyID     string
	AccessKeySecret string
	// Endpoint defaults to DefaultBailianEndpoint (Beijing public endpoint).
	Endpoint string
	// RegionID defaults to cn-beijing when Endpoint is empty.
	RegionID string
	// WorkspaceID is the business space id from the Bailian console.
	WorkspaceID string
	// IndexID is the knowledge base id (CreateIndex / console).
	IndexID string
}

type Chunk struct {
	Index    int
	Title    string
	Text     string
	Metadata map[string]any
}

type ChunkOptions struct {
	MaxChars      int
	OverlapChars  int
	MinChars      int
	DocumentTitle string
	// PreChunkClean is passed to utils.CleanText before the LLM call (UTF-8 repair, optional markdown strip, etc.).
	// If nil, StripMarkdown and DedupLines are enabled so code fences / mermaid are less likely to break JSON output.
	PreChunkClean *utils.Options
}

type Chunker interface {
	Provider() string
	Chunk(ctx context.Context, text string, opts *ChunkOptions) ([]Chunk, error)
}

type DeleteOptions struct {
	Namespace string
}

type Embedder interface {
	Embed(ctx context.Context, inputs []string) ([][]float64, error)
}

type FactoryOptions struct {
	Qdrant *QdrantOptions
	Aliyun *AliyunOptions
}

type GetOptions struct {
	Namespace string
}

type KnowledgeHandler interface {
	Provider() string

	Upsert(ctx context.Context, records []Record, options *UpsertOptions) error

	Query(ctx context.Context, text string, options *QueryOptions) ([]QueryResult, error)

	Get(ctx context.Context, ids []string, options *GetOptions) ([]Record, error)

	List(ctx context.Context, options *ListOptions) (*ListResult, error)

	Delete(ctx context.Context, ids []string, options *DeleteOptions) error
}

type KnowledgeProvider string

type LLMChunker struct {
	LLM   llm.LLMHandler
	Model string
}

type ListOptions struct {
	Namespace string
	Limit     int
	Offset    string
	Filters   map[string]any
}

type ListResult struct {
	Records    []Record
	NextOffset string
}

type NvidiaEmbedClient struct {
	BaseURL    string
	APIKey     string
	Model      string
	HTTPClient *http.Client
}

type QdrantHandler struct {
	BaseURL    string
	APIKey     string
	Collection string
	HTTPClient *http.Client
	Embedder   Embedder
}

type QdrantOptions struct {
	BaseURL    string
	APIKey     string
	Collection string
	HTTPClient *http.Client
	Embedder   Embedder
}

type QueryOptions struct {
	Namespace string
	TopK      int
	Filters   map[string]any
}

type QueryResult struct {
	Record Record
	Score  float64
}

type Record struct {
	ID        string
	Source    string
	Title     string
	Content   string
	Tags      []string
	Metadata  map[string]any
	CreatedAt time.Time
	UpdatedAt time.Time
}

type RerankResult struct {
	Index int
	Score float64
}

type SiliconFlowRerankClient struct {
	BaseURL    string
	APIKey     string
	Model      string
	HTTPClient *http.Client
}

type UpsertOptions struct {
	Namespace string
	Overwrite bool
}