klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rename _get_azure_openai_answer to _get_answer (#724)

Browse Source
This commit is contained in:
Prikshit
2023-09-29 00:13:49 +05:30
committed by GitHub
parent e73ed82ae9
commit 9951b58005
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
embedchain/llm/azure_openai.py
View File

@@ -12,10 +12,10 @@ class AzureOpenAILlm(BaseLlm):
super().__init__(config=config) super().__init__(config=config)
def get_llm_model_answer(self, prompt): 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 @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 from langchain.chat_models import AzureChatOpenAI
if not config.deployment_name: if not config.deployment_name: