feat: Add new supported model for Claude (#4377)

* feat: Add new supported model for Claude

* Fix elif

* Remove unused endpoint
This commit is contained in:
Eric Dong
2025-11-24 16:56:19 -05:00
committed by GitHub
parent e3320d2126
commit 2e049d4830
@@ -72,6 +72,9 @@
"\n",
"### Available Anthropic Claude models\n",
"\n",
"#### Claude Opus 4.5\n",
"The next generation of Anthropic's most intelligent model, Claude Opus 4.5 is an industry leader across coding, agents, computer use, and enterprise workflows.\n",
"\n",
"#### Claude Haiku 4.5\n",
"Anthropic's mid-size model with superior intelligence for high-volume uses in coding, in-depth research, agents, & more.\n",
"\n",
@@ -197,21 +200,23 @@
},
"outputs": [],
"source": [
"MODEL = \"claude-haiku-4-5@20251001\" # @param [\"claude-haiku-4-5@20251001\",\"claude-sonnet-4-5@20250929\",\"claude-opus-4-1@20250805\",\"claude-sonnet-4@20250514\",\"claude-opus-4@20250514\",\"claude-3-7-sonnet@20250219\", \"claude-3-5-sonnet-v2@20241022\", \"claude-3-5-haiku@20241022\", \"claude-3-5-sonnet@20240620\", \"claude-3-opus@20240229\", \"claude-3-haiku@20240307\" ]\n",
"if MODEL == \"claude-haiku-4-5@20251001\":\n",
" available_regions = [\"us-east5\", \"europe-west1\", \"asia-east1\", \"GLOBAL\"]\n",
"MODEL = \"claude-opus-4-5@20251101\" # @param [\"claude-opus-4-5@20251101\",\"claude-haiku-4-5@20251001\",\"claude-sonnet-4-5@20250929\",\"claude-opus-4-1@20250805\",\"claude-sonnet-4@20250514\",\"claude-opus-4@20250514\",\"claude-3-7-sonnet@20250219\", \"claude-3-5-sonnet-v2@20241022\", \"claude-3-5-haiku@20241022\", \"claude-3-5-sonnet@20240620\", \"claude-3-opus@20240229\", \"claude-3-haiku@20240307\" ]\n",
"if MODEL == \"claude-opus-4-5@20251101\":\n",
" available_regions = [\"us-east5\", \"europe-west1\", \"asia-southeast1\", \"global\"]\n",
"elif MODEL == \"claude-haiku-4-5@20251001\":\n",
" available_regions = [\"us-east5\", \"europe-west1\", \"asia-east1\", \"global\"]\n",
"elif MODEL == \"claude-sonnet-4-5@20250929\":\n",
" available_regions = [\"us-east5\", \"europe-west1\", \"asia-southeast1\", \"GLOBAL\"]\n",
" available_regions = [\"us-east5\", \"europe-west1\", \"asia-southeast1\", \"global\"]\n",
"elif MODEL == \"claude-opus-4-1@20250805\":\n",
" available_regions = [\"us-east5\", \"europe-west4\", \"GLOBAL\"]\n",
" available_regions = [\"us-east5\", \"europe-west4\", \"global\"]\n",
"elif MODEL == \"claude-sonnet-4@20250514\":\n",
" available_regions = [\"us-east5\", \"europe-west4\", \"GLOBAL\"]\n",
" available_regions = [\"us-east5\", \"europe-west4\", \"global\"]\n",
"elif MODEL == \"claude-opus-4@20250514\":\n",
" available_regions = [\"us-east5\", \"europe-west4\", \"GLOBAL\"]\n",
" available_regions = [\"us-east5\", \"europe-west4\", \"global\"]\n",
"elif MODEL == \"claude-3-7-sonnet@20250219\":\n",
" available_regions = [\"us-east5\", \"europe-west1\", \"europe-west4\", \"GLOBAL\"]\n",
" available_regions = [\"us-east5\", \"europe-west1\", \"europe-west4\", \"global\"]\n",
"elif MODEL == \"claude-3-5-sonnet-v2@20241022\":\n",
" available_regions = [\"us-east5\", \"europe-west1\", \"GLOBAL\"]\n",
" available_regions = [\"us-east5\", \"europe-west1\", \"global\"]\n",
"elif MODEL == \"claude-3-5-haiku@20241022\":\n",
" available_regions = [\"us-east5\"]\n",
"elif MODEL == \"claude-3-5-sonnet@20240620\":\n",
@@ -283,7 +288,7 @@
"source": [
"PROJECT_ID = \"[your-project-id]\" # @param {type:\"string\"}\n",
"\n",
"if LOCATION == \"GLOBAL\":\n",
"if LOCATION == \"global\":\n",
" ENDPOINT = \"https://aiplatform.googleapis.com\"\n",
"else:\n",
" ENDPOINT = f\"https://{LOCATION}-aiplatform.googleapis.com\"\n",
@@ -639,15 +644,15 @@
"source": [
"MODEL = \"claude-sonnet-4@20250514\" # @param [\"claude-opus-4-1@20250805\",\"claude-sonnet-4@20250514\",\"claude-opus-4@20250514\",\"claude-3-7-sonnet@20250219\", \"claude-3-5-sonnet-v2@20241022\", \"claude-3-5-haiku@20241022\", \"claude-3-5-sonnet@20240620\", \"claude-3-opus@20240229\", \"claude-3-haiku@20240307\"]\n",
"if MODEL == \"claude-opus-4-1@20250805\":\n",
" available_regions = [\"us-east5\", \"europe-west4\", \"GLOBAL\"]\n",
" available_regions = [\"us-east5\", \"europe-west4\", \"global\"]\n",
"elif MODEL == \"claude-sonnet-4@20250514\":\n",
" available_regions = [\"us-east5\", \"europe-west4\", \"GLOBAL\"]\n",
" available_regions = [\"us-east5\", \"europe-west4\", \"global\"]\n",
"elif MODEL == \"claude-opus-4@20250514\":\n",
" available_regions = [\"us-east5\", \"europe-west4\"]\n",
"elif MODEL == \"claude-3-7-sonnet@20250219\":\n",
" available_regions = [\"us-east5\", \"europe-west1\", \"europe-west4\", \"GLOBAL\"]\n",
" available_regions = [\"us-east5\", \"europe-west1\", \"europe-west4\", \"global\"]\n",
"elif MODEL == \"claude-3-5-sonnet-v2@20241022\":\n",
" available_regions = [\"us-east5\", \"europe-west1\", \"GLOBAL\"]\n",
" available_regions = [\"us-east5\", \"europe-west1\", \"global\"]\n",
"elif MODEL == \"claude-3-5-haiku@20241022\":\n",
" available_regions = [\"us-east5\"]\n",
"elif MODEL == \"claude-3-5-sonnet@20240620\":\n",
@@ -718,7 +723,6 @@
"outputs": [],
"source": [
"PROJECT_ID = \"[your-project-id]\" # @param {type:\"string\"}\n",
"ENDPOINT = f\"https://{LOCATION}-aiplatform.googleapis.com\"\n",
"\n",
"if not PROJECT_ID or PROJECT_ID == \"[your-project-id]\":\n",
" raise ValueError(\"Please set your PROJECT_ID\")"