AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 09:56:06 +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

2
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(
EndUser.id == message_data.from_end_user_id
).first()
if end_user_data is not None:
user_id = end_user_data.session_id
metadata["user_id"] = user_id
trace_data = LangfuseTrace(
id=message_id,

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

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