+ cohere - add stop_sequences support

This commit is contained in:
Jeremy Chone
2024-12-07 13:52:34 -08:00
parent 115a6a3c53
commit 4eed7c3216
2 changed files with 8 additions and 0 deletions

View File

@ -84,6 +84,9 @@ impl Adapter for CohereAdapter {
if let Some(temperature) = options_set.temperature() {
payload.x_insert("temperature", temperature)?;
}
if !options_set.stop_sequences().is_empty() {
payload.x_insert("stop_sequences", options_set.stop_sequences())?;
}
if let Some(max_tokens) = options_set.max_tokens() {
payload.x_insert("max_tokens", max_tokens)?;
}

View File

@ -14,6 +14,11 @@ async fn test_chat_simple_ok() -> Result<()> {
common_tests::common_test_chat_simple_ok(MODEL).await
}
#[tokio::test]
async fn test_chat_stop_sequences_ok() -> Result<()> {
common_tests::common_test_chat_stop_sequences_ok(MODEL).await
}
// endregion: --- Chat
// region: --- Chat Stream Tests