refac
This commit is contained in:
1 parent
4602abe5c6
commit
f85cb27ef8
1 file changed
+9
@@ -64,9 +64,18 @@ def get_custom_headers(custom_headers: dict, user=None, metadata: dict = None) -
|
|||||||
return {}
|
return {}
|
||||||
|
|
||||||
metadata = metadata or {}
|
metadata = metadata or {}
|
||||||
|
|
||||||
|
# Extract user_message info for tree mapping
|
||||||
|
user_message = metadata.get('user_message') or {}
|
||||||
|
user_message_id = metadata.get('user_message_id', '') or (user_message.get('id', '') if user_message else '')
|
||||||
|
user_message_parent_id = user_message.get('parentId', '') if user_message else ''
|
||||||
|
|
||||||
template_vars = {
|
template_vars = {
|
||||||
'{{CHAT_ID}}': metadata.get('chat_id', '') or '',
|
'{{CHAT_ID}}': metadata.get('chat_id', '') or '',
|
||||||
'{{MESSAGE_ID}}': metadata.get('message_id', '') or '',
|
'{{MESSAGE_ID}}': metadata.get('message_id', '') or '',
|
||||||
|
'{{USER_MESSAGE_ID}}': user_message_id or '',
|
||||||
|
'{{USER_MESSAGE_PARENT_ID}}': user_message_parent_id or '',
|
||||||
|
'{{TASK}}': metadata.get('task', '') or '',
|
||||||
'{{USER_ID}}': (user.id if user else '') or '',
|
'{{USER_ID}}': (user.id if user else '') or '',
|
||||||
'{{USER_NAME}}': (user.name if user else '') or '',
|
'{{USER_NAME}}': (user.name if user else '') or '',
|
||||||
'{{USER_EMAIL}}': (user.email if user else '') or '',
|
'{{USER_EMAIL}}': (user.email if user else '') or '',
|
||||||
|
|||||||
Reference in new issue
Block a user