.

main/main.py

@@ -3,7 +3,7 @@ from fastapi.middleware.cors import CORSMiddleware
 import logging
 import os
 import uvicorn
-from routes import router
+from .routes import router
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -35,12 +35,14 @@ def main():
     port = int(os.getenv("LLM_ENGINE_PORT", "7860"))  # Default to 7860 for Spaces
 
     # Log startup information
-    logger.info(f"Starting LLM Engine service on {host}:{port}")
+    logger.info(f"Starting LLM Engine service on {host}:{port}, or: ")
     logger.info("Available endpoints:")
     logger.info("  - /")
     logger.info("  - /health")
     logger.info("  - /initialize")
     logger.info("  - /generate")
+    logger.info("  - /initialize/custom")
+    logger.info("  - /generate/stream")
     logger.info("  - /docs")
     logger.info("  - /redoc")
     logger.info("  - /openapi.json")