refactor: update model references and remove deprecated models

notdiamond/llms/providers.py

@@ -31,7 +31,6 @@ class NDLLMProviders(Enum):
         GPT_4_1_NANO_2025_04_14 (NDLLMProvider): refers to 'gpt-4.1-nano-2025-04-14' model by OpenAI
         GPT_4o_2024_11_20 (NDLLMProvider): refers to 'gpt-4o-2024-11-20' model by OpenAI
 
-        CLAUDE_2_1 (NDLLMProvider): refers to 'claude-2.1' model by Anthropic
         CLAUDE_3_OPUS_20240229 (NDLLMProvider): refers to 'claude-3-opus-20240229' model by Anthropic
         CLAUDE_3_7_SONNET_LATEST (NDLLMProvider): refers to 'claude-3-7-sonnet-latest' model by Anthropic
         CLAUDE_3_7_SONNET_20250219 (NDLLMProvider): refers to 'claude-3-7-sonnet-20250219' model by Anthropic
@@ -48,9 +47,6 @@ class NDLLMProviders(Enum):
         CLAUDE_OPUS_4_1 (NDLLMProvider): refers to 'claude-opus-4-1' model by Anthropic
         CLAUDE_SONNET_4_0 (NDLLMProvider): refers to 'claude-sonnet-4-0' model by Anthropic
 
-        GEMINI_PRO (NDLLMProvider): refers to 'gemini-pro' model by Google
-        GEMINI_1_PRO_LATEST (NDLLMProvider): refers to 'gemini-1.0-pro-latest' model by Google
-        GEMINI_15_PRO_LATEST (NDLLMProvider): refers to 'gemini-1.5-pro-latest' model by Google
         GEMINI_15_PRO_EXP_0801 (NDLLMProvider): refers to 'gemini-1.5-pro-exp-0801' model by Google
         GEMINI_15_FLASH_LATEST (NDLLMProvider): refers to 'gemini-1.5-flash-latest' model by Google
         GEMINI_20_FLASH (NDLLMProvider): refers to 'gemini-20-flash' model by Google
@@ -119,7 +115,6 @@ class NDLLMProviders(Enum):
     CHATGPT_4o_LATEST = ("openai", "chatgpt-4o-latest")
     GPT_4o_2024_11_20 = ("openai", "gpt-4o-2024-11-20")
 
-    CLAUDE_2_1 = ("anthropic", "claude-2.1")
     CLAUDE_3_OPUS_20240229 = ("anthropic", "claude-3-opus-20240229")
     CLAUDE_3_7_SONNET_LATEST = ("anthropic", "claude-3-7-sonnet-latest")
     CLAUDE_3_7_SONNET_20250219 = ("anthropic", "claude-3-7-sonnet-20250219")
@@ -136,9 +131,6 @@ class NDLLMProviders(Enum):
     CLAUDE_OPUS_4_1 = ("anthropic", "claude-opus-4-1")
     CLAUDE_SONNET_4_0 = ("anthropic", "claude-sonnet-4-0")
 
-    GEMINI_PRO = ("google", "gemini-pro")
-    GEMINI_1_PRO_LATEST = ("google", "gemini-1.0-pro-latest")
-    GEMINI_15_PRO_LATEST = ("google", "gemini-1.5-pro-latest")
     GEMINI_15_PRO_EXP_0801 = ("google", "gemini-1.5-pro-exp-0801")
     GEMINI_15_FLASH_LATEST = ("google", "gemini-1.5-flash-latest")
     GEMINI_20_FLASH = ("google", "gemini-2.0-flash")

notdiamond/settings.py

@@ -143,7 +143,6 @@
     },
     "anthropic": {
         "models": [
-            "claude-2.1",
             "claude-3-opus-20240229",
             "claude-3-haiku-20240307",
             "claude-3-5-haiku-20241022",
@@ -179,11 +178,10 @@
             "claude-sonnet-4-0",
         ],
         "support_response_model": [
-            "claude-2.1",
             "claude-3-opus-20240229",
+            "claude-sonnet-4-5-20250929",
         ],
         "openrouter_identifier": {
-            "claude-2.1": "anthropic/claude-2.1",
             "claude-3-opus-20240229": "anthropic/claude-3-opus",
             "claude-3-haiku-20240307": "anthropic/claude-3-haiku",
             "claude-3-5-haiku-20241022": "anthropic/claude-3.5-haiku",
@@ -201,7 +199,6 @@
             "claude-sonnet-4-0": "anthropic/claude-sonnet-4",
         },
         "price": {
-            "claude-2.1": {"input": 8.0, "output": 24.0},
             "claude-3-opus-20240229": {"input": 15.0, "output": 75.0},
             "claude-3-haiku-20240307": {"input": 0.25, "output": 1.25},
             "claude-3-5-haiku-20241022": {"input": 1.0, "output": 5.0},
@@ -221,9 +218,6 @@
     },
     "google": {
         "models": [
-            "gemini-pro",
-            "gemini-1.0-pro-latest",
-            "gemini-1.5-pro-latest",
             "gemini-1.5-pro-exp-0801",
             "gemini-1.5-flash-latest",
             "gemini-2.0-flash",
@@ -233,18 +227,12 @@
         ],
         "api_key": GOOGLE_API_KEY,
         "support_tools": [
-            "gemini-pro",
-            "gemini-1.0-pro-latest",
-            "gemini-1.5-pro-latest",
             "gemini-1.5-pro-exp-0801",
             "gemini-1.5-flash-latest",
             "gemini-2.5-flash",
             "gemini-2.5-pro",
         ],
         "support_response_model": [
-            "gemini-pro",
-            "gemini-1.0-pro-latest",
-            "gemini-1.5-pro-latest",
             "gemini-1.5-pro-exp-0801",
             "gemini-1.5-flash-latest",
             "gemini-2.0-flash",
@@ -253,9 +241,6 @@
             "gemini-2.5-pro",
         ],
         "openrouter_identifier": {
-            "gemini-pro": "google/gemini-pro",
-            "gemini-1.0-pro-latest": "google/gemini-pro",
-            "gemini-1.5-pro-latest": "google/gemini-pro-1.5",  #
             "gemini-1.5-pro-exp-0801": "google/gemini-pro-1.5-exp",  #
             "gemini-1.5-flash-latest": "google/gemini-flash-1.5",  #
             "gemini-2.0-flash": "google/gemini-2.0-flash",  #
@@ -264,9 +249,6 @@
             "gemini-2.5-pro": "google/gemini-2.5-pro",  #
         },
         "price": {
-            "gemini-pro": {"input": 0.5, "output": 1.5},
-            "gemini-1.0-pro-latest": {"input": 0.5, "output": 1.5},
-            "gemini-1.5-pro-latest": {"input": 1.75, "output": 10.5},
             "gemini-1.5-pro-exp-0801": {"input": 1.75, "output": 10.5},
             "gemini-1.5-flash-latest": {"input": 0.35, "output": 1.05},
             "gemini-2.0-flash": {"input": 0.10, "output": 0.40},

tests/test_components/test_llms/cassettes/test_llm/Test_NDLLM.test_async_model_select_with_strings[_NDClientTarget.INVOKER-_NDInvokerClient].yaml

@@ -5,30 +5,30 @@ interactions:
       "is_custom": false, "context_length": null, "input_price": null, "output_price":
       null, "latency": null}, {"provider": "openai", "model": "gpt-4", "is_custom":
       false, "context_length": null, "input_price": null, "output_price": null, "latency":
-      null}, {"provider": "anthropic", "model": "claude-2.1", "is_custom": false,
+      null}, {"provider": "anthropic", "model": "claude-haiku-4-5-20251001", "is_custom":
+      false, "context_length": null, "input_price": null, "output_price": null, "latency":
+      null}, {"provider": "google", "model": "gemini-2.5-pro", "is_custom": false,
       "context_length": null, "input_price": null, "output_price": null, "latency":
-      null}, {"provider": "google", "model": "gemini-pro", "is_custom": false, "context_length":
-      null, "input_price": null, "output_price": null, "latency": null}], "metric":
-      "accuracy", "max_model_depth": 4, "hash_content": true}'
+      null}], "metric": "accuracy", "max_model_depth": 4, "hash_content": true}'
     headers:
       User-Agent:
-      - Python-SDK/0.3.41
+      - Python-SDK/0.4.6
       content-type:
       - application/json
     method: POST
     uri: https://staging-api.notdiamond.ai/v2/modelRouter/modelSelect
   response:
     body:
-      string: '{"providers":[{"provider":"openai","model":"gpt-4"}],"session_id":"7a3032f4-cf00-45d3-9f45-f1f61830abbf"}'
+      string: '{"providers":[{"provider":"google","model":"gemini-2.5-pro"}],"session_id":"a357cc02-30ae-40b2-98c9-cb717b224317"}'
     headers:
       CF-RAY:
-      - 925cfae498ddc747-VIE
+      - 99a4641bcb12e461-OTP
       Connection:
       - keep-alive
       Content-Type:
       - application/json
       Date:
-      - Tue, 25 Mar 2025 08:18:52 GMT
+      - Thu, 06 Nov 2025 11:53:34 GMT
       Server:
       - cloudflare
       Transfer-Encoding:
@@ -40,11 +40,13 @@ interactions:
       cf-cache-status:
       - DYNAMIC
       content-length:
-      - '105'
+      - '114'
       rndr-id:
-      - cd53e938-6152-4e22
+      - 84112929-6f0d-446d
       x-render-origin-server:
       - uvicorn
+      x-request-id:
+      - 84112929-6f0d-446d
     status:
       code: 200
       message: OK

tests/test_components/test_llms/cassettes/test_llm/Test_NDLLM.test_async_model_select_with_strings[_NDClientTarget.ROUTER-_NDRouterClient].yaml

@@ -5,30 +5,30 @@ interactions:
       "is_custom": false, "context_length": null, "input_price": null, "output_price":
       null, "latency": null}, {"provider": "openai", "model": "gpt-4", "is_custom":
       false, "context_length": null, "input_price": null, "output_price": null, "latency":
-      null}, {"provider": "anthropic", "model": "claude-2.1", "is_custom": false,
+      null}, {"provider": "anthropic", "model": "claude-haiku-4-5-20251001", "is_custom":
+      false, "context_length": null, "input_price": null, "output_price": null, "latency":
+      null}, {"provider": "google", "model": "gemini-2.5-pro", "is_custom": false,
       "context_length": null, "input_price": null, "output_price": null, "latency":
-      null}, {"provider": "google", "model": "gemini-pro", "is_custom": false, "context_length":
-      null, "input_price": null, "output_price": null, "latency": null}], "metric":
-      "accuracy", "max_model_depth": 4, "hash_content": true}'
+      null}], "metric": "accuracy", "max_model_depth": 4, "hash_content": true}'
     headers:
       User-Agent:
-      - Python-SDK/0.3.41
+      - Python-SDK/0.4.6
       content-type:
       - application/json
     method: POST
     uri: https://staging-api.notdiamond.ai/v2/modelRouter/modelSelect
   response:
     body:
-      string: '{"providers":[{"provider":"openai","model":"gpt-4"}],"session_id":"3625a798-925a-4435-9855-1ab93d0ed74f"}'
+      string: '{"providers":[{"provider":"google","model":"gemini-2.5-pro"}],"session_id":"d426faa6-c986-4170-a839-3a9de8c67d05"}'
     headers:
       CF-RAY:
-      - 925cfadf29645b4e-VIE
+      - 99a464140b19625c-OTP
       Connection:
       - keep-alive
       Content-Type:
       - application/json
       Date:
-      - Tue, 25 Mar 2025 08:18:52 GMT
+      - Thu, 06 Nov 2025 11:53:32 GMT
       Server:
       - cloudflare
       Transfer-Encoding:
@@ -40,11 +40,13 @@ interactions:
       cf-cache-status:
       - DYNAMIC
       content-length:
-      - '105'
+      - '114'
       rndr-id:
-      - 1e7b850b-bbb4-4ef2
+      - 02b9c1d9-4dc9-495d
       x-render-origin-server:
       - uvicorn
+      x-request-id:
+      - 02b9c1d9-4dc9-495d
     status:
       code: 200
       message: OK

tests/test_components/test_llms/cassettes/test_llm/Test_NDLLM.test_model_select_with_strings[_NDClientTarget.INVOKER-_NDInvokerClient].yaml

@@ -5,11 +5,11 @@ interactions:
       "is_custom": false, "context_length": null, "input_price": null, "output_price":
       null, "latency": null}, {"provider": "openai", "model": "gpt-4", "is_custom":
       false, "context_length": null, "input_price": null, "output_price": null, "latency":
-      null}, {"provider": "anthropic", "model": "claude-2.1", "is_custom": false,
+      null}, {"provider": "anthropic", "model": "claude-haiku-4-5-20251001", "is_custom":
+      false, "context_length": null, "input_price": null, "output_price": null, "latency":
+      null}, {"provider": "google", "model": "gemini-2.5-pro", "is_custom": false,
       "context_length": null, "input_price": null, "output_price": null, "latency":
-      null}, {"provider": "google", "model": "gemini-pro", "is_custom": false, "context_length":
-      null, "input_price": null, "output_price": null, "latency": null}], "metric":
-      "accuracy", "max_model_depth": 4, "hash_content": true}'
+      null}], "metric": "accuracy", "max_model_depth": 4, "hash_content": true}'
     headers:
       Accept:
       - '*/*'
@@ -18,25 +18,25 @@ interactions:
       Connection:
       - keep-alive
       Content-Length:
-      - '779'
+      - '798'
       User-Agent:
-      - Python-SDK/0.3.41
+      - Python-SDK/0.4.6
       content-type:
       - application/json
     method: POST
     uri: https://staging-api.notdiamond.ai/v2/modelRouter/modelSelect
   response:
     body:
-      string: '{"providers":[{"provider":"openai","model":"gpt-4"}],"session_id":"5d496dbd-2fc8-4044-a018-fec52ad9f39c"}'
+      string: '{"providers":[{"provider":"google","model":"gemini-2.5-pro"}],"session_id":"f1917af9-d1bc-4360-96f2-51eed809f159"}'
     headers:
       CF-RAY:
-      - 925cfadabdd3c1f9-VIE
+      - 99a4640c3f0cb691-OTP
       Connection:
       - keep-alive
       Content-Type:
       - application/json
       Date:
-      - Tue, 25 Mar 2025 08:18:51 GMT
+      - Thu, 06 Nov 2025 11:53:31 GMT
       Server:
       - cloudflare
       Transfer-Encoding:
@@ -46,13 +46,15 @@ interactions:
       cf-cache-status:
       - DYNAMIC
       content-length:
-      - '105'
+      - '114'
       rndr-id:
-      - 2d7f2dc4-0459-47a7
+      - e7be3cde-ba72-47e8
       vary:
       - Accept-Encoding
       x-render-origin-server:
       - uvicorn
+      x-request-id:
+      - e7be3cde-ba72-47e8
     status:
       code: 200
       message: OK

tests/test_components/test_llms/cassettes/test_llm/Test_NDLLM.test_model_select_with_strings[_NDClientTarget.ROUTER-_NDRouterClient].yaml

@@ -5,11 +5,11 @@ interactions:
       "is_custom": false, "context_length": null, "input_price": null, "output_price":
       null, "latency": null}, {"provider": "openai", "model": "gpt-4", "is_custom":
       false, "context_length": null, "input_price": null, "output_price": null, "latency":
-      null}, {"provider": "anthropic", "model": "claude-2.1", "is_custom": false,
+      null}, {"provider": "anthropic", "model": "claude-haiku-4-5-20251001", "is_custom":
+      false, "context_length": null, "input_price": null, "output_price": null, "latency":
+      null}, {"provider": "google", "model": "gemini-2.5-pro", "is_custom": false,
       "context_length": null, "input_price": null, "output_price": null, "latency":
-      null}, {"provider": "google", "model": "gemini-pro", "is_custom": false, "context_length":
-      null, "input_price": null, "output_price": null, "latency": null}], "metric":
-      "accuracy", "max_model_depth": 4, "hash_content": true}'
+      null}], "metric": "accuracy", "max_model_depth": 4, "hash_content": true}'
     headers:
       Accept:
       - '*/*'
@@ -18,25 +18,25 @@ interactions:
       Connection:
       - keep-alive
       Content-Length:
-      - '779'
+      - '798'
       User-Agent:
-      - Python-SDK/0.3.41
+      - Python-SDK/0.4.6
       content-type:
       - application/json
     method: POST
     uri: https://staging-api.notdiamond.ai/v2/modelRouter/modelSelect
   response:
     body:
-      string: '{"providers":[{"provider":"openai","model":"gpt-4"}],"session_id":"43eac436-56f1-4ec9-9dcd-4d51d1b31baf"}'
+      string: '{"providers":[{"provider":"google","model":"gemini-2.5-pro"}],"session_id":"79a85558-dc34-4cef-98e0-170f7c7a4591"}'
     headers:
       CF-RAY:
-      - 925cfad5eebad100-SOF
+      - 99a464020e1270ce-OTP
       Connection:
       - keep-alive
       Content-Type:
       - application/json
       Date:
-      - Tue, 25 Mar 2025 08:18:50 GMT
+      - Thu, 06 Nov 2025 11:53:30 GMT
       Server:
       - cloudflare
       Transfer-Encoding:
@@ -46,13 +46,15 @@ interactions:
       cf-cache-status:
       - DYNAMIC
       content-length:
-      - '105'
+      - '114'
       rndr-id:
-      - 5072108e-7aad-4c90
+      - 0521f47e-24f7-4e25
       vary:
       - Accept-Encoding
       x-render-origin-server:
       - uvicorn
+      x-request-id:
+      - 0521f47e-24f7-4e25
     status:
       code: 200
       message: OK