From 485d4b76059582d526c7b2e28aa7b782e66066c1 Mon Sep 17 00:00:00 2001
From: Carol Chen <hello@carolchen.me>
Date: Mon, 29 Mar 2021 12:08:37 -0400
Subject: [PATCH] kpppp

---
 cohere/__init__.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/cohere/__init__.py b/cohere/__init__.py
index 744f63b7b..0ec1c7930 100644
--- a/cohere/__init__.py
+++ b/cohere/__init__.py
@@ -15,11 +15,13 @@ def __init__(self, api_key):
     self.api_url = COHERE_API_URL
     self.model = None
 
-  def generate(self, model, prompt, max_tokens=20, temperature=1):
+  def generate(self, model, prompt, max_tokens=20, temperature=1, k=0, p=0.75):
     json_body = json.dumps({
         "prompt": prompt,
         "max_tokens": max_tokens,
         "temperature": temperature,
+        "k": k,
+        "p": p,
       })
     response = self.__request(json_body, GENERATE_URL, model)
     return response["text"]