vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat: add Novita AI deepseek models (#1647)

Browse Source 
Co-authored-by: Suchintan <suchintan@users.noreply.github.com>
This commit is contained in:
Jason
2025-01-28 07:51:09 +08:00
committed by GitHub
parent 81d5babcf0
commit c30eba6245
3 changed files with 33 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
skyvern/forge/sdk/api/llm/config_registry.py
View File

@@ -290,6 +290,36 @@ if settings.ENABLE_GEMINI:
)
if settings.ENABLE_NOVITA:
LLMConfigRegistry.register_config(
"NOVITA_DEEPSEEK_R1",
LLMConfig(
"openai/deepseek/deepseek-r1",
["NOVITA_API_KEY"],
supports_vision=False,
add_assistant_prefix=False,
litellm_params=LiteLLMParams(
api_base="https://api.novita.ai/v3/openai",
api_key=settings.NOVITA_API_KEY,
api_version=settings.NOVITA_API_VERSION,
model_info={"model_name": "openai/deepseek/deepseek-r1"},
),
),
)
LLMConfigRegistry.register_config(
"NOVITA_DEEPSEEK_V3",
LLMConfig(
"openai/deepseek/deepseek_v3",
["NOVITA_API_KEY"],
supports_vision=False,
add_assistant_prefix=False,
litellm_params=LiteLLMParams(
api_base="https://api.novita.ai/v3/openai",
api_key=settings.NOVITA_API_KEY,
api_version=settings.NOVITA_API_VERSION,
model_info={"model_name": "openai/deepseek/deepseek_v3"},
),
),
)
LLMConfigRegistry.register_config(
"NOVITA_LLAMA_3_3_70B",
LLMConfig(