youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-03 17:08:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

ruff format

Browse Source
This commit is contained in:
hjlarry
2025-09-10 14:25:37 +08:00
parent 21fee59b22
commit c8acc48976
6 changed files with 52 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
api/controllers/console/app/online_user.py
View File

@ -88,7 +88,7 @@ def handle_disconnect(sid):
if mapping:
data = json.loads(mapping)
workflow_id = data["workflow_id"]
# Remove this specific session
redis_client.hdel(f"workflow_online_users:{workflow_id}", sid)
redis_client.delete(f"ws_sid_map:{sid}")
@ -177,18 +177,20 @@ def broadcast_online_users(workflow_id):
"""
sessions_json = redis_client.hgetall(f"workflow_online_users:{workflow_id}")
users = []
for sid, session_info_json in sessions_json.items():
try:
session_info = json.loads(session_info_json)
# Each session appears as a separate "user" in the UI
users.append({
"user_id": session_info["user_id"],
"username": session_info["username"],
"avatar": session_info.get("avatar"),
"sid": session_info["sid"],
"connected_at": session_info.get("connected_at"),
})
users.append(
{
"user_id": session_info["user_id"],
"username": session_info["username"],
"avatar": session_info.get("avatar"),
"sid": session_info["sid"],
"connected_at": session_info.get("connected_at"),
}
)
except Exception:
continue

13
api/controllers/console/app/workflow_comment.py
View File

@ -164,10 +164,10 @@ class WorkflowCommentReplyApi(Resource):
args = parser.parse_args()
result = WorkflowCommentService.create_reply(
comment_id=comment_id,
content=args.content,
comment_id=comment_id,
content=args.content,
created_by=current_user.id,
mentioned_user_ids=args.mentioned_user_ids
mentioned_user_ids=args.mentioned_user_ids,
)
return result, 201
@ -194,10 +194,7 @@ class WorkflowCommentReplyDetailApi(Resource):
args = parser.parse_args()
reply = WorkflowCommentService.update_reply(
reply_id=reply_id,
user_id=current_user.id,
content=args.content,
mentioned_user_ids=args.mentioned_user_ids
reply_id=reply_id, user_id=current_user.id, content=args.content, mentioned_user_ids=args.mentioned_user_ids
)
return reply
@ -227,7 +224,7 @@ class WorkflowCommentMentionUsersApi(Resource):
@get_app_model
@marshal_with({"users": fields.List(fields.Nested(account_with_role_fields))})
def get(self, app_model: App):
"""Get all users in current tenant for mentions."""
"""Get all users in current tenant for mentions."""
members = TenantService.get_tenant_members(current_user.current_tenant)
return {"users": members}

12
api/controllers/console/app/workflow_draft_variable.py
View File

@ -370,7 +370,7 @@ class ConversationVariableCollectionApi(Resource):
draft_var_srv.prefill_conversation_variable_default_values(draft_workflow)
db.session.commit()
return _get_variable_list(app_model, CONVERSATION_VARIABLE_NODE_ID)
@setup_required
@login_required
@account_initialization_required
@ -379,7 +379,7 @@ class ConversationVariableCollectionApi(Resource):
# The role of the current user in the ta table must be admin, owner, or editor
if not current_user.is_editor:
raise Forbidden()
parser = reqparse.RequestParser()
parser.add_argument("conversation_variables", type=list, required=True, location="json")
args = parser.parse_args()
@ -397,7 +397,7 @@ class ConversationVariableCollectionApi(Resource):
conversation_variables=conversation_variables,
)
return { "result": "success" }
return {"result": "success"}
class SystemVariableCollectionApi(Resource):
@ -439,7 +439,7 @@ class EnvironmentVariableCollectionApi(Resource):
)
return {"items": env_vars_list}
@setup_required
@login_required
@account_initialization_required
@ -448,7 +448,7 @@ class EnvironmentVariableCollectionApi(Resource):
# The role of the current user in the ta table must be admin, owner, or editor
if not current_user.is_editor:
raise Forbidden()
parser = reqparse.RequestParser()
parser.add_argument("environment_variables", type=list, required=True, location="json")
args = parser.parse_args()
@ -466,7 +466,7 @@ class EnvironmentVariableCollectionApi(Resource):
environment_variables=environment_variables,
)
return { "result": "success" }
return {"result": "success"}
api.add_resource(