Divyanshh commited on
Commit
6a7d34e
·
verified ·
1 Parent(s): 90d8551

Deleting directory after indexing.

Browse files
Files changed (1) hide show
  1. util.py +2 -1
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,