Function writeResponsesAPIStreaming

coderd/x/chatd/chattest/openai.go:409–788

(t testing.TB, w http.ResponseWriter, r *http.Request, resp OpenAIResponse)

}

func writeResponsesAPIStreaming(t testing.TB, w http.ResponseWriter, r *http.Request, resp OpenAIResponse) {
	w.Header().Set("Content-Type", "text/event-stream")
	w.Header().Set("Cache-Control", "no-cache")
	w.Header().Set("Connection", "keep-alive")
	w.WriteHeader(http.StatusOK)

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

	responseID := resp.ResponseID
	if responseID == "" {
		responseID = fmt.Sprintf("resp_%s", uuid.New().String()[:8])
	}
	responseModel := "gpt-4"
	sequenceNumber := int64(0)
	textOffset := 0
	// outputs tracks per-output-index state so the done-event emission
	// at stream close can distinguish message items (text) from
	// function_call items (tool invocation).
	type outputItemState struct {
		itemType  string // "message" or "function_call"
		itemID    string
		text      string // accumulated text for message items
		callID    string // call_id for function_call items
		toolName  string // function name for function_call items
		arguments string // accumulated arguments for function_call items
	}
	outputs := make(map[int]*outputItemState)

	writeEvent := func(eventType string, payload map[string]interface{}) bool {
		payload["type"] = eventType
		payload["sequence_number"] = sequenceNumber
		sequenceNumber++
		if err := writeNamedSSEEvent(w, eventType, payload); err != nil {
			t.Logf("writeResponsesAPIStreaming: failed to write %s: %v", eventType, err)
			return false
		}
		flusher.Flush()
		return true
	}

	if !writeEvent("response.created", map[string]interface{}{
		"response": map[string]interface{}{
			"id":     responseID,
			"object": "response",
			"model":  responseModel,
			"status": "in_progress",
			"output": []interface{}{},
		},
	}) {
		return
	}

	if resp.Reasoning != nil {
		outputIndex := textOffset