瀏覽代碼

Fix missing user_id in trace_manager (#24024)

tags/1.8.0
Yongtao Huang 2 月之前
父節點
當前提交
8b601a983c
No account linked to committer's email address
共有 2 個文件被更改,包括 6 次插入2 次删除
  1. 3
    1
      api/core/app/apps/chat/app_generator.py
  2. 3
    1
      api/core/app/apps/completion/app_generator.py

+ 3
- 1
api/core/app/apps/chat/app_generator.py 查看文件

@@ -140,7 +140,9 @@ class ChatAppGenerator(MessageBasedAppGenerator):
)

# get tracing instance
trace_manager = TraceQueueManager(app_id=app_model.id)
trace_manager = TraceQueueManager(
app_id=app_model.id, user_id=user.id if isinstance(user, Account) else user.session_id
)

# init application generate entity
application_generate_entity = ChatAppGenerateEntity(

+ 3
- 1
api/core/app/apps/completion/app_generator.py 查看文件

@@ -124,7 +124,9 @@ class CompletionAppGenerator(MessageBasedAppGenerator):
)

# get tracing instance
trace_manager = TraceQueueManager(app_model.id)
trace_manager = TraceQueueManager(
app_id=app_model.id, user_id=user.id if isinstance(user, Account) else user.session_id
)

# init application generate entity
application_generate_entity = CompletionAppGenerateEntity(

Loading…
取消
儲存