{!isConfirming ? (
diff --git a/webview-ui/src/components/chat/checkpoints/CheckpointSaved.tsx b/webview-ui/src/components/chat/checkpoints/CheckpointSaved.tsx
index 5947b6a1fc..d5c0050e53 100644
--- a/webview-ui/src/components/chat/checkpoints/CheckpointSaved.tsx
+++ b/webview-ui/src/components/chat/checkpoints/CheckpointSaved.tsx
@@ -3,14 +3,20 @@ import { CheckpointMenu } from "./CheckpointMenu"
type CheckpointSavedProps = {
ts: number
commitHash: string
+ currentCheckpointHash?: string
}
-export const CheckpointSaved = (props: CheckpointSavedProps) => (
-
-
-
-
Checkpoint
+export const CheckpointSaved = (props: CheckpointSavedProps) => {
+ const isCurrent = props.currentCheckpointHash === props.commitHash
+
+ return (
+
+
+
+ Checkpoint
+ {isCurrent && Current}
+
+
-
-
-)
+ )
+}
diff --git a/webview-ui/src/context/ExtensionStateContext.tsx b/webview-ui/src/context/ExtensionStateContext.tsx
index 7e6a06894d..d518878aae 100644
--- a/webview-ui/src/context/ExtensionStateContext.tsx
+++ b/webview-ui/src/context/ExtensionStateContext.tsx
@@ -26,6 +26,7 @@ export interface ExtensionStateContextType extends ExtensionState {
openRouterModels: Record
openAiModels: string[]
mcpServers: McpServer[]
+ currentCheckpoint?: string
filePaths: string[]
openedTabs: Array<{ label: string; isActive: boolean; path?: string }>
setApiConfiguration: (config: ApiConfiguration) => void
@@ -126,6 +127,7 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
const [openAiModels, setOpenAiModels] = useState([])
const [mcpServers, setMcpServers] = useState([])
+ const [currentCheckpoint, setCurrentCheckpoint] = useState()
const setListApiConfigMeta = useCallback(
(value: ApiConfigMeta[]) => setState((prevState) => ({ ...prevState, listApiConfigMeta: value })),
@@ -241,6 +243,10 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
setMcpServers(message.mcpServers ?? [])
break
}
+ case "currentCheckpointUpdated": {
+ setCurrentCheckpoint(message.text)
+ break
+ }
case "listApiConfig": {
setListApiConfigMeta(message.listApiConfig ?? [])
break
@@ -265,6 +271,7 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
openRouterModels,
openAiModels,
mcpServers,
+ currentCheckpoint,
filePaths,
openedTabs,
soundVolume: state.soundVolume,