Class: Charai::OllamaConfiguration

Inherits:
Object
  • Object
show all
Defined in:
lib/charai/openai_configuration.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(endpoint_url:, model:) ⇒ OllamaConfiguration

Returns a new instance of OllamaConfiguration.



61
62
63
64
 
# File 'lib/charai/openai_configuration.rb', line 61

def initialize(endpoint_url:, model:)
  @endpoint_url = endpoint_url
  @model = model
end

Instance Attribute Details

#endpoint_url ⇒ Object (readonly)

Returns the value of attribute endpoint_url.



66
67
68
 
# File 'lib/charai/openai_configuration.rb', line 66

def endpoint_url
  @endpoint_url
end

Instance Method Details

#add_auth_header(headers) ⇒ Object 



68
69
70
71
 
# File 'lib/charai/openai_configuration.rb', line 68

def add_auth_header(headers)
  # auth header is not required.
  headers
end

#decorate_body(payload) ⇒ Object 



73
74
75
76
 
# File 'lib/charai/openai_configuration.rb', line 73

def decorate_body(payload)
  payload[:model] = @model
  payload
end