rag

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)
+