From b6eb776190c05ace0b43c03020797939d1cf2eaf Mon Sep 17 00:00:00 2001 From: better629 Date: Fri, 22 Dec 2023 17:43:10 +0800 Subject: [PATCH] update sk AzureChatCompletion from base_url to endpoint --- examples/search_with_specific_engine.py | 4 ++-- metagpt/roles/role.py | 2 ++ metagpt/utils/make_sk_kernel.py | 2 +- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/examples/search_with_specific_engine.py b/examples/search_with_specific_engine.py index 923f538ed..1a217fdf2 100644 --- a/examples/search_with_specific_engine.py +++ b/examples/search_with_specific_engine.py @@ -9,9 +9,9 @@ async def main(): # Serper API # await Searcher(engine=SearchEngineType.SERPER_GOOGLE).run(question) # SerpAPI - # await Searcher(engine=SearchEngineType.SERPAPI_GOOGLE).run(question) + await Searcher(engine=SearchEngineType.SERPAPI_GOOGLE).run(question) # Google API - await Searcher(engine=SearchEngineType.DIRECT_GOOGLE).run(question) + # await Searcher(engine=SearchEngineType.DIRECT_GOOGLE).run(question) if __name__ == "__main__": diff --git a/metagpt/roles/role.py b/metagpt/roles/role.py index b9fde7d05..e2560128b 100644 --- a/metagpt/roles/role.py +++ b/metagpt/roles/role.py @@ -204,6 +204,8 @@ class Role(BaseModel): object.__setattr__(self, "builtin_class_name", self.__class__.__name__) self.__fields__["builtin_class_name"].default = self.__class__.__name__ + self._watch(kwargs.get("watch") or [UserRequirement]) + def __init_subclass__(cls, **kwargs: Any) -> None: super().__init_subclass__(**kwargs) role_subclass_registry[cls.__name__] = cls diff --git a/metagpt/utils/make_sk_kernel.py b/metagpt/utils/make_sk_kernel.py index 5edddd618..de84e3630 100644 --- a/metagpt/utils/make_sk_kernel.py +++ b/metagpt/utils/make_sk_kernel.py @@ -22,7 +22,7 @@ def make_sk_kernel(): kernel.add_chat_service( "chat_completion", AzureChatCompletion( - deployment_name=CONFIG.deployment_name, base_url=CONFIG.openai_base_url, api_key=CONFIG.openai_api_key + deployment_name=CONFIG.deployment_name, endpoint=CONFIG.openai_base_url, api_key=CONFIG.openai_api_key ), ) else: