diff --git a/backend/danswer/llm/chat_llm.py b/backend/danswer/llm/chat_llm.py index b0384c314d07..5e893790a926 100644 --- a/backend/danswer/llm/chat_llm.py +++ b/backend/danswer/llm/chat_llm.py @@ -2,9 +2,9 @@ import abc from collections.abc import Iterator import litellm # type:ignore -from langchain.chat_models import ChatLiteLLM from langchain.chat_models.base import BaseChatModel from langchain.schema.language_model import LanguageModelInput +from langchain_community.chat_models import ChatLiteLLM from danswer.configs.app_configs import LOG_ALL_MODEL_INTERACTIONS from danswer.configs.model_configs import GEN_AI_API_ENDPOINT diff --git a/backend/requirements/default.txt b/backend/requirements/default.txt index 35c97bd6523f..423b87cfeadc 100644 --- a/backend/requirements/default.txt +++ b/backend/requirements/default.txt @@ -6,6 +6,7 @@ beautifulsoup4==4.12.2 celery==5.3.4 chardet==5.2.0 dask==2023.8.1 +ddtrace==2.6.5 distributed==2023.8.1 fastapi==0.109.2 fastapi-users==12.1.3 @@ -22,8 +23,8 @@ httpx[http2]==0.23.3 httpx-oauth==0.11.2 huggingface-hub==0.20.1 jira==3.5.1 -langchain==0.0.340 -litellm==1.7.5 +langchain==0.1.9 +litellm==1.27.10 llama-index==0.9.45 Mako==1.2.4 msal==1.26.0