Class: Ollama::Transport::NetHTTP

Inherits:
Base
  • Object
show all
Defined in:
lib/ollama/transport/net_http.rb

Overview

Net::HTTP-backed transport adapter.

Instance Method Summary collapse

Methods inherited from Base

#call, #initialize

Constructor Details

This class inherits a constructor from Ollama::Transport::Base

Instance Method Details

#capabilities ⇒ Object 



53
54
55
 
# File 'lib/ollama/transport/net_http.rb', line 53

def capabilities
  %i[request stream]
end

#request(uri:, request:, read_timeout:) ⇒ Object 



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
 
# File 'lib/ollama/transport/net_http.rb', line 7

def request(uri:, request:, read_timeout:)
  started = Process.clock_gettime(Process::CLOCK_MONOTONIC)
  req = build_raw_request(request)
  raw_response = Net::HTTP.start(
    uri.hostname,
    uri.port,
    **@config.http_connection_options(uri, read_timeout: read_timeout)
  ) { |http| http.request(req) }
  ended = Process.clock_gettime(Process::CLOCK_MONOTONIC)

  Response.new(
    status: raw_response.code,
    headers: raw_response.to_hash,
    body: raw_response.body,
    raw: raw_response,
    duration_ms: ((ended - started) * 1000.0).round(2)
  )
end

#stream(uri:, request:, read_timeout: nil, &block) ⇒ Object 



26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
 
# File 'lib/ollama/transport/net_http.rb', line 26

def stream(uri:, request:, read_timeout: nil, &block)
  started = Process.clock_gettime(Process::CLOCK_MONOTONIC)
  raw_response = nil
  req = build_raw_request(request)
  Net::HTTP.start(
    uri.hostname,
    uri.port,
    **@config.http_connection_options(uri, read_timeout: read_timeout)
  ) do |http|
    http.request(req) do |res|
      raw_response = res
      raise HTTPError, res unless res.is_a?(Net::HTTPSuccess)

      res.read_body(&block)
    end
  end
  ended = Process.clock_gettime(Process::CLOCK_MONOTONIC)

  Response.new(
    status: raw_response.code,
    headers: raw_response.to_hash,
    body: nil,
    raw: raw_response,
    duration_ms: ((ended - started) * 1000.0).round(2)
  )
end