Spaces:
Sleeping
Sleeping
Deleting directory after indexing.
Browse files
util.py
CHANGED
@@ -4,7 +4,7 @@ from langchain_community.embeddings import HuggingFaceHubEmbeddings
|
|
4 |
from langchain_community.vectorstores import Chroma
|
5 |
from langchain.chains import RetrievalQA
|
6 |
from langchain_google_genai import ChatGoogleGenerativeAI, GoogleGenerativeAIEmbeddings
|
7 |
-
|
8 |
import git
|
9 |
|
10 |
from chromadb.utils import embedding_functions
|
@@ -50,6 +50,7 @@ texts = text_splitter.split_text(context)
|
|
50 |
|
51 |
vector_index = Chroma.from_texts(texts, embeddings).as_retriever(search_kwargs={"k":5})
|
52 |
|
|
|
53 |
qa_chain = RetrievalQA.from_chain_type(
|
54 |
model,
|
55 |
retriever=vector_index,
|
|
|
4 |
from langchain_community.vectorstores import Chroma
|
5 |
from langchain.chains import RetrievalQA
|
6 |
from langchain_google_genai import ChatGoogleGenerativeAI, GoogleGenerativeAIEmbeddings
|
7 |
+
import shutil
|
8 |
import git
|
9 |
|
10 |
from chromadb.utils import embedding_functions
|
|
|
50 |
|
51 |
vector_index = Chroma.from_texts(texts, embeddings).as_retriever(search_kwargs={"k":5})
|
52 |
|
53 |
+
shutil.rmtree("githubCode")
|
54 |
qa_chain = RetrievalQA.from_chain_type(
|
55 |
model,
|
56 |
retriever=vector_index,
|