This commit is contained in:
yumoqing 2026-03-31 15:14:12 +08:00
parent bac11ef34b
commit b487bc9c34

View File

@ -10,8 +10,7 @@ if params_kw.off_peak:
userid = await get_user() userid = await get_user()
userorgid = await get_userorgid() userorgid = await get_userorgid()
if userid is None: if userid is None:
d = return_error('Authoration Error') return openai_403()
return json_response(d, status=401)
if not params_kw.prompt: if not params_kw.prompt:
d = return_error('Missing need data(prompt)') d = return_error('Missing need data(prompt)')
@ -27,14 +26,12 @@ where a.llmcatelogid=b.id
'model': params_kw.model or 'qwen3-max' 'model': params_kw.model or 'qwen3-max'
}) })
if len(recs) == 0: if len(recs) == 0:
d = return_error(f'{params_kw.model} not found') return openai_400()
return json_response(d, status=400)
params_kw.llmid = recs[0].id params_kw.llmid = recs[0].id
f = await checkCustomerBalance(params_kw.llmid, userorgid) f = await checkCustomerBalance(params_kw.llmid, userorgid)
if not f: if not f:
d = return_error('InsufficientFunds') return openai_429()
return json_response(d, status=400)
env = DictObject(**globals()) env = DictObject(**globals())
return await inference(request, env=env) return await inference(request, env=env)