feat: add analytics integration

Author: gurvinder-dhillon

package-lock.json

@@ -55,6 +55,7 @@
     "@playwright/test": "1.52.0-alpha-1743011787000",
     "dotenv": "^16.4.7",
     "playwright": "1.52.0-alpha-1743011787000",
+    "posthog-node": "^4.11.3",
     "yaml": "^2.7.1",
     "zod": "^3.24.2"
   },

package.json

@@ -0,0 +1,36 @@
+import { PostHog } from 'posthog-node';
+import { analyticsProvider, analyticsModel } from './llm';
+
+const ANALYTICS_OPT_IN = process.env.ANALYTICS_OPT_IN !== 'false'; // defaults to true
+
+let client: PostHog | null = null;
+
+if (ANALYTICS_OPT_IN)
+{
+    client = new PostHog(
+        'phc_4pwxr91oy6WYPfaD13ClVreSbT7F7ClJcAEyBpTQCOl',
+        {
+            host: 'https://us.i.posthog.com',
+            flushAt: 1, // Important for serverless environments
+            flushInterval: 0 // Important for serverless environments
+        }
+    );
+}
+
+export async function captureAutoCall() {
+    if (!client) return;
+
+    await client.capture({
+        distinctId: '120836',
+        event: 'auto_called',
+        properties: {
+            llm_provider: analyticsProvider,
+            llm_model: analyticsModel
+        }
+    });
+}
+
+export async function shutdown() {
+    if (!client) return;
+    await client.shutdown();
+}

src/analytics.ts

@@ -2,6 +2,7 @@ import { test as base } from '@playwright/test';
 import { z } from 'zod';
 import { AutoConfig } from './types';
 import { sessionManager, context } from './browser';
+import { captureAutoCall, shutdown } from './analytics';
 import { createReactAgent } from '@langchain/langgraph/prebuilt';
 import { HumanMessage } from '@langchain/core/messages';
 import { createLLMModel } from './llm';
@@ -317,6 +318,7 @@ export async function auto(
   config?: AutoConfig
 ): Promise<any> {
   console.log(`[Auto] Processing instruction: "${instruction}"`);
+  await captureAutoCall();
 
   if (config?.page)
   {
@@ -365,10 +367,16 @@ export async function auto(
   } catch (error)
   {
     console.log(`[Auto] Error processing response:`, error);
+
     throw error;
   }
 }
 
+// Ensure analytics are flushed before the process exits
+process.on('beforeExit', async () => {
+  await shutdown();
+});
+
 // Export everything needed for the package
 export { sessionManager } from './browser';
 export * from './types';

src/auto.ts

@@ -8,17 +8,23 @@ import dotenv from 'dotenv';
 // Load environment variables
 dotenv.config();
 
+// Export analytics-specific variables
+export const analyticsProvider = process.env.LLM_PROVIDER || 'openai';
+export const analyticsModel = process.env.AUTOBROWSE_LLM_MODEL || 'gpt-4o-mini';
+
 export function createLLMModel() {
   const provider = process.env.LLM_PROVIDER || 'openai';
   const model = process.env.AUTOBROWSE_LLM_MODEL || 'gpt-4o-mini';
 
-  if (provider === 'google') {
+  if (provider === 'google')
+  {
     return new ChatGoogleGenerativeAI({
       model: model
     });
   }
 
-  if (provider === 'azure') {
+  if (provider === 'azure')
+  {
     return new AzureChatOpenAI({
       temperature: 0,
       maxRetries: 2,
@@ -30,21 +36,24 @@ export function createLLMModel() {
     });
   }
 
-  if (provider === 'anthropic') {
+  if (provider === 'anthropic')
+  {
     return new ChatAnthropic({
       model: model,
       temperature: 0,
       anthropicApiKey: process.env.ANTHROPIC_API_KEY
     });
   }
 
-  if (provider === 'vertex') {
+  if (provider === 'vertex')
+  {
     return new ChatVertexAI({
       model: model,
       temperature: 0
     });
   }
-  if (provider === 'ollama') {
+  if (provider === 'ollama')
+  {
     return new ChatOllama({
       model: model,
       temperature: 0,