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

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

diff --git a/rag/init.py b/rag/init.py
index 89a2cc3..567e6a5 100644
--- a/rag/init.py
+++ b/rag/init.py
@@ -1,3 +1,14 @@
+from appPublic.log import debug, error, info
+from ahserver.serverenv import ServerEnv
+import aiohttp
+from aiohttp import ClientSession, ClientTimeout
+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_by_catelog
+from appPublic.registerfunction import RegisterFunction
+from sqlor.dbpools import DBPools
 from appPublic.registerfunction import RegisterFunction
 from rag.ragapi import docs, get_kdbs, fusedsearch
 
@@ -5,8 +16,18 @@ def load_rag():
 	"""
 	初始化 ServerEnv，绑定 MilvusConnection 的所有功能。
 	"""
+	env = ServerEnv()
+	env.docs = docs
+	env.RagFileMgr = RagFileMgr
+	env.set_program = set_program
+	env.get_rag_programs = get_rag_programs
+	env.get_ragllms_by_catelog = get_ragllms_by_catelog
+	env.get_user_kdbs = get_user_kdbs
 	rf = RegisterFunction()
 	rf.register('docs', docs)
 	rf.register('get_kdbs', get_kdbs)
 	rf.register('fusedsearch', fusedsearch)
+	rf.register('add_user_messages', add_user_messages)
+	rf.register('get_user_memories', get_user_memories)
+