This commit is contained in:
yumoqing 2025-08-20 15:46:34 +08:00
parent 594b38e5dc
commit b95722ec01
2 changed files with 39 additions and 0 deletions

View File

@ -0,0 +1,33 @@
db = DBPools()
dbname = get_module_dbname('rag')
async with db.sqlorContext(dbname) as sor:
u = await get_session_userinfo(request)
sql="select * from ragquote where orgid = ${orgid}$ order by enabled_date"
qs = await sor.sqlExe(sql, {'orgid': u.userorgid})
today = curDateString()
if len(qs) == 0:
if params_kw.program_type == 'free':
expired_date = dateadd(today, months=1)
else:
expired_date = '9999-12-31'
ns = {
"id": uuid(),
"orgid": u.userorgid,
"enabled_date": today,
"expired_date": expired_date,
'quota': params_kw.quota
}
await sor.C('ragquota', ns)
else:
lastq = qs[-1]
lastq.expired_date = today
await sor.U('ragquote', lastq.copy())
ns = {
"id": uuid(),
"orgid": u.userorgid,
"enabled_date": today,
"expired_date": expired_date,
'quota': params_kw.quota
}
await sor.C('ragquota', ns)

View File

@ -0,0 +1,6 @@
debug(f'{params=}')
fiid = params_kw.ffid
query = params_kw.query
rf = RegisteFunction('fusedsearch')
x = await rf.exe('fusedsearch', params_kw, DictObject(**globals()))
return x