�
���g� � � � U d dl Z d dlmZ d dlmZ i Zeee j f ed< i Z d defd�Z
d d�Z defd�Z d � Z
d
� Zdefd �ZdS )
� N)�Dict)�uuid4�tasks�task_idc � � t � | d� � |rg| t � |g � � v rLt | � | � � t | s!t � |d� � dS dS dS dS )zQ
Remove a completed or canceled task from the global `tasks` dictionary.
N)r �pop�
chat_tasks�get�remove)r �ids �`/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/open_webui/tasks.py� cleanup_taskr s� � �
�I�I�g�t� � � �
� %�g�� � ��B�/�/�/�/��2�� � �g�&�&�&��"�~� %� �N�N�2�t� $� $� $� $� $�%� %�/�/� %� %� c �> ��� t t � � � � �t j | � � }|� ��fd�� � |t
�< t � �� � r!t � � �� � n �gt �< �|fS )zM
Create a new asyncio task and add it to the global task dictionary.
c �$ �� t ��� � S �N)r )�tr r s ��r
�<lambda>z create_task.<locals>.<lambda> s �� �\�'�2�%>�%>� r ) �strr �asyncio� create_task�add_done_callbackr r r
�append)� coroutiner �taskr s ` @r
r r s� ��� � �%�'�'�l�l�G� � �y� )� )�D� ���>�>�>�>�>�?�?�?��E�'�N� �~�~�b��� #��2�� � �g�&�&�&�&�!��
�2�� �D�=�r c �6 � t � | � � S )z)
Retrieve a task by its task ID.
)r r
)r s r
�get_taskr , s � � �9�9�W� � � r c �N � t t � � � � � S )z-
List all currently active task IDs.
)�listr �keys� r r
�
list_tasksr" 3 s � � ��
�
� � � � � r c �8 � t � | g � � S )z7
List all tasks associated with a specific ID.
)r r
)r s r
�list_task_ids_by_chat_idr$ : s � � �>�>�"�b� !� !�!r c � �$ K � t � | � � }|st d| � d�� � �|� � � |� d{V �� n9# t j $ r' t � | d� � dd| � d�d�cY S w xY wdd | � d
�d�S ) zH
Cancel a running task and remove it from the global task list.
z
Task with ID z not found.NTzTask z successfully stopped.)�status�messageFzFailed to stop task �.)r r
�
ValueError�cancelr �CancelledErrorr )r r s r
� stop_taskr, A s� � � � � �9�9�W� � �D� � ?��=��=�=�=�>�>�>��K�K�M�M�M�T��
�
�
�
�
�
�
�
�� � !� T� T� T�
� � �'�4� � � � �+R�7�+R�+R�+R�S�S�S�S�S�T����
�(I�w�(I�(I�(I� J� J�Js �A �3B�Br )r �typingr �uuidr r r �Task�__annotations__r r r r r" r$ r, r! r r
�<module>r1 s� �� ����� � � � � � � � � � � � � "$��t�C�� � � � #� #� #�
�
�
%�
%�#�
%�
%�
%�
%�� � � �( �c� � � � � � � �"� "� "�K�S� K� K� K� K� K� Kr