From 83fcb104a27277ed1ce917d7bf8f4fd57fa32e95 Mon Sep 17 00:00:00 2001 From: Katon Minhas Date: Tue, 2 Dec 2025 17:34:15 +0000 Subject: [PATCH] Merged in feature/deprecate-haiku-3 (pull request #793) Feature/deprecate haiku 3 * Update config - remove Haiku 3 * Update llm_utils * Merged main into feature/deprecate-haiku-3 Approved-by: Siddhant Medar --- fieldExtraction/src/config.py | 7 ++----- fieldExtraction/src/utils/llm_utils.py | 2 +- fieldExtraction/tests/test_llm_utils.py | 4 ++-- 3 files changed, 5 insertions(+), 8 deletions(-) diff --git a/fieldExtraction/src/config.py b/fieldExtraction/src/config.py index ccd2b93..0ec494f 100644 --- a/fieldExtraction/src/config.py +++ b/fieldExtraction/src/config.py @@ -230,20 +230,17 @@ if WRITE_TO_S3: # Claude model IDs # Note: Claude 2 support has been removed - only Claude 3+ models are supported -MODEL_ID_CLAUDE3_HAIKU = "anthropic.claude-3-haiku-20240307-v1:0" MODEL_ID_CLAUDE35_SONNET = "anthropic.claude-3-5-sonnet-20240620-v1:0" MODEL_ID_CLAUDE35_SONNET_V2 = "anthropic.claude-3-5-sonnet-20241022-v2:0" MODEL_ID_CLAUDE37_SONNET = "us.anthropic.claude-3-7-sonnet-20250219-v1:0" MODEL_ID_CLAUDE4_SONNET = "us.anthropic.claude-sonnet-4-20250514-v1:0" MODEL_ID_CLAUDE45_SONNET = "us.anthropic.claude-sonnet-4-5-20250929-v1:0" +MODEL_ID_CLAUDE45_HAIKU = "us.anthropic.claude-haiku-4-5-20251001-v1:0" # Model aliases for easy upgrades MODEL_ALIASES = { - "sonnet_37" : MODEL_ID_CLAUDE37_SONNET, - "sonnet_4" : MODEL_ID_CLAUDE4_SONNET, - "sonnet_45": MODEL_ID_CLAUDE45_SONNET, "sonnet_latest": MODEL_ID_CLAUDE4_SONNET, - "haiku_latest": MODEL_ID_CLAUDE3_HAIKU, + "haiku_latest": MODEL_ID_CLAUDE45_HAIKU, "legacy_sonnet": MODEL_ID_CLAUDE35_SONNET, } diff --git a/fieldExtraction/src/utils/llm_utils.py b/fieldExtraction/src/utils/llm_utils.py index 3c7eb63..ec5b0a5 100644 --- a/fieldExtraction/src/utils/llm_utils.py +++ b/fieldExtraction/src/utils/llm_utils.py @@ -29,7 +29,7 @@ _cache_warming_lock = threading.Lock() # Supported models (from config.py) # Add new models here when they're added to config.py _SUPPORTED_MODELS = { - config.MODEL_ID_CLAUDE3_HAIKU, + config.MODEL_ID_CLAUDE45_HAIKU, config.MODEL_ID_CLAUDE35_SONNET, config.MODEL_ID_CLAUDE37_SONNET, config.MODEL_ID_CLAUDE4_SONNET, diff --git a/fieldExtraction/tests/test_llm_utils.py b/fieldExtraction/tests/test_llm_utils.py index 478f8e0..b789e0f 100644 --- a/fieldExtraction/tests/test_llm_utils.py +++ b/fieldExtraction/tests/test_llm_utils.py @@ -307,8 +307,8 @@ class TestLLMUtils(unittest.TestCase): # Should not raise an exception try: llm_utils._validate_model_support( - config.MODEL_ID_CLAUDE3_HAIKU, - config.MODEL_ID_CLAUDE3_HAIKU + config.MODEL_ID_CLAUDE45_HAIKU, + config.MODEL_ID_CLAUDE45_HAIKU ) except ValueError: self.fail("_validate_model_support raised ValueError for supported Claude 3 Haiku model")