Kalletlamadhav/sql-optimization-env
0
1�
2 �w�i � � � d dl mZmZ d dlmZmZmZmZ d dlm Z G d� de3e � � Z G d� de� � Z G d� d e� � Z
G d4� d e� � Z G d � d
e� � Z G d� de� � Z G d� de� � ZdS )� )� BaseModel�Field)�Optional�List�Dict�Any)�Enumc �. � e Zd ZdZdZdZdZdZdZdZ dZ5d S )6�AntiPatternType�7N_PLUS_ONE�CARTESIAN_PRODUCT�
MISSING_INDEX� SELECT_STAR�LEADING_WILDCARD�
IMPLICIT_CAST�UNBOUNDED_AGGREGATION�NONEN) �__name__�8__module__� __qualname__r r
r r r r r r � � �9C:\open_env_sql_opt\sql-optimization-env\server\models.pyr r s9 � � � � � � �J�+��#�M� �K�)��#�M� 3�� �D�D�Dr r c �v � e Zd ZU eed< eed< eed< eed< dZee ed< dZ ee ed< eed< dS ) �
ExecutionPlan� operation�
rows_examined�
rows_returned�
cost_estimateN� using_index�missing_index_hint� explain_raw)9r r r �str�__annotations__�int�floatr r r! r r r r r st � � � � � � ��N�N�N�������������!%�K��#� �%�%�%�(,����
�,�,�,������r r c � � e Zd ZU eed< eed< eed< eed< eed< eed< eed< eed< eee f ed < eed10< d Z11e e ed < d Z e e ed
< d S )�SQLOptObservation�task_id� step_number�goal�12schema_ddl�
current_query�execution_plan�execution_time_ms� row_count�db_stats�curriculum_levelN�anti_pattern_hint�
error_message)
r r r r# r$ r% r r& r r r3 r r4 r r r r( r( s� � � � � � � �
�L�L�L�����13
�I�I�I��O�O�O�����!�!�!�!� � � � ��N�N�N��3��8�n� � � �����'+��x��}�+�+�+�#'�M�8�C�=�'�'�'�'�'r r( c �| � e Zd ZU eed< ej Zeed< dZeed< e e14�� � Z e e ed< dZ
eed< dS ) � SQLOptAction�optimized_query�identified_pattern� � explanation��default_factory�index_statements�schema_analysisN)r r r r# r$ r r r8 r: r �listr= r r> r r r r6 r6 , st � � � � � � �����*9�*>���>�>�>��K�����"'�%��"=�"=�"=��d�3�i�=�=�=� �O�S� � � � � r r6 c � � e Zd ZU eed< eed< eed< eed< eed< eed< eed< eed< eed < d15Zee ed < d16S ) � SQLOptReward�total�
speedup_score�equivalence_score�
pattern_score� index_score�simplicity_score� penalties�
speedup_ratio�
hack_detectedN� hack_type) r r r r& r$ �boolrK r r# r r r rA rA 4 s� � � � � � � � �L�L�L����� � � � ������������������������� #�I�x��}�#�#�#�#�#r rA c �j � e Zd ZU eed< eed< eed< eed< ee �� � Z17e e e
f ed< dS )�18StepResult� observation�reward�
reward_detail�doner; �infoN)r r r r( r$ r& rA rL r �dictrS r r# r r r r rN rN A sa � � � � � � �"�"�"�"� �M�M�M� � � � �19�J�J�J� �5��6�6�6�D�$�s�C�x�.�6�6�6�6�6r rN c � � e Zd ZU eed< eed< eed< eed< eed< ee�� � Ze e20 ed< eed< e ed < d21S ) �EnvironmentState�22episode_id�current_task_id� current_step� max_stepsr2 r; �episode_rewards�total_episodes�23is_runningN) r r r r# r$ r% r r? r[ r r& rL r r r rV rV I s� � � � � � � ��O�O�O����������N�N�N�����#(�5��#>�#>�#>�O�T�%�[�>�>�>����������r rV N)�pydanticr r �typingr r r r �enumr r# r r r( r6 rA rN rV r r r �<module>ra s� �� &� %� %� %� %� %� %� %� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,� � � � � � �� � � � �c�4� � � �� � � � �I� � � � (� (� (� (� (� � (� (� (� � � � � �9� � � �24$� 25$� 26$� 27$� 28$�9� 29$� 30$� 31$�7� 7� 7� 7� 7�� 7� 7� 7�� � � � �y� � � � � r