Skip to content

Commit a4e4602

Browse files
authored
Update moonshot model
1 parent 971883d commit a4e4602

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

src/utils/utils.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,8 @@
2020
"anthropic": "Anthropic",
2121
"deepseek": "DeepSeek",
2222
"google": "Google",
23-
"alibaba": "Alibaba"
23+
"alibaba": "Alibaba",
24+
"moonshot": "MoonShot"
2425
}
2526

2627
def get_llm_model(provider: str, **kwargs):
@@ -149,6 +150,14 @@ def get_llm_model(provider: str, **kwargs):
149150
base_url=base_url,
150151
api_key=api_key,
151152
)
153+
154+
elif provider == "moonshot":
155+
return ChatOpenAI(
156+
model=kwargs.get("model_name", "moonshot-v1-32k-vision-preview"),
157+
temperature=kwargs.get("temperature", 0.0),
158+
base_url=os.getenv("MOONSHOT_ENDPOINT"),
159+
api_key=os.getenv("MOONSHOT_API_KEY"),
160+
)
152161
else:
153162
raise ValueError(f"Unsupported provider: {provider}")
154163

0 commit comments

Comments
 (0)