diff --git a/backend/app/api/routes/utils.py b/backend/app/api/routes/utils.py index 5be4db2..dbb6ad8 100644 --- a/backend/app/api/routes/utils.py +++ b/backend/app/api/routes/utils.py @@ -2,6 +2,7 @@ from fastapi import APIRouter, Depends from pydantic.networks import EmailStr import httpx from pydantic import BaseModel +import re from app.api.deps import get_current_active_superuser from app.models import Message @@ -37,10 +38,15 @@ class ClassifyRequest(BaseModel): @router.post("/bentoml_classifiy") async def bentoml_classifiy(data: ClassifyRequest): + text = data.text + cleaned_text = text.strip().replace("\n", "") + cleaned_text = re.sub(r'[\x00-\x1F]', '', cleaned_text) + + async with httpx.AsyncClient() as client: response = await client.post( "http://120.76.41.122:3000/classify", - json={"text": f"{data.text}"} + json={"text": f"{cleaned_text}"} ) result = response.json() return fast_json_response(data=result)