diff --git a/internal/app/app.go b/internal/app/app.go index 20f7bfe0..dacbf0fd 100644 --- a/internal/app/app.go +++ b/internal/app/app.go @@ -16,6 +16,7 @@ import ( "cyberstrike-ai/internal/c2" "cyberstrike-ai/internal/config" "cyberstrike-ai/internal/database" + "cyberstrike-ai/internal/einoobserve" "cyberstrike-ai/internal/handler" "cyberstrike-ai/internal/knowledge" "cyberstrike-ai/internal/logger" @@ -558,6 +559,10 @@ func (a *App) RunWithContext(ctx context.Context) error { // Shutdown 关闭应用 func (a *App) Shutdown() { + shutdownCtx, shutdownCancel := context.WithTimeout(context.Background(), 5*time.Second) + _ = einoobserve.ShutdownOtel(shutdownCtx) + shutdownCancel() + // 停止钉钉/飞书长连接 a.robotMu.Lock() if a.dingCancel != nil { diff --git a/internal/handler/agent.go b/internal/handler/agent.go index 805dd5c7..b338c04f 100644 --- a/internal/handler/agent.go +++ b/internal/handler/agent.go @@ -1249,6 +1249,10 @@ func (h *AgentHandler) createProgressCallback(runCtx context.Context, cancelRun eventType != "response_start" && eventType != "response_delta" && eventType != "tool_result_delta" && + eventType != "eino_trace_run" && + eventType != "eino_trace_start" && + eventType != "eino_trace_end" && + eventType != "eino_trace_error" && eventType != "eino_agent_reply_stream_start" && eventType != "eino_agent_reply_stream_delta" && eventType != "eino_agent_reply_stream_end" {