API: Add log options for initialization

Make each API log their respective URLs to help inform users.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri
2024-07-26 21:32:05 -04:00
parent e8fc13a1f6
commit 884b6f5ecd
3 changed files with 30 additions and 10 deletions

View File

@@ -22,7 +22,12 @@ from endpoints.Kobold.utils.generation import (
from endpoints.core.utils.model import get_current_model
api_name = "KoboldAI"
router = APIRouter(prefix="/api")
urls = {
"Generation": "http://{host}:{port}/api/v1/generate",
"Streaming": "http://{host}:{port}/api/extra/generate/stream",
}
@router.post(