SwarmSell/run.py at main · adityasingh2400/SwarmSell · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
"""Entry point for the v2 pipeline."""
import logging
import sys

import uvicorn
from config import settings

_DEBUG_BANNER = """
================================================================================
  SWARMA DEBUG TRACES
  • Terminal (python run.py): copy lines starting with "SWARMA |"
  • Browser: DevTools → Console → filter "SWARMA:fe"
================================================================================
"""


def main():
    logging.basicConfig(
        level=logging.DEBUG,
        format="%(asctime)s | %(levelname)-5s | %(name)s | %(message)s",
        datefmt="%H:%M:%S",
        stream=sys.stdout,
        force=True,
    )
    logging.getLogger("swarma.trace").setLevel(logging.DEBUG)
    logging.getLogger("swarmsell.server").setLevel(logging.DEBUG)
    logging.getLogger("httpx").setLevel(logging.WARNING)
    logging.getLogger("httpcore").setLevel(logging.WARNING)
    logging.getLogger("hpack").setLevel(logging.WARNING)
    logging.getLogger("sentence_transformers").setLevel(logging.WARNING)
    logging.getLogger("huggingface_hub").setLevel(logging.WARNING)
    logging.getLogger("watchfiles").setLevel(logging.WARNING)
    logging.getLogger("watchfiles.main").setLevel(logging.WARNING)
    print(_DEBUG_BANNER, flush=True)

    settings.ensure_dirs()

    log_config = uvicorn.config.LOGGING_CONFIG
    log_config["loggers"]["watchfiles"] = {"level": "WARNING", "handlers": []}
    log_config["loggers"]["watchfiles.main"] = {"level": "WARNING", "handlers": []}

    uvicorn.run(
        "backend.server:app",
        host=settings.api_host,
        port=settings.api_port,
        reload=True,
        log_level="info",
        log_config=log_config,
    )


if __name__ == "__main__":
    main()