AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 15:15:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: langsmith message_trace end_user_data session_id error (#5759)

Browse Source
This commit is contained in:
Joe 2024-06-30 01:12:16 +08:00 committed by GitHub
parent ffb07eb24b
commit f637ae4794
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/ops/langfuse_trace/langfuse_trace.py
View File

@ -215,7 +215,9 @@ class LangFuseDataTrace(BaseTraceInstance):
end_user_data: EndUser = db.session.query(EndUser).filter( end_user_data: EndUser = db.session.query(EndUser).filter(
EndUser.id == message_data.from_end_user_id EndUser.id == message_data.from_end_user_id
).first() ).first()
user_id = end_user_data.session_id if end_user_data is not None:
user_id = end_user_data.session_id
metadata["user_id"] = user_id
trace_data = LangfuseTrace( trace_data = LangfuseTrace(
id=message_id, id=message_id,

10
api/core/ops/langsmith_trace/langsmith_trace.py
View File

@ -183,13 +183,15 @@ class LangSmithDataTrace(BaseTraceInstance):
message_id = message_data.id message_id = message_data.id
user_id = message_data.from_account_id user_id = message_data.from_account_id
metadata["user_id"] = user_id
if message_data.from_end_user_id: if message_data.from_end_user_id:
end_user_data: EndUser = db.session.query(EndUser).filter( end_user_data: EndUser = db.session.query(EndUser).filter(
EndUser.id == message_data.from_end_user_id EndUser.id == message_data.from_end_user_id
).first().session_id ).first()
end_user_id = end_user_data.session_id if end_user_data is not None:
metadata["end_user_id"] = end_user_id end_user_id = end_user_data.session_id
metadata["user_id"] = user_id metadata["end_user_id"] = end_user_id
message_run = LangSmithRunModel( message_run = LangSmithRunModel(
input_tokens=trace_info.message_tokens, input_tokens=trace_info.message_tokens,