BCh\ddlZddlmZmZmZmZej eZGddZ y)N)DictAnyOptionalListc eZdZdZdZdZddedeeefdede eeefd eeeff d Z d eeefded ed eeeffd Z deeefd eeeffdZ y) AIModelBasezBase class for AI modelsbasecy)zInitialize the AI modelN)selfs )/var/www/html/app/models/ai_model_base.py__init__zAIModelBase.__init__ s Nprompt schema_datadb_type db_configreturnctd)aGenerate SQL query from natural language prompt Args: prompt: Natural language prompt schema_data: Database schema data db_type: Database type (mysql, postgresql, sqlserver, oracle) db_config: Optional database configuration settings Returns: Dictionary with generated SQL query and metadata %Subclasses must implement this methodNotImplementedError)r rrrrs r generate_sqlzAIModelBase.generate_sqls""IJJr explain_data sql_queryctd)z9Analyze database explain plan and provide recommendationsrr)r rrrs r analyze_explain_planz AIModelBase.analyze_explain_plans!"IJJrresultc|jdds|S|jddjjgd}|D]8}j|s|jdd}d|d<||d<d|d<|cSrt dkrd|d<d |d<d|d<|Sgd }t fd |D}|sd|d<d |d<d|d<|S|S) aValidate SQL query generation results This method checks if the generated SQL is valid or if it contains patterns that indicate a failure to generate a proper query despite the model returning a success status. Args: result: The result dictionary from the AI model Returns: Updated result dictionary with corrected success status if needed successFr) z i'm sorryzi cannotz i apologizez unable tozcan't generatezcannot generatez sorry, butzsorry i cannotz i don't havez i don't knowz not able toz$Failed to generate a valid SQL queryerror z5Generated SQL query is empty or too short to be valid)selectfromwherejoinzgroup byzorder byhavinglimitc3&K|]}|v ywNr ).0keywordrs r z6AIModelBase.validate_sql_generation..YsPw)3Psz6Generated text does not appear to be a valid SQL query)getstriplower startswithlenany)r rfailure_indicators indicator error_message sql_keywordshas_sql_keywordsrs @r validate_sql_generationz#AIModelBase.validate_sql_generation"s zz)U+MJJ{B/557==?   , I##I. & ;8^ _ %*y!"/w&({#  C NR/ %F9 UF7O"$F; Mf P<PP %F9 VF7O"$F; M rr+) __name__ __module__ __qualname____doc__ model_typerstrrrrrrr:r rr rrs"J  K3 KT#s(^ Kc K^fgkloqtltgu^v KCGHKMPHPCQ KKc3hK#KZ]KbfgjlogobpK?d38n?c3h?rr) loggingtypingrrrr getLoggerr;loggerrr rr rEs,,,   8 $[[r