Ifeanyi commited on
Commit
cffb55a
·
verified ·
1 Parent(s): cb0ae37

Update retriever.py

Browse files
Files changed (1) hide show
  1. 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.0-flash",
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 = ""