From 40629285fcd454fad313a27faea8936751e1c96a Mon Sep 17 00:00:00 2001 From: Matthew Zhou Date: Wed, 18 Jun 2025 17:23:52 -0700 Subject: [PATCH] feat: Add claude 4 config and add more models to CI (#2747) Co-authored-by: Charles Packer --- tests/configs/llm_model_configs/claude-4-sonnet.json | 8 ++++++++ tests/configs/llm_model_configs/openai-o4-mini.json | 8 ++++++++ tests/integration_test_send_message.py | 3 ++- 3 files changed, 18 insertions(+), 1 deletion(-) create mode 100644 tests/configs/llm_model_configs/claude-4-sonnet.json create mode 100644 tests/configs/llm_model_configs/openai-o4-mini.json 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",