@@ -161,6 +161,10 @@ export interface ExtensionStateContextType extends ExtensionState {
161161 setMaxDiagnosticMessages : ( value : number ) => void
162162 includeTaskHistoryInEnhance ?: boolean
163163 setIncludeTaskHistoryInEnhance : ( value : boolean ) => void
164+ includeCurrentTime ?: boolean
165+ setIncludeCurrentTime : ( value : boolean ) => void
166+ includeCurrentCost ?: boolean
167+ setIncludeCurrentCost : ( value : boolean ) => void
164168}
165169
166170export const ExtensionStateContext = createContext < ExtensionStateContextType | undefined > ( undefined )
@@ -270,6 +274,8 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
270274 maxDiagnosticMessages : 50 ,
271275 openRouterImageApiKey : "" ,
272276 openRouterImageGenerationSelectedModel : "" ,
277+ includeCurrentTime : true ,
278+ includeCurrentCost : true ,
273279 } )
274280
275281 const [ didHydrateState , setDidHydrateState ] = useState ( false )
@@ -290,6 +296,8 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
290296 } )
291297 const [ includeTaskHistoryInEnhance , setIncludeTaskHistoryInEnhance ] = useState ( true )
292298 const [ prevCloudIsAuthenticated , setPrevCloudIsAuthenticated ] = useState ( false )
299+ const [ includeCurrentTime , setIncludeCurrentTime ] = useState ( true )
300+ const [ includeCurrentCost , setIncludeCurrentCost ] = useState ( true )
293301
294302 const setListApiConfigMeta = useCallback (
295303 ( value : ProviderSettingsEntry [ ] ) => setState ( ( prevState ) => ( { ...prevState , listApiConfigMeta : value } ) ) ,
@@ -327,6 +335,14 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
327335 if ( ( newState as any ) . includeTaskHistoryInEnhance !== undefined ) {
328336 setIncludeTaskHistoryInEnhance ( ( newState as any ) . includeTaskHistoryInEnhance )
329337 }
338+ // Update includeCurrentTime if present in state message
339+ if ( ( newState as any ) . includeCurrentTime !== undefined ) {
340+ setIncludeCurrentTime ( ( newState as any ) . includeCurrentTime )
341+ }
342+ // Update includeCurrentCost if present in state message
343+ if ( ( newState as any ) . includeCurrentCost !== undefined ) {
344+ setIncludeCurrentCost ( ( newState as any ) . includeCurrentCost )
345+ }
330346 // Handle marketplace data if present in state message
331347 if ( newState . marketplaceItems !== undefined ) {
332348 setMarketplaceItems ( newState . marketplaceItems )
@@ -575,6 +591,10 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
575591 } ,
576592 includeTaskHistoryInEnhance,
577593 setIncludeTaskHistoryInEnhance,
594+ includeCurrentTime,
595+ setIncludeCurrentTime,
596+ includeCurrentCost,
597+ setIncludeCurrentCost,
578598 }
579599
580600 return < ExtensionStateContext . Provider value = { contextValue } > { children } </ ExtensionStateContext . Provider >
0 commit comments