bugfix

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
 

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 []
+