46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
from appPublic.log import debug, error, info
|
||
from ahserver.serverenv import ServerEnv
|
||
import aiohttp
|
||
from aiohttp import ClientSession, ClientTimeout
|
||
import json
|
||
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, textinsert
|
||
|
||
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 的所有功能。
|
||
"""
|
||
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('textinsert', textinsert)
|
||
|
||
|
||
|