From 38ea9143f3b906cc2abfa97a4c623f1a6879a269 Mon Sep 17 00:00:00 2001 From: mrT23 Date: Thu, 14 Dec 2023 09:05:53 +0200 Subject: [PATCH] Make LangChain dependency optional in pr-agent and update requirements.txt --- pr_agent/algo/ai_handlers/langchain_ai_handler.py | 7 +++++-- requirements.txt | 2 +- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/pr_agent/algo/ai_handlers/langchain_ai_handler.py b/pr_agent/algo/ai_handlers/langchain_ai_handler.py index 5c793f2b..3e31bcb8 100644 --- a/pr_agent/algo/ai_handlers/langchain_ai_handler.py +++ b/pr_agent/algo/ai_handlers/langchain_ai_handler.py @@ -1,5 +1,8 @@ -from langchain.chat_models import ChatOpenAI -from langchain.schema import SystemMessage, HumanMessage +try: + from langchain.chat_models import ChatOpenAI + from langchain.schema import SystemMessage, HumanMessage +except: # we don't enforce langchain as a dependency, so if it's not installed, just move on + pass from pr_agent.algo.ai_handlers.base_ai_handler import BaseAiHandler from pr_agent.config_loader import get_settings diff --git a/requirements.txt b/requirements.txt index 1f97ed7f..b293f3b3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -23,4 +23,4 @@ starlette-context==0.3.6 tiktoken==0.5.2 ujson==5.8.0 uvicorn==0.22.0 -langchain==0.0.349 +# langchain==0.0.349 # uncomment this to support language LangChainOpenAIHandler