Add support seed

patterns-ai-core · Apr 6, 2024 · 4c217a8 · 4c217a8
1 parent 6015ca2
commit 4c217a8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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