Spaces:
Running
Running
Update util.py
Browse files
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=
|
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 = []
|