WeKnora deployment source with the reviewed wiki identity, summary and recovery fixes

internal/application/service/chat_pipeline/chat_pipeline.go

@@ -0,0 +1,140 @@
+package chatpipeline
+
+import (
+	"context"
+
+	"github.com/Tencent/WeKnora/internal/types"
+)
+
+// Plugin defines the interface for chat pipeline plugins
+// Plugins can handle specific events in the chat pipeline
+type Plugin interface {
+	// OnEvent handles the event with given context and chat management object
+	OnEvent(
+		ctx context.Context,
+		eventType types.EventType,
+		chatManage *types.ChatManage,
+		next func() *PluginError,
+	) *PluginError
+	// ActivationEvents returns the event types this plugin can handle
+	ActivationEvents() []types.EventType
+}
+
+// EventManager manages plugins and their event handling
+type EventManager struct {
+	// Map of event types to registered plugins
+	listeners map[types.EventType][]Plugin
+	// Map of event types to handler functions
+	handlers map[types.EventType]func(context.Context, types.EventType, *types.ChatManage) *PluginError
+}
+
+// NewEventManager creates and initializes a new EventManager
+func NewEventManager() *EventManager {
+	return &EventManager{
+		listeners: make(map[types.EventType][]Plugin),
+		handlers:  make(map[types.EventType]func(context.Context, types.EventType, *types.ChatManage) *PluginError),
+	}
+}
+
+// Register adds a plugin to the EventManager and sets up its event handlers
+func (e *EventManager) Register(plugin Plugin) {
+	if e.listeners == nil {
+		e.listeners = make(map[types.EventType][]Plugin)
+	}
+	if e.handlers == nil {
+		e.handlers = make(map[types.EventType]func(context.Context, types.EventType, *types.ChatManage) *PluginError)
+	}
+	for _, eventType := range plugin.ActivationEvents() {
+		e.listeners[eventType] = append(e.listeners[eventType], plugin)
+		e.handlers[eventType] = e.buildHandler(e.listeners[eventType])
+	}
+}
+
+// buildHandler constructs a handler chain for the given plugins
+func (e *EventManager) buildHandler(plugins []Plugin) func(
+	ctx context.Context, eventType types.EventType, chatManage *types.ChatManage,
+) *PluginError {
+	next := func(context.Context, types.EventType, *types.ChatManage) *PluginError { return nil }
+	for i := len(plugins) - 1; i >= 0; i-- {
+		current := plugins[i]
+		prevNext := next
+		next = func(ctx context.Context, eventType types.EventType, chatManage *types.ChatManage) *PluginError {
+			return current.OnEvent(ctx, eventType, chatManage, func() *PluginError {
+				return prevNext(ctx, eventType, chatManage)
+			})
+		}
+	}
+	return next
+}
+
+// Trigger invokes the handler for the specified event type
+func (e *EventManager) Trigger(ctx context.Context,
+	eventType types.EventType, chatManage *types.ChatManage,
+) *PluginError {
+	if handler, ok := e.handlers[eventType]; ok {
+		return handler(ctx, eventType, chatManage)
+	}
+	return nil
+}
+
+// PluginError represents an error in plugin execution
+type PluginError struct {
+	Err         error  // Original error
+	Description string // Human-readable description
+	ErrorType   string // Error type identifier
+}
+
+// Predefined plugin errors
+var (
+	ErrSearchNothing = &PluginError{
+		Description: "No relevant content found",
+		ErrorType:   "search_nothing",
+	}
+	ErrSearch = &PluginError{
+		Description: "Failed to search knowledge base",
+		ErrorType:   "search_failed",
+	}
+	ErrRerank = &PluginError{
+		Description: "Reranking failed",
+		ErrorType:   "rerank_failed",
+	}
+	ErrGetRerankModel = &PluginError{
+		Description: "Failed to get rerank model",
+		ErrorType:   "get_rerank_model_failed",
+	}
+	ErrGetChatModel = &PluginError{
+		Description: "Failed to get chat model",
+		ErrorType:   "get_chat_model_failed",
+	}
+	ErrTemplateParse = &PluginError{
+		Description: "Failed to parse context template",
+		ErrorType:   "template_parse_failed",
+	}
+	ErrTemplateExecute = &PluginError{
+		Description: "Failed to generate search content",
+		ErrorType:   "template_execution_failed",
+	}
+	ErrModelCall = &PluginError{
+		Description: "Failed to call model",
+		ErrorType:   "model_call_failed",
+	}
+	ErrGetHistory = &PluginError{
+		Description: "Failed to get conversation history",
+		ErrorType:   "get_history_failed",
+	}
+)
+
+// clone creates a copy of the PluginError
+func (p *PluginError) clone() *PluginError {
+	return &PluginError{
+		Description: p.Description,
+		ErrorType:   p.ErrorType,
+	}
+}
+
+// WithError attaches an error to the PluginError and returns a new instance
+func (p *PluginError) WithError(err error) *PluginError {
+	pp := p.clone()
+	pp.Err = err
+	return pp
+}