From 26eb54114756a23c148f311984847712da031c9f Mon Sep 17 00:00:00 2001
From: Manuel Saelices <msaelices@gmail.com>
Date: Tue, 3 Sep 2024 11:35:29 +0200
Subject: [PATCH] Fix broken langchain import. It's broken using modern
 langchain versions

---
 promptimize/prompt_cases.py | 2 +-
 requirements.txt            | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/promptimize/prompt_cases.py b/promptimize/prompt_cases.py
index 6f61889..4444e92 100644
--- a/promptimize/prompt_cases.py
+++ b/promptimize/prompt_cases.py
@@ -1,7 +1,7 @@
 import os
 from typing import Any, Callable, List, Optional, Union
 
-from langchain.llms import OpenAI
+from langchain_community.llms import OpenAI
 from langchain.callbacks import get_openai_callback
 
 from box import Box
diff --git a/requirements.txt b/requirements.txt
index e67d69c..bb52360 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,6 +5,7 @@ click
 pygments
 pyyaml
 langchain
+langchain-community
 tabulate
 pandas
 openai