rag

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 的所有功能。