diff --git a/embedchain/llm/vertex_ai.py b/embedchain/llm/vertex_ai.py index b6224e0c..af33a085 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)