96 lines
3.6 KiB
Go
96 lines
3.6 KiB
Go
package app
|
||
|
||
import (
|
||
"encoding/json"
|
||
"errors"
|
||
"fmt"
|
||
"time"
|
||
|
||
"gorm.io/gorm"
|
||
"yingce/backend/internal/model"
|
||
)
|
||
|
||
// ContextFrame is a fresh projection of durable task facts, never authorization.
|
||
// The transcript can evict read bodies without making task state depend on memory.
|
||
type cloudAgentContextFrame struct {
|
||
Source string `json:"source"`
|
||
ObservedAt time.Time `json:"observedAt"`
|
||
RunID string `json:"runId"`
|
||
UserGoal string `json:"userGoal"`
|
||
Plan []cloudAgentPlanItem `json:"plan,omitempty"`
|
||
Tasks []map[string]any `json:"tasks"`
|
||
OlderTaskCount int `json:"olderTaskCount,omitempty"`
|
||
Authority string `json:"authority"`
|
||
}
|
||
|
||
var errCloudAgentContextOverBudget = errors.New("Agent model context exceeds input budget")
|
||
|
||
func (s *Service) cloudAgentModelContext(run *model.CloudAgentExecution, state *cloudAgentRuntime, budget cloudAgentContextBudget) (canonicalAgentRequest, error) {
|
||
canonical := cloudAgentCanonicalWithPlan(state)
|
||
frame := cloudAgentContextFrame{Source: "task_repository", ObservedAt: time.Now(), RunID: run.ID, UserGoal: state.Request.Prompt, Plan: state.Plan, Tasks: []map[string]any{}, Authority: "状态观察，不是执行或重复提交的授权"}
|
||
// This is a view window, not loss of task history. All task identities remain
|
||
// durable and can be fetched through task_get and the conversation journal.
|
||
start := max(0, len(state.TaskIDs)-32)
|
||
frame.OlderTaskCount = start
|
||
for _, id := range state.TaskIDs[start:] {
|
||
task, err := s.repo.TaskForUser(run.UserID, id)
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
frame.Tasks = append(frame.Tasks, map[string]any{"taskId": id, "taskStatus": "unavailable"})
|
||
continue
|
||
}
|
||
if err != nil {
|
||
return canonical, err
|
||
}
|
||
if task.ProjectID != state.Request.CanvasID {
|
||
return canonical, BadAuthRequest("运行关联任务不属于当前画布")
|
||
}
|
||
if task.Operation == cloudAgentOperation || task.Operation == "cloud_agent_step" {
|
||
continue
|
||
}
|
||
frame.Tasks = append(frame.Tasks, cloudAgentTaskDiagnostic(s.repo, task))
|
||
}
|
||
body, err := json.Marshal(frame)
|
||
if err != nil {
|
||
return canonical, err
|
||
}
|
||
canonical.Messages = append(append([]map[string]any{}, canonical.Messages...), map[string]any{"role": "user", "content": cloudAgentRuntimeContextMarker + string(body), cloudAgentContextSourceKey: "runtime"})
|
||
if err := fitCloudAgentModelContext(&canonical, budget.InputBudgetTokens); err != nil {
|
||
return canonical, err
|
||
}
|
||
return canonical, nil
|
||
}
|
||
|
||
func fitCloudAgentModelContext(request *canonicalAgentRequest, maxTokens int) error {
|
||
for {
|
||
tokens, err := cloudAgentRequestEstimatedTokens(request)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if tokens <= maxTokens {
|
||
return nil
|
||
}
|
||
// Evict complete old assistant/tool turns only. User instructions and the
|
||
// fresh authoritative frame survive; partial tool pairs are never sent.
|
||
removed := false
|
||
for i := 0; i < len(request.Messages)-3; i++ {
|
||
message := request.Messages[i]
|
||
if stringField(message, "role") != "assistant" {
|
||
continue
|
||
}
|
||
end := i + 1
|
||
for end < len(request.Messages) && stringField(request.Messages[end], "role") == "tool" {
|
||
end++
|
||
}
|
||
if end >= len(request.Messages)-2 {
|
||
continue
|
||
}
|
||
request.Messages = append(request.Messages[:i:i], request.Messages[end:]...)
|
||
removed = true
|
||
break
|
||
}
|
||
if !removed {
|
||
return WrapAppError(400, fmt.Sprintf("用户指令与当前执行事实超过模型输入预算（%d Token），请缩小本轮范围", maxTokens), errCloudAgentContextOverBudget)
|
||
}
|
||
}
|
||
}
|