Spaces:
Running
Running
Commit ·
3cbcbf3
1
Parent(s): 4fc045a
Update limit
Browse files
app.py
CHANGED
|
@@ -18,7 +18,7 @@ from generate import Generator
|
|
| 18 |
from retrieve import USE_CHROMA_CLOUD
|
| 19 |
|
| 20 |
_state: dict = {}
|
| 21 |
-
limiter = Limiter(key_func=get_remote_address, default_limits=["
|
| 22 |
|
| 23 |
|
| 24 |
def get_client_ip(request: Request) -> str:
|
|
@@ -34,7 +34,7 @@ def get_client_ip(request: Request) -> str:
|
|
| 34 |
|
| 35 |
limiter = Limiter(
|
| 36 |
key_func=get_client_ip,
|
| 37 |
-
default_limits=["
|
| 38 |
)
|
| 39 |
|
| 40 |
|
|
@@ -91,7 +91,7 @@ def health():
|
|
| 91 |
|
| 92 |
|
| 93 |
@app.post("/api/chat", response_model=ChatResponse)
|
| 94 |
-
@limiter.limit("
|
| 95 |
def chat(req: ChatRequest, request: Request):
|
| 96 |
gen = get_generator()
|
| 97 |
tokens = list(gen.answer(req.question, top_n=req.top_n))
|
|
@@ -99,7 +99,7 @@ def chat(req: ChatRequest, request: Request):
|
|
| 99 |
|
| 100 |
|
| 101 |
@app.post("/api/chat/stream")
|
| 102 |
-
@limiter.limit("
|
| 103 |
def chat_stream(req: ChatRequest, request: Request):
|
| 104 |
gen = get_generator()
|
| 105 |
|
|
|
|
| 18 |
from retrieve import USE_CHROMA_CLOUD
|
| 19 |
|
| 20 |
_state: dict = {}
|
| 21 |
+
limiter = Limiter(key_func=get_remote_address, default_limits=["5/minute"])
|
| 22 |
|
| 23 |
|
| 24 |
def get_client_ip(request: Request) -> str:
|
|
|
|
| 34 |
|
| 35 |
limiter = Limiter(
|
| 36 |
key_func=get_client_ip,
|
| 37 |
+
default_limits=["5/minute"],
|
| 38 |
)
|
| 39 |
|
| 40 |
|
|
|
|
| 91 |
|
| 92 |
|
| 93 |
@app.post("/api/chat", response_model=ChatResponse)
|
| 94 |
+
@limiter.limit("5/minute")
|
| 95 |
def chat(req: ChatRequest, request: Request):
|
| 96 |
gen = get_generator()
|
| 97 |
tokens = list(gen.answer(req.question, top_n=req.top_n))
|
|
|
|
| 99 |
|
| 100 |
|
| 101 |
@app.post("/api/chat/stream")
|
| 102 |
+
@limiter.limit("5/minute")
|
| 103 |
def chat_stream(req: ChatRequest, request: Request):
|
| 104 |
gen = get_generator()
|
| 105 |
|