diff --git a/rag/init.py b/rag/init.py index 21a412c..becdec1 100644 --- a/rag/init.py +++ b/rag/init.py @@ -6,6 +6,7 @@ import json from .file import file_uploaded, file_deleted from .folderinfo import RagFileMgr from .ragprogram import set_program, get_rag_programs +from ragllm_utils import get_ragllms from appPublic.registerfunction import RegisterFunction async def _make_connection_request(action: str, params: dict = None) -> dict: @@ -155,4 +156,5 @@ def load_rag(): env.RagFileMgr = RagFileMgr env.set_program = set_program env.get_rag_programs = get_rag_programs + env.get_ragllms = get_ragllms diff --git a/rag/ragllm_utils.py b/rag/ragllm_utils.py new file mode 100644 index 0000000..101b853 --- /dev/null +++ b/rag/ragllm_utils.py @@ -0,0 +1,14 @@ +from sqlor.dbpools import DBPools + +async def get_ragllms(request, **params): + env = request._run_ns + userid = await env.get_user() + userorgid = await env.get_userorgid() + dbname = env.get_module_dbname('rag') + db = DBPools() + async with db.sqlorContext(dbname) as sor: + sql = '''select * from ragllm where userscope = 'unlimited' or ownerid = ${userorgid}$''' + recs = await sor.sqlExe(sql, {'userorgid': userorgid}) + return recs + return [] +