bugfix:修复一些bug
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user