Merge pull request #205 from wcmolin/fix/zhipu-api-key
[Fix-204]: use correct ZAI_API_KEY for Zhipu/GLM models #204
This commit is contained in:
commit
771c918770
@ -52,7 +52,7 @@ class LiteLLMProvider(LLMProvider):
|
|||||||
elif "gemini" in default_model.lower():
|
elif "gemini" in default_model.lower():
|
||||||
os.environ.setdefault("GEMINI_API_KEY", api_key)
|
os.environ.setdefault("GEMINI_API_KEY", api_key)
|
||||||
elif "zhipu" in default_model or "glm" in default_model or "zai" in default_model:
|
elif "zhipu" in default_model or "glm" in default_model or "zai" in default_model:
|
||||||
os.environ.setdefault("ZHIPUAI_API_KEY", api_key)
|
os.environ.setdefault("ZAI_API_KEY", api_key)
|
||||||
elif "groq" in default_model:
|
elif "groq" in default_model:
|
||||||
os.environ.setdefault("GROQ_API_KEY", api_key)
|
os.environ.setdefault("GROQ_API_KEY", api_key)
|
||||||
elif "moonshot" in default_model or "kimi" in default_model:
|
elif "moonshot" in default_model or "kimi" in default_model:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user