mirror of
https://github.com/AutoMaker-Org/automaker.git
synced 2026-02-03 08:53:36 +00:00
feat(ui): add React Query hooks for data fetching
- Add useFeatures, useFeature, useAgentOutput for feature data - Add useGitHubIssues, useGitHubPRs, useGitHubValidations, useGitHubIssueComments - Add useClaudeUsage, useCodexUsage with polling intervals - Add useRunningAgents, useRunningAgentsCount - Add useWorktrees, useWorktreeInfo, useWorktreeStatus, useWorktreeDiffs - Add useGlobalSettings, useProjectSettings, useCredentials - Add useAvailableModels, useCodexModels, useOpencodeModels - Add useSessions, useSessionHistory, useSessionQueue - Add useIdeationPrompts, useIdeas - Add CLI status queries (claude, cursor, codex, opencode, github) - Add useCursorPermissionsQuery, useWorkspaceDirectories - Add usePipelineConfig, useSpecFile, useSpecRegenerationStatus Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
134
apps/ui/src/hooks/queries/use-models.ts
Normal file
134
apps/ui/src/hooks/queries/use-models.ts
Normal file
@@ -0,0 +1,134 @@
|
||||
/**
|
||||
* Models Query Hooks
|
||||
*
|
||||
* React Query hooks for fetching available AI models.
|
||||
*/
|
||||
|
||||
import { useQuery } from '@tanstack/react-query';
|
||||
import { getElectronAPI } from '@/lib/electron';
|
||||
import { queryKeys } from '@/lib/query-keys';
|
||||
import { STALE_TIMES } from '@/lib/query-client';
|
||||
|
||||
interface CodexModel {
|
||||
id: string;
|
||||
label: string;
|
||||
description: string;
|
||||
hasThinking: boolean;
|
||||
supportsVision: boolean;
|
||||
tier: 'premium' | 'standard' | 'basic';
|
||||
isDefault: boolean;
|
||||
}
|
||||
|
||||
interface OpencodeModel {
|
||||
id: string;
|
||||
name: string;
|
||||
modelString: string;
|
||||
provider: string;
|
||||
description: string;
|
||||
supportsTools: boolean;
|
||||
supportsVision: boolean;
|
||||
tier: string;
|
||||
default?: boolean;
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch available models
|
||||
*
|
||||
* @returns Query result with available models
|
||||
*/
|
||||
export function useAvailableModels() {
|
||||
return useQuery({
|
||||
queryKey: queryKeys.models.available(),
|
||||
queryFn: async () => {
|
||||
const api = getElectronAPI();
|
||||
const result = await api.model.getAvailable();
|
||||
if (!result.success) {
|
||||
throw new Error(result.error || 'Failed to fetch available models');
|
||||
}
|
||||
return result.models ?? [];
|
||||
},
|
||||
staleTime: STALE_TIMES.MODELS,
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch Codex models
|
||||
*
|
||||
* @param refresh - Force refresh from server
|
||||
* @returns Query result with Codex models
|
||||
*/
|
||||
export function useCodexModels(refresh = false) {
|
||||
return useQuery({
|
||||
queryKey: queryKeys.models.codex(),
|
||||
queryFn: async (): Promise<CodexModel[]> => {
|
||||
const api = getElectronAPI();
|
||||
const result = await api.codex.getModels(refresh);
|
||||
if (!result.success) {
|
||||
throw new Error(result.error || 'Failed to fetch Codex models');
|
||||
}
|
||||
return (result.models ?? []) as CodexModel[];
|
||||
},
|
||||
staleTime: STALE_TIMES.MODELS,
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch OpenCode models
|
||||
*
|
||||
* @param refresh - Force refresh from server
|
||||
* @returns Query result with OpenCode models
|
||||
*/
|
||||
export function useOpencodeModels(refresh = false) {
|
||||
return useQuery({
|
||||
queryKey: queryKeys.models.opencode(),
|
||||
queryFn: async (): Promise<OpencodeModel[]> => {
|
||||
const api = getElectronAPI();
|
||||
const result = await api.setup.getOpencodeModels(refresh);
|
||||
if (!result.success) {
|
||||
throw new Error(result.error || 'Failed to fetch OpenCode models');
|
||||
}
|
||||
return (result.models ?? []) as OpencodeModel[];
|
||||
},
|
||||
staleTime: STALE_TIMES.MODELS,
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch OpenCode providers
|
||||
*
|
||||
* @returns Query result with OpenCode providers
|
||||
*/
|
||||
export function useOpencodeProviders() {
|
||||
return useQuery({
|
||||
queryKey: queryKeys.models.opencodeProviders(),
|
||||
queryFn: async () => {
|
||||
const api = getElectronAPI();
|
||||
const result = await api.setup.getOpencodeProviders();
|
||||
if (!result.success) {
|
||||
throw new Error(result.error || 'Failed to fetch OpenCode providers');
|
||||
}
|
||||
return result.providers ?? [];
|
||||
},
|
||||
staleTime: STALE_TIMES.MODELS,
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch model providers status
|
||||
*
|
||||
* @returns Query result with provider status
|
||||
*/
|
||||
export function useModelProviders() {
|
||||
return useQuery({
|
||||
queryKey: queryKeys.models.providers(),
|
||||
queryFn: async () => {
|
||||
const api = getElectronAPI();
|
||||
const result = await api.model.checkProviders();
|
||||
if (!result.success) {
|
||||
throw new Error(result.error || 'Failed to fetch providers');
|
||||
}
|
||||
return result.providers ?? {};
|
||||
},
|
||||
staleTime: STALE_TIMES.MODELS,
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user