fix: langsmith message_trace end_user_data session_id error (#5759)

2024-06-30 01:12:16 +08:00
parent ffb07eb24b
commit f637ae4794
2 changed files with 9 additions and 5 deletions
--- a/api/core/ops/langfuse_trace/langfuse_trace.py
+++ b/api/core/ops/langfuse_trace/langfuse_trace.py
@@ -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()
-            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(
            id=message_id,