Spaces:
Sleeping
Sleeping
Upload indexer.py
Browse files- indexer.py +3 -2
indexer.py
CHANGED
|
@@ -10,7 +10,8 @@ import os
|
|
| 10 |
|
| 11 |
def index_text():
|
| 12 |
|
| 13 |
-
os.environ["NVIDIA_API_KEY"] ="
|
|
|
|
| 14 |
|
| 15 |
nvidia_embeddings = NVIDIAEmbeddings(
|
| 16 |
model="nvidia/llama-3.2-nv-embedqa-1b-v2",
|
|
@@ -31,7 +32,7 @@ QUERY:
|
|
| 31 |
Use the provided context to answer the user query. Only use the provided context to answer the query.
|
| 32 |
If you do not know the answer, or it's not contained in the provided context, respond with "I don't know".
|
| 33 |
"""
|
| 34 |
-
os.environ["GEMINI_API_KEY"] = "
|
| 35 |
client = genai.Client()
|
| 36 |
# Get relevant documents
|
| 37 |
retriever = vectorstore.as_retriever()
|
|
|
|
| 10 |
|
| 11 |
def index_text():
|
| 12 |
|
| 13 |
+
os.environ["NVIDIA_API_KEY"] =os.getenv("NVIDIA_API_KEY")
|
| 14 |
+
|
| 15 |
|
| 16 |
nvidia_embeddings = NVIDIAEmbeddings(
|
| 17 |
model="nvidia/llama-3.2-nv-embedqa-1b-v2",
|
|
|
|
| 32 |
Use the provided context to answer the user query. Only use the provided context to answer the query.
|
| 33 |
If you do not know the answer, or it's not contained in the provided context, respond with "I don't know".
|
| 34 |
"""
|
| 35 |
+
os.environ["GEMINI_API_KEY"] = os.getenv("GEMINI_API_KEY")
|
| 36 |
client = genai.Client()
|
| 37 |
# Get relevant documents
|
| 38 |
retriever = vectorstore.as_retriever()
|