Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
48 changes: 48 additions & 0 deletions electron/src/handlers/oagent-sessions.handler.ts
Original file line number Diff line number Diff line change
Expand Up @@ -142,6 +142,26 @@ function inferProvider(llmProvider: StartOptions["llmProvider"], model?: string)
return "agent";
}

async function checkOllamaReachability(endpointRaw?: string): Promise<{ ok: true } | { ok: false; error: string }> {
const endpoint = (endpointRaw || "http://localhost:11434").trim().replace(/\/+$/, "");
const controller = new AbortController();
const timeout = setTimeout(() => controller.abort(), 2000);
try {
const res = await fetch(`${endpoint}/api/tags`, { signal: controller.signal });
if (!res.ok) {
return { ok: false, error: `Ollama endpoint returned ${res.status}. Start Ollama or update Settings → Ollama Endpoint.` };
}
return { ok: true };
} catch {
return {
ok: false,
error: `Ollama is not reachable at ${endpoint}. Start Ollama, fix the endpoint, or switch to OpenRouter in Settings.`,
};
} finally {
clearTimeout(timeout);
}
}

function applyProviderOptions(
queryOptions: Record<string, unknown>,
{
Expand Down Expand Up @@ -450,6 +470,34 @@ export function register(getMainWindow: () => BrowserWindow | null): void {
ipcMain.on(`agent:${suffix}`, handler);
};

handleBoth("check-provider-setup", async (_event, options: {
llmProvider?: "openrouter" | "ollama";
model?: string;
openRouterKey?: string;
ollamaEndpoint?: string;
}) => {
const provider = inferProvider(options?.llmProvider, options?.model);
if (provider === "openrouter") {
const key = (options?.openRouterKey || "").trim();
if (!key) {
return {
ok: false,
provider,
error: "OpenRouter API key missing. Add it in Settings → Models → OpenRouter API Key, or switch provider to Ollama.",
};
}
return { ok: true, provider };
}
if (provider === "ollama") {
const result = await checkOllamaReachability(options?.ollamaEndpoint);
if (!result.ok) {
return { ok: false, provider, error: result.error };
}
return { ok: true, provider };
}
return { ok: true, provider };
});

handleBoth("start", async (_event, options: StartOptions = {}) => {
const sessionId = options.resume || crypto.randomUUID();
const query = await getSDK();
Expand Down
1 change: 1 addition & 0 deletions electron/src/preload.ts
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@ const clientCoreApi = {
ipcRenderer.invoke("oagent:mcp-reconnect", { sessionId, serverName }),
restartSession: (sessionId: string, mcpServers?: unknown[]) =>
ipcRenderer.invoke("oagent:restart-session", { sessionId, mcpServers }),
checkProviderSetup: (options: unknown) => ipcRenderer.invoke("oagent:check-provider-setup", options),
readFile: (filePath: string) => ipcRenderer.invoke("file:read", filePath),
openInEditor: (filePath: string, line?: number) => ipcRenderer.invoke("file:open-in-editor", { filePath, line }),
generateTitle: (
Expand Down
44 changes: 42 additions & 2 deletions src/core/workspace/hooks/useWorkspaceSessions.ts
Original file line number Diff line number Diff line change
Expand Up @@ -160,6 +160,29 @@ export function useSessionManager(projects: Project[], settings: Settings) {
return projectsRef.current.find((p) => p.id === projectId) ?? null;
}, []);

const pushSystemMessage = useCallback((setMessagesFn: (updater: (prev: UIMessage[]) => UIMessage[]) => void, content: string) => {
setMessagesFn((prev) => [
...prev,
{
id: `system-setup-${Date.now()}`,
role: "system" as const,
content,
timestamp: Date.now(),
},
]);
}, []);

const checkProviderSetup = useCallback(async (model?: string) => {
const settings = settingsRef.current;
const result = await window.clientCore.checkProviderSetup({
llmProvider: settings.llmProvider,
model,
openRouterKey: settings.openRouterKey,
ollamaEndpoint: settings.ollamaEndpoint,
});
return result;
}, []);

// Eagerly start a Agent SDK session for immediate MCP status display
const eagerStartSession = useCallback(async (projectId: string, options?: StartOptions) => {
const project = projectsRef.current.find((p) => p.id === projectId);
Expand Down Expand Up @@ -1043,7 +1066,17 @@ export function useSessionManager(projects: Project[], settings: Settings) {
...(inferredProvider === "ollama" ? { ollamaEndpoint: settingsRef.current.ollamaEndpoint } : {}),
};

const setup = await checkProviderSetup(session.model);
if (!setup.ok) {
pushSystemMessage(engine.setMessages, `Setup required: ${setup.error}`);
return;
}

const result = await window.clientCore.start(startPayload);
if ("error" in result && result.error) {
pushSystemMessage(engine.setMessages, `Setup required: ${result.error}`);
return;
}
const newSessionId = result.sessionId;

if (newSessionId !== oldId) {
Expand Down Expand Up @@ -1111,14 +1144,21 @@ export function useSessionManager(projects: Project[], settings: Settings) {
},
]);
},
[engine.setMessages, findProject],
[checkProviderSetup, engine.setMessages, findProject, pushSystemMessage],
);

const send = useCallback(
async (text: string, images?: ImageAttachment[]) => {
const activeId = activeSessionIdRef.current;
if (activeId === DRAFT_ID) {
const draftEngine = startOptionsRef.current.engine ?? "agent";
if (draftEngine !== "oap") {
const setup = await checkProviderSetup(startOptionsRef.current.model);
if (!setup.ok) {
pushSystemMessage(agent.setMessages, `Setup required: ${setup.error}`);
return;
}
}
const sessionId = await materializeDraft(text);
if (!sessionId) return;
await new Promise((resolve) => setTimeout(resolve, 50));
Expand Down Expand Up @@ -1211,7 +1251,7 @@ export function useSessionManager(projects: Project[], settings: Settings) {
return;
}
},
[agent.send, agent.setMessages, oap.send, oap.setMessages, oap.setIsProcessing, engine.setMessages, materializeDraft, reviveSession],
[agent.send, agent.setMessages, checkProviderSetup, oap.send, oap.setMessages, oap.setIsProcessing, engine.setMessages, materializeDraft, pushSystemMessage, reviveSession],
);

const deselectSession = useCallback(async () => {
Expand Down
6 changes: 6 additions & 0 deletions src/types/window.d.ts
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,12 @@ declare global {
mcpStatus: (sessionId: string) => Promise<{ servers: McpServerStatus[]; error?: string }>;
mcpReconnect: (sessionId: string, serverName: string) => Promise<{ ok?: boolean; error?: string; restarted?: boolean }>;
restartSession: (sessionId: string, mcpServers?: McpServerConfig[]) => Promise<{ ok?: boolean; error?: string; restarted?: boolean }>;
checkProviderSetup: (options: {
llmProvider?: "openrouter" | "ollama";
model?: string;
openRouterKey?: string;
ollamaEndpoint?: string;
}) => Promise<{ ok: boolean; provider: "openrouter" | "ollama"; error?: string }>;
readFile: (filePath: string) => Promise<{ content?: string; error?: string }>;
openInEditor: (filePath: string, line?: number) => Promise<{ ok?: boolean; editor?: string; error?: string }>;
generateTitle: (
Expand Down
Loading