Method writeRaw

go/copilot_request_handler.go:773–800

(data string, binary bool)

}

func (s *responseSink) writeRaw(data string, binary bool) error {
	s.exchange.mu.Lock()
	started := s.exchange.started
	finished := s.exchange.finished
	s.exchange.mu.Unlock()
	if !started {
		return fmt.Errorf("CopilotRequestHandler response sink Write() called before Start()")
	}
	if finished {
		return fmt.Errorf("CopilotRequestHandler response sink Write() called after End()/Error()")
	}
	api, err := s.rpcAPI()
	if err != nil {
		return err
	}
	end := false
	chunk := &rpc.LlmInferenceHTTPResponseChunkRequest{
		RequestID: s.requestID,
		Data:      data,
		End:       &end,
	}
	if binary {
		b := true
		chunk.Binary = &b
	}
	_, err = api.HttpResponseChunk(context.Background(), chunk)
	return err
}

func (s *responseSink) end() error {
	s.exchange.mu.Lock()