cQbj!ddlmZddlmZmZddlmZddlZddlZddl m Z ddl m Z ddl m Z eGd d Z Gd d ejZeGd dZeGddZd!dZd!dZ d"dZ d#dZedeGddZedeGddZedeGddZedeGdd Zy)$) annotations) event_class T_JSON_DICT) dataclassN)dom)page)runtimecReZdZUdZdZded<dZded<dZded<dZe dZ y) Annotationz Tool annotations Nztyping.Optional[bool] read_onlyuntrusted_content autosubmitct}|j|j|d<|j|j|d<|j|j|d<|S)NreadOnlyuntrustedContentr)dictr rrselfjsons b/var/www/html/venv/lib/python3.12/site-packages/selenium/webdriver/common/devtools/v150/web_mcp.pyto_jsonzAnnotation.to_jsons[v >> %#~~D   ! ! -'+'='=D# $ ?? &!%D  c|d|vrt|dndd|vrt|dndd|vrt|dSdS)Nrrr)r rr)boolclsrs r from_jsonzAnnotation.from_json)sc0:d0Bd4 +,@RVZ@Zd4(:#;<`d3?43GtD./  NR  r) __name__ __module__ __qualname____doc__r __annotations__rrr classmethodrrrr r sH(,I$+04,3)-J%,  rr c2eZdZdZdZdZdZdZedZ y)InvocationStatusz5 Represents the status of a tool invocation. CompletedCanceledErrorc|jSN)value)rs rrzInvocationStatus.to_json:s zzrc||Sr,r%rs rrzInvocationStatus.from_json=s 4yrN) rr r!r" COMPLETEDCANCELEDERRORrr$rr%rrr'r'2s1IH Err'c~eZdZUdZded<ded<ded<dZded <dZd ed <dZd ed <dZded<dZ e dZ y)Toolz3 Definition of a tool that can be invoked. strname description page.FrameIdframe_idNztyping.Optional[dict] input_schemaztyping.Optional[Annotation]rz"typing.Optional[dom.BackendNodeId]backend_node_idz#typing.Optional[runtime.StackTrace] stack_tracect}|j|d<|j|d<|jj |d<|j |j |d<|j |j j |d<|j|jj |d<|j|jj |d<|S)Nr5r6frameId inputSchemar backendNodeId stackTrace) rr5r6r8rr9rr:r;rs rrz Tool.to_json\svyyV "..]--//1Y    ("&"3"3D     '"&"2"2":":"rr?r@)r5r6r8r9rr:r;) r4r FrameIdrrr r BackendNodeIdr StackTracers rrzTool.from_jsonksT&\"D/0\\++DO<6Ct6Kd=12QUERVZEZ ,,T--@A`dRaeiRiC--77_8MNosLX\`L`**44T,5GH  gk  r) rr r!r"r#r9rr:r;rr$rr%rrr3r3Bsk I+/L'.04K,3;?O7>8D c#Kddi}|}yw)z% Disables the WebMCP domain. rKzWebMCP.disableNr%rLs rdisablerQs "H>DrOc#Kt}|j|d<||d<||d<d|d}|}t|dSw)a< Invokes a registered tool. :param frame_id: Frame in which to invoke the tool. :param tool_name: Name of the tool to invoke. :param input_: Input parameters for the tool, matching the tool's inputSchema. :returns: Unique identifier for this invocation. Response is sent before tool events. r=toolNameinputzWebMCP.invokeToolrKparams invocationId)rrr4)r8 tool_nameinput_rVrMrs r invoke_toolrZs[&F ((*F9"F:F7O%H>D tN# $$s?Ac#<Kt}||d<d|d}|}yw)zh Cancels a pending tool invocation. :param invocation_id: Invocation identifier to cancel. rWzWebMCP.cancelInvocationrUN)r) invocation_idrVrMrs rcancel_invocationr]s/&F*F>+H>DszWebMCP.toolsAddedc.eZdZUdZded<eddZy) ToolsAddedz/ Event fired when new tools are added. ztyping.List[Tool]toolscf||dDcgc]}tj|c}Scc}wNr`)r`)r3rrris rrzToolsAdded.from_jsons-.27m<4>>!$<  <.N)rrreturnr_rr r!r"r#r$rr%rrr_r_s!   rr_zWebMCP.toolsRemovedc.eZdZUdZded<eddZy) ToolsRemovedz- Event fired when tools are removed. ztyping.List[RemovedTool]r`cf||dDcgc]}tj|c}Scc}wrb)rFrrcs rrzToolsRemoved.from_jsons/59']C;((+C  CreN)rrrfrirgr%rrriris! $#  rrizWebMCP.toolInvokedcLeZdZUdZded<ded<ded<ded<ed dZy ) ToolInvokedz4 Event fired when a tool invocation starts. r4rXr7r8r\rYc |t|dtjj|dt|dt|dS)NrSr=rWrT)rXr8r\rYrIrs rrzToolInvoked.from_jsonsK$z*+\\++DO<d>23tG}%   rN)rrrfrlrgr%rrrlrls1N K  rrlzWebMCP.toolRespondedcVeZdZUdZded<ded<ded<ded <d ed <edd Zy ) ToolRespondedz@ Event fired when a tool invocation completes or fails. r4r\r'statusztyping.Optional[typing.Any]outputztyping.Optional[str] error_textz%typing.Optional[runtime.RemoteObject] exceptionc |t|dtj|dd|vr|dndd|vrt|dndd|vr'tjj|dSdS)NrWrprq errorTextrs)r\rprqrrrs)r4r'rr RemoteObjectrs rrzToolResponded.from_jsonsd>23#--d8n=%-%54>41<1Ds4 ,-$KVZ^K^g**44T+5FG   ei   rN)rrrfrorgr%rrroro s;  ('$$44  rro)rf0typing.Generator[T_JSON_DICT, T_JSON_DICT, None])r8r7rXr4rYrrfz/typing.Generator[T_JSON_DICT, T_JSON_DICT, str])r\r4rfrw) __future__rutilrr dataclassesrenumtypingrr r r Enumr'r3rFrNrQrZr]r_rirlror%rrrs`#*!      @ tyy   2 2  2 j     0%%%% 7 %2 7" !     "   "#     $   !"    # . #$    % r