Rename _get_azure_openai_answer to _get_answer (#724)

This commit is contained in:
Prikshit
2023-09-29 00:13:49 +05:30
committed by GitHub
parent e73ed82ae9
commit 9951b58005

View File

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