kKj;8PdZddlmZ ddlmZmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZddlmZmZmZddlmZmZm Z m!Z!dd l$Z$dd l%m%Z%m&Z&dd l'm(Z(dd l)m*Z*ddl+m,Z,e-dZ.ddZ/ddZ0ddZ1Gdde*Z2y #e"$rZ#e"d e#d Z#[#wwxYw)u,FileUpload — a Provider that adds drag-and-drop file upload to any server. Lets users upload files directly to the server through an interactive UI, bypassing the LLM context window entirely. The LLM can then read and work with uploaded files through model-visible tools. Requires ``fastmcp[apps]`` (prefab-ui). Usage:: from fastmcp import FastMCP from fastmcp.apps import FileUpload mcp = FastMCP("My Server") mcp.add_provider(FileUpload()) For custom persistence, override the storage methods:: class S3Upload(FileUpload): def on_store(self, files, ctx): # write to S3, return summaries ... def on_list(self, ctx): # list from S3 ... def on_read(self, name, ctx): # read from S3 ... ) annotations)SetState ShowToast)CallTool) PrefabApp)H3BadgeButtonCard CardContent CardFooter CardHeaderColumnDropZoneMutedRow SeparatorSmallText)ElseForEachIf)ERRORRESULTSTATERxzHFileUpload requires prefab-ui. Install with: pip install 'fastmcp[apps]'N)datetimetimezone)Any) FastMCPApp)Context)z.csvz.jsonz.txtz.mdz.pyz.yamlz.ymlz.tomlcxt|}|dk(ry|jdtd|dz }|dzdz|z S)zLReturn the exact decoded byte-length of a base64 string without decoding it.r=)lencountmax)b64npaddings i/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastmcp/apps/file_upload.py_b64_decoded_sizer.JsB CAAviiSAE]+G q5A: cD|dkr|dS|dkr |dz ddS|dz ddS)Niz Biz.1fz KBz MB)sizes r- _format_sizer3SsF d{r{ +c"#&&+&s+3//r/cB|d|d|dt|d|ddS)Nnametyper2 uploaded_at)r5r6r2 size_displayr7)r3)entrys r- _make_summaryr:\s5f f f $U6]3]+  r/ceZdZdZ dddddd dfdZddZdd Z dd Zdd Zdd Z dd Z xZ S) FileUploadaA Provider that adds file upload capabilities to a server. Registers a drag-and-drop UI tool, a backend storage tool, and model-visible tools for listing and reading uploaded files. Files are scoped by MCP session and stored in memory by default. Override ``on_store``, ``on_list``, and ``on_read`` for custom persistence (filesystem, S3, database, etc.). Each method receives the current ``Context``, giving access to session ID, auth tokens, and request metadata for partitioning and authorization. **Session scoping:** The default storage uses ``ctx.session_id`` to isolate files by session. This works with stdio, SSE, and stateful HTTP transports. In **stateless HTTP** mode, each request creates a new session, so files won't persist across requests. For stateless deployments, override the storage methods to partition by a stable identifier from the auth context:: class UserScopedUpload(FileUpload): def on_store(self, files, ctx): user_id = ctx.access_token["sub"] ... Example:: from fastmcp import FastMCP from fastmcp.apps.file_upload import FileUpload mcp = FastMCP("My Server") mcp.add_provider(FileUpload()) iz File UploadzsDrop files to upload them to the server. The model can then read and analyze them without using the context window.zDrop files here) max_file_sizetitle description drop_labelct||||_||_||_||_i|_|jy)N)super__init___max_file_size_title _description _drop_label_store_register_tools)selfr5r=r>r?r@ __class__s r-rCzFileUpload.__init__sF + '%=?  r/c"d|jdS)Nz FileUpload())r5)rJs r-__repr__zFileUpload.__repr__sTYYM++r/c: |jS#t$rYywxYw)aReturn the key used to partition file storage. Defaults to ``ctx.session_id``, which is stable for stdio, SSE, and stateful HTTP. The default ``on_store``/``on_list``/``on_read`` implementations call this to partition the in-memory store. Override to scope by user, tenant, or any other dimension:: def _get_scope_key(self, ctx): return ctx.access_token["sub"] __default__) session_id RuntimeError)rJctxs r-_get_scope_keyzFileUpload._get_scope_keys# !>> ! !  !s  cX|j|}|jj|i}|D]M}|d|d|d|dtjt j jdd||d<O|jDcgc] }t|c}Scc}w)atStore uploaded files and return summaries. Args: files: List of file dicts, each with ``name``, ``size``, ``type``, and ``data`` (base64-encoded content). ctx: The current request context. Use for session ID, auth tokens, or any metadata needed for partitioning. Override this method for custom persistence. The default implementation stores files in memory, scoped by ``_get_scope_key(ctx)``. Returns: List of file summary dicts (``name``, ``type``, ``size``, ``size_display``, ``uploaded_at``). r5r2r6dataseconds)timespec)r5r2r6rVr7) rTrH setdefaultrnowrutc isoformatvaluesr:)rJfilesrSscope session_filesfes r-on_storezFileUpload.on_stores*##C( ..ub9 A& & & & '||HLL9CCYCW (M!F) $+8*>*>*@A*@Q a *@AAAsB'c|j|}|jj|i}|jDcgc] }t |c}Scc}w)aList all stored files. Args: ctx: The current request context. Override this method for custom persistence. The default implementation returns files from the current scope. Returns: List of file summary dicts. )rTrHgetr]r:)rJrSr_r`rbs r-on_listzFileUpload.on_listsP##C( r2 *7*>*>*@A*@Q a *@AAAsAc|j|}|jj|i}||vr*t|j }t d|d|||ddddd}dj dxstfd tD}|r- tjd jd |d <|Sd d ddz|d<|S#t$rd d ddz|d<Y|SwxYw)a>Read a file's contents by name. Args: name: The filename to read. ctx: The current request context. Override this method for custom persistence. The default implementation reads from the current scope's in-memory store. Text files are decoded from base64; binary files return a truncated base64 preview. Returns: Dict with file metadata and ``content`` (text) or ``content_base64`` (binary preview). Raises: ValueError: If the file is not found. File z not found. Available: r5r2r6r7)r5r2r6r7ztext/c3FK|]}dj|yw)r5N)endswith).0extr9s r- z%FileUpload.on_read..s%; 3CCE&M " "3 '3Cs!rVzutf-8contentNz...content_base64) rTrHrelistkeys ValueError startswithany_TEXT_EXTENSIONSbase64 b64decodedecodeUnicodeDecodeError) rJr5rSr_r` availableresultis_textr9s @r-on_readzFileUpload.on_readsG&##C( r2 } $]//12IuTH,CI;OP Pd#&M&M&M / "  -**73 s; 3C; 8   G$*$4$4U6]$C$J$J7$Sy!  (-V}Tc':U'BF# $ & G+0=#+>+F'(  Gs&*C""C=<C=c||jdfd }|jddfd }|jdd fd }|jd fd }y) Nc  |D]m}t|jdd}|jkDs.td|jdddt |dt jdj ||S) zQStore uploaded files. Receives file objects with name, size, type, data (base64).rVrhr5?z exceeds max size (z > rM)r.rerDrsr3rc)r^rSra actual_sizeproviders r- store_filesz/FileUpload._register_tools..store_filess0fb0AB !8!88$fc 256(56c'(?(?@AD $$UC0 0r/T)modelc&j|S)z&List all uploaded files with metadata.)rf)rSrs r- list_filesz.FileUpload._register_tools..list_files)s##C( (r/c(j||S)z)Read an uploaded file's contents by name.)r~)r5rSrs r- read_filez-FileUpload._register_tools..read_file.s##D#. .r/c$ td5}t5tdd5tjt t jj5tt jjddddddddddt5td 5tjtd d jd t!j"dj"t t j$j5td 5t'd 5tdd5td 5t)t+dtt+ddddddddddt-dt/ddt+d it1dt2t1d gt5ddgt5t6dddddddt t jj5t9t;ddt'd5}tddd5td 5t)|j<t|j>dddtd 5t|j@dt|jBd ddddddddddddddddddtE5tdd!"5t t jj5tt jjd#t jjjGd$d%dddtI5td&ddddddddddddtKgjM|d'(S#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYZxYw#1swY_xYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swY"xYw#1swYxYw#1swYxYw#1swYxYw#1swY xYw))zDUpload and manage files. Drop files here to send them to the server.zmax-w-2xl mx-auto) css_classr$center)gapalign secondary)variantNr&)rpendinginboxzAny file type, up to T)r5iconlabelr?multiplemax_sizerz $item.namez $item.typezUpload to Serverrr^storedzFiles uploaded!successerror) arguments on_successon_error)on_clickUploadedzfont-medium text-smzjustify-between)rrroutlinezw-full)rr filez on serverzNo files uploaded yet)rr)viewstate)'r rrrrErrrlengthr r rrrFrrGr3rDrrrrr rrrrrrrr5r7r6r8r pluralizerrrf)rSrrars r- file_managerz0FileUpload._register_tools..file_manager3s 34\31H#=x'ELL//12!LL//1$/3$>\!]FqM(//0&$&223+H,C,CDEG"&!)!8!8 EMM0023V]#I.AX6"qM!"\"23!"\"23*7/.%- -$+R ]+"%-Xv$>$,Y$;$-(909%&,"*3$),3*"&6C3>ELL//12! &&; $H-$%&.*; "(A %aff %amm 4"/"% %affk B %$%NN,5!"",.3]%2]L \3X#JELL//12$||2245 3 3 5 ? ? GH()3 56 $K\_5r!&..s3 m32$>#=\\6*M76/.6C]33Z"/",.-32]%2M]]N32  $K#J\\_54s TP(=P$*P PP(T0 S=A8R75 Q)  Q Q  Q))P5QQ "A"Q Q) /R7;#R* R - R :+Q6%R 9/R(R 0R 8R* R7ST"S80(S+AS 1S+ S S+S8TP PP% P((P2 -T5P?:QQ Q QQ Q&!Q) )Q3 .R76R;R R R RR R'"R* *R4 /R77S <SS TS S+S( #S++S5 0S88T =TT)r^ list[dict]rSr!returnr)rSr!rr)r5strrSr!rdict)rSr!rr)toolui)rJrrrrrs @r-rIzFileUpload._register_toolssz  1  1   )  )   /  / a  a r/)Files) r5rr=intr>rr?rr@rrNone)rr)rSr!rr)r^list[dict[str, Any]]rSr!rr)rSr!rr)r5rrSr!rdict[str, Any])rr) __name__ __module__ __qualname____doc__rCrNrTrcrfr~rI __classcell__)rKs@r-r<r<fsD." 0,      0,!"B#BB  BBB )^~r/r<)r*rrr)r2rrr)r9rrr)3r __future__rprefab_ui.actionsrrprefab_ui.actions.mcpr prefab_ui.apprprefab_ui.componentsrr r r r r rrrrrrrr!prefab_ui.components.control_flowrrr prefab_ui.rxrrrr ImportError_excrwrrtypingrfastmcp.apps.appr fastmcp.server.contextr! frozensetrvr.r3r:r<r1r/r-rs@#5.' DC55 ''*E  0oo[ R  sA BB% B  B%