SHUBHAMOS/meta-pytorch-hackathon
0
1�
2 ��i� � �� � S r SSKJr SSKJr SSKJrJrJrJrJ r SSK3J r J r \S r
\S r\S r\S r " S4 S \ 5 r " S S
\ 5 r " S S\ 5 r " S S\ 5 r " S S\ 5 r " S S\ 5 rg)u 5SHUBHAMOS: AI Email Operations & Triage Environment6models.py — All typed Pydantic v2 data models7 8These models define the data contract between environment, agents, and graders.9- Email: full internal representation with ground truth labels10- ObservationEmail: agent-facing truncated view (no ground truth)11- Action: structured agent action with type dispatch12- Observation: what the agent receives each step13- State: full internal state (includes ground truth, used by graders)14- StepInfo: per-step metadata15� )� annotations)�datetime)�Literal�Optional�List�Dict�Any)� BaseModel�Field)�spam�general_inquiry�
billing_issue�urgent_complaint� tech_support)�low�medium�high�unknown)�positive�neutral�negative)�classify_email� set_priority� draft_reply�
mark_resolved�escalate_email� ignore_emailc � � \ rS rSr% Sr\" SSS9rS\S' \" SSS9rS\S ' \" SS16S9r S\S ' \" SS S9r17S\S
' \" SSS9r S\S' \" SSS9r S\S' \" SSS9r
S\S' \" SSS9rS\S' \" SSS9rS \S ' \" S S S9rS\S ' \" S!S"S9rS#\S$' \" S!S%S9rS#\S&' \" S!S'S9rS#\S(' \" S!S)S9rS#\S*' \" SS+S9rS,\S-' S0S. jrS/rg)1�Email�# zwFull internal email representation. Ground truth labels are stored here18and never exposed to the agent via Observation..z(Unique email identifier e.g. 'email_001'�� description�str�idzEmail subject line�subjectzFull email body text�bodyzSender email address�senderzArrival timestampr � timestampzTrue category for grading�
EmailCategory� gt_categoryz(True priority for grading (no 'unknown')�Priority� gt_priorityzBody sentiment� Sentiment� sentimentNzAgent-assigned category�Optional[EmailCategory]�categoryr zAgent-assigned priority�priorityFzMarked resolved by agent�bool�resolvedzEscalated by agent� escalatedzIgnored by agent�ignoredzReply has been drafted�
reply_draftedzDrafted reply text�
Optional[str]�19reply_textc �f � U R =( d U R =( d U R $ )zHTrue if email has been terminally acted on (resolved/escalated/ignored).)r3 r4 r5 )�selfs �server/models.py� is_processed�Email.is_processed= s � ��}�}�>����>�$�,�,�>� � )�returnr2 )�__name__�20__module__� __qualname__�__firstlineno__�__doc__r r$ �__annotations__r% r& r' r( r* r, r. r0 r1 r3 r4 r5 r6 r8 r< �__static_attributes__r? r> r; r r # s# � �7� �C�%O�P�B��P���*>�?�G�S�?��c�'=�>�D�#�>���)?�@�F�C�@� ��1D�E�I�x�E� "'�s�8S�!T�K� �T� !�#�3]� ^�K��^� ��2B�C�I�y�C� ).�d�@Y�(Z�H�%�Z� �y�6O�P�H�h�P� �5�.H�I�H�d�I��E�/C�D�I�t�D��%�-?�@�G�T�@� ��3K�L�M�4�L� %�d�8L� M�J�
�M�?r> r c �� � \ rS rSr% SrS\S' S\S' S\S' S\S' S \S21' \" S S S
9rS\S' \" SSS
9rS\S' \" SSS
9r S\S' Sr22S\S' Sr S\S' Sr S\S' Sr
S\S ' S rg) �ObservationEmail�D zKTruncated email view exposed to the agent. Contains no ground truth labels.r# r$ r% r' r r( r- r. .z First 200 chars of body onlyr! � body_previewNz Agent-assigned category so farr/ r0 r z Agent-assigned priority so farr+ r1 Fr2 r3 r4 r5 r6 r? )rA rB rC rD rE rF r rK r0 r1 r3 r4 r5 r6 rG r? r> r; rI rI D s � �U� �G�
�L� �K����� �c�/M�N�L�#�N�(-�d�@`�(a�H�%�a� �y�6V�W�H�h�W��H�d���I�t���G�T�� �M�4� r> rI c � � \ rS rSr% Sr\" SSS9rS\S' \" SSS9rS \S23' \" S S S9r S
\S' \" S SS9r24S\S' \" S SS9r S\S' Sr g )�Action�W a^ Structured agent action. action_type determines which payload fields apply.25 26Action types and their required fields:27- classify_email: email_id, category (required)28- set_priority: email_id, level (required)29- draft_reply: email_id, text (required)30- mark_resolved: email_id only31- escalate_email: email_id only32- ignore_email: email_id only33.zWhich action to performr! �34ActionType� action_typezTarget email IDr# �email_idNzUsed by classify_emailr/ r0 zUsed by set_priorityzOptional[Priority]�levelzUsed by draft_replyr7 �textr? )
rA rB rC rD rE r rP rF rQ r0 rR rS rG r? r> r; rM rM W se � � � $�C�5N� O�K��O��#�+<�=�H�c�=�(-�d�@X�(Y�H�%�Y� %�d�8N� O�E� �O� ��2G�H�D�-�Hr> rM c �� � \ rS rSr% Sr\" SSS9rS\S' \" SSS9rS \S35' \" SS S9r S \S ' \" SS
S9r36S \S' \" SSS9r S \S' \" SSS9r S \S' \" SSS9r
S \S' \" SSS9rS \S' \" SSS9rS\S' Srg) � Observation�l zAWhat the agent sees after each step or on reset. No ground truth..zCurrent inbox stater! zList[ObservationEmail]�emailsz Total emails in this episode�int� total_emailsz#Emails not yet terminally processed�
pending_countz Emails successfully resolved�resolved_countzEmails escalated�escalated_countzEmails ignored�
ignored_countz'Steps taken so far (0-indexed at reset)�37step_countz'Maximum steps before forced termination� max_stepsz"step_count / max_steps, 0.0 to 1.0�float�
elapsed_ratior? N)rA rB rC rD rE r rW rF rY rZ r[ r\ r] r^ r_ ra rG r? r> r; rU rU l s� � �K�%*�3�<Q�%R�F� "�R� �c�/M�N�L�#�N� �s�0U�V�M�3�V� ��1O�P�N�C�P� ��2D�E�O�S�E� �s�0@�A�M�3�A��C�-V�W�J��W��3�,U�V�I�s�V� ��2V�W�M�5�Wr> rU c �� � \ rS rSr% Sr\" SSS9rS\S' \" \SS 9r S38\S ' \" S S
S9r39S\S' \" S SS9r S\S' \" SSS9r S\S' \" SSS9r
S\S' \" S SS9rS\S' \" S S S9rS \S ' S rg!)"�State�| zMComplete internal environment state including ground truth. Graders use this..z#All emails with ground truth labelsr! z List[Email]rW z.Ordered list of all actions taken this episode)�default_factoryr" zList[Dict[str, Any]]�action_historyr z Steps taken in current episoderX r^ zMax steps for current taskr_ Fz Whether episode has terminatedr2 �done� z Cumulative reward this episoder` � total_rewardz!Random seed used for this episode�seed� z%Task identifier: easy / medium / hardr# �task_idr? N)rA rB rC rD rE r rW rF �listrf r^ r_ rg ri rj rl rG r? r> r; rc rc | s� � �W� ��1V�W�F�K�W�+0� �D�,�N�(� � �A�+K�L�J��L��1�*F�G�I�s�G��u�*J�K�D�$�K� ��1Q�R�L�%�R��a�%H�I�D�#�I���)P�Q�G�S�Qr> rc c � � \ rS rSr% Sr\" SSS9rS\S' \" SSS9rS\S ' \" SS40S9r S \S ' \" S
SS9r41S\S' \" SSS9r S\S' \" SSS9r S \S' Sr
g
)�StepInfo� z6Structured info returned in the info dict from step()..zAction that was appliedr! r# rP zEmail that was targetedrQ z Whether the action was validr2 �validNzError message if invalidr7 �errorrh zReward earned this stepr` � step_rewardFz%Whether episode ended after this steprg r? )rA rB rC rD rE r rP rF rQ rq rr rs rg rG r? r> r; ro ro � sq � �@� �S�.G�H�K��H��#�+D�E�H�c�E���)G�H�E�4�H� ��3M�N�E�=�N� �s�0I�J�K��J��u�*Q�R�D�$�Rr> ro N)rE �42__future__r r �typingr r r r r �pydanticr43 r r) r+ r- rO r rI rM rU rc ro r? r> r; �<module>rw s� �� � #� � 5� 5� %� �f�g�
� �5� 6�� �5� 6� �
���44� ?�I� ?�B �y� �&I�Y� I�* X�)� X�
R�I�
R�$S�y� Sr>