Method handleAnthropicWithLabels

scaletest/llmmock/server.go:374–435

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

}

func (s *Server) handleAnthropicWithLabels(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	requestID := uuid.New()

	var req llmRequest

	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
		s.logger.Error(ctx, "failed to parse LLM request", slog.Error(err))
		http.Error(w, "invalid request body", http.StatusBadRequest)
		return
	}

	if s.artificialLatency > 0 {
		time.Sleep(s.artificialLatency)
	}

	var resp anthropicResponse
	resp.ID = fmt.Sprintf("msg_%s", requestID.String()[:8])
	resp.Type = "message"
	resp.Role = "assistant"

	var responseText string
	if s.responsePayloadSize > 0 {
		pattern := "x"
		repeated := strings.Repeat(pattern, s.responsePayloadSize)
		responseText = repeated[:s.responsePayloadSize]
	} else {
		responseText = "This is a mock response from Anthropic."
	}

	resp.Content = []struct {
		Type string `json:"type"`
		Text string `json:"text"`
	}{
		{
			Type: "text",
			Text: responseText,
		},
	}
	resp.Model = req.Model
	resp.StopReason = "end_turn"
	resp.Usage.InputTokens = 10
	resp.Usage.OutputTokens = 5

	responseBody, _ := json.Marshal(resp)

	if req.Stream {
		s.sendAnthropicStream(ctx, w, resp)
	} else {
		w.Header().Set("Content-Type", "application/json")
		w.Header().Set("anthropic-version", "2023-06-01")
		w.WriteHeader(http.StatusOK)
		if _, err := w.Write(responseBody); err != nil {
			s.logger.Error(ctx, "failed to write Anthropic response",
				slog.F("request_id", requestID),
				slog.Error(err),
				slog.F("error_type", "write_error"),
				slog.F("likely_cause", "network_error"),