Skip to content

Commit 86f11ea

Browse files
authored
Merge pull request #62 from OrenZhang/fix_usage
fix(usage): usage is not used
2 parents 578e18a + 68aca75 commit 86f11ea

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

app/api/v1/outlet/route.ts

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -94,12 +94,12 @@ export async function POST(req: Request) {
9494
let inputTokens: number;
9595
let outputTokens: number;
9696
if (
97-
lastMessage.info &&
98-
lastMessage.info.prompt_tokens &&
99-
lastMessage.info.completion_tokens
97+
lastMessage.usage &&
98+
lastMessage.usage.prompt_tokens &&
99+
lastMessage.usage.completion_tokens
100100
) {
101-
inputTokens = lastMessage.info.prompt_tokens;
102-
outputTokens = lastMessage.info.completion_tokens;
101+
inputTokens = lastMessage.usage.prompt_tokens;
102+
outputTokens = lastMessage.usage.completion_tokens;
103103
} else {
104104
outputTokens = encode(lastMessage.content).length;
105105
const totalTokens = data.body.messages.reduce(

0 commit comments

Comments
 (0)