Update moonshot model

This commit is contained in:
fyq163
2025-02-15 01:12:15 +08:00
committed by GitHub
parent 971883d739
commit a4e4602f25

View File

@@ -20,7 +20,8 @@ PROVIDER_DISPLAY_NAMES = {
"anthropic": "Anthropic",
"deepseek": "DeepSeek",
"google": "Google",
"alibaba": "Alibaba"
"alibaba": "Alibaba",
"moonshot": "MoonShot"
}
def get_llm_model(provider: str, **kwargs):
@@ -149,6 +150,14 @@ def get_llm_model(provider: str, **kwargs):
base_url=base_url,
api_key=api_key,
)
elif provider == "moonshot":
return ChatOpenAI(
model=kwargs.get("model_name", "moonshot-v1-32k-vision-preview"),
temperature=kwargs.get("temperature", 0.0),
base_url=os.getenv("MOONSHOT_ENDPOINT"),
api_key=os.getenv("MOONSHOT_API_KEY"),
)
else:
raise ValueError(f"Unsupported provider: {provider}")