apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-07-02 16:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix config when open llm model hosts as openai interface

Browse source
This commit is contained in:
better629 2023-11-20 14:46:31 +08:00
parent 6ef3b213c3
commit f8f938f333
2 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
metagpt/llm.py
View file

@ -17,13 +17,15 @@ from metagpt.provider.spark_api import SparkAPI
def LLM() -> "BaseGPTAPI":
""" initialize different LLM instance according to the key field existence"""
# TODO a little trick, can use registry to initialize LLM instance further
if CONFIG.openai_api_key and CONFIG.openai_api_key.starswith("sk-"):
if CONFIG.openai_api_key and CONFIG.openai_api_key.startswith("sk-"):
llm = OpenAIGPTAPI()
elif CONFIG.claude_api_key:
llm = Claude()
elif CONFIG.spark_api_key:
llm = SparkAPI()
elif CONFIG.zhipuai_api_key:
elif CONFIG.zhipuai_api_key and CONFIG.zhipuai_api_key != "YOUR_API_KEY":
llm = ZhiPuAIGPTAPI()
else:
raise RuntimeError("You should config a LLM configuration first")
return llm