From 06e072c7799356834cce9c385c6f6cbbd533bac1 Mon Sep 17 00:00:00 2001
From: yorkeking <34937020+yorkeking@users.noreply.github.com>
Date: Tue, 4 Apr 2023 02:18:41 +0800
Subject: [PATCH] Update requests.ts

---
 app/requests.ts | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/app/requests.ts b/app/requests.ts
index a8ba4e9fd..1a32407d8 100644
--- a/app/requests.ts
+++ b/app/requests.ts
@@ -52,6 +52,7 @@ export function requestOpenaiClient(path: string) {
       method,
       headers: {
         "Content-Type": "application/json",
+        "Cache-Control": "no-cache",
         path,
         ...getHeaders(),
       },
@@ -73,21 +74,28 @@ export async function requestChat(messages: Message[]) {
 }
 
 export async function requestUsage() {
+ const formatDate = (d: Date) =>
+    `${d.getFullYear()}-${(d.getMonth() + 1).toString().padStart(2, "0")}-${d
+      .getDate()
+      .toString()
+      .padStart(2, "0")}`;
+  const ONE_DAY = 24 * 60 * 60 * 1000;
+  const now = new Date(Date.now() + ONE_DAY);
+  const startOfMonth = new Date(now.getFullYear(), now.getMonth(), 1);
+  const startDate = formatDate(startOfMonth);
+  const endDate = formatDate(now);
   const res = await requestOpenaiClient(
-    "dashboard/billing/credit_grants?_vercel_no_cache=1",
+    `dashboard/billing/usage?start_date=${startDate}&end_date=${endDate}`,
   )(null, "GET");
 
   try {
     const response = (await res.json()) as {
-      total_available: number;
-      total_granted: number;
-      total_used: number;
+      total_usage: number;
     };
-    return response;
+    return Math.round(response.total_usage) / 100;
   } catch (error) {
     console.error("[Request usage] ", error, res.body);
   }
-}
 
 export async function requestChatStream(
   messages: Message[],