sachin commited on
Commit
ac28f35
·
1 Parent(s): 64ec71a
Files changed (1) hide show
  1. src/server/main.py +5 -5
src/server/main.py CHANGED
@@ -406,7 +406,6 @@ async def translate(
406
 
407
  from fastapi import FastAPI, File, HTTPException, Request, UploadFile, Form, Query
408
  from pydantic import BaseModel, Field
409
-
410
  class VisualQueryResponse(BaseModel):
411
  answer: str
412
 
@@ -474,11 +473,11 @@ async def visual_query(
474
  response.raise_for_status()
475
 
476
  response_data = response.json()
477
- answer = response_data.get("answer", "")
478
 
479
  if not answer:
480
- logger.warning(f"Empty answer received from external API: {response_data}")
481
- raise HTTPException(status_code=500, detail="No answer provided by visual query service")
482
 
483
  logger.info(f"Visual query successful: {answer}")
484
  return VisualQueryResponse(answer=answer)
@@ -492,7 +491,8 @@ async def visual_query(
492
  except ValueError as e:
493
  logger.error(f"Invalid JSON response: {str(e)}")
494
  raise HTTPException(status_code=500, detail="Invalid response format from visual query service")
495
-
 
496
  from enum import Enum
497
 
498
  class SupportedLanguage(str, Enum):
 
406
 
407
  from fastapi import FastAPI, File, HTTPException, Request, UploadFile, Form, Query
408
  from pydantic import BaseModel, Field
 
409
  class VisualQueryResponse(BaseModel):
410
  answer: str
411
 
 
473
  response.raise_for_status()
474
 
475
  response_data = response.json()
476
+ answer = response_data.get("response", "")
477
 
478
  if not answer:
479
+ logger.warning(f"Empty or missing 'response' field in external API response: {response_data}")
480
+ raise HTTPException(status_code=500, detail="No valid response provided by visual query service")
481
 
482
  logger.info(f"Visual query successful: {answer}")
483
  return VisualQueryResponse(answer=answer)
 
491
  except ValueError as e:
492
  logger.error(f"Invalid JSON response: {str(e)}")
493
  raise HTTPException(status_code=500, detail="Invalid response format from visual query service")
494
+
495
+
496
  from enum import Enum
497
 
498
  class SupportedLanguage(str, Enum):