From fc829edc45571f9ca3b5d3212a4f49e46d77a4eb Mon Sep 17 00:00:00 2001 From: geekan Date: Tue, 19 Dec 2023 18:04:12 +0800 Subject: [PATCH] make registry work --- metagpt/llm.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/metagpt/llm.py b/metagpt/llm.py index 60f110a00..8763642f0 100644 --- a/metagpt/llm.py +++ b/metagpt/llm.py @@ -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)