apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-02 14:45:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

make registry work

Browse source
This commit is contained in:
geekan 2023-12-19 18:04:12 +08:00 • committed by better629
parent af59323a69
commit fc829edc45
1 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
metagpt/llm.py
View file

@ -6,7 +6,7 @@
@File : llm.py
"""
from metagpt.config import CONFIG
from metagpt.config import CONFIG, LLMProviderEnum
from metagpt.provider.base_gpt_api import BaseGPTAPI
from metagpt.provider.human_provider import HumanProvider
from metagpt.provider.llm_provider_registry import LLM_REGISTRY
@ -14,7 +14,6 @@ from metagpt.provider.llm_provider_registry import LLM_REGISTRY
_ = HumanProvider() # Avoid pre-commit error
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
return LLM_REGISTRY.get_provider(CONFIG.get_default_llm_provider_enum())
def LLM(provider: LLMProviderEnum = CONFIG.get_default_llm_provider_enum()) -> BaseGPTAPI:
"""get the default llm provider"""
return LLM_REGISTRY.get_provider(provider)