ibdZddlmZddlmZmZmZdZdd ZddZ dddZ dddZ ddZ dS)z@Tracking for SDK helper usage via the x-stainless-helper header.) annotations)AnyDictcast_stainless_helperobjrnamestrreturnNonecz t|t|dS#ttf$rYdSwxYw)z0Mark an object as created by a named SDK helper.N)object __setattr__ _HELPER_ATTRAttributeError TypeError)rr s ]/root/projects/qq-shell/venv/lib/python3.11/site-packages/anthropic/lib/_stainless_helpers.py tag_helperr sL 3 d33333 I &     s !%::r str | Nonec.t|tdS)z+Get the helper name from an object, if any.N)getattrr)rs rget_helper_tagrs 3 d + ++Ntoolsmessages list[str]cnt}|r+|D](}t|}|||)|r|D]}t|}|||t|tr;t t ttf| d}nt|dd}t|trIt tt|D](}t|}|||)t|S)z:Collect deduplicated helper names from tools and messages.Ncontent) setradd isinstancedictrrr rgetrlistr)rrhelperstooltagmessageblocksblocks rcollect_helpersr+s1 G ! ! !D &&C C   ) ) )G ))C C   '4(( ;"4S>7;;?? JJ )T::&$'' )!$v,77))E(//C C((( ==rdict[str, str]cXt||}|siSdd|iS)zvBuild x-stainless-helper header dict from tools and messages. Returns an empty dict if no helpers are found. x-stainless-helperz, )r+join)rrr%s rstainless_helper_headerr08s7eX..G  $))G"4"4 55rfilec0t|}|iSd|iS)z8Build x-stainless-helper header dict from a file object.Nr.)r)r1r's r!stainless_helper_header_from_filer3Fs$   C { # &&r)rrr r r r )rrr r)NN)rrrrr r)rrrrr r,)r1rr r,) __doc__ __future__rtypingrrrrrrr+r0r3rrr8sFF"""""""""""""""""     ,,,, D 6 6 6 6 6''''''r