redis for logging

app/utils/cloud/redisClient.ts

@@ -1,34 +0,0 @@
-// redisClient.ts
-import Redis from 'ioredis';
-
-const redis = new Redis(process.env.UPSTASH_REDIS_URL);
-
-export const incrementSignInCount = async (email: string | undefined, dateKey: string) => {
-  if (!email) {
-    console.error('Email is undefined, cannot increment sign-in count.');
-    return;
-  }
-  await redis.hincrby(`sign_ins:${email}`, dateKey, 1);
-};
-
-export const incrementSessionRefreshCount = async (email: string | undefined, dateKey: string) => {
-  if (!email) {
-    console.error('Email is undefined, cannot increment session refresh count.');
-    return;
-  }
-  await redis.hincrby(`session_refreshes:${email}`, dateKey, 1);
-};
-
-export const incrementTokenCounts = async (
-  email: string | undefined,
-  dateKey: string,
-  completionTokens: number,
-  promptTokens: number
-) => {
-  if (!email) {
-    console.error('Email is undefined, cannot increment token counts.');
-    return;
-  }
-  await redis.hincrby(`tokens:${email}`, `${dateKey}:completion_tokens`, completionTokens);
-  await redis.hincrby(`tokens:${email}`, `${dateKey}:prompt_tokens`, promptTokens);
-};

app/utils/cloud/redisRestClient.ts

@@ -0,0 +1,76 @@
+// redisRestClient.ts
+import fetch from 'node-fetch'; // or any other fetch-compatible API
+
+const redisRestUrl = process.env.UPSTASH_REDIS_URL;
+const redisRestToken = process.env.UPSTASH_REDIS_REST_TOKEN;
+
+if (!redisRestUrl || !redisRestToken) {
+  console.error('The Upstash Redis URL and token must be provided.');
+  // Handle the lack of Redis client here, e.g., by disabling certain features
+}
+
+const headers = {
+  Authorization: `Bearer ${redisRestToken}`,
+  'Content-Type': 'application/json',
+};
+
+export const incrementSessionRefreshCount = async (email: string | undefined, dateKey: string) => {
+  if (!email) {
+    console.error('Email is undefined, cannot increment session refresh count.');
+    return;
+  }
+
+  const response = await fetch(`${redisRestUrl}/hincrby`, {
+    method: 'POST',
+    headers: headers,
+    body: JSON.stringify({
+      key: `session_refreshes:${email}`,
+      field: dateKey,
+      increment: 1
+    }),
+  });
+
+  if (!response.ok) {
+    console.error('Failed to increment session refresh count in Redis via REST API');
+  }
+};
+
+export const incrementTokenCounts = async (
+  email: string | undefined,
+  dateKey: string,
+  completionTokens: number,
+  promptTokens: number
+) => {
+  if (!email) {
+    console.error('Email is undefined, cannot increment token counts.');
+    return;
+  }
+
+  const incrementCompletionTokensResponse = await fetch(`${redisRestUrl}/hincrby`, {
+    method: 'POST',
+    headers: headers,
+    body: JSON.stringify({
+      key: `tokens:${email}`,
+      field: `${dateKey}:completion_tokens`,
+      increment: completionTokens
+    }),
+  });
+
+  const incrementPromptTokensResponse = await fetch(`${redisRestUrl}/hincrby`, {
+    method: 'POST',
+    headers: headers,
+    body: JSON.stringify({
+      key: `tokens:${email}`,
+      field: `${dateKey}:prompt_tokens`,
+      increment: promptTokens
+    }),
+  });
+
+  if (!incrementCompletionTokensResponse.ok) {
+    console.error('Failed to increment completion token count in Redis via REST API');
+  }
+
+  if (!incrementPromptTokensResponse.ok) {
+    console.error('Failed to increment prompt token count in Redis via REST API');
+  }
+};