Sha256: b8b00e236bc1c83c0da82e559e034b6077ebd893c63a5909e09a5887dddb1621
Contents?: true
Size: 426 Bytes
Versions: 11
Compression:
Stored size: 426 Bytes
Contents
# lib/ai_client/embed.rb
class AiClient
######################################
# OmniAI Params
# model [String] required
#
def embed(input, **params)
@client.embed(input, model: @model, **params)
end
def batch_embed(inputs, batch_size: 100, **params)
inputs.each_slice(batch_size).flat_map do |batch|
sleep 1 # DEBUG rate limits being exceeded
embed(batch, **params)
end
end
end
Version data entries
11 entries across 11 versions & 1 rubygems