KidZddlZddlZddlmZddlZddlmZddlm Z ddl m Z e dZ eZ ejdd Zdd Zej" dd eedzej&dfd dfdZy)z-FastMCP tasks CLI for Docket task management.N) Annotated)Console)load_and_merge_config) get_loggerz cli.taskstasksz,Manage FastMCP background tasks using Docket)namehelpreturncddl}|jjj}|j dr+t j dtjdyy)zCheck if Docket is configured with a distributed backend. The CLI worker runs as a separate process, so it needs Redis/Valkey to coordinate with the main server process. Raises: SystemExit: If using memory:// URL rNz memory://up[bold red]✗ In-memory backend not supported by CLI[/bold red] Your Docket configuration uses an in-memory backend (memory://) which only works within a single process. To use [cyan]fastmcp tasks[/cyan] CLI commands (which run in separate processes), you need a distributed backend: [bold]1. Install Redis or Valkey:[/bold] [dim]macOS:[/dim] brew install redis [dim]Ubuntu:[/dim] apt install redis-server [dim]Valkey:[/dim] See https://valkey.io/ [bold]2. Start the service:[/bold] redis-server [bold]3. Configure Docket URL:[/bold] [dim]Environment variable:[/dim] export FASTMCP_DOCKET_URL=redis://localhost:6379/0 [bold]4. Try again[/bold] The memory backend works great for single-process servers, but the CLI commands need a distributed backend to coordinate across processes. Need help? See: [cyan]https://gofastmcp.com/docs/tasks[/cyan]) fastmcpsettingsdocketurl startswithconsoleprintsysexit)r docket_urls ]/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/cli/tasks.pycheck_distributed_backendrsR!!((,,J[)  L (  +* server_speczWPython file to run, optionally with :object suffix, or None to auto-detect fastmcp.json)r cddlt t|\}}t jjjfd} t j|y#t$rt j dYowxYw#t$r-tjdt j dYywxYw)aLStart an additional worker to process background tasks. Connects to your Docket backend and processes tasks in parallel with any other running workers. Configure via environment variables (FASTMCP_DOCKET_*). Example: fastmcp tasks worker server.py fastmcp tasks worker examples/tasks/server.py rNr c8Kj4d{tjdjdtjdjj jtjdjj j tjdjj j tjdd{77#1d{7swYyxYww)z'Enter server lifespan and camp forever.Nu7[bold green]✓[/bold green] Starting worker for [cyan]z[/cyan]z Docket: z Backend: z Concurrency: i) _lifespan_managerrrrrrr concurrencyasynciosleep)r serversr run_workerzworker..run_worker\s++- * * MMI&++V]^  MMJw'7'7'>'>'C'C&DE F MMK(8(8(?(?(C(C'DE F MMOG,<,<,C,C,O,O+PQ RmmD))) ** * * *sEDDDCD;D<DDDD D DDz [yellow]Worker stopped[/yellow]) r rrFileNotFoundErrorrrrrunsource load_serverKeyboardInterruptrr)rconfig_resolved_specr"r r!s @@rworkerr*<s&!6{!C [[224 5F * JL!+   ,  9:  s#A/B/BB3C C )r N)N)__doc__rrtypingrcyclopts rich.consolerfastmcp.utilities.clirfastmcp.utilities.loggingrloggerrApp tasks_apprcommandstr Parameterr*rrr8s3  70 K  ) HLL  7  #L  1 d j  1 11r