diff --git a/web/default/src/components/ai-elements/context.tsx b/web/default/src/components/ai-elements/context.tsx
index ec7afafa..5541b476 100644
--- a/web/default/src/components/ai-elements/context.tsx
+++ b/web/default/src/components/ai-elements/context.tsx
@@ -347,7 +347,7 @@ export const ContextReasoningUsage = ({
 }: ContextReasoningUsageProps) => {
   const { t } = useTranslation()
   const { usage, modelId } = useContextValue()
-  const reasoningTokens = usage?.reasoningTokens ?? 0
+  const reasoningTokens = usage?.outputTokenDetails.reasoningTokens ?? 0
 
   if (children) {
     return children
@@ -388,7 +388,7 @@ export const ContextCacheUsage = ({
 }: ContextCacheUsageProps) => {
   const { t } = useTranslation()
   const { usage, modelId } = useContextValue()
-  const cacheTokens = usage?.cachedInputTokens ?? 0
+  const cacheTokens = usage?.inputTokenDetails.cacheReadTokens ?? 0
 
   if (children) {
     return children