o i/@sddlmZmZddlZddlmZmZddlmZddl m Z ddl m Z ddl mZmZmZddlmZdd lmZdd lmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#d d l$m%Z%m&Z&m'Z'ed Z(zHddl)m*Z*ddl+m,Z,m-Z-ddl.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6mZ7ddl8m9Z9m:Z:ddl;mZ>ddl?m@Z@ddlAmBZBmCZCWneDyZEzeDdeEdZE[EwwedAdd ZFdddd!d"dddd#ddddd$ dBd=d>ZGe Gd?d@d@eStorage | Noner? Broker | Noner@ str | NonerAstrrBrCrDAgentProvider | NonerElist[Skill] | NonerFboolrGSequence[Route] | NonerHSequence[Middleware] | NonerI"dict[Any, ExceptionHandler] | NonerJLifespan[FastA2A] | Nonec Cs\|pt}|p t}t|||d}| ptt||d} t|||p!|j|||||| | | | | d S)z&Create a FastA2A server from an agent.)r4r?r>)r3r4r=)r+r" AgentWorkerrr<r r@)r4r>r?r@rArBrCrDrErFrGrHrIrJr3r:r:r; agent_to_a2aJs&  rWc@sjeZdZUdZded<d&dd Zd'd d Zd(ddZd)ddZd*ddZ d+ddZ d,d d!Z d-d#d$Z d%S).rVz-A worker that uses an agent to execute tasks.z(AbstractAgent[AgentDepsT, WorkerOutputT]r4paramsr*r6Nonec s|j|dIdH}|durtd|dd|dddkr3td|dd|ddd|jj|dd d IdH|j|d IdHpLg}|||d gzF|jj |d IdH}|j |d | IdHg}| D]}t |trqx||j}|r|td|dttdqx||j}Wnty|jj|ddd IdHw|jj|dd||dIdHdS)NidzTask z not foundstatusstate submittedz$ has already been processed (state: )working)r\ context_idhistory)message_historyr4message)rolepartskind message_idfailed completed)r\ new_artifacts new_messages)r> load_task ValueError update_task load_contextextendbuild_message_historygetr4r9update_context all_messagesrk isinstancer_response_parts_to_a2areappendr&rNuuiduuid4build_artifactsoutput Exception) selfrXtaskrbresult a2a_messagesrc a2a_parts artifactsr:r:r;run_task{s@     zAgentWorker.run_taskr)csdS)Nr:)r}rXr:r:r; cancel_taskszAgentWorker.cancel_taskrrlist[Artifact]cCs(tt}||}t|d|gdgS)a Build artifacts from agent result. All agent outputs become artifacts to mark them as durable task outputs. For string results, we use TextPart. For structured data, we use DataPart. Metadata is included to preserve type information. r) artifact_idr@re)rNrxry_convert_result_to_partr$)r}rrpartr:r:r;rzs  zAgentWorker.build_artifactsr'cCsVt|tr td|dSt|}t|}|j|dd}|jdd}tdd|id|id S) zConvert agent result to a Part (TextPart or DataPart). For string results, returns a TextPart. For structured data, returns a DataPart with properly serialized data. textrfrjson)mode serializationdatar json_schema)rfrmetadata)rurN A2ATextParttyper dump_pythonrr%)r}r output_type type_adapterrrr:r:r;rs   z#AgentWorker._convert_result_to_partra list[Message]list[ModelMessage]cCsTg}|D]#}|ddkr|t||ddq|t||ddq|S)Nrduserre)re)rwr_request_parts_from_a2ar_response_parts_from_a2a)r}ramodel_messagesrcr:r:r;rqs  z!AgentWorker.build_message_historyre list[Part]list[ModelRequestPart]c Cs g}|D]}|ddkr|t|ddq|ddkru|d}d|vrB|dd}|dd}t||d }|t|gdq|d }ttttfD]} | |d }z|j Wn t yaYqLwnt d ||t|gdq|dd krt dt |q|S)aConvert A2A Part objects to pydantic-ai ModelRequestPart objects. This handles the conversion from A2A protocol parts (text, file, data) to pydantic-ai's internal request parts (UserPromptPart with various content types). Args: parts: List of A2A Part objects from incoming messages Returns: List of ModelRequestPart objects for the pydantic-ai agent rfrcontentfilebyteszutf-8 mime_typezapplication/octet-stream)r media_typeuri)rAzUnsupported file type: r!Data parts are not supported yet.) rwrencoderrrrr rrrrmNotImplementedErrorr ) r}re model_partsr file_contentrrrrAurl_clsr:r:r;rs4          z#AgentWorker._request_parts_from_a2alist[ModelResponsePart]cCsdg}|D]+}|ddkr|t|ddq|ddkr!td|ddkr+tdt|q|S)aConvert A2A Part objects to pydantic-ai ModelResponsePart objects. This handles the conversion from A2A protocol parts (text, file, data) to pydantic-ai's internal response parts. Currently only supports text parts as agent responses in A2A are expected to be text-based. Args: parts: List of A2A Part objects from stored agent messages Returns: List of ModelResponsePart objects for message history rfrrrz!File parts are not supported yet.rr)rwrrr )r}rerrr:r:r;rs     z$AgentWorker._response_parts_from_a2aSequence[ModelResponsePart]c Cslg}|D]/}t|tr|td|jdqt|tr-|td|jd|j|jddqt|tr3 q|S)afConvert pydantic-ai ModelResponsePart objects to A2A Part objects. This handles the conversion from pydantic-ai's internal response parts to A2A protocol parts. Different part types are handled as follows: - TextPart: Converted directly to A2A TextPart - ThinkingPart: Converted to TextPart with metadata indicating it's thinking - ToolCallPart: Skipped (internal to agent execution) Args: parts: List of ModelResponsePart objects from agent response Returns: List of A2A Part objects suitable for sending via A2A protocol rrthinking)r thinking_id signature)rfrr) rurrwrrrrZrr)r}rerrr:r:r;rvs    z"AgentWorker._response_parts_to_a2aN)rXr*r6rY)rXr)r6rY)rrr6r)rrr6r')rarr6r)rerr6r)rerr6r)rerr6r) __name__ __module__ __qualname____doc____annotations__rrrzrrqrrrvr:r:r:r;rVus   /   *rV)r2r r3r-r4r5r6r7)r4r5r>rKr?rLr@rMrArNrBrNrCrMrDrOrErPrFrQrGrRrHrSrIrTrJrUr6r )J __future__r _annotationsrxcollections.abcrr contextlibr dataclassesr functoolsrtypingrr r pydanticr typing_extensionsr pydantic_air rrrrrrrrrrrrrr4rrrrfasta2a.applicationsr fasta2a.brokerr!r"fasta2a.schemar#r$r%r&r'r(r)r*rfasta2a.storager+r,fasta2a.workerr-starlette.middlewarer.starlette.routingr/starlette.typesr0r1 ImportError _import_errorr<rWlistrVr:r:r:r;s^     @ ,    +(