Sameel/flags new

.gitignore

@@ -83,3 +83,5 @@ target/
 # pnpm
 
 test/
+.vercel
+.env*.local

app/api/agent/stream/route.ts

@@ -200,4 +200,4 @@ export async function GET(request: Request) {
       "X-Accel-Buffering": "no",
     },
   });
-}
+}

app/api/session/route.ts

@@ -1,106 +1,185 @@
-import { NextResponse } from "next/server";
 import Browserbase from "@browserbasehq/sdk";
+import { getAll } from "@vercel/edge-config";
+import { NextResponse } from "next/server";
 
 type BrowserbaseRegion =
   | "us-west-2"
   | "us-east-1"
   | "eu-central-1"
   | "ap-southeast-1";
 
-// Exact timezone matches for east coast cities
-const exactTimezoneMap: Record<string, BrowserbaseRegion> = {
-  "America/New_York": "us-east-1",
-  "America/Detroit": "us-east-1",
-  "America/Toronto": "us-east-1",
-  "America/Montreal": "us-east-1",
-  "America/Boston": "us-east-1",
-  "America/Chicago": "us-east-1",
+// Timezone abbreviation to region mapping
+const timezoneAbbreviationMap: Record<string, BrowserbaseRegion> = {
+  // US East Coast
+  EST: "us-east-1",
+  EDT: "us-east-1",
+
+  // US West Coast
+  PST: "us-west-2",
+  PDT: "us-west-2",
+
+  // US Mountain/Central - route to appropriate region
+  MST: "us-west-2",
+  MDT: "us-west-2",
+  CST: "us-east-1",
+  CDT: "us-east-1",
+
+  // Europe
+  GMT: "eu-central-1",
+  BST: "eu-central-1",
+  CET: "eu-central-1",
+  CEST: "eu-central-1",
+  EET: "eu-central-1",
+  EEST: "eu-central-1",
+  WET: "eu-central-1",
+  WEST: "eu-central-1",
+
+  // Asia-Pacific
+  JST: "ap-southeast-1", // Japan Standard Time
+  KST: "ap-southeast-1", // Korea Standard Time
+  IST: "ap-southeast-1", // India Standard Time
+  AEST: "ap-southeast-1", // Australian Eastern Standard Time
+  AEDT: "ap-southeast-1", // Australian Eastern Daylight Time
+  AWST: "ap-southeast-1", // Australian Western Standard Time
+  NZST: "ap-southeast-1", // New Zealand Standard Time
+  NZDT: "ap-southeast-1", // New Zealand Daylight Time
 };
 
-// Prefix-based region mapping
-const prefixToRegion: Record<string, BrowserbaseRegion> = {
-  America: "us-west-2",
-  US: "us-west-2",
-  Canada: "us-west-2",
-  Europe: "eu-central-1",
-  Africa: "eu-central-1",
-  Asia: "ap-southeast-1",
-  Australia: "ap-southeast-1",
-  Pacific: "ap-southeast-1",
+// Default fallback distributions if edge config is not available
+const defaultDistributions: Record<
+  BrowserbaseRegion,
+  Record<BrowserbaseRegion, number>
+> = {
+  "us-west-2": {
+    "us-west-2": 100,
+    "us-east-1": 0,
+    "eu-central-1": 0,
+    "ap-southeast-1": 0,
+  },
+  "us-east-1": {
+    "us-east-1": 100,
+    "us-west-2": 0,
+    "eu-central-1": 0,
+    "ap-southeast-1": 0,
+  },
+  "eu-central-1": {
+    "eu-central-1": 100,
+    "us-west-2": 0,
+    "us-east-1": 0,
+    "ap-southeast-1": 0,
+  },
+  "ap-southeast-1": {
+    "ap-southeast-1": 100,
+    "us-west-2": 0,
+    "us-east-1": 0,
+    "eu-central-1": 0,
+  },
 };
 
-// Offset ranges to regions (inclusive bounds)
-const offsetRanges: {
-  min: number;
-  max: number;
-  region: BrowserbaseRegion;
-}[] = [
-  { min: -24, max: -4, region: "us-west-2" }, // UTC-24 to UTC-4
-  { min: -3, max: 4, region: "eu-central-1" }, // UTC-3 to UTC+4
-  { min: 5, max: 24, region: "ap-southeast-1" }, // UTC+5 to UTC+24
-];
-
-function getClosestRegion(timezone?: string): BrowserbaseRegion {
-  try {
-    if (!timezone) {
-      return "us-west-2"; // Default if no timezone provided
-    }
+function selectRegionWithProbability(
+  baseRegion: BrowserbaseRegion,
+  distributions: Record<BrowserbaseRegion, Record<BrowserbaseRegion, number>>
+): BrowserbaseRegion {
+  const distribution = distributions[baseRegion];
+  if (!distribution) {
+    return baseRegion;
+  }
+
+  const random = Math.random() * 100; // Generate random number between 0-100
 
-    // Check exact matches first
-    if (timezone in exactTimezoneMap) {
-      return exactTimezoneMap[timezone];
+  let cumulativeProbability = 0;
+  for (const [region, probability] of Object.entries(distribution)) {
+    cumulativeProbability += probability;
+    if (random < cumulativeProbability) {
+      return region as BrowserbaseRegion;
     }
+  }
+
+  // Fallback to base region if something goes wrong
+  return baseRegion;
+}
 
-    // Check prefix matches
-    const prefix = timezone.split("/")[0];
-    if (prefix in prefixToRegion) {
-      return prefixToRegion[prefix];
+function getRegionFromTimezoneAbbr(timezoneAbbr?: string): BrowserbaseRegion {
+  try {
+    if (!timezoneAbbr) {
+      return "us-west-2"; // Default if no timezone provided
     }
 
-    // Use offset-based fallback
-    const date = new Date();
-    // Create a date formatter for the given timezone
-    const formatter = new Intl.DateTimeFormat("en-US", { timeZone: timezone });
-    // Get the timezone offset in minutes
-    const timeString = formatter.format(date);
-    const testDate = new Date(timeString);
-    const hourOffset = (testDate.getTime() - date.getTime()) / (1000 * 60 * 60);
-
-    const matchingRange = offsetRanges.find(
-      (range) => hourOffset >= range.min && hourOffset <= range.max
-    );
+    // Direct lookup from timezone abbreviation
+    const region = timezoneAbbreviationMap[timezoneAbbr.toUpperCase()];
+    if (region) {
+      return region;
+    }
 
-    return matchingRange?.region ?? "us-west-2";
+    // Fallback to us-west-2 for unknown abbreviations
+    return "us-west-2";
   } catch {
     return "us-west-2";
   }
 }
 
+interface EdgeConfig {
+  advancedStealth: boolean | undefined;
+  proxies: boolean | undefined;
+  regionDistribution:
+    | Record<BrowserbaseRegion, Record<BrowserbaseRegion, number>>
+    | undefined;
+}
+
 async function createSession(timezone?: string) {
   const bb = new Browserbase({
     apiKey: process.env.BROWSERBASE_API_KEY!,
   });
 
-  const browserSettings: Browserbase.Sessions.SessionCreateParams.BrowserSettings = {
-    viewport: {
-      width: 2560,
-      height: 1440,
-    },
-    // @ts-expect-error - os is not a valid property
-    os: "windows",
-    blockAds: true,
-    advancedStealth: true
-  };
-
-  console.log("timezone ", timezone);
-  console.log("getClosestRegion(timezone)", getClosestRegion(timezone));
+  const config = await getAll<EdgeConfig>();
+
+  const {
+    advancedStealth: advancedStealthConfig,
+    proxies: proxiesConfig,
+    regionDistribution: distributionsConfig,
+  } = config;
+
+  const advancedStealth: boolean = advancedStealthConfig ?? false;
+  const proxies: boolean = proxiesConfig ?? true;
+
+  // Build browserSettings conditionally
+  const browserSettings: Browserbase.Sessions.SessionCreateParams.BrowserSettings =
+    {
+      viewport: {
+        width: 2560,
+        height: 1440,
+      },
+      blockAds: true,
+      advancedStealth,
+      // Only set os if advancedStealth is true
+      ...(advancedStealth
+        ? {
+            os: "mac",
+          }
+        : {
+            os: "linux",
+          }),
+    };
+
+  // Use timezone abbreviation to determine base region
+  const closestRegion = getRegionFromTimezoneAbbr(timezone);
+
+  // Get distributions from config or use default
+  const distributions = distributionsConfig ?? defaultDistributions;
+
+  // Apply probability routing for potential load balancing
+  const finalRegion = selectRegionWithProbability(closestRegion, distributions);
+
+  console.log("timezone abbreviation:", timezone);
+  console.log("mapped to region:", closestRegion);
+  console.log("final region after probability routing:", finalRegion);
 
   const session = await bb.sessions.create({
     projectId: process.env.BROWSERBASE_PROJECT_ID!,
-    proxies: true,
+    proxies,
     browserSettings,
     keepAlive: true,
-    region: getClosestRegion(timezone),
+    region: finalRegion,
   });
   return {
     session,

app/components/BrowserSessionContainer.tsx

@@ -318,4 +318,4 @@ const BrowserSessionContainer: React.FC<BrowserSessionContainerProps> = ({
   );
 };
 
-export default BrowserSessionContainer;
+export default BrowserSessionContainer;

app/components/ChatFeed.tsx

@@ -373,4 +373,4 @@ export default function ChatFeed({
       </main>
     </motion.div>
   );
-}
+}

app/components/NavBar.tsx

@@ -31,7 +31,7 @@ export default function NavBar({
     >
       <div className="flex items-center gap-2">
         <a
-          href="https://www.browserbase.com/cua/google"
+          href="https://browserbase.com/computer-use"
           target="_blank"
           rel="noopener noreferrer"
           className="flex items-center gap-3 hover:opacity-90 transition-opacity duration-200"
@@ -50,7 +50,7 @@ export default function NavBar({
       </div>
       <div className="flex items-center gap-2">
         <a
-          href="https://www.browserbase.com/cua/google"
+          href="https://browserbase.com/computer-use"
           target="_blank"
           rel="noopener noreferrer"
         >
@@ -69,7 +69,7 @@ export default function NavBar({
         </a>
         {showGitHubButton && (
           <a
-            href="https://github.com/browserbase/google-cua-browser"
+            href="https://github.com/browserbase/cua-browser"
             target="_blank"
             rel="noopener noreferrer"
           >