From 68aca75caf1fcac8bb39b12666cd3f7714e0ff97 Mon Sep 17 00:00:00 2001
From: orenzhang <orenzhang@ovinc.cn>
Date: Thu, 27 Feb 2025 21:24:59 +0800
Subject: [PATCH] fix(usage): usage is not used

---
 app/api/v1/outlet/route.ts | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/app/api/v1/outlet/route.ts b/app/api/v1/outlet/route.ts
index f3c3abb..626f253 100644
--- a/app/api/v1/outlet/route.ts
+++ b/app/api/v1/outlet/route.ts
@@ -94,12 +94,12 @@ export async function POST(req: Request) {
     let inputTokens: number;
     let outputTokens: number;
     if (
-      lastMessage.info &&
-      lastMessage.info.prompt_tokens &&
-      lastMessage.info.completion_tokens
+      lastMessage.usage &&
+      lastMessage.usage.prompt_tokens &&
+      lastMessage.usage.completion_tokens
     ) {
-      inputTokens = lastMessage.info.prompt_tokens;
-      outputTokens = lastMessage.info.completion_tokens;
+      inputTokens = lastMessage.usage.prompt_tokens;
+      outputTokens = lastMessage.usage.completion_tokens;
     } else {
       outputTokens = encode(lastMessage.content).length;
       const totalTokens = data.body.messages.reduce(