playmak3r commited on
Commit
89891b8
·
1 Parent(s): 4abb970

enable Sugoi compatible route

Browse files
Files changed (1) hide show
  1. opus_api/app_uvi.py +5 -11
opus_api/app_uvi.py CHANGED
@@ -52,18 +52,12 @@ async def translate_post(request_body: PostRequestBody):
52
  return Response(status_code= 400)
53
 
54
 
55
- """ @app.post('/')
56
- async def default_post(request_body: DefaultSugoiRequestBody):
57
  if isinstance(request_body.content, list) and len(request_body.content) and request_body.message == "translate sentences":
58
- task_id = str(uuid.uuid4())
59
- redis_client.lpush(queue_key, json.dumps({
60
- "id": task_id,
61
- "input": request_body.content
62
- })) # left push
63
- result = query_translation(task_id)
64
- if result is not None: return { "translations": ast.literal_eval(result) }
65
- else: return JSONResponse(status_code= 529, content={"error": "Server overloaded"})
66
- return Response(status_code= 400) """
67
 
68
 
69
  """ @app.get("/api/translate/{task_id}")
 
52
  return Response(status_code= 400)
53
 
54
 
55
+ @app.post('/')
56
+ async def sugoi_default_post(request_body: DefaultSugoiRequestBody):
57
  if isinstance(request_body.content, list) and len(request_body.content) and request_body.message == "translate sentences":
58
+ status_code, result = await translate_batch(request_body.content)
59
+ return JSONResponse(status_code=status_code, content=result)
60
+ return Response(status_code= 400)
 
 
 
 
 
 
61
 
62
 
63
  """ @app.get("/api/translate/{task_id}")