Update util.py

util.py

@@ -12,7 +12,6 @@ import google.generativeai as genai
 
 import git  # pip install gitpython
 
-os.environ['GOOGLE_API_KEY'] = "AIzaSyCS1bV6_bfizb1tAcQEB9BvCTtqCCLlGFo"
 genai.configure(api_key = os.environ['GOOGLE_API_KEY'])
 
 # quantization_config = BitsAndBytesConfig(
@@ -55,6 +54,7 @@ def get_folder_paths(directory = "githubCode"):
 
 directory_paths = get_folder_paths()
 directory_paths.append("Code")
+print("directory_paths: ", directory_paths)
 
 with open("Code.txt", "w", encoding='utf-8') as output:
     for directory_path in directory_paths:
@@ -84,7 +84,7 @@ pages = loader.load_and_split()
 # Split data into chunks
 text_splitter = RecursiveCharacterTextSplitter(
 chunk_size = 4000,
-chunk_overlap  = 200,
+chunk_overlap  = 20,
 length_function = len,
 add_start_index = True,
 )
@@ -98,7 +98,7 @@ db.persist()
 vectordb = Chroma(persist_directory="test_index", embedding_function = embeddings)
 
 # Load the retriver
-retriever = vectordb.as_retriever()
+retriever = vectordb.as_retriever(search_kwargs = {"k": 3})
 
 # Function to generate assistant's response using ask function
 def generate_assistant_response(question):