From cef8fd7e7447f35c0468ec41f1e894fecad79bef Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 7 Jan 2024 20:06:27 -0300 Subject: [PATCH] Update import statements in ChatOllamaEndpoint.py --- src/backend/langflow/components/llms/ChatOllamaEndpoint.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/components/llms/ChatOllamaEndpoint.py b/src/backend/langflow/components/llms/ChatOllamaEndpoint.py index 3201ff432..34e9e0bd7 100644 --- a/src/backend/langflow/components/llms/ChatOllamaEndpoint.py +++ b/src/backend/langflow/components/llms/ChatOllamaEndpoint.py @@ -1,8 +1,8 @@ from typing import Any, Dict, List, Optional # from langchain_community.chat_models import ChatOllama -from langchain.chat_models import ChatOllama -from langchain.chat_models.base import BaseChatModel +from langchain_community.chat_models import ChatOllama +from langchain_core.language_models.chat_models import BaseChatModel # from langchain.chat_models import ChatOllama from langflow import CustomComponent