From 25b8a6dcef768ed1e45489e2dd3a5462f37fd593 Mon Sep 17 00:00:00 2001 From: geekan Date: Tue, 19 Dec 2023 18:02:51 +0800 Subject: [PATCH] make registry work --- metagpt/llm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/metagpt/llm.py b/metagpt/llm.py index e0c0716de..60f110a00 100644 --- a/metagpt/llm.py +++ b/metagpt/llm.py @@ -9,7 +9,7 @@ from metagpt.config import CONFIG from metagpt.provider.base_gpt_api import BaseGPTAPI from metagpt.provider.human_provider import HumanProvider -from metagpt.provider.llm_provider_registry import LLMProviderRegistry +from metagpt.provider.llm_provider_registry import LLM_REGISTRY _ = HumanProvider() # Avoid pre-commit error @@ -17,4 +17,4 @@ _ = 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 LLMProviderRegistry.get_provider(CONFIG.get_default_llm_provider_enum()) + return LLM_REGISTRY.get_provider(CONFIG.get_default_llm_provider_enum())