From 498199b37ddd1e29736b0e41b1ac79d73c049a27 Mon Sep 17 00:00:00 2001 From: wenyifan Date: Fri, 20 Mar 2026 16:38:48 +0800 Subject: [PATCH] fix code quality --- relay/channel/openai/relay-openai.go | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/relay/channel/openai/relay-openai.go b/relay/channel/openai/relay-openai.go index 02387fe0..222bb28f 100644 --- a/relay/channel/openai/relay-openai.go +++ b/relay/channel/openai/relay-openai.go @@ -703,13 +703,19 @@ func extractLlamaCachedTokensFromBody(body []byte) (int, bool) { } var payload struct { - Usage struct { - CachedTokens *int `json:"cache_n"` + Timings struct { + Usage struct { + CachedTokens *int `json:"cache_n"` + } `json:"usage"` } `json:"timings"` } if err := common.Unmarshal(body, &payload); err != nil { return 0, false } - return *payload.Usage.CachedTokens, true + + if payload.Timings.Usage.CachedTokens == nil { + return 0, false + } + return *payload.Timings.Usage.CachedTokens, true }