feat: update slang prediction endpoint to support batch input

Author: cjeongmin

apps/classifier/app.py

@@ -5,7 +5,7 @@
     PredictionRequest,
     PredictionResponse,
 )
-from model import predict
+from model import predict, predict_batch
 
 app = FastAPI()
 
@@ -26,6 +26,8 @@ async def improve_reply_predict(data: PredictionRequest):
 
 @app.post("/slang-predict", response_model=SlangPredictionResponse)
 async def slang_predict(data: SlangPredictionRequest):
-    text = data.input
-    predicted = predict(text, type="slang")
-    return {"predicted": predicted[0], "probability": predicted[1]}
+    text = data.inputs
+    predicted = predict_batch(text, type="slang")
+    return {
+        "predictions": [{"predicted": p[0], "probability": p[1]} for p in predicted]
+    }

apps/classifier/schemas.py

@@ -1,27 +1,29 @@
 from pydantic import BaseModel
+from typing import List
 
 
 class SlangPredictionRequest(BaseModel):
-    input: str
+    inputs: List[str]
 
     class Config:
         json_schema_extra = {
             "example": {
-                "input": "X같네",
+                "inputs": ["X같네"],
             }
         }
 
 
-class SlangPredictionResponse(BaseModel):
+class SlangPredictionItem(BaseModel):
     predicted: str
     probability: float
 
+
+class SlangPredictionResponse(BaseModel):
+    predictions: List[SlangPredictionItem]
+
     class Config:
         json_schema_extra = {
-            "example": {
-                "predicted": "욕설",
-                "probability": 0.99,
-            }
+            "example": {"predictions": [{"predicted": "욕설", "probability": 0.99}]}
         }