SyncShift/sql-correction-env

1+
2�b�i���j�^R
It^RI
HtHtHtRRltRRltRRltR	R3ltRRlt	R
Rlt4R
#)�N��	SQLAction�SQLTask�	SQLRewardc

�0�V^8�dQ
h
R
\R\/#���query�return)
�str)
�formats
"�DC:\Users\srava\Downloads\SQL corrector agent files\sql_env\grader.py�__annotate__rs��

�

�c�

�c�

�c
��VP
4P4p
\P!R
RV
4p
V
P	R4
P
4p
V
#)z:Uppercase, collapse whitespace, strip trailing semicolons.z\s+�
 �
;)�strip�upper�re�sub�rstrip)r5�
qs& r�6_normalizersC��
���
����A�7���v�s�A��A�	����
����A��Hrc

�0�V^8�dQ
h
R
\R\/#r�r�set)
r
s
"rrr
s��
8�
8�S�
8�S�
8rc
��\
V4
p
\P!RR
V
4p
\\P!RV
44
#)z'[^']*'�__STR__z[A-Z0-9_'*.=><]+)rrrr�findall)r8�normeds& r�	_tokenizer"
s5��
��
�F�
�V�V�J�	�6�
2�F��r�z�z�-�v�6�7�7rc

�0�V^8�dQ
h
R
\R\/#rr)
r
s
"rrrs��

�

��

��

rc
�>
�0Rk
R
k
Rk
Rk
Rk
Rk
Rk
Rk
Rk
R	k
R9k
Rk
Rk
R
k
Rk
Rk
Rk
Rk
Rk
Rk
Rk
Rk
Rk
Rk
Rk
Rk
Rk
Rk
Rk
Rk
Rk
p
\
V4
p\4pV
FAp\P!RV,R,V4'gK0VP	V4
KC	V#) �SELECT�FROM�WHERE�GROUP�BY�HAVING�ORDER�JOIN�INNER�LEFT�RIGHT�OUTER�BETWEEN�DESC�ASC�LIMIT�COUNT�SUM�AVG�MAX�MIN�AS�ON�AND�OR�NOT�IN�LIKE�IS�NULL�DISTINCTz\b)rrr�search�add)r10�keywordsr!�found�kws&    r�_sql_keywords_presentrIsp
�������!��#*��,0��2:������ ��"(��*1��3:��	����!��#*��,3��5:��	�	��	��	�"�	�$(�	�*/�	�15�	�7<�	�11	
��12��13��14#��15%/��H���
�F��E�E���
�9�9�U�R�Z�%�'��0�0��I�I�b�M���Lrc

�0�V^8�dQ
h
R
\R\/#)r	�valuer)
�float)
r
s
"rrr#s��
'�
'�%�
'�E�
'rc
�.�\
R
\RV44#)zHEnsure reward is strictly within (0, 1) as required by the OpenEnv spec.�{�G�z�?�G�z��?)�max�min�
rKs
&r�_clamprS#s���t�S��u�%�&�&rc

�<�V^8�dQ
h
R
\R\R\/#)r	�action�taskrr)
r
s
"rrr(s+��@

N
�@

N
�)�@

N
�7�@

N
�y�@

N
rc�B�\
VP4
p\
V
P4
pW#8Xd\\	R
4
RR7#\VP4
p\V
P4
pWE8Xd\\	R4
RR7#\
V
P4
p\
VP4
p\Wg,
4
\\V4
^
4,p\WE,
4
\\V4
^
4,p	VR8�d'V	R8�d \\	R4
R	VR16
RV	R17
R2R7#VR
8�d'V	R8�d \\	R4
RVR18
RV	R19
R2R7#RV9dRV9d\\	R4
RR7#\\	R4
RR7#)u�
205-level grader with partial progress signals.21All scores are clamped to [0.01, 0.99] — strictly between 0 and 1.22 230.99 — exact normalized match (perfect fix)240.70 — same token set, minor structural/whitespace differences250.40 — most SQL keywords correct AND high token overlap260.30 — partial keyword and structure match270.20 — basic SELECT/FROM structure present280.01 — not recognizable SQL29rOu#Exact match — perfect correction.)rK�reason�ffffff�?z:All correct tokens present but structure differs slightly.g333333�?g�?皙�����?zMost keywords correct (z.0%z keyword match, z token match).g�������?g�?�333333�?z%Partial keyword and structure match (r%r&皙�����?zBBasic SELECT/FROM structure present but significant errors remain.rNzResponse is not valid SQL.)	r�corrected_query�canonical_answerrrSr"rI�lenrP)30rUrV�agent�correct�agent_tokens�correct_tokens�correct_kws�	agent_kws�31kw_overlap�
token_overlaps32&&        r�graderh(s�
��
�v�-�-�.�E���.�.�/�G�
�����,�8�33�	34��V�3�3�4�L��t�4�4�5�N��%����,�O�35�	36�(��(=�(=�>�K�%�f�&<�&<�=�I��[�,�-��C��4D�a�0H�H�J���5�6��S��=P�RS�9T�T�M��T��m�t�3����,�
��s�#�#3�M�#�3F�n�
V
�37�	38��T��m�t�3����,�
��s�#�#3�M�#�3F�n�
V
�39�	40��5��V�u�_����,�W�41�	42��6�$�<�0L�M�Mrc

�H�V^8�dQ
h
R
\R\R\R\/#)r	rUrV�rewardr)rrrr)
r
s
"rrrks.��
V
�
V
�i�
V
�w�
V
�	�
V
�c�
V
rc���VPR
8�dR#VPR8�dR#VPR8�dR#VPR8�dR#VPR	8�dR43#R#)z6Human-readable feedback shown in the next observation.rOz!Correct! Query matches perfectly.rYu9Very close — check spacing or minor clause differences.rZu[Good progress — most keywords are right, but check for typos in keywords or column names.r[uJPartial match — right direction but several keywords or columns are off.r\uKBasic structure is there — look carefully at every SQL keyword for typos.zHThe response doesn't look like valid SQL. Start with SELECT ... FROM ...rR)rUrVrjs&&&r�generate_feedbackrlksa��
�|�|�t��2�
�|�|�t��J�
�|�|�t��

?�	44��|�|�t��[�
�|�|�t��Y�	45�V
r)r�sql_env.modelsrrrrr"rIrSrhrl�rr�<module>ro
s0�


�	�8�8�

�
8�

� 
'�46@

N
�F
V
r