[Feature] OpenAI Function Calling (#1224)

2024-02-12 02:58:11 +01:00
parent 38e212c721
commit 41bd258b93
31 changed files with 259 additions and 213 deletions
--- a/embedchain/llm/vertex_ai.py
+++ b/embedchain/llm/vertex_ai.py
@@ -24,7 +24,7 @@ class VertexAILlm(BaseLlm):

    @staticmethod
    def _get_answer(prompt: str, config: BaseLlmConfig) -> str:
-        from langchain.chat_models import ChatVertexAI
+        from langchain_community.chat_models import ChatVertexAI

        chat = ChatVertexAI(temperature=config.temperature, model=config.model)