upgrade config model gemini

app.py

@@ -120,7 +120,8 @@ def initialize_llm():
         llm = ChatGoogleGenerativeAI(
             model="gemini-2.0-flash",
             temperature=0,
-            google_api_key=google_api_key
+            google_api_key=google_api_key,
+            convert_system_message_to_human=True  # Convert system messages to human messages
         )
         
         # Test the model with a simple prompt
@@ -170,7 +171,6 @@ def create_agent():
         logger.error(error_msg)
         return None, error_msg
     
-    # Step 3: Create SQL agent
     try:
         logger.info("Creating SQL agent...")
         agent = create_sql_agent(
@@ -196,6 +196,56 @@ def create_agent():
         logger.error(error_msg, exc_info=True)  # Include full stack trace
         return None, error_msg
 
+def create_agent(llm, db_connection):
+    """Create and return a SQL database agent."""
+    if not llm:
+        error_msg = "Cannot create agent: LLM is not available"
+        logger.error(error_msg)
+        return None, error_msg
+    
+    if not db_connection:
+        error_msg = "Cannot create agent: Database connection is not available"
+        logger.error(error_msg)
+        return None, error_msg
+        
+    try:
+        logger.info("Creating SQL agent...")
+        
+        # Create the database toolkit with additional configuration
+        toolkit = SQLDatabaseToolkit(
+            db=db_connection,
+            llm=llm
+        )
+        
+        # Create the agent with more detailed configuration
+        agent = create_sql_agent(
+            llm=llm,
+            toolkit=toolkit,
+            agent_type=AgentType.OPENAI_FUNCTIONS,
+            verbose=True,
+            handle_parsing_errors=True,  # Better error handling for parsing
+            max_iterations=10,  # Limit the number of iterations
+            early_stopping_method="generate"  # Stop early if the agent is stuck
+        )
+        
+        # Test the agent with a simple query
+        logger.info("Testing agent with a simple query...")
+        try:
+            test_query = "SELECT 1"
+            test_result = agent.run(test_query)
+            logger.info(f"Agent test query successful: {str(test_result)[:200]}...")
+        except Exception as e:
+            logger.warning(f"Agent test query failed (this might be expected): {str(e)}")
+            # Continue even if test fails, as it might be due to model limitations
+        
+        logger.info("SQL agent created successfully")
+        return agent, ""
+        
+    except Exception as e:
+        error_msg = f"Error creating SQL agent: {str(e)}"
+        logger.error(error_msg, exc_info=True)
+        return None, error_msg
+
 # Inicializar el agente
 logger.info("="*50)
 logger.info("Starting application initialization...")