180 lines
11 KiB
Plaintext
180 lines
11 KiB
Plaintext
|
o
|
|||
|
|
¸‰Úi[A ã @ s¢ d Z ddlZddlZddlmZmZmZmZ ddlm Z m
|
|||
|
|
Z
|
|||
|
|
m Z m Z m
|
|||
|
|
Z
|
|||
|
|
mZmZmZmZmZ ddlmZ ddlmZ ddlmZ dd lmZmZ G d
|
|||
|
|
d „ d ƒZ dS ) uX
|
|||
|
|
规划Agent (Orchestrator) - æ„�图ç�†è§£ã€�任务拆分ã€�ç«žæ ‡ç®¡ç�†ã€�结果验è¯�
|
|||
|
|
é N)ÚDictÚListÚOptionalÚAnyé )
|
|||
|
|
ÚTaskÚTaskTypeÚ
|
|||
|
|
TaskStatusÚ TaskGraphÚBidÚ ExecutionÚAttemptÚValidationResultÚ UserRequestÚ AgentProfile)Ú TaskBoard)Ú BidEvaluator)ÚExecutionMonitorManager)Ú LLMClientÚdefault_clientc @ s\ e Zd ZdZ d0dedefdd„Zdedefd d
|
|||
|
|
„Z d edede
|
|||
|
|
fd d
|
|||
|
|
„Z de de
|
|||
|
|
e fdd„Z d0de de
|
|||
|
|
e dede
|
|||
|
|
e fdd„Zde de
|
|||
|
|
e defdd„Z d0de dedede
|
|||
|
|
e def
|
|||
|
|
dd „Zd ede fd d „Zd ede de
|
|||
|
|
e fd d!„Zd ede d"efd#d$„Zde d ed%efd&d'„Z de d(e defd)d*„Z dede fd+d,„Z d-edefd.d/„Z!dS )1Ú Orchestratoru 规划Agent - æ ¸å¿ƒè°ƒåº¦å¼•æ“ŽNÚ
|
|||
|
|
task_boardÚ
|
|||
|
|
llm_clientc C s< || _ |pt| _tƒ | _tƒ | _d| _d| _d| _ d| _
|
|||
|
|
d S )Né r é i ) r r Úllmr Ú
|
|||
|
|
bid_evaluatorr Úexecution_managerÚbidding_windowÚmax_retry_per_agentÚmax_total_attemptsÚmax_total_time)Úselfr r © r# úR/home/xian/.openclaw/workspace-coder/works/multi-agent-bidding/app/orchestrator.pyÚ__init__ s
|
|||
|
|
|
|||
|
|
zOrchestrator.__init__Ú user_requestÚreturnc
|
|||
|
|
C sF z | j |¡}|W S ty" } z
|
|||
|
|
|g dg dœW Y d}~S d}~ww )u›
|
|||
|
|
�图�解
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
user_request: 用户原始请求
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
�图分�结果
|
|||
|
|
F)ÚintentÚkeywordsÚneed_clarificationÚ questionsN)r Úanalyze_intentÚ Exception)r" r& r( Úer# r# r$ Úunderstand_intent( s
|
|||
|
|
ü€þz Orchestrator.understand_intentr( c
|
|||
|
|
C s z=| j ||¡}tƒ }|D ]-}| d¡dkrtjntj}t|| dd¡| di ¡| di ¡| dg ¡d�}| |¡ q
|
|||
|
|
|W S t y` } ztƒ }ttj
|
|||
|
|
|d �}| |¡ |W Y d
|
|||
|
|
}~S d
|
|||
|
|
}~ww ) u¹
|
|||
|
|
任务拆分
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
intent: �图分�结果
|
|||
|
|
user_request: 用户原始请求
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
TaskGraph
|
|||
|
|
ÚtypeÚparallelÚ descriptionÚ Ú input_schemaÚ
|
|||
|
|
output_schemaÚ dependencies)r0 r2 Ú
|
|||
|
|
input_datar5 r6 )r0 r2 N) r Ú split_tasksr
|
|||
|
|
Úgetr ÚPARALLELÚSERIALr Úadd_taskr- Ú INDEPENDENT) r" r( r& Ú task_listÚgraphÚtÚ task_typeÚtaskr. r# r# r$ r8 @ s0
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
û þ
|
|||
|
|
€øzOrchestrator.split_tasksrB c C s$ t j|_| j |¡ | j |¡}|S )u•
|
|||
|
|
�布任务
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
task: 任务
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
能处�该任务的Agent列表
|
|||
|
|
)r Ú PUBLISHEDÚstatusr r< Úfind_capable_agents)r" rB Úcapable_agentsr# r# r$ Ú publish_taski s zOrchestrator.publish_taskÚagentsÚwindow_secondsc
|
|||
|
|
|
|||
|
|
C sÐ |p| j }tj|_g }|D ]X}zD| j | ¡ | ¡ ¡}t|j|jt |
|
|||
|
|
dd¡ƒt |
|
|||
|
|
dd¡ƒt |
|
|||
|
|
dd¡ƒ|
|
|||
|
|
dd¡|
|
|||
|
|
dg ¡|
|
|||
|
|
d g ¡d
|
|||
|
|
�}| |¡ | j
|
|||
|
|
|¡ W q
|
|||
|
|
tye } zW Y d } ~ q
|
|||
|
|
d } ~ ww |S ) uÐ
|
|||
|
|
æ”¶é›†ç«žæ ‡
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
task: 任务
|
|||
|
|
agents: Agent列表
|
|||
|
|
window_seconds: ç«žæ ‡çª—å�£æ—¶é—´
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
ç«žæ ‡åˆ—è¡¨
|
|||
|
|
Úcapability_matchç à?Úestimated_timeé< Ú
|
|||
|
|
confidenceÚapproachr3 Ú
|
|||
|
|
prerequisitesÚalternative_approaches)Útask_idÚagent_idrJ rL rN rO rP rQ N)r r ÚBIDDINGrD r Ú generate_bidÚto_dictr ÚidÚfloatr9 ÚintÚappendr Úadd_bidr- )
|
|||
|
|
r" rB rH rI ÚwindowÚbidsÚagentÚbid_dataÚbidr. r# r# r$ Ú collect_bids~ s.
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
ø
|
|||
|
|
€þzOrchestrator.collect_bidsr] c C sR dd„ | j ¡ D ƒ}| j ||¡\}}}|r$| j |||j¡}|||fS ddg fS )uÁ
|
|||
|
|
è¯„ä¼°ç«žæ ‡
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
task: 任务
|
|||
|
|
bids: ç«žæ ‡åˆ—è¡¨
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
(selected_agent, selected_bid, backup_agents)
|
|||
|
|
c S s i | ]}|j |“qS r# )rW ©Ú.0Úar# r# r$ Ú
|
|||
|
|
<dictcomp>º s ÿz.Orchestrator.evaluate_bids.<locals>.<dictcomp>N)r Ú list_agentsr Úselect_best_bidÚget_backup_agentsrW )r" rB r] Ú agents_dictÚbest_bidÚ
|
|||
|
|
best_agentÚscoresÚ
|
|||
|
|
backup_agentsr# r# r$ Ú
|
|||
|
|
evaluate_bids¯ s ÿÿ
|
|||
|
|
|
|||
|
|
zOrchestrator.evaluate_bidsr^ r` rm c sŠ t jˆ_‡fdd„}ˆjjˆ||‡‡fdd„‡ ‡‡fdd„‡‡fdd„dœd�}ˆ r4ˆ D ]}| |¡ q,t jˆ_| |¡}ˆj |¡ |S ) uî
|
|||
|
|
分�并执行任务
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
task: 任务
|
|||
|
|
agent: 执行Agent
|
|||
|
|
bid: ç«žæ ‡ä¹¦
|
|||
|
|
backup_agents: 备选Agent
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Execution对象
|
|||
|
|
c
|
|||
|
|
sB zˆ j | ¡ |j¡}| d¡p|W S ty } z|‚d }~ww )NÚresult)r Ú execute_taskrV rO r9 r- )r@ rd Úbro r. )r" r# r$ Ú execute_funcá s €ÿz5Orchestrator.assign_and_execute.<locals>.execute_funcc s ˆ | ˆ¡S ©N)Ú_on_execution_complete©Úexec©r" rB r# r$ Ú<lambda>î s z1Orchestrator.assign_and_execute.<locals>.<lambda>c s ˆ | ˆˆ ¡S rs )Ú_on_execution_timeoutru ©rm r" rB r# r$ rx ï ó c s ˆ | ˆ|¡S rs )Ú_on_execution_error)rv r. rw r# r$ rx ð r{ )ÚcompleteÚtimeoutÚerror)rB r^ r` Ú callbacks)
|
|||
|
|
r ÚASSIGNEDrD r Úcreate_monitorÚadd_backup_agentÚ EXECUTINGÚstartr Ú
|
|||
|
|
add_execution) r" rB r^ r` rm rr ÚmonitorÚbackupÚ executionr# rz r$ Úassign_and_executeË s$ ýü
|
|||
|
|
z Orchestrator.assign_and_executer‰ c C sp t j|_| ||j¡}|jr/t j|_| j |j |j¡ |j
|
|||
|
|
|j }| jj |j
|
|||
|
|
d||jd� dS | |||¡ dS )u 执行完æˆ�回调T)ÚsuccessÚdurationÚ
|
|||
|
|
quality_scoreN)r Ú
|
|||
|
|
VALIDATINGrD Úvalidate_resultro ÚpassedÚ COMPLETEDr Úupdate_task_statusrW Úend_timeÚ
|
|||
|
|
start_timeÚupdate_agent_statsrS ÚscoreÚ_handle_validation_failure)r" r‰ rB Ú
|
|||
|
|
validationrΠr# r# r$ rt s
|
|||
|
|
üz#Orchestrator._on_execution_completec sÄ t |j|j|dd|j|j d�}| j |¡ |rS| j |j¡| jk rS|D ].‰ ‡ fdd„| j |j¡D ƒ}t
|
|||
|
|
|ƒ| j k rRt |jˆ jddddd �}|
|
|||
|
|
|ˆ |g ¡ d
|
|||
|
|
S q$tj|_| j |j|j¡ d
|
|||
|
|
S ) u 执行超时回调Fr~ )rR rS r‰ r‹ Ú failure_typerŒ c s g | ]
|
|||
|
|
}|j ˆ jkr|‘qS r# )rS rW rb ©rˆ r# r$ Ú
|
|||
|
|
<listcomp>) ó z6Orchestrator._on_execution_timeout.<locals>.<listcomp>rK rM u 备选执行)rR rS rJ rL rN rO N)r
|
|||
|
|
rW rS r“ r” r Ú add_attemptÚget_attempt_countr Ú get_attemptsÚlenr r rŠ r ÚFAILEDrD r’ )r" r‰ rB rm ÚattemptÚattemptsr` r# rš r$ ry s4
|
|||
|
|
ú úõz"Orchestrator._on_execution_timeoutr c C sR t |j|j|ddt ¡ |j t|ƒd�}| j |¡ tj |_
|
|||
|
|
| j |j|j
|
|||
|
|
¡ dS )u 执行错误回调Fr )rR rS r‰ r‹ r™ rŒ ÚfeedbackN) r
|
|||
|
|
rW rS Útimer” Ústrr r� r r¡ rD r’ )r" r‰ rB r r¢ r# r# r$ r| ; s ù z Orchestrator._on_execution_errorr˜ c s¸ t |jˆ jˆ |ddˆ jˆ j t |j¡d�}| j |¡ ‡ fdd„| j
|
|||
|
|
|j¡D ƒ}t |ƒ| j k rMt
|
|||
|
|
j|_| j |j|j¡ t
|
|||
|
|
j|_| j |j|j¡ dS t
|
|||
|
|
j|_| j |j|j¡ dS )u 处ç�†éªŒè¯�失败Fr˜ )rR rS r‰ r˜ r‹ r™ rŒ r¤ c s g | ]
|
|||
|
|
}|j ˆ j kr|‘qS r# )rS rb ©r‰ r# r$ r› \ rœ z;Orchestrator._handle_validation_failure.<locals>.<listcomp>N)r
|
|||
|
|
rW rS r“ r” ÚjsonÚdumpsÚissuesr r� rŸ r r r ÚRETRYrD r’ r¡ )r" rB r‰ r˜ r¢ Úagent_attemptsr# r§ r$ r— L s&
|
|||
|
|
|
|||
|
|
ø
|
|||
|
|
z'Orchestrator._handle_validation_failurero c
|
|||
|
|
C st z!| j | ¡ |¡}t| dd¡| dg ¡t| dd¡ƒd�}|W S ty9 } z tddd�W Y d }~S d }~ww )
|
|||
|
|
u¦
|
|||
|
|
验�结果
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
task: 任务
|
|||
|
|
result: 执行结果
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
ValidationResult
|
|||
|
|
r� Frª r– rK )r� rª r– T)r� r– N)r r� rV r r9 rX r- )r" rB ro Úvalidation_datar˜ r. r# r# r$ r� n s
|
|||
|
|
|
|||
|
|
ý€þz Orchestrator.validate_resultc
|
|||
|
|
C s | j |¡}zá| |¡}| d¡r'd|_t | dg ¡¡|_| j |¡ |W S | ||¡}||_
|
|||
|
|
d|_| j |¡ | ¡ }t ƒ }i }|D ]”}i } |D ]ˆ}
|
|||
|
|
|
|
|||
|
|
|
|
|||
|
|
¡} | sXtj|
|
|||
|
|
_qJ| |
|
|||
|
|
| ¡} | setj|
|
|||
|
|
_qJ| |
|
|||
|
|
| ¡\}
|
|||
|
|
}}|
|
|||
|
|
sutj|
|
|||
|
|
_qJ| |
|