mirror of
https://github.com/Ed1s0nZ/CyberStrikeAI.git
synced 2026-10-20 06:27:53 +02:00
Sycun
983c981b92
Mark generated execution prompts, populate deterministic user sections from actual human entries, preserve canonical ledger on overflow/transient restarts, fit long human entries under their formatted budget, and keep unchanged historical keys editable. Add real reduction-to-snapshot Cookie recovery and production entry-point regression tests.
80 lines
2.8 KiB
Go
80 lines
2.8 KiB
Go
package multiagent
|
|
|
|
import (
|
|
"cyberstrike-ai/internal/agent"
|
|
"cyberstrike-ai/internal/config"
|
|
"github.com/cloudwego/eino/adk"
|
|
"github.com/cloudwego/eino/schema"
|
|
)
|
|
|
|
func isGeneratedExecutionUserMessage(msg adk.Message) bool {
|
|
if msg == nil || msg.Role != schema.User {
|
|
return false
|
|
}
|
|
kind, _ := msg.Extra[agent.GeneratedContextTypeKey].(string)
|
|
return agent.IsGeneratedExecutionContext(kind)
|
|
}
|
|
|
|
// Templates contain plan/model/tool text. Carry human intent from the original
|
|
// messages before flattening/budgeting, while keeping the execution prompt intact.
|
|
func withPlanExecuteInputProvenance(msgs, originals []adk.Message, appCfg *config.Config, mwCfg *config.MultiAgentEinoMiddlewareConfig, kind string) []adk.Message {
|
|
out := make([]adk.Message, len(msgs))
|
|
for i, msg := range msgs {
|
|
if msg == nil || msg.Role != schema.User {
|
|
out[i] = msg
|
|
continue
|
|
}
|
|
clone := *msg
|
|
clone.Extra = cloneAnyMap(msg.Extra)
|
|
if clone.Extra == nil {
|
|
clone.Extra = make(map[string]interface{})
|
|
}
|
|
clone.Extra[agent.GeneratedContextTypeKey] = kind
|
|
out[i] = &clone
|
|
}
|
|
maxRunes, entryMax := config.DefaultSummarizationUserIntentLedgerMaxRunes, config.DefaultSummarizationUserIntentLedgerEntryMaxRunes
|
|
if mwCfg != nil {
|
|
maxRunes = mwCfg.SummarizationUserIntentLedgerMaxRunesEffective()
|
|
entryMax = mwCfg.SummarizationUserIntentLedgerEntryMaxRunesEffective()
|
|
}
|
|
maxTotal := 120000
|
|
if appCfg != nil && appCfg.OpenAI.MaxTotalTokens > 0 {
|
|
maxTotal = appCfg.OpenAI.MaxTotalTokens
|
|
}
|
|
maxRunes = minPositiveInt(maxRunes, modelFacingRuneBudget(maxTotal, 0.20))
|
|
return mergeMessageIntoLeadingSystem(out, buildOriginalUserIntentLedgerMessage(originals, maxRunes, minPositiveInt(entryMax, maxRunes)))
|
|
}
|
|
|
|
// Eino's deterministic <all_user_messages> replacement knows only its own
|
|
// summary flags. For host templates, give that replacement actual human entries
|
|
// from the ledger, keeping model-generated execution data outside user authority.
|
|
// Summary model input, snapshots and working trails still contain the template.
|
|
func summarizationDefaultFinalizeInput(msgs, originals []adk.Message) []adk.Message {
|
|
hasTemplates := false
|
|
out := make([]adk.Message, 0, len(msgs))
|
|
liveHuman := make(map[string]bool)
|
|
for _, msg := range msgs {
|
|
if isGeneratedExecutionUserMessage(msg) {
|
|
hasTemplates = true
|
|
clone := *msg
|
|
clone.Role = schema.Assistant
|
|
out = append(out, &clone)
|
|
} else {
|
|
out = append(out, msg)
|
|
if msg != nil && msg.Role == schema.User && !isInternalSummarizationUserMessage(msg) {
|
|
liveHuman[normalizeUserIntentLedgerText(adkUserMessageText(msg))] = true
|
|
}
|
|
}
|
|
}
|
|
if !hasTemplates {
|
|
return msgs
|
|
}
|
|
var carried []adk.Message
|
|
for _, entry := range collectOriginalUserIntentEntries(originals) {
|
|
if !liveHuman[normalizeUserIntentLedgerText(entry)] {
|
|
carried = append(carried, schema.UserMessage(entry))
|
|
}
|
|
}
|
|
return append(carried, out...)
|
|
}
|