diff --git a/embedchain/llm/azure_openai.py b/embedchain/llm/azure_openai.py index 7d9b2f8f..34c4f081 100644 --- a/embedchain/llm/azure_openai.py +++ b/embedchain/llm/azure_openai.py @@ -12,10 +12,10 @@ class AzureOpenAILlm(BaseLlm): super().__init__(config=config) def get_llm_model_answer(self, prompt): - return AzureOpenAILlm._get_azure_openai_answer(prompt=prompt, config=self.config) + return AzureOpenAILlm._get_answer(prompt=prompt, config=self.config) @staticmethod - def _get_azure_openai_answer(prompt: str, config: BaseLlmConfig) -> str: + def _get_answer(prompt: str, config: BaseLlmConfig) -> str: from langchain.chat_models import AzureChatOpenAI if not config.deployment_name: