jacob-mathison/doczyai-pipelines
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Activity

Merged in feature/deprecate-haiku-3 (pull request #793)

Browse Source 
Feature/deprecate haiku 3

* Update config - remove Haiku 3

* Update llm_utils

* Merged main into feature/deprecate-haiku-3


Approved-by: Siddhant Medar
This commit is contained in:
Katon Minhas
2025-12-02 17:34:15 +00:00
parent 407b4bd55e
commit 83fcb104a2
3 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fieldExtraction/src/config.py
+2 -5
View File
@@ -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,
}
fieldExtraction/src/utils/llm_utils.py
+1 -1
View File
@@ -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,
fieldExtraction/tests/test_llm_utils.py
+2 -2
View File
@@ -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")