o wi@sUddlmZddlmZddlmZmZddlmZddl m Z m Z m Z ddl mZmZddlmZddlmZed Ze d Zd Zd ed <dZd ed<eddGdddZedZded< dddZd ddZdS)!) annotations) ContextVar) dataclassfield)wraps)CallableFinalTypeVar)UUIDuuid4) ParamSpec)nameofPRzX-Wandb-Python-Funcz Final[str]X_WANDB_PYTHON_FUNCzX-Wandb-Python-Call-IdX_WANDB_PYTHON_CALL_IDT)frozenc@s6eZdZUded< eedZded< d dd Zd S) TrackedFuncInfostrfunc)default_factoryr call_idreturndict[str, str]cCst|jtt|jiS)N)rrrrr)selfrD/home/ubuntu/.local/lib/python3.10/site-packages/wandb/_analytics.py to_headerss zTrackedFuncInfo.to_headersN)rr)__name__ __module__ __qualname____annotations__rr rrrrrrrs r _current_funczContextVar[TrackedFuncInfo]rCallable[P, R]rcs0jdttd fdd }|S) zA decorator to inject the calling function name into any GraphQL request headers. If a tracked function calls another tracked function, only the outermost function in the call stack will be tracked. .argsP.argskwargsP.kwargsrrc sLtr |i|Sttd}z |i|Wt|St|w)N)r) tracked_funcr"setrreset)r%r'tokenrfunc_namespacerrwrapper0s ztracked..wrapperN)r%r&r'r(rr)rr r)rr/rr-rtracked(s r0TrackedFuncInfo | NonecCs tdS)zEReturns info on the current tracked function, if any, otherwise None.N)r"getrrrrr)?s r)N)rr#rr#)rr1) __future__r contextvarsr dataclassesrr functoolsrtypingrrr uuidr r typing_extensionsr wandb._strutilsr rrrr!rrr"r0r)rrrrs$