Update src/rag_pipeline.py

src/rag_pipeline.py

@@ -43,20 +43,21 @@ class RAGState(TypedDict):
 # --- LLM Wrapper ---
 class GeminiLLMWrapper:
     """
-    A simple wrapper around google-generativeai chat API to mimic
-    the ChatGoogleGenerativeAI interface for compatibility with app.py.
+    Wrapper around Google Gemini API to mimic ChatGoogleGenerativeAI interface.
     """
     def invoke(self, prompt: str):
-        response = genai.chat.create(
+        # Create a chat-style response using generate_text
+        response = genai.models.generate_text(
             model="models/gemini-2.5-flash",
-            messages=[{"role": "user", "content": prompt}]
+            prompt=prompt,
+            temperature=0.7,
+            max_output_tokens=512
         )
-        # Wrap the response to have a .content attribute
+        # The response text is under response.text
         class Result:
-            content = response.last
+            content = response.text
         return Result()
 
-
 def build_rag_pipeline():
     """Builds a LangGraph-based RAG pipeline compatible with LangChain 1.x."""