Gonalb/multi_agentic_sql_generator
1
1�
2 g��g� � � � S SK r S rg)� Nc � � U R SS5 nU( d SSS0S.$ SU S3n[ R R R SS US3./S S 9nUR4 S
R R R 5 nUR S5 ( a0 UR SS5 R SS5 R 5 nUSS0S.$ )z&Validates and optimizes the SQL query.�
generated_sql� �errorzNo SQL query to validate.)�
optimized_sql�validation_resultz�5 Validate and optimize the following SQL query. Check for:6 1. Syntax correctness7 2. Proper table and column references8 3. Performance optimization opportunities9 10 ```sql11 a� 12 ```13 14 **Rules:**15 - DO NOT create or suggest any indexes.16 - DO NOT include any CREATE INDEX, ALTER TABLE, or similar DDL statements.17 - Focus only on query structure optimization without modifying the database schema.18 - Optimize filters and joins.19 - Use appropriate WHERE clauses and join conditions.20 - Output only the SQL query without markdown formatting.21 - The table name must be the full ecommerceaie5.table_name format.22 z gpt-4o-mini�user)�role�contentg )�model�messages� temperaturer z```sqlz```�validT) �get�openai�chat� completions�create�choices�messager �strip�23startswith�replace)�state� sql_query�prompt�response�optimized_querys �I/home/usuario/Escritorio/Multi-Agentic-SQL-Generator/agents/validation.py�!query_validation_and_optimizationr s� � �� � �/�2�.�I� �!#�7�D_�:`�a�a�� �K� ��F�& �{�{�&�&�-�-��!�f�5�6�� .� �H� �&�&�q�)�1�1�9�9�?�?�A�O��!�!�(�+�+�)�1�1�(�B�?�G�G��r�R�X�X�Z��(� %�t�_� � � )r r � r! r �<module>r# s
��
�(r!