knowledgebase

type ExportKnowledgeBaseOptions struct {
	KnowledgeBaseId                 int64
	ExportKnowledgeBaseEntryHandler func(entry schema.KnowledgeBaseEntry) (schema.KnowledgeBaseEntry, error)
	ExportRAGDocumentHandler        func(doc schema.VectorStoreDocument) (schema.VectorStoreDocument, error)
	OnProgressHandler               func(percent float64, message string, messageType string)
	ExtraDataReader                 io.Reader // 额外数据的 reader，会在导出时一起导出
	VectorStoreName                 string
}

type ImportKnowledgeBaseOptions struct {
	ImportKnowledgeBaseEntryHandler func(entry schema.KnowledgeBaseEntry) (schema.KnowledgeBaseEntry, error)
	ImportRAGDocumentHandler        func(doc schema.VectorStoreDocument) (schema.VectorStoreDocument, error)
	OverwriteExisting               bool
	NewKnowledgeBaseName            string
	OnProgressHandler               func(percent float64, message string, messageType string)
	ExtraDataHandler                func(extraData io.Reader) error // 额外数据处理回调函数
	RAGID                           string
}

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

type KnowledgeEntryWithScore struct {
	Entry *schema.KnowledgeBaseEntry `json:"entry"`
	Score float64                    `json:"score"`
}

type QueryConfig struct {
	Ctx                  context.Context
	Limit                int
	CollectionName       string
	CollectionNumLimit   int
	CollectionScoreLimit int
	EnableAISummary      bool
	Filter               func(key string, docGetter func() *vectorstore.Document, knowledgeBaseEntryGetter func() (*schema.KnowledgeBaseEntry, error)) bool
	EnhancePlan          string
	MsgCallBack          func(*SearchKnowledgebaseResult)
}

type QueryOption func(*QueryConfig)

type SearchKnowledgebaseResult struct {
	Message string
	Data    any
	Type    string
}

type SyncResult struct {
	TotalDBEntries    int      `json:"total_db_entries"`    // 数据库中的总条目数
	TotalRAGDocuments int      `json:"total_rag_documents"` // RAG中的总文档数
	AddedToRAG        []string `json:"added_to_rag"`        // 添加到RAG的条目ID列表
	DeletedFromRAG    []string `json:"deleted_from_rag"`    // 从RAG删除的文档ID列表
	SyncErrors        []string `json:"sync_errors"`         // 同步过程中的错误列表
}

type SyncStatus struct {
	DatabaseEntries int  `json:"database_entries"` // 数据库中的条目数
	RAGDocuments    int  `json:"rag_documents"`    // RAG中的文档数
	InSync          bool `json:"in_sync"`          // 是否同步
}