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

Rename _get_athrophic_answer to _get_answer (#775)

Browse Source
This commit is contained in:
Prikshit
2023-10-07 14:52:49 +05:30
committed by GitHub
parent 1741d3bef6
commit 1a048390fd
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
embedchain/llm/vertex_ai.py
View File

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