chore: update import statements in utils.py and llm.py to use langcha…

…in.vectorstores.FAISS
rsaryev · Nov 5, 2023 · 53de4f1 · 53de4f1
1 parent ee32d80
commit 53de4f1
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "talk-codebase"
-version = "0.1.48"
+version = "0.1.49"
 description = "talk-codebase is a powerful tool for querying and analyzing codebases."
 authors = ["Saryev Rustam <[email protected]>"]
 readme = "README.md"

diff --git a/talk_codebase/llm.py b/talk_codebase/llm.py
@@ -5,7 +5,7 @@
 import gpt4all
 import questionary
 from halo import Halo
-from langchain import FAISS
+from langchain.vectorstores import FAISS
 from langchain.callbacks.manager import CallbackManager
 from langchain.chains import RetrievalQA
 from langchain.chat_models import ChatOpenAI

diff --git a/talk_codebase/utils.py b/talk_codebase/utils.py
@@ -2,7 +2,7 @@
 
 import tiktoken
 from git import Repo
-from langchain import FAISS
+from langchain.vectorstores import FAISS
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 
 from talk_codebase.consts import LOADER_MAPPING, EXCLUDE_FILES