Spaces:
Paused
Paused
sachin
commited on
Commit
·
ac28f35
1
Parent(s):
64ec71a
update
Browse files- 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("
|
478 |
|
479 |
if not answer:
|
480 |
-
logger.warning(f"Empty
|
481 |
-
raise HTTPException(status_code=500, detail="No
|
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):
|