From 4c217a82a76697c367f8b435f499a5bc44a99940 Mon Sep 17 00:00:00 2001 From: naoa Date: Sat, 6 Apr 2024 10:05:06 +0900 Subject: [PATCH] Add support seed --- lib/cohere/client.rb | 2 ++ 1 file changed, 2 insertions(+) diff --git a/lib/cohere/client.rb b/lib/cohere/client.rb index a21083f..03a8624 100644 --- a/lib/cohere/client.rb +++ b/lib/cohere/client.rb @@ -29,6 +29,7 @@ def chat( max_tokens: nil, k: nil, p: nil, + seed: nil, frequency_penalty: nil, presence_penalty: nil, tools: [], @@ -54,6 +55,7 @@ def chat( req.body[:max_tokens] = max_tokens if max_tokens req.body[:k] = k if k req.body[:p] = p if p + req.body[:seed] = seed if seed req.body[:frequency_penalty] = frequency_penalty if frequency_penalty req.body[:presence_penalty] = presence_penalty if presence_penalty req.body[:tools] = tools if tools