Update rag.py

rag.py

@@ -67,7 +67,7 @@ REFINED QUERY:
 """
     # Use Gemini to generate the refined query
     genai.configure(api_key=GEMINI_API_KEY)
-    model = genai.GenerativeModel(model_name='gemini-1.5-flash-latest')
+    model = genai.GenerativeModel(model_name='gemini-3-flash')
     response = model.generate_content(refine_prompt)
     refined_query = response.text.strip()
     
@@ -153,7 +153,7 @@ def generate_chat_answer(query):
         
         # Configure Gemini API
         genai.configure(api_key=GEMINI_API_KEY)
-        model = genai.GenerativeModel(model_name='gemini-1.5-flash-latest')
+        model = genai.GenerativeModel(model_name='gemini-3-flash')
         
         # Retrieve context and sources from Chroma DB using refined query
         context, sources = get_relevant_context_from_db(query, conversation.get_history())