54 lines
9.3 KiB
Plaintext
54 lines
9.3 KiB
Plaintext
|
o
|
|||
|
|
:‰Úi^ ã @ sz d Z ddlZddlZddlZddlmZmZmZmZ ddl m
|
|||
|
|
Z
|
|||
|
|
ddl m Z m
|
|||
|
|
Z
|
|||
|
|
mZmZmZmZmZmZ G dd„ dƒZdS ) u5
|
|||
|
|
任务公告æ�¿ - 任务状æ€�管ç�†ã€�ä¿¡æ�¯å˜å‚¨
|
|||
|
|
é N)ÚDictÚListÚOptionalÚAny)ÚPathé )ÚTaskÚBidÚ ExecutionÚAttemptÚ
|
|||
|
|
TaskStatusÚ TaskGraphÚ UserRequestÚ AgentProfilec @ s e Zd ZdZdPdefdd„Zdd„ Zdd „ Zd
|
|||
|
|
ed efd d
|
|||
|
|
„Z ded e
|
|||
|
|
e fdd„Z defdd„Z dQde
|
|||
|
|
d ee fdd„Zdefdd„Zded e
|
|||
|
|
e fd d „Zded efd d „Zd ed ee fd!d"„ZdRde
|
|||
|
|
d ee fd$d%„Zd&efd'd(„Zded ee fd)d*„Zdefd+d,„Zd-efd.d/„Z d0ed e
|
|||
|
|
e fd1d2„Z d-efd3d4„Z d5e fd6d7„Z ded ee fd8d9„Z!ded e
|
|||
|
|
fd:d;„Z"d<e#fd=d>„Z$d?efd@dA„Z%d?ed e
|
|||
|
|
e# fdBdC„Z&d ee# fdDdE„Z' FdSd?edGe(dHe)dIe)fdJdK„Z*ded ee# fdLdM„Z+d e,fdNdO„Z-dS )TÚ TaskBoardu3 任务公告æ�¿ - å˜å‚¨å’Œç®¡ç�†æ‰€æœ‰ä»»åŠ¡çŠ¶æ€�NÚ storage_pathc C sf t |p
|
|||
|
|
t ¡ d d ƒ| _| jjjddd� i | _i | _i | _i | _i | _ i | _
|
|||
|
|
t ¡ | _
|
|||
|
|
| ¡ d S )Nz .multi_agentztask_board.jsonT)ÚparentsÚexist_ok)r Úhomer ÚparentÚmkdirÚ
|
|||
|
|
user_requestsÚtasksÚbidsÚ
|
|||
|
|
executionsÚattemptsÚagentsÚ threadingÚLockÚ_lockÚ_load)Úselfr © r" úP/home/xian/.openclaw/workspace-coder/works/multi-agent-bidding/app/task_board.pyÚ__init__ s
|
|||
|
|
zTaskBoard.__init__c C sd | j ¡ r0z t| j dƒ�}t |¡}W d ƒ W dS 1 s w Y W dS ty/ Y dS w dS )u ä»Žæ–‡ä»¶åŠ è½½ÚrN)r ÚexistsÚopenÚjsonÚloadÚ Exception)r! ÚfÚdatar" r" r# r " s
|
|||
|
|
&ÿ ÿûzTaskBoard._loadc
|
|||
|
|
C sÀ zEdd„ | j ¡ D ƒdd„ | j ¡ D ƒdd„ | j ¡ D ƒt ¡ dœ}t| jdƒ�}tj||ddd � W d
|
|||
|
|
ƒ W d
|
|||
|
|
S 1 s>w Y W d
|
|||
|
|
S t y_ } zt
|
|||
|
|
d |› �ƒ W Y d
|
|||
|
|
}~d
|
|||
|
|
S d
|
|||
|
|
}~ww ) u ä¿�å˜åˆ°æ–‡ä»¶c S ó i | ] \}}|| ¡ “qS r" ©Úto_dict©Ú.0ÚkÚvr" r" r# Ú
|
|||
|
|
<dictcomp>0 ó z#TaskBoard._save.<locals>.<dictcomp>c S r- r" r. r0 r" r" r# r4 1 r5 c S r- r" r. r0 r" r" r# r4 2 r5 )r r r Úsaved_atÚwFé )Ú ensure_asciiÚindentNu ä¿�å˜å¤±è´¥: ) r Úitemsr r Útimer' r r( Údumpr* Úprint)r! r, r+ Úer" r" r# Ú_save, s ü&ÿ €ÿzTaskBoard._saveÚcontentÚreturnc C sJ t |d�}| j� || j|j< | ¡ W d ƒ |S 1 s w Y |S )u 创建用户请求)rA N)r r r Úidr@ )r! rA Úrequestr" r" r# Úcreate_request< s
|
|||
|
|
|
|||
|
|
|
|||
|
|
þýzTaskBoard.create_requestÚ
|
|||
|
|
request_idc C ó | j |¡S )u 获å�–用户请求)r Úget)r! rF r" r" r# Ú get_requestD ó zTaskBoard.get_requestrD c C ó@ | j � || j|j< | ¡ W d ƒ dS 1 sw Y dS )u 更新用户请求N)r r rC r@ )r! rD r" r" r# Úupdate_requestH ó
|
|||
|
|
"þzTaskBoard.update_requesté2 Úlimitc C ó, t | j ¡ ƒ}|jdd„ dd� |d|… S )u 列出用户请求c S ó | j S ©N©Ú
|
|||
|
|
created_at)r% r" r" r# Ú<lambda>Q ó z)TaskBoard.list_requests.<locals>.<lambda>T©ÚkeyÚreverseN)Úlistr ÚvaluesÚsort)r! rO Úrequestsr" r" r# Ú
|
|||
|
|
list_requestsN ó zTaskBoard.list_requestsÚtaskc C sX | j � || j|j< i | j|j< g | j|j< | ¡ W d ƒ dS 1 s%w Y dS )u æ·»åŠ ä»»åŠ¡N)r r rC r r r@ )r! r` r" r" r# Úadd_taskV s
|
|||
|
|
"üzTaskBoard.add_taskÚtask_idc C rG )u 获å�–任务)r rH ©r! rb r" r" r# Úget_task^ rJ zTaskBoard.get_taskÚstatusc C sZ | j � || jv r|| j| _| ¡ W d ƒ dS W d ƒ dS 1 s&w Y dS )u 更新任务状æ€�N)r r re r@ )r! rb re r" r" r# Úupdate_task_statusb s
|
|||
|
|
|
|||
|
|
ý"ÿz TaskBoard.update_task_statusc s ‡ fdd„| j ¡ D ƒS )u 获å�–指定状æ€�的任务c ó g | ] }|j ˆ kr|‘qS r" ©re ©r1 Útrh r" r# Ú
|
|||
|
|
<listcomp>k r5 z1TaskBoard.get_tasks_by_status.<locals>.<listcomp>)r r[ )r! re r" rh r# Úget_tasks_by_statusi s z TaskBoard.get_tasks_by_statuséd c C rP )u 列出所有任务c S rQ rR rS )rj r" r" r# rU p rV z&TaskBoard.list_tasks.<locals>.<lambda>TrW N)rZ r r[ r\ )r! rO r r" r" r# Ú
|
|||
|
|
list_tasksm r_ zTaskBoard.list_tasksÚbidc C sZ | j � |j| jv r|| j|j |j< W d ƒ dS W d ƒ dS 1 s&w Y dS )u æ·»åŠ ç«žæ ‡N)r rb r rC )r! ro r" r" r# Úadd_bidu ó þ"ÿzTaskBoard.add_bidc C s || j v rt| j | ¡ ƒS g S )u 获å�–ä»»åŠ¡çš„ç«žæ ‡åˆ—è¡¨)r rZ r[ rc r" r" r# Úget_bids{ s
|
|||
|
|
zTaskBoard.get_bidsc C sP | j � || jv ri | j|< W d ƒ dS W d ƒ dS 1 s!w Y dS )u æ¸…é™¤ä»»åŠ¡çš„ç«žæ ‡N)r r rc r" r" r# Ú
|
|||
|
|
clear_bids� s
|
|||
|
|
þ"ÿzTaskBoard.clear_bidsÚ executionc C s8 | j � || j|j< W d ƒ dS 1 sw Y dS )u æ·»åŠ æ‰§è¡Œè®°å½•N)r r rC ©r! rt r" r" r# Ú
|
|||
|
|
add_execution‰ s "ÿzTaskBoard.add_executionÚ execution_idc C rG )u 获å�–执行记录)r rH )r! rw r" r" r# Ú
|
|||
|
|
get_executionŽ rJ zTaskBoard.get_executionc C rK )u 更新执行记录N)r r rC r@ ru r" r" r# Úupdate_execution’ rM zTaskBoard.update_executionÚattemptc C sZ | j � |j| jv r| j|j |¡ W d ƒ dS W d ƒ dS 1 s&w Y dS )u æ·»åŠ å°�试记录N)r rb r Úappend)r! rz r" r" r# Ú add_attemptš rq zTaskBoard.add_attemptc C s || j v r
|
|||
|
|
| j | S g S )u 获å�–任务的å°�试记录)r rc r" r" r# Ú get_attempts s
|
|||
|
|
|
|||
|
|
zTaskBoard.get_attemptsc C s t | |¡ƒS )u 获å�–任务的å°�试次数)Úlenr} rc r" r" r# Úget_attempt_count¦ ó zTaskBoard.get_attempt_countÚagentc C rK )u 注册AgentN)r r rC r@ )r! r� r" r" r# Úregister_agent¬ rM zTaskBoard.register_agentÚagent_idc C sB | j � | j |d¡ | ¡ W d ƒ dS 1 sw Y dS )u 注销AgentN)r r Úpopr@ ©r! rƒ r" r" r# Úunregister_agent² s
|