rag-gateway.git

			@@ -1,29 +1,61 @@
			from datetime import datetime

			from app.models import KnowledgeModel, GroupModel, DialogModel, ConversationModel, group_dialog_table
			from sqlalchemy import or_

			from app.config.const import Dialog_STATSU_DELETE
			from app.models import KnowledgeModel, GroupModel, DialogModel, ConversationModel, group_dialog_table, LabelWorkerModel, \
			LabelModel
			from app.models.user_model import UserModel
			from Log import logger


			async def get_dialog_list(db, user_id):
			async def get_dialog_list(db, user_id, keyword, label, status, page_size, page_index):
			user = db.query(UserModel).filter(UserModel.id == user_id).first()
			if user is None:
			return {"rows": []}
			if user.permission == "admin":
			query = db.query(DialogModel)
			query = db.query(DialogModel)
			if status:
			query = query.filter(DialogModel.status == status)
			else:
			group_list = [i.id for i in user.groups]
			query = db.query(DialogModel)
			query = query.filter(DialogModel.tenant_id == user_id)
			query = query.union(
			db.query(DialogModel).join(
			group_dialog_table,
			DialogModel.id == group_dialog_table.c.dialog_id
			).filter(
			group_dialog_table.c.group_id.in_(group_list)
			)
			)
			return {"rows": [kld.to_json() for kld in query.all()]}
			query = query.filter(DialogModel.status != Dialog_STATSU_DELETE)
			id_list = []
			# if label:
			# id_list = [i.object_id for i in db.query(LabelWorkerModel).filter(LabelWorkerModel.label_id==label).all()]
			if user.permission != "admin":
			dia_list = [j.id for i in user.groups for j in i.dialogs]
			query = query.filter(or_(DialogModel.tenant_id == user_id, DialogModel.id.in_(dia_list)))
			# else:
			if label:
			id_list = [i.object_id for i in db.query(LabelWorkerModel).filter(LabelWorkerModel.label_id == label).all()]
			query = query.filter(DialogModel.id.in_(id_list))

			if keyword:
			query = query.filter(DialogModel.name.like('%{}%'.format(keyword)))


			query = query.order_by(DialogModel.update_date.desc())
			total = query.count()
			if page_size:
			query = query.limit(page_size).offset((page_index - 1) * page_size)

			rows = []
			user_id_set = set()
			dialog_id_set = set()
			label_dict = {}
			for kld in query.all():
			user_id_set.add(kld.tenant_id)
			dialog_id_set.add(kld.id)
			rows.append(kld.to_json())
			user_dict = {str(i.id): i.to_dict() for i in db.query(UserModel).filter(UserModel.id.in_(user_id_set)).all()}
			for i in db.query(LabelModel.id, LabelModel.name, LabelWorkerModel.object_id).outerjoin(LabelWorkerModel,
			LabelModel.id == LabelWorkerModel.label_id).filter(
			LabelWorkerModel.object_id.in_(dialog_id_set)).all():

			label_dict[i.object_id] = label_dict.get(i.object_id, []) +[{"labelId": i.id, "labelName": i.name}]
			for r in rows:
			r["user"] = user_dict.get(r["user_id"], {})
			r["label"] = label_dict.get(r["id"], [])
			return {"total": total, "rows": rows}


			async def update_session_history(db, data: dict, user_id):
			@@ -54,8 +86,102 @@
			db.rollback()


			async def get_session_history(db, user_id, dialog_id):
			async def get_session_history(db, user_id, dialog_id, page, limit):
			session_list = db.query(ConversationModel).filter(ConversationModel.tenant_id.__eq__(user_id),
			ConversationModel.dialog_id.__eq__(dialog_id)).order_by(
			ConversationModel.update_time.desc()).all()
			ConversationModel.update_time.desc()).limit(limit).offset((page - 1) * limit).all()
			return [i.to_json() for i in session_list]



			async def create_dialog_service(db, dialog_id, dialog_name, description, icon, dialog_type, mode, user_id):
			try:
			dialog_model = DialogModel(id=dialog_id,name=dialog_name, description=description,icon=icon, dialog_type=dialog_type, tenant_id=user_id, mode=mode,update_date=datetime.now(),create_date=datetime.now())
			db.add(dialog_model)
			db.commit()
			db.refresh(dialog_model)
			except Exception as e:
			logger.error(e)
			db.rollback()
			return False
			return True


			async def update_dialog_status_service(db, dialog_id, status):
			try:
			db.query(DialogModel).filter_by(id=dialog_id).update({"status":status, "update_date": datetime.now()})
			db.commit()
			except Exception as e:
			logger.error(e)
			db.rollback()
			return False
			return True


			async def delete_dialog_service(db, dialog_id):
			try:
			db.query(DialogModel).filter_by(id=dialog_id).update({"status":Dialog_STATSU_DELETE, "update_date": datetime.now()})
			db.commit()
			except Exception as e:
			logger.error(e)
			db.rollback()
			return False
			return True


			async def update_dialog_icon_service(db, dialog_id, icon):
			try:
			db.query(DialogModel).filter_by(id=dialog_id).update({"icon":icon, "update_date": datetime.now()})
			db.commit()
			except Exception as e:
			logger.error(e)
			db.rollback()
			return False
			return True


			async def get_dialog_manage_list(db, user_id, keyword, label, status, page_size, page_index, mode):
			user = db.query(UserModel).filter(UserModel.id == user_id).first()
			if user is None:
			return {"rows": []}
			query = db.query(DialogModel).filter(DialogModel.status != Dialog_STATSU_DELETE)
			if user.permission != "admin":
			dia_list = [j.id for i in user.groups for j in i.dialogs]
			query = query.filter(or_(DialogModel.tenant_id == user_id, DialogModel.id.in_(dia_list)))

			if label:
			id_list = set(
			[i.object_id for i in db.query(LabelWorkerModel).filter(LabelWorkerModel.label_id.in_(label)).all()])
			query = query.filter(DialogModel.id.in_(id_list))

			if keyword:
			query = query.filter(DialogModel.name.like('%{}%'.format(keyword)))

			if status:
			# print(status)
			query = query.filter(DialogModel.status == status)
			if mode:
			query = query.filter(DialogModel.mode == mode)
			query = query.order_by(DialogModel.update_date.desc())
			total = query.count()
			if page_size:
			query = query.limit(page_size).offset((page_index - 1) * page_size)

			rows = []
			user_id_set = set()
			dialog_id_set = set()
			label_dict = {}
			for kld in query.all():
			user_id_set.add(kld.tenant_id)
			dialog_id_set.add(kld.id)
			rows.append(kld.to_json())
			user_dict = {str(i.id): i.to_dict() for i in db.query(UserModel).filter(UserModel.id.in_(user_id_set)).all()}
			for i in db.query(LabelModel.id, LabelModel.name, LabelWorkerModel.object_id).outerjoin(LabelWorkerModel,
			LabelModel.id == LabelWorkerModel.label_id).filter(
			LabelWorkerModel.object_id.in_(dialog_id_set)).all():

			label_dict[i.object_id] = label_dict.get(i.object_id, []) +[{"labelId": i.id, "labelName": i.name}]
			for r in rows:
			r["user"] = user_dict.get(r["user_id"], {})
			r["label"] = label_dict.get(r["id"], [])
			return {"total": total, "rows": rows}