inital requirements

claims_agent.py

@@ -1,5 +1,5 @@
 import os
-import google.generativeai as genai
+from google import genai
 from dotenv import load_dotenv
 from PIL import Image
 
@@ -11,8 +11,7 @@ class ClaimsIntakeAgent:
         if not api_key:
             print("Warning: GEMINI_API_KEY not found in environment variables.")
         else:
-            genai.configure(api_key=api_key)
-            self.model = genai.GenerativeModel('gemini-2.5-flash')
+            self.client = genai.Client(api_key=api_key)
 
     def process_claim(self, user_input, image=None, document_path=None):
         """
@@ -24,7 +23,7 @@ class ClaimsIntakeAgent:
         Returns:
             str: The agent's response.
         """
-        if not hasattr(self, 'model'):
+        if not hasattr(self, 'client'):
              return "Error: Gemini API key not configured. Please check your .env file."
 
         from utils import convert_doc_to_markdown
@@ -65,7 +64,10 @@ class ClaimsIntakeAgent:
             if image:
                 inputs.append(image)
             
-            response = self.model.generate_content(inputs)
+            response = self.client.models.generate_content(
+                model='gemini-2.5-flash',
+                contents=inputs
+            )
             return response.text
         except Exception as e:
             return f"Error processing claim: {str(e)}"

requirements.txt

@@ -1,3 +1,3 @@
-google-generativeai
+google-genai
 python-dotenv
 markitdown