refactor: body data for cohere v2 chat api (#1048)

1 files changed, 6 insertions, 1 deletions

diff --git a/src/client/cohere.rs b/src/client/cohere.rs
index 457ea66..5f61454 100644
--- a/src/client/cohere.rs
+++ b/src/client/cohere.rs
@@ -49,7 +49,12 @@ fn prepare_chat_completions(
         .unwrap_or_else(|_| API_BASE.to_string());
 
     let url = format!("{}/chat", api_base.trim_end_matches('/'));
-    let body = openai_build_chat_completions_body(data, &self_.model);
+    let mut body = openai_build_chat_completions_body(data, &self_.model);
+    if let Some(obj) = body.as_object_mut() {
+        if let Some(top_p) = obj.remove("top_p") {
+            obj.insert("p".to_string(), top_p);
+        }
+    }
 
     let mut request_data = RequestData::new(url, body);