pi-mono

src/apps/officer-web/state/useModels.ts

@@ -66,3 +66,24 @@ export const useVisibleOpenCodeModels = () => {
     [models, providers, enabled],
   );
 };
+
+export const usePiMonoModels = () => {
+  const client = useClient();
+  const { isAuthenticated } = useAuth();
+
+  const { data: models = [] } = useQuery<ModelOption[]>({
+    queryKey: ['PI_MONO_MODELS'],
+    enabled: isAuthenticated,
+    queryFn: () => client.get<ModelOption[]>('/pi-mono/models'),
+    staleTime: 5 * 60 * 1000,
+  });
+
+  return models;
+};
+
+export const useVisiblePiMonoModels = () => {
+  const models = usePiMonoModels();
+  const { settings } = useSettings();
+  const enabled = settings.ai?.enabledModels ?? [];
+  return useMemo(() => models.filter((m) => enabled.includes(modelKey(m))), [models, enabled]);
+};