sachin commited on
Commit
78be025
·
1 Parent(s): 45d1513
Files changed (1) hide show
  1. src/server/main.py +3 -2
src/server/main.py CHANGED
@@ -537,8 +537,9 @@ class VisualQueryRequest(BaseModel):
537
  class VisualQueryResponse(BaseModel):
538
  answer: str
539
 
 
540
  @app.post("/v1/visual_query", response_model=VisualQueryResponse)
541
- @limiter.limit(settings.chat_rate_limit) # Reusing chat rate limit
542
  async def visual_query(
543
  request: Request,
544
  query: str = Form(...),
@@ -557,7 +558,7 @@ async def visual_query(
557
  logger.info("Processing visual query request", extra={
558
  "endpoint": "/v1/visual_query",
559
  "query_length": len(query),
560
- "filename": file.filename,
561
  "client_ip": get_remote_address(request),
562
  "src_lang": src_lang,
563
  "tgt_lang": tgt_lang
 
537
  class VisualQueryResponse(BaseModel):
538
  answer: str
539
 
540
+
541
  @app.post("/v1/visual_query", response_model=VisualQueryResponse)
542
+ @limiter.limit(settings.chat_rate_limit)
543
  async def visual_query(
544
  request: Request,
545
  query: str = Form(...),
 
558
  logger.info("Processing visual query request", extra={
559
  "endpoint": "/v1/visual_query",
560
  "query_length": len(query),
561
+ "file_name": file.filename, # Changed from "filename" to "file_name"
562
  "client_ip": get_remote_address(request),
563
  "src_lang": src_lang,
564
  "tgt_lang": tgt_lang