Spaces:
Sleeping
Sleeping
Ashmi Banerjee
commited on
Commit
Β·
c59035e
1
Parent(s):
420fa8a
updates with gemini
Browse files- .gitignore +2 -2
- build_rag.py +1 -1
- models/gemini.py +1 -1
- {setups β setup}/db_setup.py +0 -0
- {setups β setup}/vertex_ai_setup.py +0 -0
.gitignore
CHANGED
@@ -180,5 +180,5 @@ gradio_cached_examples/
|
|
180 |
__pycache__/
|
181 |
gradio_cached_examples/
|
182 |
models/__pycache__/
|
183 |
-
|
184 |
-
|
|
|
180 |
__pycache__/
|
181 |
gradio_cached_examples/
|
182 |
models/__pycache__/
|
183 |
+
setup/__pycache__/
|
184 |
+
setup/gcp-default-creds.json
|
build_rag.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
from sentence_transformers import SentenceTransformer
|
2 |
-
from
|
3 |
|
4 |
|
5 |
def get_embedding(text: str) -> list[float]:
|
|
|
1 |
from sentence_transformers import SentenceTransformer
|
2 |
+
from setup.db_setup import get_mongo_client, get_mongo_url
|
3 |
|
4 |
|
5 |
def get_embedding(text: str) -> list[float]:
|
models/gemini.py
CHANGED
@@ -2,7 +2,7 @@ from typing import Optional
|
|
2 |
import sys
|
3 |
|
4 |
sys.path.append("../")
|
5 |
-
from
|
6 |
from vertexai.preview.generative_models import GenerativeModel
|
7 |
|
8 |
|
|
|
2 |
import sys
|
3 |
|
4 |
sys.path.append("../")
|
5 |
+
from setup.vertex_ai_setup import initialize_vertexai_params
|
6 |
from vertexai.preview.generative_models import GenerativeModel
|
7 |
|
8 |
|
{setups β setup}/db_setup.py
RENAMED
File without changes
|
{setups β setup}/vertex_ai_setup.py
RENAMED
File without changes
|