Merge branch 'andreibondarev:main' into timeout

patterns-ai-core · Apr 6, 2024 · 3a11e5e · 3a11e5e
2 parents 2890784 + f6ddad2
commit 3a11e5e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/cohere/client.rb b/lib/cohere/client.rb
@@ -30,6 +30,7 @@ def chat(
       max_tokens: nil,
       k: nil,
       p: nil,
+      seed: nil,
       frequency_penalty: nil,
       presence_penalty: nil,
       tools: [],
@@ -55,6 +56,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