Spaces:
Sleeping
Sleeping
Update retriever.py
Browse files- retriever.py +2 -2
retriever.py
CHANGED
|
@@ -42,10 +42,10 @@ def retriever(prompt, history):
|
|
| 42 |
Search results in natural language.
|
| 43 |
"""
|
| 44 |
context = "\n\n".join(documents)
|
| 45 |
-
super_prompt = f"{context}\n\nRetrieve information:\n\nQuestion: {prompt}"
|
| 46 |
|
| 47 |
response = client.models.generate_content(
|
| 48 |
-
model="gemini-2.
|
| 49 |
contents=super_prompt)
|
| 50 |
|
| 51 |
stream_text = ""
|
|
|
|
| 42 |
Search results in natural language.
|
| 43 |
"""
|
| 44 |
context = "\n\n".join(documents)
|
| 45 |
+
super_prompt = f"{context}\n\nRetrieve information:\n\nQuestion: {prompt}. DO NOT precede your response with 'Answer:'. Just give the answer."
|
| 46 |
|
| 47 |
response = client.models.generate_content(
|
| 48 |
+
model="gemini-2.5-flash",
|
| 49 |
contents=super_prompt)
|
| 50 |
|
| 51 |
stream_text = ""
|