rishitpant commited on
Commit
3cbcbf3
·
1 Parent(s): 4fc045a

Update limit

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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=["10/minute"])
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=["2/minute"],
38
  )
39
 
40
 
@@ -91,7 +91,7 @@ def health():
91
 
92
 
93
  @app.post("/api/chat", response_model=ChatResponse)
94
- @limiter.limit("2/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,7 +99,7 @@ def chat(req: ChatRequest, request: Request):
99
 
100
 
101
  @app.post("/api/chat/stream")
102
- @limiter.limit("2/minute")
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