update sk AzureChatCompletion from base_url to endpoint

This commit is contained in:
better629 2023-12-22 17:43:10 +08:00
parent a6346c7bce
commit b6eb776190
3 changed files with 5 additions and 3 deletions

View file

@ -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__":

View file

@ -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

View file

@ -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: