diff --git a/pyproject.toml b/pyproject.toml index b98587b..d53ef98 100644 --- 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 "] readme = "README.md" diff --git a/talk_codebase/llm.py b/talk_codebase/llm.py index 2303336..66316e9 100644 --- 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 index 2622e6a..29a076b 100644 --- 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