Fallback model

app.py

@@ -31,7 +31,7 @@ GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 HF_TOKEN = os.getenv("HF_TOKEN")
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 GROQ_MODEL = "meta-llama/llama-4-scout-17b-16e-instruct"
-GOOGLE_MODEL = "gemini-2.5-flash"
+GOOGLE_MODEL = "gemini-2.5-flash-lite"
 ist_tz = timezone("Asia/Kolkata")
 
 class TransactionParser(BaseModel):
@@ -143,12 +143,14 @@ def write_message(data: dict, header: str = Header()):
     message = data['message']
 
     try:
-        model = ChatGoogleGenerativeAI(model=GOOGLE_MODEL, max_retries=3, callbacks = [OpikTracer()])
+        model = ChatGoogleGenerativeAI(model=GOOGLE_MODEL, callbacks = [OpikTracer()])
+        transaction_bot = Agent(model, system=prompt)
+        transaction_bot.graph.invoke({"messages": [message]})
     except Exception as e: #fallback model
         model = ChatGroq(model=GROQ_MODEL, temperature=1, callbacks = [OpikTracer()])
-    # model = ChatOllama(model="gemma3:1b", temperature=1)
-    transaction_bot = Agent(model, system=prompt)
-    transaction_bot.graph.invoke({"messages": [message]})
+        transaction_bot = Agent(model, system=prompt)
+        transaction_bot.graph.invoke({"messages": [message]})
+    
     return {"message": "Transaction completed successfully"}
 
 if __name__ == "__main__":