github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2026-07-17 11:29:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: don't allow to run worker without celery (#2243)

Browse Source 
(cherry picked from commit 162b9087b4cf3fb66ae0c8ae465bd90e8dacc0f2)
This commit is contained in:
Javier Martinez
2026-06-03 06:52:41 +02:00
committed by GitHub
parent aa5262154f
commit 86ceea5de8
1 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
private_gpt/cli/main.py
View File

@@ -5,7 +5,15 @@ import typer
from private_gpt.cli.commands.run import run_command
from private_gpt.cli.commands.serve import serve_command
from private_gpt.cli.commands.worker import worker_command
try:
import celery as _celery # noqa: F401
from private_gpt.cli.commands.worker import worker_command
_CELERY_AVAILABLE = True
except ImportError:
_CELERY_AVAILABLE = False
app = typer.Typer(
name="private-gpt",
@@ -16,7 +24,8 @@ app = typer.Typer(
)
app.command("serve")(serve_command)
app.command("worker")(worker_command)
if _CELERY_AVAILABLE:
app.command("worker")(worker_command)
app.command(
"run",
context_settings={"allow_extra_args": True, "ignore_unknown_options": True},
@@ -38,7 +47,9 @@ def help_command(
raise SystemExit(0)
_KNOWN_COMMANDS = ["serve", "worker", "run", "help"]
_KNOWN_COMMANDS = ["serve", "run", "help"]
if _CELERY_AVAILABLE:
_KNOWN_COMMANDS.append("worker")
def main() -> None: