59 lines
1.6 KiB
Ruby
59 lines
1.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class AnthropicCompletionStubs
|
|
class << self
|
|
def response(content)
|
|
{
|
|
completion: content,
|
|
stop: "\n\nHuman:",
|
|
stop_reason: "stop_sequence",
|
|
truncated: false,
|
|
log_id: "12dcc7feafbee4a394e0de9dffde3ac5",
|
|
model: "claude-2",
|
|
exception: nil,
|
|
}
|
|
end
|
|
|
|
def stub_response(prompt, response_text, req_opts: {})
|
|
WebMock
|
|
.stub_request(:post, "https://api.anthropic.com/v1/complete")
|
|
.with(
|
|
body: { model: "claude-2", prompt: prompt, max_tokens_to_sample: 2000 }.merge(
|
|
req_opts,
|
|
).to_json,
|
|
)
|
|
.to_return(status: 200, body: JSON.dump(response(response_text)))
|
|
end
|
|
|
|
def stream_line(delta, finish_reason: nil)
|
|
+"data: " << {
|
|
completion: delta,
|
|
stop: finish_reason ? "\n\nHuman:" : nil,
|
|
stop_reason: finish_reason,
|
|
truncated: false,
|
|
log_id: "12b029451c6d18094d868bc04ce83f63",
|
|
model: "claude-2",
|
|
exception: nil,
|
|
}.to_json
|
|
end
|
|
|
|
def stub_streamed_response(prompt, deltas, model: nil, req_opts: {})
|
|
chunks =
|
|
deltas.each_with_index.map do |_, index|
|
|
if index == (deltas.length - 1)
|
|
stream_line(deltas[index], finish_reason: "stop_sequence")
|
|
else
|
|
stream_line(deltas[index])
|
|
end
|
|
end
|
|
|
|
chunks = chunks.join("\n\n")
|
|
|
|
WebMock
|
|
.stub_request(:post, "https://api.anthropic.com/v1/complete")
|
|
.with(body: { model: model || "claude-2", prompt: prompt }.merge(req_opts).to_json)
|
|
.to_return(status: 200, body: chunks)
|
|
end
|
|
end
|
|
end
|