Skip to content

Commit f1836a4

Browse files
committed
catch failures when reading token counts
1 parent 37f34fa commit f1836a4

1 file changed

Lines changed: 10 additions & 7 deletions

File tree

packages/ai-providers/server-ai-openai/src/ldai_openai/openai_agent_graph_runner.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -97,14 +97,17 @@ async def run(self, input: Any) -> AgentGraphResult:
9797
tracker.track_path(path)
9898
tracker.track_latency(duration)
9999
tracker.track_invocation_success()
100-
usage = result.context_wrapper.usage
101-
tracker.track_total_tokens(
102-
TokenUsage(
103-
total=usage.total_tokens,
104-
input=usage.input_tokens,
105-
output=usage.output_tokens,
100+
try:
101+
usage = result.context_wrapper.usage
102+
tracker.track_total_tokens(
103+
TokenUsage(
104+
total=usage.total_tokens,
105+
input=usage.input_tokens,
106+
output=usage.output_tokens,
107+
)
106108
)
107-
)
109+
except Exception:
110+
pass
108111

109112
return AgentGraphResult(
110113
output=str(result.final_output),

0 commit comments

Comments
 (0)