diff --git a/tests/configs/llm_model_configs/claude-4-sonnet.json b/tests/configs/llm_model_configs/claude-4-sonnet.json new file mode 100644 index 00000000..12a582e0 --- /dev/null +++ b/tests/configs/llm_model_configs/claude-4-sonnet.json @@ -0,0 +1,8 @@ +{ + "model": "claude-sonnet-4-20250514", + "model_endpoint_type": "anthropic", + "model_endpoint": "https://api.anthropic.com/v1", + "model_wrapper": null, + "context_window": 200000, + "put_inner_thoughts_in_kwargs": true +} diff --git a/tests/configs/llm_model_configs/openai-o4-mini.json b/tests/configs/llm_model_configs/openai-o4-mini.json new file mode 100644 index 00000000..7579f56d --- /dev/null +++ b/tests/configs/llm_model_configs/openai-o4-mini.json @@ -0,0 +1,8 @@ +{ + "context_window": 128000, + "model": "o4-mini", + "model_endpoint_type": "openai", + "model_endpoint": "https://api.openai.com/v1", + "model_wrapper": null, + "temperature": 1.0 +} diff --git a/tests/integration_test_send_message.py b/tests/integration_test_send_message.py index c331f92a..a469ca2d 100644 --- a/tests/integration_test_send_message.py +++ b/tests/integration_test_send_message.py @@ -101,8 +101,9 @@ all_configs = [ "openai-o1.json", "openai-o1-mini.json", "openai-o3.json", - "openai-o3-mini.json", + "openai-o4-mini.json", "azure-gpt-4o-mini.json", + "claude-4-sonnet.json", "claude-3-5-sonnet.json", "claude-3-7-sonnet.json", "claude-3-7-sonnet-extended.json",