From e6c5e89f09637b8d9ebca6895d781663f12646d6 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 17 十二月 2024 16:43:55 +0800
Subject: [PATCH] dify 文档智能
---
app/service/dialog.py | 28 +++++++++++++++++++++++++++-
1 files changed, 27 insertions(+), 1 deletions(-)
diff --git a/app/service/dialog.py b/app/service/dialog.py
index 623477a..d00ad53 100644
--- a/app/service/dialog.py
+++ b/app/service/dialog.py
@@ -27,8 +27,9 @@
query = query.filter(DialogModel.name.like('%{}%'.format(keyword)))
if status:
- print(status)
+ # print(status)
query = query.filter(DialogModel.status == status)
+ 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)
@@ -86,3 +87,28 @@
ConversationModel.dialog_id.__eq__(dialog_id)).order_by(
ConversationModel.update_time.desc()).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
\ No newline at end of file
--
Gitblit v1.8.0