chore(code): put glm in cc behind flag

Author: adboio

packages/shared/src/flags.ts

@@ -8,3 +8,4 @@ export const DISCOVERY_RUN_FLAG = "posthog-code-discovery-run";
 // routes, channels and dashboards.
 export const PROJECT_BLUEBIRD_FLAG = "project-bluebird";
 export const TASKS_PREWARM_SANDBOX_FLAG = "tasks-prewarm-sandbox";
+export const GLM_MODEL_FLAG = "posthog-code-glm-model";

packages/ui/src/features/sessions/components/ModelSelector.tsx

@@ -13,6 +13,9 @@ import {
   DropdownMenuTrigger,
   MenuLabel,
 } from "@posthog/quill";
+import { GLM_MODEL_FLAG } from "@posthog/shared";
+import { useFeatureFlag } from "@posthog/ui/features/feature-flags/useFeatureFlag";
+import { stripGlmModelOption } from "@posthog/ui/features/sessions/modelOptionFilters";
 import {
   flattenSelectOptions,
   useModelConfigOptionForTask,
@@ -34,7 +37,12 @@ export function ModelSelector({
 }: ModelSelectorProps) {
   const sessionService = useService<SessionService>(SESSION_SERVICE);
   const session = useSessionForTask(taskId);
-  const modelOption = useModelConfigOptionForTask(taskId);
+  const rawModelOption = useModelConfigOptionForTask(taskId);
+  const glmEnabled = useFeatureFlag(GLM_MODEL_FLAG);
+  const modelOption =
+    glmEnabled || !rawModelOption
+      ? rawModelOption
+      : stripGlmModelOption(rawModelOption);
 
   const selectOption = modelOption?.type === "select" ? modelOption : undefined;
   const options = selectOption

packages/ui/src/features/sessions/modelOptionFilters.ts

@@ -0,0 +1,27 @@
+import type { SessionConfigOption } from "@agentclientprotocol/sdk";
+import { isSelectGroup } from "@posthog/shared";
+
+export function stripGlmModelOption(
+  option: SessionConfigOption,
+): SessionConfigOption {
+  if (option.type !== "select") return option;
+
+  if (isSelectGroup(option.options)) {
+    return {
+      ...option,
+      options: option.options.map((group) => ({
+        ...group,
+        options: group.options.filter(
+          (o) => !o.value.toLowerCase().includes("glm"),
+        ),
+      })),
+    };
+  }
+
+  return {
+    ...option,
+    options: option.options.filter(
+      (o) => !o.value.toLowerCase().includes("glm"),
+    ),
+  };
+}

packages/ui/src/features/task-detail/hooks/usePreviewConfig.ts

@@ -6,10 +6,12 @@ import {
   deriveInitialConfig,
 } from "@posthog/core/task-detail/previewConfig";
 import { useHostTRPCClient } from "@posthog/host-router/react";
-import { getCloudUrlFromRegion } from "@posthog/shared";
+import { GLM_MODEL_FLAG, getCloudUrlFromRegion } from "@posthog/shared";
+import { stripGlmModelOption } from "@posthog/ui/features/sessions/modelOptionFilters";
 import { useCallback, useEffect, useMemo, useRef, useState } from "react";
 import { logger } from "../../../shell/logger";
 import { useAuthStateValue } from "../../auth/store";
+import { useFeatureFlag } from "../../feature-flags/useFeatureFlag";
 import { useSettingsStore } from "../../settings/settingsStore";
 
 const log = logger.scope("preview-config");
@@ -42,6 +44,7 @@ export function usePreviewConfig(
   adapter: "claude" | "codex",
 ): PreviewConfigResult {
   const hostClient = useHostTRPCClient();
+  const glmEnabled = useFeatureFlag(GLM_MODEL_FLAG);
   const cloudRegion = useAuthStateValue((state) => state.cloudRegion);
   const apiHost = useMemo(
     () => (cloudRegion ? getCloudUrlFromRegion(cloudRegion) : null),
@@ -74,9 +77,13 @@ export function usePreviewConfig(
 
     hostClient.agent.getPreviewConfigOptions
       .query({ apiHost, adapter }, { signal: abort.signal })
-      .then((options) => {
+      .then((serverOptions) => {
         if (abort.signal.aborted) return;
 
+        const options = glmEnabled
+          ? serverOptions
+          : serverOptions.map(stripGlmModelOption);
+
         const {
           defaultInitialTaskMode,
           lastUsedInitialTaskMode,
@@ -134,7 +141,7 @@ export function usePreviewConfig(
     return () => {
       abort.abort();
     };
-  }, [adapter, apiHost, hostClient, hasHydrated]);
+  }, [adapter, apiHost, hostClient, hasHydrated, glmEnabled]);
 
   const setConfigOption = useCallback(
     (configId: string, value: string) => {