We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 971883d commit a4e4602Copy full SHA for a4e4602
src/utils/utils.py
@@ -20,7 +20,8 @@
20
"anthropic": "Anthropic",
21
"deepseek": "DeepSeek",
22
"google": "Google",
23
- "alibaba": "Alibaba"
+ "alibaba": "Alibaba",
24
+ "moonshot": "MoonShot"
25
}
26
27
def get_llm_model(provider: str, **kwargs):
@@ -149,6 +150,14 @@ def get_llm_model(provider: str, **kwargs):
149
150
base_url=base_url,
151
api_key=api_key,
152
)
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
+ )
161
else:
162
raise ValueError(f"Unsupported provider: {provider}")
163
0 commit comments