mlogclub
164121836d
- Replaced the existing runtime executor with a new eino package for better modularity. - Updated Service struct to use the new RuntimeExecutor from eino. - Adjusted Run and Resume methods to accommodate changes in input types and execution logic. - Introduced new types in the eino package to align with the previous executor's functionality. - Refactored tooling preparation logic to streamline tool definitions and improve clarity. - Added new context builders and event consumers to enhance runtime event handling. - Removed legacy executor code and ensured all references are updated to the new eino package.
59 lines
2.2 KiB
Go
59 lines
2.2 KiB
Go
package executor
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"cs-agent/internal/ai/runtime/internal/impl/adapter"
|
|
"cs-agent/internal/ai/runtime/internal/impl/callbacks"
|
|
"cs-agent/internal/ai/runtime/internal/impl/retrievers"
|
|
"cs-agent/internal/pkg/utils"
|
|
|
|
"github.com/cloudwego/eino/schema"
|
|
)
|
|
|
|
func buildRunMessages(ctx context.Context, req RunInput, summary *RunResult, collector *callbacks.RuntimeTraceCollector) []*schema.Message {
|
|
if req.Conversation == nil || req.UserMessage == nil {
|
|
return nil
|
|
}
|
|
history := adapter.BuildHistoryMessages(req.Conversation.ID, req.UserMessage.ID, 12)
|
|
if summary != nil {
|
|
summary.HistoryMessageCount = len(history.Messages)
|
|
}
|
|
if collector != nil {
|
|
collector.Data.Input.HistoryMessageCount = len(history.Messages)
|
|
if req.AIAgent != nil {
|
|
collector.Data.Input.KnowledgeBaseIDs = utils.SplitInt64s(req.AIAgent.KnowledgeIDs)
|
|
}
|
|
collector.Data.Input.CurrentUserMessagePreview = preview(req.UserMessage.Content, 120)
|
|
}
|
|
messages := make([]*schema.Message, 0, len(history.Messages)+3)
|
|
messages = append(messages, history.Messages...)
|
|
appendRetrievedContext(ctx, req, summary, collector, &messages)
|
|
messages = append(messages, schema.UserMessage(strings.TrimSpace(req.UserMessage.Content)))
|
|
return messages
|
|
}
|
|
|
|
func appendRetrievedContext(ctx context.Context, req RunInput, summary *RunResult, collector *callbacks.RuntimeTraceCollector, messages *[]*schema.Message) {
|
|
if req.AIAgent == nil || req.UserMessage == nil || messages == nil {
|
|
return
|
|
}
|
|
retriever := retrievers.NewKnowledgeRetriever(req.AIAgent)
|
|
retrieveOptions := retrievers.DefaultKnowledgeRetrieveOptions()
|
|
retrieveOptions.QueryPreview = preview(req.UserMessage.Content, 120)
|
|
retrieveResult, retrieveErr := retriever.RetrieveContextByOptions(ctx, retrieveOptions, strings.TrimSpace(req.UserMessage.Content))
|
|
if retrieveErr != nil || retrieveResult == nil {
|
|
return
|
|
}
|
|
if summary != nil {
|
|
summary.RetrieverCount = len(retrieveResult.Hits)
|
|
}
|
|
if collector != nil {
|
|
collector.SetRetrieverSummary(retrieveResult.TraceSummary)
|
|
collector.Data.Retriever.Items = append(collector.Data.Retriever.Items, retrieveResult.TraceItems...)
|
|
}
|
|
if strings.TrimSpace(retrieveResult.ContextText) != "" {
|
|
*messages = append(*messages, schema.SystemMessage(retrieveResult.ContextText))
|
|
}
|
|
}
|