From 24a7a1087df68ac80e978150e6e294047a42dc44 Mon Sep 17 00:00:00 2001
From: wangmeihua <13383952685@163.com>
Date: Fri, 10 Oct 2025 14:19:19 +0800
Subject: [PATCH] rag

---
 rag/init.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

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