diff --git a/src/body.v b/src/body.v index 476971b..7b7175f 100644 --- a/src/body.v +++ b/src/body.v @@ -8,7 +8,7 @@ struct Body { temperature f32 top_p f32 n u8 - stop string + stop ?string presence_penalty f32 frequency_penalty f32 best_of int diff --git a/src/client.v b/src/client.v index 4d4a9ea..5c48e09 100644 --- a/src/client.v +++ b/src/client.v @@ -49,12 +49,13 @@ pub fn (c Client) generate_multiple(prompt string, n u8, config GenerationConfig data: dump(json.encode(Body{ prompt: prompt max_tokens: config.max_tokens - stop: config.stop or { '' } - n: n temperature: config.temperature top_p: config.top_p - frequency_penalty: config.frequency_penalty + n: n + stop: config.stop presence_penalty: config.presence_penalty + frequency_penalty: config.frequency_penalty + best_of: config.best_of })) )! }