pass through new generate params (#82)

* pass through new generate params

* multi turn

* update version

.gitignore

@@ -6,6 +6,7 @@ __pycache__/
 *.py[cod]
 *$py.class
 .idea
+*.ipynb
 
 # C extensions
 *.so

arcee/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "1.3.6"
+__version__ = "1.3.7"
 
 import os
 

arcee/api.py

@@ -472,8 +472,26 @@ def deployment_status(deployment_name: str) -> Dict[str, str]:
     return make_request("get", Route.deployment + "/status", data)
 
 
-def generate(deployment_name: str, query: str) -> Dict[str, str]:
-    data = {"deployment_name": deployment_name, "query": query}
+def generate(
+    deployment_name: str,
+    query: str | None = None,
+    messages: List[Dict[str, str]] | None = None,
+    repetition_penalty: float | None = None,
+    top_k: int | None = None,
+    max_new_tokens: int | None = None,
+    temperature: float | None = None,
+    top_p: float | None = None,
+) -> Dict[str, str]:
+    data = {
+        "deployment_name": deployment_name,
+        "query": query,
+        "repetition_penalty": repetition_penalty,
+        "top_k": top_k,
+        "max_new_tokens": max_new_tokens,
+        "temperature": temperature,
+        "messages": messages,
+        "top_p": top_p,
+    }
     return make_request("post", Route.deployment + "/generate", data)