Divyanshh commited on
Commit
88b5d3f
1 Parent(s): 1729ee6

Update util.py

Browse files
Files changed (1) hide show
  1. util.py +1 -1
util.py CHANGED
@@ -11,7 +11,7 @@ import git
11
  from chromadb.utils import embedding_functions
12
  embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001", google_api_key=os.environ['GOOGLE_API_KEY'], task_type="retrieval_document")
13
 
14
- model = ChatGoogleGenerativeAI(model="gemini-pro",google_api_key=os.environ['GOOGLE_API_KEY'],temperature=0.2,convert_system_message_to_human=False)
15
 
16
  def get_folder_paths(directory = "githubCode"):
17
  folder_paths = []
 
11
  from chromadb.utils import embedding_functions
12
  embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001", google_api_key=os.environ['GOOGLE_API_KEY'], task_type="retrieval_document")
13
 
14
+ model = ChatGoogleGenerativeAI(model="gemini-pro",google_api_key=os.environ['GOOGLE_API_KEY'],temperature=0.2,convert_system_message_to_human=True)
15
 
16
  def get_folder_paths(directory = "githubCode"):
17
  folder_paths = []