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

sigoden · Dec 10, 2024 · 2e7ff29 · 2e7ff29
1 parent 6dbc328
commit 2e7ff29
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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);