diff --git a/rag/init.py b/rag/init.py index 567e6a5..461292d 100644 --- a/rag/init.py +++ b/rag/init.py @@ -12,6 +12,19 @@ from sqlor.dbpools import DBPools from appPublic.registerfunction import RegisterFunction from rag.ragapi import docs, get_kdbs, fusedsearch +async def get_user_kdbs(request): + env = request._run_ns + db = env.DBPools() + dbname = env.get_module_dbname('rag') + userorgid = await env.get_userorgid() + async with db.sqlorContext(dbname) as sor: + sql = "select * from kdb where ownerid = ${orgid}$" + sql = "select * from kdb where orgid = ${orgid}$" + recs = await sor.sqlExe(sql, {'orgid': userorgid}) + return recs + return recs + return [] + def load_rag(): """ 初始化 ServerEnv,绑定 MilvusConnection 的所有功能。