ig UddlZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z m Z ddlmZddlmZddlmZmZmZddlmZddlZdd lmZeej6d ej8Zej<eZ e!d d iZ"e!e#d <da$e!dze#d<da%e!dze#d<de!dzfdZ&de!dzfdZ'da(edZ)edZ*edZ+GddZ,d:de-deee+e)fgee+e)fffdZ. d:de-deee+e eee)ffgee+e eee)ffffdZ/dZ0e1fde2e-de3fdZ4de-de3fd Z5d!e-de3fd"Z6d;d!e-d#e-d$e3de3fd%Z7dd=d/e-d0e?dzde-fd1Z@dddd2d3e eee*fd4e-dzd5ejdzd6e3deje*f d7ZCd8e-de-fd9ZDy)>N)Callable Coroutine)fnmatch)cachewraps)Path)stderr)Any ParamSpecTypeVar)urlparse) load_dotenvzMhttps?://[^\s<>"\']+|www\.[^\s<>"\']+|[^\s<>"\']+\.[a-z]{2,}(?:/[^\s<>"\']*)?_ImportNotFound_IMPORT_NOT_FOUND_openai_bad_request_error_groq_bad_request_errorreturncrt ddlm}|atturtSdS#t$r taY%wxYw)z'Lazy loader for OpenAI BadRequestError.NrBadRequestError)ropenair ImportErrorrrs [/mnt/e/genesis-system/.venvs/browser-army/lib/python3.12/site-packages/browser_use/utils.py_get_openai_bad_request_errorr"sF%1%.&?FW%W!a]aa 101 $66crt ddlm}|atturtSdS#t$r taY%wxYw)z%Lazy loader for Groq BadRequestError.Nrr)rgroqrrrrs r_get_groq_bad_request_errorr/sF#/#,$;BS#S]Y]] /./rFRTPceZdZdZ ddej dzdegdfdzdegdfdzdegdfdzdedee dzf d Z dd Z dd Z dd Z dd ZddZddZddZddZddZy) SignalHandlera A modular and reusable signal handling system for managing SIGINT (Ctrl+C), SIGTERM, and other signals in asyncio applications. This class provides: - Configurable signal handling for SIGINT and SIGTERM - Support for custom pause/resume callbacks - Management of event loop state across signals - Standardized handling of first and second Ctrl+C presses - Cross-platform compatibility (with simplified behavior on Windows) Nlooppause_callbackresume_callbackcustom_exit_callbackexit_on_second_intinterruptible_task_patternsc|xstj|_||_||_||_||_|xsgd|_tjdk(|_ |jd|_ d|_ y)a4 Initialize the signal handler. Args: loop: The asyncio event loop to use. Defaults to current event loop. pause_callback: Function to call when system is paused (first Ctrl+C) resume_callback: Function to call when system is resumed custom_exit_callback: Function to call on exit (second Ctrl+C or SIGTERM) exit_on_second_int: Whether to exit on second SIGINT (Ctrl+C) interruptible_task_patterns: List of patterns to match task names that should be canceled on first Ctrl+C (default: ['step', 'multi_act', 'get_next_action']) )step multi_actget_next_actionWindowsN)asyncioget_event_loopr%r&r'r(r)r*platformsystem is_windows_initialize_loop_stateoriginal_sigint_handleroriginal_sigterm_handler)selfr%r&r'r(r)r*s r__init__zSignalHandler.__init__Rsz*.g,,.$)&$($2$.$%@%lDl$"OO%2$/"&$"&$c`t|jddt|jddy)z:Initialize loop state attributes used for signal handling.ctrl_c_pressedFwaiting_for_inputN)setattrr%r8s rr5z$SignalHandler._initialize_loop_statevs$ $))%u- $))(%0r:cd jr/fd}tjtj|_yjj tjfd_jj tj fd_y#t$rYywxYw)z0Register signal handlers for SIGINT and SIGTERM.ctdtjrjtjdy)Nu5 🛑 Got Ctrl+C. Exiting immediately on Windows... filer)printr r(os_exit)sigframer8s rwindows_handlerz/SignalHandler.register..windows_handlers. EFS !! !XXa[r:c$jSN)sigint_handlerr?srz(SignalHandler.register..sW[WjWjWlr:c$jSrK)sigterm_handlerr?srrMz(SignalHandler.register..sY]YmYmYor:N) r4signalSIGINTr6r%add_signal_handlerSIGTERMr7 Exception)r8rIs` rregisterzSignalHandler.register{s oo$*==#PD $(99#?#? Ol#mD $(II$@$@Qo$pD!   s:B#A$B## B/.B/c |jr;|jrtjtj|jy|jj tj|jj tj |jr.tjtj|j|jr/tjtj |jyyy#t$r"}tjd|Yd}~yd}~wwxYw)zEUnregister signal handlers and restore original handlers if possible.z+Error while unregistering signal handlers: N) r4r6rPrQr%remove_signal_handlerrSr7rTloggerwarningr8es r unregisterzSignalHandler.unregistersE oo ## ]]6==$">">? II##FMM2II##FNN3 ## ]]6==$">">? $$ ]]6>>4#@#@A%$ E >>?sCDDEsAD CD D=D88D=cdtsda|jr |jt dt t dddt t ddd t d ddt t d dd t d ddt t d dd t d ddt t d dd t d ddt t d dd t dt tjdy#t$r#}tj d|Yd}~d}~wwxYw)z Handle a second Ctrl+C press by performing cleanup and exiting. This is shared logic used by both sigint_handler and wait_for_resume. TzError in exit callback: Nu2 🛑 Got second Ctrl+C. Exiting immediately... rBz[?25hendflushrC)r`raz[?1lz[?2004l zI(tip: press [Enter] once to fix escape codes appearing after chrome exit)r) _exitingr(rTrXerrorrDr rErFrZs r_handle_second_ctrl_cz#SignalHandler._handle_second_ctrl_cs 8 2  ?fM  4f5 4( rF3 r& $V4 $' 2T7 2T*"Dv."D!SZ`a((1+C 2 \\,QC0112sD D/ D**D/ctrtjdt|jddr4t|jddry|j r|j t|jdd|j|jr |jtdt y#t$r"}tjd|Yd}~8d}~wwxYw) z SIGINT (Ctrl+C) handler. First Ctrl+C: Cancel current step and pause. Second Ctrl+C: Exit immediately if exit_on_second_int is True. rr<Fr=NTzError in pause callback: zF----------------------------------------------------------------------rB)rdrErFgetattrr%r)rfr>_cancel_interruptible_tasksr&rTrXrerDr rZs rrLzSignalHandler.sigint_handlers88A; TYY(%0 dii,e4    $))%t,""$ 2 PW]^ 2 LL,QC0112sB== C(C##C(cts/datdt|jr|jt j dy)z> SIGTERM handler. Always exits the program completely. Tu1 🛑 SIGTERM received. Exiting immediately... rBrN)rdrDr r(rErFr?s rrOzSignalHandler.sigterm_handlers9 8 @vN ((1+r:ctj|j}tj|jD]}||k7s |j rt |dr|j n t|tfd|jDs`tjd|j|jd|r|j sot |dr|j n t|tfd|jDr)tjd|jyyyy)z2Cancel current tasks that should be interruptible.get_namec3&K|]}|v ywrKr.0pattern task_names r z.s PG7i  PzCancelling task: cF|jr|jSdSrK) cancelled exception)ts rrMz;SignalHandler._cancel_interruptible_tasks..sq{{}akkm$r:c3&K|]}|v ywrKrrns rrrz. s O7'Y  OrszCancelling current task: N)r0 current_taskr% all_tasksdonehasattrrlstranyr*rXdebugcanceladd_done_callback)r8rytaskrqs @rriz)SignalHandler._cancel_interruptible_taskss%%dii0, *Pd l499;#*4#< #d)I Pt/O/O PP \\%i[12 [[] NOP,++-*1, *K|$$&QTUaQb9 Od.N.N OO LL,YK89P.\r:ct|jddtjtj} tjtjtj d}d}d}d}d} td|d|d |d |d |d |d ddtt|jr|j tjtj|t|jddy#t $rYwxYw#t$r|jYgwxYw#t$rYywxYw# tjtj|t|jddw#t$rYwwxYwxYw)z Wait for user input to resume or exit. This method should be called after handling the first Ctrl+C. It temporarily restores default signal handling to allow catching a second Ctrl+C directly. r=Tzzzrbu➡️ Press z[Enter]z to resume or z[Ctrl+C]z again to exitz... r^r_FN)r>r%rP getsignalrQdefault_int_handler ValueErrorrDr inputr'KeyboardInterruptrfrT)r8original_handlergreenredblinkunblinkresets rwait_for_resumezSignalHandler.wait_for_resume$sq $))($/%%fmm4 == : :; %# % ' % UG75'uHUG>Z_Y``cdkcllmn     7    MM&--!12 DII*E2= .        MM&--!12 DII*E2   sl2D9A D;D0 D DD-*D?,D--D?0 D<;D<?F ;E=<F = F F F  F ct|jdrt|jddt|jdrt|jddyy)zReset state after resuming.r<Fr=N)r|r%r>r?s rrzSignalHandler.resetVsI TYY() 499&. TYY+, 499)51-r:)NNNNTN)rN)__name__ __module__ __qualname____doc__r0AbstractEventLooprboollistr}r9r5rUr\rfrLrOrirrrr:rr$r$Es ,0.2/348!26"'!!D("'2t8$t+"'BH%, "' !T*T1 "'  "' $Cy4/"'H1 2E(/b$_L"(0 d2r:r$additional_textc`dtttfdtttfffd }|S)Nfuncrctdtjdtjdtffd }|S)Nargskwargsrctj}|i|}tj|z }|dkDr|xrt|ddd}|rt|dd}nAd|vrt|dd}n-d|vrt|dd}ntjt}|j dj dd|d d |S Ng?rrXagentbrowser_sessionu⏳ -z() took z.2fstimerhlogging getLoggerrrstrip rr start_timeresultexecution_timeself_has_loggerrXrrs rwrapperz7time_execution_sync..decorator..wrapperas : $ !& !6IIK*,>t?wtAw$?Od1gx(V F fWox0V f $f./:V)V LL4--c238N3;OqQR =r:rr"rrr rrrs` r decoratorz&time_execution_sync..decorator`s=;QVVqxxA$ .r:)rr"r rrs` rtime_execution_syncr_s.Xad^A, r:c dttttttffdttttttffffd }|S)Nrrctdtjdtjdtffd }|S)NrrrcKtj}|i|d{}tj|z }|dkDr|xrt|ddd}|rt|dd}nAd|vrt|dd}n-d|vrt|dd}ntjt}|j dj dd|d d |S7wrrrs rrz8time_execution_async..decorator..wrapper}s :'' '6IIK*,>t?wtAw$?Od1gx(V F fWox0V f $f./:V)V LL4--c238N3;OqQR = (s CCB2Crrs` rrz'time_execution_async..decorator|s=;166QXX!& .r:)rr"rr r rs` rtime_execution_asyncrysLXa3Q;!778XaSVX[]^S^I_F_=`. r:cdgfd}|S)Nc.d |i|d<dS)Nrr)rrclsinstances rrzsingleton..wrappers* a[d%f%8A; !r:r)rrrs` @r singletonrs F r:keysc |d|DS)z3Check if all required environment variables are setc3dK|](}tj|dj*yw)r^N)rEgetenvr)rokeys rrrz&check_env_variables..s$>#299S"%++->s.0r)r any_or_alls rcheck_env_variablesrs>>>>r:rpcd|vr|jdd\}}|jddjdd}d|vS)z Check if a domain pattern has complex wildcards that could match too many domains. Args: pattern: The domain pattern to check Returns: bool: True if the pattern has unsafe wildcards, False otherwise ://.*r^*.*)splitreplace)rp_ bare_domains ris_unsafe_patternrsL W}}UA&*!WtR(00r: {r:urlc |dvS)z Check if a URL is a new tab page (about:blank, chrome://new-tab-page, or chrome://newtab). Args: url: The URL to check Returns: bool: True if the URL is a new tab page, False otherwise )z about:blankzchrome://new-tab-page/zchrome://new-tab-pagezchrome://newtab/zchrome://newtabr)rs ris_new_tab_pagers xxxr:domain_pattern log_warningsc  t|ryt|}|jr|jjnd}|jr|jjnd}|r|sy|j}d|vr|j dd\}}nd}|}d|vr&|j ds|j ddd}t||sy|dk(s||k(ry d|vr|jd dkDs|jd dkDr1|r.tjt}|jd |d y|jd r1|r.tjt}|jd|dy|jd d} d| vr1|r.tjt}|jd|dy|j d r|dd} || k(s t|| ry t||ry y#t$rS} tjt}|jd|d|dt!| jd| Yd} ~ yd} ~ wwxYw)a Check if a URL matches a domain pattern. SECURITY CRITICAL. Supports optional glob patterns and schemes: - *.example.com will match sub.example.com and example.com - *google.com will match google.com, agoogle.com, and www.google.com - http*://example.com will match http://example.com, https://example.com - chrome-extension://* will match chrome-extension://aaaaaaaaaaaa and chrome-extension://bbbbbbbbbbbbb When no scheme is specified, https is used by default for security. For example, 'example.com' will match 'https://example.com' but not 'http://example.com'. Note: New tab pages (about:blank, chrome://new-tab-page) must be handled at the callsite, not inside this function. Args: url: The URL to check domain_pattern: Domain pattern to match against log_warnings: Whether to log warnings about unsafe patterns Returns: bool: True if the URL matches the pattern, False otherwise Fr^rrhttps:rrTrru&⛔️ Multiple wildcards in pattern=[z] are not supportedu&⛔️ Wildcard TLDs like in pattern=[z ] are not supported for securityuE⛔️ Only *.domain style patterns are supported, ignoring pattern=[]Nu⛔️ Error matching URL z with pattern : )rr schemelowerhostnamer startswithrcountrrrreendswithrrTtype) rrr parsed_urlrdomainpattern_schemepattern_domainrXr parent_domainr[s rmatch_url_with_domain_patternrss.LS }*)3(9(9:   " " $r&*4*=*=:   $ $ &2& v "''). n$2$8$8$B!>>>"> N>#<#  ( sf6  N4 1$(<(:JJ]^_ d#)V \\:>:JJjkl  ''b1; [)V \\YZhYiijkl %"12&M '&-"@ fn%     X &&,,+C5~>NbQUVWQXQaQaPbbdefdghi sL G;AG;.A$G; G;AG;AG;'G;- G;; IA IIabpath.c |D]}||vrt||tr4t||tr!t|||||t|fzNt||tr"t||tr||||z||<||||k7st ddj |t|fzz||||<|S)Nz Conflict at .) isinstancedict merge_dictsr}rrTjoin)rrrrs rrr.s SAX3:afd#;##C{ 231S64 Z#%= sVaf_AcF #!C& NSXXdc#h[.@%AA BB cF1S6  r:cL ttjj}|dz }|jrpddl}t |d5}|j }|jd|}|r0|jd}|tjd<|cdddS ddddd l m }t|d }|tjd<|S#1swY5xYw#t$r8}tj!d t#|j$d |Yd}~y d}~wwxYw)zeGet the browser-use package version using the same logic as Agent._set_browser_use_version_and_sourcezpyproject.tomlrNutf-8)encodingz!version\s*=\s*["\']([^"\']+)["\']rLIBRARY_VERSION)versionz browser-usez%Error detecting browser-use version: runknown)r__file__parentexistsreopenreadsearchgrouprEenvironimportlib.metadatarr}rTrXrrr) package_rootpyproject_pathrfcontentmatchr get_versionr[s rget_browser_use_versionr<s h&&--,"22.  ^g.!ffhG BII:G DE ++a.!W%,RZZ!"   8  M* +'")"**  . ,,6tAw7G7G6H1#NO s7A C" A C C""3C"CC"" D#+.DD#cfK tjd4d{}|jdd{}|jdk(r)|j }|ddcdddd{Sdddd{y7f7O77 #1d{7swYyxYw#t $rYywxYww)zCheck the latest version of browser-use from PyPI asynchronously. Returns: The latest version string if successful, None if failed g@)timeoutNz&https://pypi.org/pypi/browser-use/jsoninfor)httpx AsyncClientget status_codejsonrT)clientresponsedatas r check_latest_browser_use_versionr[s    s +##vJJGHH8c! ==?D < " ##### #H#####   sB1B"BB"B B*B " B".B /B"3B14 B"?B B"B1B"B  B" B" BB BB"B1B"" B.+B1-B..B1c ddl}ttjj}|dz }|j sy|j gd||j jj}|j gd||j jj}|j gd||j jj}|j gd||j jj}||||d S#t$r8}tjd t|jd |Yd}~yd}~wwxYw) z4Get git information if installed from git repositoryrNz.git)git rev-parseHEAD)cwdr )rrz --abbrev-refr)rconfigz--getzremote.origin.url)rshowz-sz --format=%cir) commit_hashbranch remote_urlcommit_timestampzError getting git info: r) subprocessrrrr check_outputDEVNULLdecoderrTrXrrr)r rgit_dirrrrrr[s r get_git_infor%msM%h&&--, 6 !'   7\R\RdRdellnttv G\blbtbtu FH EG JP\eoewewx FH EG Hlcmcucuv FH EG %jfv ww,,)$q'*:*:);2aSAB s~rr") r}rrrrrrhomerresolve)r pretty_paths r_log_pretty_pathr.s CIOO%  4#t% T$Z ! ##4y  TYY[!137??DHHJDVDVDX@Y[^_C;.K="+r:rmax_lenc|jddjddjdd}|t||kDr|d|dzS|S)zXTruncate/pretty-print a URL with a maximum length, removing the protocol and www. prefixzhttps://r^zhttp://zwww.Nu…)rlen)rr/s r_log_pretty_urlr2sSYYz2&&y"5==fbI CFW, 8Gu  r:)namelogger_instancesuppress_exceptionscoror3r4r5ctj||}|xstdtjtddffd }|j ||S)a Create an asyncio task with proper exception handling to prevent "Task exception was never retrieved" warnings. Args: coro: The coroutine to wrap in a task name: Optional name for the task (useful for debugging) logger_instance: Optional logger instance to use. If None, uses module logger. suppress_exceptions: If True, logs exceptions at ERROR level. If False, logs at WARNING level and exceptions remain retrievable via task.exception() if the caller awaits the task. Default False. Returns: asyncio.Task: The created task with exception handling callback Example: # Fire-and-forget with suppressed exceptions create_task_with_error_handling(some_async_function(), name="my_task", suppress_exceptions=True) # Task with retrievable exceptions (if you plan to await it) task = create_task_with_error_handling(critical_function(), name="critical") result = await task # Will raise the exception if one occurred )r3rwrNc $d} |j}|t|dr|jnd}r0jd|dt |j d||n1j d|dt |j d|||}||y#tj$rYt$rU}t|dr|jnd}jd|dt |j d|Yd}~rd}~wwxYw) z"Callback to handle task exceptionsNrlunnamedzException in background task [z]: r)exc_infoz"Error handling exception in task [) rvr|rlrerrrYr0CancelledErrorrT)rw exc_to_raiseexcrqr[logr5s r_handle_task_exceptionz?create_task_with_error_handling.._handle_task_exceptions',Y 3 o ': 6 IIYY/ {#d3i>P>P=QQSTWSXYdgYh [[&ykT#Y5G5G4H3%P\     Y&q*5qzz|99991)CQ@P@P?QQSTUSV WXXYsBBD2D:A D  D)r0 create_taskrXTaskr!r)r6r3r4r5rr?r>s ` @rcreate_task_with_error_handlingrBsR: Dt, &w||A4:./ r:textcF|jddjdS)a'Remove surrogate characters that can't be encoded in UTF-8. Surrogate pairs (U+D800 to U+DFFF) are invalid in UTF-8 when unpaired. These often appear in DOM content from mathematical symbols or emojis. Args: text: The text to sanitize Returns: Text with surrogate characters removed rignore)errors)encoder#)rCs rsanitize_surrogatesrHs" GH -44W==r:)r^)F)r))Er0rrEr2rrPrcollections.abcrrr functoolsrrpathlibrsysr typingr r r urllib.parser r dotenvrcompile IGNORECASE URL_PATTERNrrrXrr__annotations__rrrrrdr r!r"r$r}rrrallrrrrrrrtuplerrrr%r.intr2LoggerrArBrHrr:rrYs  /"**!  bjjikmkxkxy    8 $0"b949)-4$;-'++ btd{ b ^TD[ ^  CL CL cNW2W2thA?OQYZ[]^Z^Q_?_6`6 x9S#q[1123Xa3PSUV;AW>W5XXY:58?d3i?D? st* y y ycscCctc`dcL  4  D  c3h  < d  $'d38nt+''T3:,, s S4Z  *." > c1>  Tz> ..4' >  >  \\!_ > B >c >c >r: