hybrid-rag-parser/api_server.py at main · tahaislam/hybrid-rag-parser · 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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#!/usr/bin/env python
"""
Entry point for running the Hybrid RAG Parser API server

Usage:
    python api_server.py                # Start with default settings
    python api_server.py --host 0.0.0.0 --port 8000  # Custom host/port
    python api_server.py --reload       # Development mode with hot reload

Environment Variables:
    API_HOST: Server host (default: 0.0.0.0)
    API_PORT: Server port (default: 8000)
    DEBUG: Enable debug mode (default: False)
    REDIS_URL: Redis connection URL for caching (optional)
    CACHE_TTL_SECONDS: Cache TTL in seconds (default: 3600)
"""

import argparse
import os
import sys
import logging

try:
    import uvicorn
except ImportError:
    print("Error: uvicorn is not installed. Install it with:")
    print("    pip install uvicorn[standard]")
    sys.exit(1)

from src.api.config import config

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
)
logger = logging.getLogger(__name__)


def main():
    """Run the API server"""
    parser = argparse.ArgumentParser(
        description="Hybrid RAG Parser API Server",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog=__doc__,
    )

    parser.add_argument(
        "--host",
        type=str,
        default=config.HOST,
        help=f"Server host (default: {config.HOST})",
    )

    parser.add_argument(
        "--port",
        type=int,
        default=config.PORT,
        help=f"Server port (default: {config.PORT})",
    )

    parser.add_argument(
        "--reload",
        action="store_true",
        help="Enable auto-reload mode for development",
    )

    parser.add_argument(
        "--workers",
        type=int,
        default=1,
        help="Number of worker processes (default: 1)",
    )

    parser.add_argument(
        "--log-level",
        type=str,
        choices=["critical", "error", "warning", "info", "debug"],
        default="info",
        help="Logging level (default: info)",
    )

    args = parser.parse_args()

    logger.info("=" * 70)
    logger.info("Hybrid RAG Parser API Server")
    logger.info("=" * 70)
    logger.info(f"Host: {args.host}")
    logger.info(f"Port: {args.port}")
    logger.info(f"Reload: {args.reload}")
    logger.info(f"Workers: {args.workers}")
    logger.info(f"Log Level: {args.log_level.upper()}")
    logger.info(f"Cache Enabled: {config.CACHE_ENABLED}")
    if config.REDIS_URL:
        logger.info(f"Redis URL: {config.REDIS_URL}")
    logger.info("=" * 70)
    logger.info("")
    logger.info("Server starting at:")
    logger.info(f"  http://{args.host}:{args.port}")
    logger.info("")
    logger.info("Documentation available at:")
    logger.info(f"  http://{args.host}:{args.port}/docs (Swagger UI)")
    logger.info(f"  http://{args.host}:{args.port}/redoc (ReDoc)")
    logger.info("")

    # Run server
    uvicorn.run(
        "src.api.main:app",
        host=args.host,
        port=args.port,
        reload=args.reload,
        workers=args.workers if not args.reload else 1,
        log_level=args.log_level,
    )


if __name__ == "__main__":
    main()