mariano/nova
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix Langfuse SDK: use start_as_current_observation API

Browse Source
This commit is contained in:
buenosairesam
2026-04-15 23:34:47 -03:00
parent f03e47d9ba
commit 1b1b1383a1
3 changed files with 56 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
agents/fce.py
View File

@@ -18,7 +18,7 @@ async def run_fce(
flight_id: str,
mcp: MCPMultiClient,
on_event: Any = None,
trace: Any = None,
langfuse: Any = None,
) -> dict:
run_start = time.time()
errors = []
@@ -67,27 +67,35 @@ async def run_fce(
async def _call(server, tool, args, is_live=False, timeout=15.0):
t = time.time()
span = trace.span(name=tool, input=args, metadata={"server": server, "is_live": is_live}) if trace else None
ctx = langfuse.start_as_current_observation(
name=tool, as_type="tool",
input=args, metadata={"server": server, "is_live": is_live},
) if langfuse else None
if ctx:
ctx.__enter__()
try:
result = await asyncio.wait_for(
mcp.call_tool(server, tool, args), timeout=timeout,
)
lat = int((time.time() - t) * 1000)
if span:
span.end(output=result, metadata={"latency_ms": lat})
if ctx:
langfuse.update_current_span(output=result, metadata={"latency_ms": lat})
ctx.__exit__(None, None, None)
await emit("tool_call_end", tool=tool, latency_ms=lat, is_live=is_live)
return result
except asyncio.TimeoutError:
lat = int((time.time() - t) * 1000)
if span:
span.end(output={"error": "timeout"}, level="ERROR")
if ctx:
langfuse.update_current_span(output={"error": "timeout"}, level="ERROR")
ctx.__exit__(None, None, None)
await emit("tool_call_error", tool=tool, error="timeout", latency_ms=lat)
errors.append(f"{tool}: timeout after {timeout}s")
return None
except Exception as e:
lat = int((time.time() - t) * 1000)
if span:
span.end(output={"error": str(e)}, level="ERROR")
if ctx:
langfuse.update_current_span(output={"error": str(e)}, level="ERROR")
ctx.__exit__(None, None, None)
await emit("tool_call_error", tool=tool, error=str(e), latency_ms=lat)
errors.append(f"{tool}: {e}")
return None