From 1ed74123db9bc20b1797fac15949a12663b583d5 Mon Sep 17 00:00:00 2001 From: Claude Date: Tue, 23 Jun 2026 15:45:56 +0000 Subject: [PATCH] Add DeepSeek V4 models to TEE_LLM enum Add DEEPSEEK_V4_FLASH and DEEPSEEK_V4_PRO to the TEE_LLM enum, served through BytePlus ModelArk (bytedance/ prefix). These are already supported by the tee-gateway model registry. Co-Authored-By: Claude Opus 4.8 Claude-Session: https://claude.ai/code/session_01Y4REtNaefr2VdrkunjPhQA --- docs/opengradient/types.md | 2 ++ src/opengradient/types.py | 4 ++++ 2 files changed, 6 insertions(+) diff --git a/docs/opengradient/types.md b/docs/opengradient/types.md index 5542493..6cf8d44 100644 --- a/docs/opengradient/types.md +++ b/docs/opengradient/types.md @@ -483,6 +483,8 @@ auditability and tamper-proof AI inference. * static `CLAUDE_OPUS_4_8` * static `CLAUDE_SONNET_4_5` * static `CLAUDE_SONNET_4_6` +* static `DEEPSEEK_V4_FLASH` +* static `DEEPSEEK_V4_PRO` * static `GEMINI_2_5_FLASH` * static `GEMINI_2_5_FLASH_IMAGE` * static `GEMINI_2_5_FLASH_LITE` diff --git a/src/opengradient/types.py b/src/opengradient/types.py index 2846ad1..f2c6259 100644 --- a/src/opengradient/types.py +++ b/src/opengradient/types.py @@ -598,6 +598,10 @@ class TEE_LLM(str, Enum): SEED_1_8 = "bytedance/seed-1.8" SEED_2_0_LITE = "bytedance/seed-2.0-lite" + # DeepSeek models via TEE (served through BytePlus ModelArk) + DEEPSEEK_V4_FLASH = "bytedance/deepseek-v4-flash" + DEEPSEEK_V4_PRO = "bytedance/deepseek-v4-pro" + # Nous Research Hermes models via TEE (Nous Portal) HERMES_4_405B = "nous/hermes-4-405b" HERMES_4_70B = "nous/hermes-4-70b"