Make LangChain dependency optional in pr-agent and update requirement…

…s.txt
Codium-ai · Dec 14, 2023 · 38ea914 · 38ea914
1 parent 246be61
commit 38ea914
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git 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
@@ -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