�
{�8i� � �� � d dl mZ d dlmZ d dlmZ d dlZ G d� de� Z G d� de� Zed e d
e d e fd �� Z
ed
e fd�� Z eddee d ee de fd�� Z
eddee d ee fd�� Zy)� )�Enum)�tool)�OptionalNc �, � e Zd ZdZdZdZdZdZdZdZ dZ
y )
�Category�work�health�learning�personal�finance�social�hobby� self_careN) �__name__�
__module__� __qualname__�WORK�HEALTH�LEARNING�PERSONAL�FINANCE�SOCIAL�HOBBY� SELF_CARE� � �/C:\Users\Workspace\ProductivityTracker\tools.pyr r s* � � �D�
�F��H��H��G�
�F� �E��Ir r c � � e Zd ZdZdZdZdZy)�Status�todo�blocked�started� completedN)r r r �TODO�BLOCKED�STARTED� COMPLETEDr r r r r s � � �D��G��G��Ir r � task_name�category�statusc
�t ��� t �fd�t D � � }t �fd�t D � � }t d�� d|� d�� d|� �� t j
| |j |j � d| � d|j � d|j � d �S # t $ r}d
t |� � �cY d }~S d }~ww xY w) aQ
Log a new task with its category and current status.
Use this tool when the user wants to record or track a task they're working on or planning to do.
Common phrases that should use this tool:
- "Log a task..."
- "Add a task..."
- "I'm starting/working on..."
- "Track this task..."
Args:
task_name: Name or description of the task
category: Type of task
- work: Professional tasks, meetings, projects, deadlines
- health: Exercise, medical appointments, nutrition, wellness
- learning: Study, courses, reading, skill development
- personal: Errands, household chores, organization
- finance: Budgeting, bills, expenses, financial planning
- social: Time with friends, family interactions, social events
- hobby: Creative projects, recreational activities, entertainment
- self_care: Meditation, journaling, relaxation, personal growth
status: Current state - todo (planned), started (in progress), completed (finished), blocked (waiting on something)
Returns:
Confirmation message with task details
Examples:
- Log workout task: category=health, status=completed
- Log code review: category=work, status=started
- Log bill payment: category=finance, status=todo
c 3 �^ �K � | ]$ }|j �j � k( s�!|�� �& y �w�N��value�lower)�.0�cr) s �r � <genexpr>zlog_task.<locals>.<genexpr>> s$ �� �� � P��1�A�G�G�x�~�~�?O�4O�Q��� �"-�-c 3 �^ �K � | ]$ }|j �j � k( s�!|�� �& y �wr- r. )r1 �sr* s �r r3 zlog_task.<locals>.<genexpr>? s# �� �� �J�f�����6�<�<�>�0I�1�f�r4 z Converting: z -> z, zTask 'z' logged under category 'z' with status 'z'.�Error logging task: N) �nextr r �print�database� log_task_dbr/ � Exception�str)r( r) r* �
category_enum� status_enum�es `` r �log_taskrA s� �� �H/� � P�� P�P�
��J�f�J�J� �
� �X�J�d�=�/��F�8�4� �}�U�V�� � �Y�
�(;�(;�[�=N�=N�O�� �{�";�M�<O�<O�;P�P_�`k�`q�`q�_r�rt�u�u�� � /�%�c�!�f�X�.�.��/�s �BB � B7�
B2�,B7�2B7�periodc �< � t j | � }d| � d|� �S )a
Retrieve a summary of logged tasks for a specific time period.
Use this tool when the user asks about their progress, productivity, or wants to see what they've accomplished.
Common questions that should use this tool:
- "How is my day/week/month going?"
- "What have I done today?"
- "Show me my progress"
- "What tasks did I complete?"
Args:
period: Time period to summarize - "today", "week", or "month"
Returns:
Dictionary with task counts by category and status
z Summary for z : summary: )r: �get_summary_db)rB �summarys r � get_summaryrF F s' � �* �%�%�f�-�G�
�&���W�I� 6�6r �task_id�
new_statusc �� �� t �fd�t D � � }|r| s|sy t j | ||j � }|S # t
$ r}dt
|� � �cY d}~S d}~ww xY w)a�
Update the status of an existing task.
Use this tool when the user wants to change the status of a task they have already logged.
Common phrases that should use this tool:
- "Update task status..."
- "Change status of..."
- "Mark task as..."
- "Set task to..."
Args:
task_id: Unique identifier of the task (optional if task_name is provided)
task_name: Name or description of the task (optional if task_id is provided)
new_status: New status to set - todo (planned), started (in progress), completed (finished), blocked (waiting on something)
Returns:
Confirmation message with updated task details
c 3 �^ �K � | ]$ }|j �j � k( s�!|�� �& y �wr- r. )r1 r6 rH s �r r3 z%update_task_status.<locals>.<genexpr>s s& �� �� �N�f�����:�;K�;K�;M�0M�1�f�r4 z8Must provide new_status and either task_id or task_name.r7 N)r8 r r: �update_task_status_dbr/ r< r= )rG r( rH �new_status_enum�resultr@ s ` r �update_task_statusrN ^ sd �� �* �N�f�N�N�O� �7�9�I�/��/�/���O�DY�DY�Z���
�� � /�%�c�!�f�X�.�.��/�s �"A � A&�
A!�A&�!A&c �< � | s|syt j | |� }|S )a(
Remove an existing task from the database.
Use this tool when the user wants to delete a task they have logged.
Common phrases that should use this tool:
- "Remove task..."
- "Delete task..."
- "Clear task..."
- "Erase task..."
- "remove duplicate tasks..."
Args:
task_id: Unique identifier of the task (optional if task_name is provided)
task_name: Name or description of the task (optional if task_id is provided)
Returns:
Confirmation message with details of the removed task
z:Must provide either task_id or task_name to remove a task.)r: �remove_task_db)rG r( rM s r � remove_taskrQ s% � �* �9�K� �
$�
$�W�i�
8�F� �Mr )NNN)NN)�enumr �langchain.toolsr �typingr r: r r r= rA rF �intrN rQ r r r �<module>rV s� �� � � � � �t� � �T� � �+/�� +/�s� +/�C� +/� �+/�Z �7�� 7� �7�. � /���
� /��#� � /�cf� /� � /�@ ���#� � �(�3�-� � �r
