Class: Anthropic::Client::Streaming

Inherits:
Base
  • Object
show all
Defined in:
lib/anthropic/client/streaming.rb

Overview

Provides a client for sending streaming HTTP requests.

Class Method Summary collapse

Class Method Details

.post(url, data, headers = {}) ⇒ Object

rubocop:disable Metrics/AbcSize, Metrics/MethodLength, Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity



9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
 
# File 'lib/anthropic/client/streaming.rb', line 9

def self.post(url, data, headers = {})
  response = HTTPX.plugin(:stream).with(
    headers: {
      'Content-Type' => 'application/json',
      'x-api-key' => Anthropic.api_key,
      'anthropic-version' => Anthropic.api_version
    }.merge(headers)
  ).post(url, json: data, stream: true)

  response.each_line do |line|
    type, event = line.split(/(\w+\b:\s)/)[1..2]
    next unless type&.start_with?('data') && event

    response_data = build_response(event)
    yield response_data unless %w[ping error].include?(response_data.body[:type])
  end
rescue HTTPX::HTTPError => error
  case error.response.status
  when 400
    raise Anthropic::Client::InvalidRequestError
  when 401
    raise Anthropic::Client::AuthenticationError
  when 403
    raise Anthropic::Client::PermissionError
  when 404
    raise Anthropic::Client::NotFoundError
  when 409
    raise Anthropic::Client::ConflictError
  when 422
    raise Anthropic::Client::UnprocessableEntityError
  when 429
    raise Anthropic::Client::RateLimitError
  when 500
    raise Anthropic::Client::ApiError
  when 529
    raise Anthropic::Client::OverloadedError
  end
end