ateetvatan commited on
Commit
ef83977
·
1 Parent(s): e6a07cc

modified default root get

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -3,7 +3,7 @@ app.py FastAPI API for Quantized OpenChat 3.5 (GGUF) using ctransformers
3
  """
4
 
5
  import logging
6
- from fastapi import FastAPI, HTTPException
7
  from pydantic import BaseModel
8
  from model_loader import model
9
  import uvicorn
@@ -37,7 +37,9 @@ class ChatResponse(BaseModel):
37
 
38
 
39
  @app.get("/")
40
- def root():
 
 
41
  return {"message": "MASX OpenChat API is running"}
42
 
43
 
 
3
  """
4
 
5
  import logging
6
+ from fastapi import FastAPI, HTTPException, Request
7
  from pydantic import BaseModel
8
  from model_loader import model
9
  import uvicorn
 
37
 
38
 
39
  @app.get("/")
40
+ def root(request: Request):
41
+ if request.query_params.get("logs") == "container":
42
+ return {"status": "log-polling"} # optional message
43
  return {"message": "MASX OpenChat API is running"}
44
 
45