Add force_model_selection option to intelligent file updates for manual model selection

Author: robertpiosik

apps/editor/src/views/panel/backend/message-handlers/handle-intelligent-update-file-in-preview.ts

@@ -24,7 +24,7 @@ export const handle_intelligent_update_file_in_preview = async (
   panel_provider: PanelProvider,
   message: IntelligentUpdateFileInPreviewMessage
 ): Promise<void> => {
-  const { file_path, workspace_name } = message
+  const { file_path, workspace_name, force_model_selection } = message
   const file_name = path.basename(file_path)
 
   const original_states = panel_provider.context.workspaceState.get<
@@ -95,7 +95,7 @@ export const handle_intelligent_update_file_in_preview = async (
   )
   const config_result = await get_intelligent_update_config(
     api_providers_manager,
-    false,
+    force_model_selection ?? false,
     panel_provider.context
   )
   if (!config_result) return

apps/editor/src/views/panel/frontend/Panel.tsx

@@ -587,7 +587,8 @@ export const Panel = () => {
                   post_message(vscode, {
                     command: 'INTELLIGENT_UPDATE_FILE_IN_PREVIEW',
                     file_path: file.file_path,
-                    workspace_name: file.workspace_name
+                    workspace_name: file.workspace_name,
+                    force_model_selection: file.force_model_selection
                   })
                 }}
                 on_cancel_intelligent_update={(file) => {

apps/editor/src/views/panel/types/messages.ts

@@ -347,6 +347,7 @@ export interface IntelligentUpdateFileInPreviewMessage extends BaseMessage {
   command: 'INTELLIGENT_UPDATE_FILE_IN_PREVIEW'
   file_path: string
   workspace_name?: string
+  force_model_selection?: boolean
 }
 
 export interface CancelIntelligentUpdateFileInPreviewMessage extends BaseMessage {

packages/ui/src/components/editor/panel/ResponsePreview/ResponsePreview.tsx

@@ -13,6 +13,7 @@ type Props = {
   on_intelligent_update: (file: {
     file_path: string
     workspace_name?: string
+    force_model_selection?: boolean
   }) => void
   on_cancel_intelligent_update: (file: {
     file_path: string
@@ -299,10 +300,11 @@ export const ResponsePreview: FC<Props> = (props) => {
                       content: file.ai_content!
                     })
                   }
-                  on_intelligent_update={() =>
+                  on_intelligent_update={(force_model_selection) =>
                     props.on_intelligent_update({
                       file_path: file.file_path,
-                      workspace_name: file.workspace_name
+                      workspace_name: file.workspace_name,
+                      force_model_selection
                     })
                   }
                   on_cancel_intelligent_update={() =>

packages/ui/src/components/editor/panel/ResponsePreview/components/FileItem/FileItem.tsx

@@ -14,7 +14,7 @@ type Props = {
   on_toggle: (checked: boolean) => void
   on_discard_user_changes: () => void
   on_preview_generated_code: () => void
-  on_intelligent_update: () => void
+  on_intelligent_update: (force_model_selection?: boolean) => void
   on_cancel_intelligent_update: () => void
   on_go_to_file: () => void
 }
@@ -163,7 +163,9 @@ export const FileItem: FC<Props> = (props) => {
                       title="Fix with Intelligent Update API tool"
                       on_click={(e) => {
                         e.stopPropagation()
-                        props.on_intelligent_update()
+                        props.on_intelligent_update(
+                          !!props.file.fixed_with_intelligent_update
+                        )
                       }}
                     />
                   )}
@@ -227,7 +229,11 @@ export const FileItem: FC<Props> = (props) => {
               {!props.file.is_applying && (
                 <div
                   className={styles['message__actions__item']}
-                  onClick={props.on_intelligent_update}
+                  onClick={() =>
+                    props.on_intelligent_update(
+                      !!props.file.fixed_with_intelligent_update
+                    )
+                  }
                   title={'Fix with Intelligent Update API tool'}
                 >
                   <span className="codicon codicon-sparkle" />