Update llm.py

rm 'sk-' verification
This commit is contained in:
Sirui Hong 2023-11-21 17:35:33 +08:00 committed by GitHub
parent 9940022096
commit dfe9168509
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -18,7 +18,7 @@ from metagpt.provider.human_provider import HumanProvider
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.startswith("sk-"):
if CONFIG.openai_api_key:
llm = OpenAIGPTAPI()
elif CONFIG.claude_api_key:
llm = Claude()