Update main.py

main.py

@@ -177,8 +177,8 @@ def get_gemini_model(model_name: str):
     if "thinking" in model_name_lower:
         return Model.G_3_FLASH_THINKING_AI_FREE
     elif "pro" in model_name_lower:
-        # Padrão ou o modelo mais inteligente que criamos (3.1 Pro)
-        return "gemini-3.1-pro"
+        # Padrão ou o modelo mais inteligente que criamos
+        return Model.G_3_PRO_AI_FREE
     
     # Default flash
     return Model.G_3_FLASH_AI_FREE