chore: update import statements in utils.py and llm.py to use langchain.vectorstores.FAISS

pull/36/head
rsaryev 7 months ago
parent ee32d8087e
commit 53de4f1fae
No known key found for this signature in database

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "talk-codebase" name = "talk-codebase"
version = "0.1.48" version = "0.1.49"
description = "talk-codebase is a powerful tool for querying and analyzing codebases." description = "talk-codebase is a powerful tool for querying and analyzing codebases."
authors = ["Saryev Rustam <rustam1997@gmail.com>"] authors = ["Saryev Rustam <rustam1997@gmail.com>"]
readme = "README.md" readme = "README.md"

@ -5,7 +5,7 @@ from typing import Optional
import gpt4all import gpt4all
import questionary import questionary
from halo import Halo from halo import Halo
from langchain import FAISS from langchain.vectorstores import FAISS
from langchain.callbacks.manager import CallbackManager from langchain.callbacks.manager import CallbackManager
from langchain.chains import RetrievalQA from langchain.chains import RetrievalQA
from langchain.chat_models import ChatOpenAI from langchain.chat_models import ChatOpenAI

@ -2,7 +2,7 @@ import sys
import tiktoken import tiktoken
from git import Repo from git import Repo
from langchain import FAISS from langchain.vectorstores import FAISS
from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
from talk_codebase.consts import LOADER_MAPPING, EXCLUDE_FILES from talk_codebase.consts import LOADER_MAPPING, EXCLUDE_FILES

Loading…
Cancel
Save