module RubyLLM::Providers::OpenAI::Streaming

# File lib/ruby_llm/providers/openai/streaming.rb, line 16
def build_chunk(data)
  usage = data['usage'] || {}
  delta = data.dig('choices', 0, 'delta') || {}
  content_source = delta['content'] || data.dig('choices', 0, 'message', 'content')
  content, thinking_from_blocks = OpenAI::Chat.extract_content_and_thinking(content_source)

  Chunk.new(
    role: :assistant,
    model_id: data['model'],
    content: content,
    thinking: Thinking.build(
      text: thinking_from_blocks || delta['reasoning_content'] || delta['reasoning'],
      signature: delta['reasoning_signature']
    ),
    tool_calls: parse_tool_calls(delta['tool_calls'], parse_arguments: false),
    input_tokens: OpenAI::Chat.input_tokens(usage),
    output_tokens: OpenAI::Chat.output_tokens(usage),
    cached_tokens: OpenAI::Chat.cache_read_tokens(usage),
    cache_creation_tokens: OpenAI::Chat.cache_write_tokens(usage),
    thinking_tokens: OpenAI::Chat.thinking_tokens(usage)
  )
end

# File lib/ruby_llm/providers/openai/streaming.rb, line 39
def parse_streaming_error(data)
  error_data = JSON.parse(data)
  return unless error_data['error']

  case error_data.dig('error', 'type')
  when 'server_error'
    [500, error_data['error']['message']]
  when 'rate_limit_exceeded', 'insufficient_quota'
    [429, error_data['error']['message']]
  else
    [400, error_data['error']['message']]
  end
end

# File lib/ruby_llm/providers/openai/streaming.rb, line 12
def stream_url
  completion_url
end