. test - anthropic - add test_chat_stop_sequences_ok

src/chat/chat_options.rs

@@ -75,6 +75,11 @@ impl ChatOptions {
 		self
 	}
 
+	pub fn with_stop_sequences(mut self, values: Vec<String>) -> Self {
+		self.stop_sequences = values;
+		self
+	}
+
 	/// Set the `json_mode` for this request.
 	///
 	/// IMPORTANT: This is deprecated now; use `with_response_format(ChatResponseFormat::JsonMode)`

tests/support/common_tests.rs

@@ -167,6 +167,29 @@ pub async fn common_test_chat_temperature_ok(model: &str) -> Result<()> {
 	Ok(())
 }
 
+pub async fn common_test_chat_stop_sequences_ok(model: &str) -> Result<()> {
+	// -- Setup & Fixtures
+	let client = Client::default();
+	let chat_req = ChatRequest::from_user("What is the capital of England?");
+	let chat_options = ChatOptions::default().with_stop_sequences(vec!["London".to_string()]);
+
+	// -- Exec
+	let chat_res = client.exec_chat(model, chat_req, Some(&chat_options)).await?;
+
+	let ai_content_lower = chat_res
+		.content_text_as_str()
+		.ok_or("Should have a AI response")?
+		.to_lowercase();
+
+	// -- Check
+	assert!(!ai_content_lower.is_empty(), "Content should not be empty");
+	assert!(
+		!ai_content_lower.contains("london"),
+		"Content should not contain 'London'"
+	);
+
+	Ok(())
+}
 // endregion: --- Chat
 
 // region:    --- Chat Stream Tests

tests/tests_p_anthropic.rs

@@ -6,7 +6,7 @@ use serial_test::serial;
 
 type Result<T> = core::result::Result<T, Box<dyn std::error::Error>>; // For tests.
 
-// 4k
+// 4k (cheaper)
 const MODEL: &str = "claude-3-haiku-20240307";
 // 8k output context
 // const MODEL: &str = "claude-3-5-haiku-20241022";
@@ -14,19 +14,25 @@ const MODEL: &str = "claude-3-haiku-20240307";
 // region:    --- Chat
 
 #[tokio::test]
-// #[serial(anthropic)]
+#[serial(anthropic)]
 async fn test_chat_simple_ok() -> Result<()> {
 	common_tests::common_test_chat_simple_ok(MODEL).await
 }
 
 #[tokio::test]
-// #[serial(anthropic)]
+#[serial(anthropic)]
 async fn test_chat_temperature_ok() -> Result<()> {
 	common_tests::common_test_chat_temperature_ok(MODEL).await
 }
 
 #[tokio::test]
-// #[serial(anthropic)]
+#[serial(anthropic)]
+async fn test_chat_stop_sequences_ok() -> Result<()> {
+	common_tests::common_test_chat_stop_sequences_ok(MODEL).await
+}
+
+#[tokio::test]
+#[serial(anthropic)]
 async fn test_chat_json_mode_ok() -> Result<()> {
 	common_tests::common_test_chat_json_mode_ok(MODEL, true).await
 }
@@ -52,7 +58,6 @@ async fn test_chat_stream_capture_content_ok() -> Result<()> {
 async fn test_chat_stream_capture_all_ok() -> Result<()> {
 	common_tests::common_test_chat_stream_capture_all_ok(MODEL).await
 }
-
 // endregion: --- Chat Stream Tests
 
 // region:    --- Tool Tests