Function handleChatCompletionRequest

wo2api.go:881–943

处理聊天补全请求

(w http.ResponseWriter, r *http.Request)

// 处理聊天补全请求
func handleChatCompletionRequest(w http.ResponseWriter, r *http.Request) {
	reqID := generateRequestID()
	startTime := time.Now()
	logInfo("[reqID:%s] 处理聊天补全请求", reqID)

	// 从请求头中提取令牌
	token, err := extractToken(r)
	if err != nil {
		logError("[reqID:%s] 提取令牌失败: %v", reqID, err)
		http.Error(w, fmt.Sprintf("Authorization error: %v", err), http.StatusUnauthorized)
		return
	}

	// 解析请求体
	var apiReq APIRequest
	if err := json.NewDecoder(r.Body).Decode(&apiReq); err != nil {
		logError("[reqID:%s] 解析请求失败: %v", reqID, err)
		http.Error(w, "Invalid request body", http.StatusBadRequest)
		return
	}

	// 验证消息格式
	valid, errMsg := validateMessages(apiReq.Messages)
	if !valid {
		logError("[reqID:%s] 消息格式验证失败: %s", reqID, errMsg)
		http.Error(w, errMsg, http.StatusBadRequest)
		return
	}

	// 获取最后一条用户消息和历史记录
	userMessage, history := extractMessages(apiReq.Messages)
	if userMessage == "" {
		logError("[reqID:%s] 未找到有效的用户消息", reqID)
		http.Error(w, "No valid user message found", http.StatusBadRequest)
		return
	}

	// 转发请求到 WoCloud API
	var responseErr error
	if apiReq.Stream {
		responseErr = handleStreamingRequest(w, r, userMessage, history, token, reqID)
	} else {
		responseErr = handleNonStreamingRequest(w, r, userMessage, history, token, reqID)
	}
	
	// 请求处理完成，更新指标
	elapsed := time.Since(startTime).Milliseconds()
	
	// 更新延迟直方图
	bucketIndex := min(int(elapsed/100), 9)
	atomic.AddInt64(&latencyHistogram[bucketIndex], 1)
	
	// 更新平均响应时间
	atomic.AddInt64(&avgResponseTime, elapsed)
	
	if responseErr == nil {
		// 成功计数增加
		atomic.AddInt64(&successCounter, 1)