Method ChatStreamHandler

pkg/server/server.go:480–603

ChatStreamHandler 提供基于 Server-Sent Events (SSE) 的流式 Chat 接口。 路径示例: POST /v1/agents/chat/stream 请求体与 ChatHandler 相同,但响应为 text/event-stream。 每个事件均为一行 JSON: data: {"cursor":1,"bookmark":{...},"event":{...}}\n\n 前端可以根据 event 字段中的具体类型(decoding types.*Event)来渲染流式思考/文本/工具调用等。

()

//
// 前端可以根据 event 字段中的具体类型(decoding types.*Event)来渲染流式思考/文本/工具调用等。
func (s *Server) ChatStreamHandler() http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		start := time.Now()

		if r.Method != http.MethodPost {
			http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
			return
		}

		flusher, ok := w.(http.Flusher)
		if !ok {
			http.Error(w, "streaming unsupported", http.StatusInternalServerError)
			return
		}

		var req ChatRequest
		if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
			http.Error(w, "invalid JSON body", http.StatusBadRequest)
			return
		}
		if req.TemplateID == "" || req.Input == "" {
			http.Error(w, "template_id and input are required", http.StatusBadRequest)
			return
		}

		ctx := r.Context()
		ctx, cancel := context.WithTimeout(ctx, 5*time.Minute)
		defer cancel()

		agentConfig := &types.AgentConfig{
			TemplateID:     req.TemplateID,
			ModelConfig:    req.ModelConfig,
			Sandbox:        req.Sandbox,
			Middlewares:    req.Middlewares,
			Metadata:       req.Metadata,
			RoutingProfile: req.RoutingProfile,
		}

		if req.SkillsPackage != nil {
			agentConfig.SkillsPackage = req.SkillsPackage
		} else if len(req.Skills) > 0 {
			agentConfig.SkillsPackage = &types.SkillsPackageConfig{
				Source:        "local",
				Path:          ".",
				SkillsDir:     "skills",
				EnabledSkills: req.Skills,
			}
		}

		ag, err := agent.Create(ctx, agentConfig, s.deps)
		if err != nil {
			logging.Error(ctx, "http.chat_stream.error", map[string]any{
				"template_id":     req.TemplateID,
				"routing_profile": req.RoutingProfile,
				"stage":           "create_agent",
				"error":           err.Error(),
				"latency_ms":      time.Since(start).Milliseconds(),
			})