Method: Cohere::Client#generate

Defined in:
lib/cohere/client.rb

#generate(prompt:, model: nil, num_generations: nil, max_tokens: nil, preset: nil, temperature: nil, k: nil, p: nil, frequency_penalty: nil, presence_penalty: nil, end_sequences: nil, stop_sequences: nil, return_likelihoods: nil, logit_bias: nil, truncate: nil) ⇒ Object

This endpoint generates realistic text conditioned on a given input.



64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
 
# File 'lib/cohere/client.rb', line 64

def generate(
  prompt:,
  model: nil,
  num_generations: nil,
  max_tokens: nil,
  preset: nil,
  temperature: nil,
  k: nil,
  p: nil,
  frequency_penalty: nil,
  presence_penalty: nil,
  end_sequences: nil,
  stop_sequences: nil,
  return_likelihoods: nil,
  logit_bias: nil,
  truncate: nil
)
  response = v1_connection.post("generate") do |req|
    req.body = {prompt: prompt}
    req.body[:model] = model if model
    req.body[:num_generations] = num_generations if num_generations
    req.body[:max_tokens] = max_tokens if max_tokens
    req.body[:preset] = preset if preset
    req.body[:temperature] = temperature if temperature
    req.body[:k] = k if k
    req.body[:p] = p if p
    req.body[:frequency_penalty] = frequency_penalty if frequency_penalty
    req.body[:presence_penalty] = presence_penalty if presence_penalty
    req.body[:end_sequences] = end_sequences if end_sequences
    req.body[:stop_sequences] = stop_sequences if stop_sequences
    req.body[:return_likelihoods] = return_likelihoods if return_likelihoods
    req.body[:logit_bias] = logit_bias if logit_bias
    req.body[:truncate] = truncate if truncate
  end
  response.body
end