import error fix

Aggregate-Intellect · Jul 2, 2024 · d5d4df5 · d5d4df5
1 parent 764dcdc
commit d5d4df5
Show file tree

Hide file tree

Showing 4 changed files with 46 additions and 16 deletions.
diff --git a/src/log.txt b/src/log.txt
diff --git a/src/poetry.lock b/src/poetry.lock
diff --git a/src/pyproject.toml b/src/pyproject.toml
@@ -12,7 +12,8 @@ langchain = "0.2.5"
 langchain-community = "0.2.5"
 python-dotenv = "^1.0.0"
 unstructured = "^0.10.11"
-openai = "^0.28.0"
+langchain-openai = "0.1.13"
+openai = "1.35.7"
 chromadb = "^0.4.8"
 tiktoken = ">=0.6.0,<1.0"
 pinecone-client = "^2.2.2"

diff --git a/src/sherpa_ai/test_utils/llms.py b/src/sherpa_ai/test_utils/llms.py
@@ -2,8 +2,8 @@
 import re
 
 import pytest  # type: ignore
-from langchain_community.chat_models import BaseLanguageModel  # type: ignore
-from langchain_community.llms import ChatOpenAI  # type: ignore
+from langchain_core.language_models import BaseLanguageModel  # type: ignore
+from langchain_community.chat_models import ChatOpenAI  # type: ignore
 from langchain_core.language_models import FakeListLLM  # type: ignore
 
 from sherpa_ai.models.chat_model_with_logging import ChatModelWithLogging
@@ -16,7 +16,8 @@ def get_fake_llm(filename) -> FakeListLLM:
         for line in f:
             responses.append(json.loads(line))
     # restore new line characters
-    responses = [response["output"].replace("\\n", "\n") for response in responses]
+    responses = [response["output"].replace(
+        "\\n", "\n") for response in responses]
     return FakeListLLM(responses=responses)