feat: add usage metrics

different-ai · Apr 19, 2024 · 23affaa · 23affaa
1 parent 2bda5fd
commit 23affaa
Show file tree

Hide file tree

Showing 4 changed files with 81 additions and 2 deletions.
diff --git a/app/lib/posthog.js b/app/lib/posthog.js
@@ -0,0 +1,15 @@
+// app/posthog.js
+import { PostHog } from 'posthog-node'
+
+export default function PostHogClient() {
+    if (!process.env.NEXT_PUBLIC_POSTHOG_KEY) {
+        return null
+    }
+
+    const posthogClient = new PostHog(process.env.NEXT_PUBLIC_POSTHOG_KEY, {
+        host: process.env.NEXT_PUBLIC_POSTHOG_HOST || 'https://app.posthog.com',
+        flushAt: 1,
+        flushInterval: 0
+    })
+    return posthogClient
+}
diff --git a/app/package-lock.json b/app/package-lock.json
diff --git a/app/package.json b/app/package.json
@@ -26,6 +26,7 @@
     "next": "13.5.4",
     "openai": "^4.33.1",
     "postcss": "8.4.31",
+    "posthog-node": "^4.0.0",
     "react": "18.2.0",
     "react-dom": "18.2.0",
     "tailwind-merge": "^1.14.0",

diff --git a/app/pages/api/name.ts b/app/pages/api/name.ts
@@ -1,5 +1,6 @@
 import { verifyKey } from "@unkey/api";
 import type { NextApiRequest, NextApiResponse } from "next";
+import PosthogClient from "../../lib/posthog";
 
 type ResponseData = {
   message: string;
@@ -11,7 +12,7 @@ export default async function handler(
 ) {
   // if ENABLE_USER_MANAGEMENT=true in .env file, then we need to check for the Authorization header
   if (process.env.ENABLE_USER_MANAGEMENT == "true") {
-    console.log("ENABLE_USER_MANAGEMENT", process.env.ENABLE_USER_MANAGEMENT)
+    console.log("ENABLE_USER_MANAGEMENT", process.env.ENABLE_USER_MANAGEMENT);
 
     const header = req.headers.authorization;
     console.log("header", header);
@@ -21,6 +22,16 @@ export default async function handler(
     const token = header.replace("Bearer ", "");
     const { result, error } = await verifyKey(token);
 
+    const client = PosthogClient();
+
+    if (client && result?.ownerId) {
+      client.capture({
+        distinctId: result?.ownerId,
+        event: "call-api",
+        properties: { endpoint: "name" },
+      });
+    }
+
     if (error) {
       console.error(error.message);
       return res.status(500).json({ message: "Internal Server Error" });
@@ -32,7 +43,7 @@ export default async function handler(
   }
   try {
     const apiKey = process.env.OPENAI_API_KEY || "";
-    console.log("apiKey", apiKey)
+    console.log("apiKey", apiKey);
     const model = "gpt-3.5-turbo";
     const data = {
       ...req.body,