pi-mono

src/apps/officer-web/Screens/Dashboard/Settings/ProfileSettings/TaskDefaults.tsx

@@ -12,12 +12,13 @@ import {
   SelectValue,
 } from '@/components/ui/select';
 import { useSettings } from '@/state/useSettings';
-import { useVisibleClaudeModels, useVisibleOpenCodeModels } from '@/state/useModels';
+import { useVisibleClaudeModels, useVisibleOpenCodeModels, useVisiblePiMonoModels } from '@/state/useModels';
 
 export const TaskDefaults = () => {
   const { settings, saveSettings } = useSettings();
   const claudeModels = useVisibleClaudeModels();
   const openCodeModels = useVisibleOpenCodeModels();
+  const piMonoModels = useVisiblePiMonoModels();
   const [isSaving, setIsSaving] = useState(false);
 
   const [model, setModel] = useState<string | null>(settings.tasks.defaultModel);
@@ -26,24 +27,28 @@ export const TaskDefaults = () => {
     setModel(settings.tasks.defaultModel);
   }, [settings]);
 
-  const openCodeGroups = useMemo(() => {
+  const buildGroups = (models: { id: string; name: string; provider?: string }[], fallback: string) => {
     const groups: Record<string, { id: string; name: string }[]> = {};
-    for (const m of openCodeModels) {
-      const provider = m.provider ?? 'OpenCode';
+    for (const m of models) {
+      const provider = m.provider ?? fallback;
       if (!groups[provider]) groups[provider] = [];
       groups[provider].push({ id: m.id, name: m.name });
     }
     return Object.entries(groups)
       .sort(([a], [b]) => a.localeCompare(b))
       .map(([provider, models]) => ({ provider, models: models.sort((a, b) => a.name.localeCompare(b.name)) }));
-  }, [openCodeModels]);
+  };
+
+  const openCodeGroups = useMemo(() => buildGroups(openCodeModels, 'OpenCode'), [openCodeModels]);
+  const piMonoGroups = useMemo(() => buildGroups(piMonoModels, 'Pi'), [piMonoModels]);
 
   const handleSave = async () => {
     if (isSaving) return;
     setIsSaving(true);
     try {
+      const isPiMono = piMonoModels.some((m) => m.id === model);
       const isOpenCode = openCodeModels.some((m) => m.id === model);
-      const defaultProvider = isOpenCode ? ('opencode' as const) : ('claude' as const);
+      const defaultProvider = isPiMono ? ('pi-mono' as const) : isOpenCode ? ('opencode' as const) : ('claude' as const);
       await saveSettings({ ...settings, tasks: { defaultProvider, defaultModel: model } });
       toast.success('Task defaults saved');
     } catch {
@@ -82,6 +87,16 @@ export const TaskDefaults = () => {
                 ))}
               </SelectGroup>
             ))}
+            {piMonoGroups.map(({ provider, models }) => (
+              <SelectGroup key={`pi-${provider}`}>
+                <SelectLabel>{provider} (Pi)</SelectLabel>
+                {models.map((m) => (
+                  <SelectItem key={`pi:${provider}:${m.id}`} value={m.id}>
+                    {m.name}
+                  </SelectItem>
+                ))}
+              </SelectGroup>
+            ))}
           </SelectContent>
         </Select>
       </Label>