Xj"ddlmZddlmZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZmZmZmZmZmZddlmZGd d e ZeGd d eeeeeefZe dde eeeeeefe fded zdeeeeeeefd ffdZy )) Generator)contextmanager) dataclassfield)Generic) BaseModel)LifespanContextTRequestContext) BaseSessionReceiveNotificationTReceiveRequestTSendNotificationT SendRequestT SendResultT) ProgressTokenc(eZdZUeed<edzed<y)ProgressprogressNtotal)__name__ __module__ __qualname__float__annotations__g/mnt/ssd/data/Dropbox/adrian/vault-secondbrain/venv/lib/python3.12/site-packages/mcp/shared/progress.pyrrsO 4<rrc|eZdZUeeeeeefe d<e e d<e dze d<e ddZ e e d<d d e d edzd dfd Zy)ProgressContextsessionprogress_tokenNrgF)defaultinitcurrentamountmessagereturncK|xj|z c_|jj|j|j|j|d{y7w)N)rr&)r$r send_progress_notificationr!r)selfr%r&s rrzProgressContext.progress sL  ll55   TZZ6   sAA AA N)rrrr rrrr r rrrrr$strrrrrrrsY '8+Xll mm!! 4<3U3GU3 U S4Z 4 rrNctxrr'c#K|j|jj tdt|j|jj|} |y#wxYww)NzNo progress token provided)meta progressToken ValueErrorrr )r-r progress_ctxs rrr(sZ xx388119566"3;;0F0FNL  sAA&A! A&!A##A&r+)collections.abcr contextlibr dataclassesrrtypingrpydanticrmcp.shared.contextr r mcp.shared.sessionr r r rrr mcp.typesrrrrrrrrr;s%%(?$y    gl,={O]qqr        L"3[/Sggh   4<  L"3[/Sggh    r