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
8 changes: 6 additions & 2 deletions apps/desktop/src/app/index.tsx
Original file line number Diff line number Diff line change
@@ -1,9 +1,13 @@
import { Layout } from "./layout";
import { Page } from "./page";

export function App() {
export function App({
totalMemoryBytes,
}: {
totalMemoryBytes: number | null;
}) {
return (
<Layout>
<Layout totalMemoryBytes={totalMemoryBytes}>
<Page />
</Layout>
);
Expand Down
43 changes: 26 additions & 17 deletions apps/desktop/src/app/layout.tsx
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
import "@fontsource-variable/geist/index.css";
import "@fontsource-variable/geist-mono/index.css";
import { MessageVirtualizationProvider } from "@llm-space/ui/components/message-virtualization-provider";
import {
ThemeProvider,
useTheme,
Expand All @@ -16,25 +17,33 @@ import { PLAYGROUND_LABELS } from "@/i18n/playground-labels";

import { QueryProvider } from "./query-provider";

export function Layout({ children }: { children: React.ReactNode }) {
export function Layout({
children,
totalMemoryBytes,
}: {
children: React.ReactNode;
totalMemoryBytes: number | null;
}) {
return (
<ThemeProvider>
<I18nProvider>
<PlaygroundLabelsBridge>
<ExperimentalProvider>
<QueryProvider>
<TooltipProvider delayDuration={1000}>
<PluginCommandExecutionProvider>
<div className="flex size-full flex-col">
<ThemedToaster />
{children}
</div>
</PluginCommandExecutionProvider>
</TooltipProvider>
</QueryProvider>
</ExperimentalProvider>
</PlaygroundLabelsBridge>
</I18nProvider>
<MessageVirtualizationProvider totalMemoryBytes={totalMemoryBytes}>
<I18nProvider>
<PlaygroundLabelsBridge>
<ExperimentalProvider>
<QueryProvider>
<TooltipProvider delayDuration={1000}>
<PluginCommandExecutionProvider>
<div className="flex size-full flex-col">
<ThemedToaster />
{children}
</div>
</PluginCommandExecutionProvider>
</TooltipProvider>
</QueryProvider>
</ExperimentalProvider>
</PlaygroundLabelsBridge>
</I18nProvider>
</MessageVirtualizationProvider>
</ThemeProvider>
);
}
Expand Down
27 changes: 27 additions & 0 deletions apps/desktop/src/app/page.tsx
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import type { Thread } from "@llm-space/core";
import type { EditorCommitScopeHandle } from "@llm-space/ui/components/code-editor/editor-commit-scope";
import { FirecrawlLimitDialog } from "@llm-space/ui/components/firecrawl-limit-dialog";
import {
useModels,
Expand Down Expand Up @@ -299,6 +300,27 @@ function PageWorkspace({
);
}, []);
const tabs = useThreadTabs({ canPruneRestoredTab });
const viewCommitHandlesRef = useRef(
new Map<string, EditorCommitScopeHandle>()
);
const commitView = useCallback((paneId: string) => {
viewCommitHandlesRef.current.get(paneId)?.commitAll();
}, []);
const commitViews = useCallback(
(closingTabs: AppTab[]) => {
for (const tab of closingTabs) {
commitView(paneIdForTab(tab));
}
},
[commitView]
);
const handleViewCommitScopeReady = useCallback(
(paneId: string, handle: EditorCommitScopeHandle | null) => {
if (handle) viewCommitHandlesRef.current.set(paneId, handle);
else viewCommitHandlesRef.current.delete(paneId);
},
[]
);
const { executeCommand } = useCommands();
const models = useModels();
const refreshModels = useRefreshModels();
Expand Down Expand Up @@ -749,6 +771,7 @@ function PageWorkspace({
tabs: tabs.tabs,
targetId: target,
onBlocked: () => showRuntimeRunBlocked("closing this tab"),
commitViews,
close,
});
},
Expand All @@ -764,6 +787,7 @@ function PageWorkspace({
keepId: target,
runtimeId: targetRuntimeId,
onBlocked: () => showRuntimeRunBlocked("closing other tabs"),
commitViews,
closeOthers: closeOthersInRuntime,
});
},
Expand All @@ -774,6 +798,7 @@ function PageWorkspace({
tabs: tabs.tabs,
runtimeId,
onBlocked: () => showRuntimeRunBlocked("closing all tabs"),
commitViews,
closeAll: closeAllInRuntime,
});
},
Expand Down Expand Up @@ -1150,6 +1175,8 @@ function PageWorkspace({
onToggleSidebar={handleToggleSidebar}
lifecycleHost={paneLifecycleHost}
mutationRevision={mutationRevision}
commitView={commitView}
onViewCommitScopeReady={handleViewCommitScopeReady}
onThreadStateChange={handleThreadStateChange}
toolbarSlot={<UpdateIndicator />}
/>
Expand Down
Loading
Loading