From 53de4f1faea9bfbeb961fb35b0fe7b0dba39f179 Mon Sep 17 00:00:00 2001 From: rsaryev Date: Sun, 5 Nov 2023 12:10:28 +0200 Subject: [PATCH] chore: update import statements in utils.py and llm.py to use langchain.vectorstores.FAISS --- pyproject.toml | 2 +- talk_codebase/llm.py | 2 +- talk_codebase/utils.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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 @@ from typing import Optional 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 sys 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