From 1a048390fd19623c7629c65abef2360a05b49ce2 Mon Sep 17 00:00:00 2001 From: Prikshit <101416953+Prikshit7766@users.noreply.github.com> Date: Sat, 7 Oct 2023 14:52:49 +0530 Subject: [PATCH] Rename _get_athrophic_answer to _get_answer (#775) --- embedchain/llm/vertex_ai.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/embedchain/llm/vertex_ai.py b/embedchain/llm/vertex_ai.py index b6224e0c3..af33a085c 100644 --- a/embedchain/llm/vertex_ai.py +++ b/embedchain/llm/vertex_ai.py @@ -12,10 +12,10 @@ class VertexAiLlm(BaseLlm): super().__init__(config=config) def get_llm_model_answer(self, prompt): - return VertexAiLlm._get_athrophic_answer(prompt=prompt, config=self.config) + return VertexAiLlm._get_answer(prompt=prompt, config=self.config) @staticmethod - def _get_athrophic_answer(prompt: str, config: BaseLlmConfig) -> str: + def _get_answer(prompt: str, config: BaseLlmConfig) -> str: from langchain.chat_models import ChatVertexAI chat = ChatVertexAI(temperature=config.temperature, model=config.model)