bugfix:修复一些bug

This commit is contained in:
2026-10-02 02:36:11 -07:00
parent 68588d1f92
commit 5b1e6c5654
14 changed files with 37 additions and 81 deletions
-20
View File
@@ -55,26 +55,6 @@ def flush_tracing() -> None:
logger.opt(exception=exc).warning("Langfuse flush 失败")
async def ainvoke_screen(
graph: Any,
payload: Any,
*,
thread_id: str,
session_id: str | None = None,
user_id: int | None = None,
) -> Any:
"""每轮调用记录一个根 span,返回、等待用户或异常时结束。
session_id 由调用方在首次与恢复时一致传入(通常为 conversation_id)。
未传时始终使用 thread_id,不从仅首次请求才有的 payload 推断。
user_id 为用户中心的用户 ID,接口调用时传入,本地脚本不传。
"""
with _screen_trace(payload, thread_id, session_id, user_id) as (config, root):
result = await graph.ainvoke(payload, config=config)
_finish_trace(root, result)
return result
async def astream_screen(graph, payload, *, thread_id, session_id=None, user_id=None):
"""同一轮流保持同一个追踪上下文,关闭流时结束追踪"""
from contextlib import aclosing