Class: Cloudflare::AI::Binding

Inherits:
Object
  • Object
show all
Defined in:
lib/homura/runtime/ai.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(js) ⇒ Binding

Returns a new instance of Binding.



48
49
50
 
# File 'lib/homura/runtime/ai.rb', line 48

def initialize(js)
  @js = js
end

Instance Attribute Details

#js ⇒ Object (readonly)

Returns the value of attribute js.



46
47
48
 
# File 'lib/homura/runtime/ai.rb', line 46

def js
  @js
end

Instance Method Details

#available? ⇒ Boolean

Returns:

  • (Boolean) 


52
53
54
55
 
# File 'lib/homura/runtime/ai.rb', line 52

def available?
  js = @js
  !!`(#{js} !== null && #{js} !== undefined && #{js} !== Opal.nil)`
end

#chat(prompt = nil, messages: nil, system: nil, model: DEFAULT_CHAT_MODEL, options: nil, **input_options) ⇒ Object 



77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
 
# File 'lib/homura/runtime/ai.rb', line 77

def chat(
  prompt = nil,
  messages: nil,
  system: nil,
  model: DEFAULT_CHAT_MODEL,
  options: nil,
  **input_options
)
  chat_options = Cloudflare::AI.chat_input_options(model, input_options)
  payload = {
    messages: Cloudflare::AI.build_messages(
      prompt,
      messages: messages,
      system: system
    )
  }.merge(chat_options)
  run(model, payload, options: options)
end

#chat_text(prompt = nil, messages: nil, system: nil, model: DEFAULT_CHAT_MODEL, options: nil, **input_options) ⇒ Object 



96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
 
# File 'lib/homura/runtime/ai.rb', line 96

def chat_text(
  prompt = nil,
  messages: nil,
  system: nil,
  model: DEFAULT_CHAT_MODEL,
  options: nil,
  **input_options
)
  response = chat(
    prompt,
    messages: messages,
    system: system,
    model: model,
    options: options,
    **input_options
  )
  response = response.__await__ if Cloudflare.js_promise?(response)
  Cloudflare::AI.extract_text(response)
end

#run(model, inputs = nil, options: nil, **input_options) ⇒ Object 



57
58
59
60
61
62
63
64
65
 
# File 'lib/homura/runtime/ai.rb', line 57

def run(model, inputs = nil, options: nil, **input_options)
  payload = inputs || input_options
  if inputs.is_a?(Hash) &&
      !input_options.empty?
    payload = payload.merge(input_options)
  end

  Cloudflare::AI.run(model, payload, binding: @js, options: options)
end

#run_stream(model, inputs = nil, **input_options) ⇒ Object 



67
68
69
70
71
72
73
74
75
 
# File 'lib/homura/runtime/ai.rb', line 67

def run_stream(model, inputs = nil, **input_options)
  payload = inputs || input_options
  if inputs.is_a?(Hash) &&
      !input_options.empty?
    payload = payload.merge(input_options)
  end

  run(model, payload.merge(stream: true))
end

#speak(text, model: DEFAULT_SPEAK_MODEL, options: nil, **input_options) ⇒ Object 



139
140
141
142
143
144
 
# File 'lib/homura/runtime/ai.rb', line 139

def speak(text, model: DEFAULT_SPEAK_MODEL, options: nil, **input_options)
  payload = {text: text.to_s}.merge(input_options)
  response = Cloudflare::AI.speak(model, payload, binding: @js, options: options)
  response = response.__await__ if Cloudflare.js_promise?(response)
  response
end

#speak_data_url(text, model: DEFAULT_SPEAK_MODEL, options: nil, **input_options) ⇒ Object 



146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
 
# File 'lib/homura/runtime/ai.rb', line 146

def speak_data_url(
  text,
  model: DEFAULT_SPEAK_MODEL,
  options: nil,
  **input_options
)
  payload = {text: text.to_s}.merge(input_options)
  response = Cloudflare::AI.speak_data_url(
    model,
    payload,
    binding: @js,
    options: options
  )
  response = response.__await__ if Cloudflare.js_promise?(response)
  response.to_s
end

#transcribe(audio, model: DEFAULT_TRANSCRIBE_MODEL, options: nil, **input_options) ⇒ Object 



116
117
118
119
120
121
122
123
124
125
126
 
# File 'lib/homura/runtime/ai.rb', line 116

def transcribe(
  audio,
  model: DEFAULT_TRANSCRIBE_MODEL,
  options: nil,
  **input_options
)
  payload = {audio: Cloudflare::AI.audio_input(audio)}.merge(
    input_options
  )
  run(model, payload, options: options)
end

#transcribe_text(audio, model: DEFAULT_TRANSCRIBE_MODEL, options: nil, **input_options) ⇒ Object 



128
129
130
131
132
133
134
135
136
137
 
# File 'lib/homura/runtime/ai.rb', line 128

def transcribe_text(
  audio,
  model: DEFAULT_TRANSCRIBE_MODEL,
  options: nil,
  **input_options
)
  response = transcribe(audio, model: model, options: options, **input_options)
  response = response.__await__ if Cloudflare.js_promise?(response)
  Cloudflare::AI.extract_text(response)
end