use a setting in core and set the location to notebook (microsoft#257548)

use a setting in core and set the location from the start

Author: amunger

src/vs/workbench/contrib/inlineChat/browser/inlineChatController.ts

@@ -1240,6 +1240,7 @@ export class InlineChatController2 implements IEditorContribution {
 		@IChatAttachmentResolveService private readonly _chatAttachmentResolveService: IChatAttachmentResolveService,
 		@IEditorService private readonly _editorService: IEditorService,
 		@IInlineChatSessionService inlineChatService: IInlineChatSessionService,
+		@IConfigurationService configurationService: IConfigurationService,
 	) {
 
 		const ctxInlineChatVisible = CTX_INLINE_CHAT_VISIBLE.bindTo(contextKeyService);
@@ -1270,6 +1271,18 @@ export class InlineChatController2 implements IEditorContribution {
 					if (codeEditor === this._editor) {
 						location.location = ChatAgentLocation.Notebook;
 						notebookEditor = editor;
+						// set location2 so that the notebook agent intent is used
+						if (configurationService.getValue(InlineChatConfigKeys.notebookAgent)) {
+							location.resolveData = () => {
+								assertType(this._editor.hasModel());
+
+								return {
+									type: ChatAgentLocation.Notebook,
+									sessionInputUri: this._editor.getModel().uri,
+								};
+							};
+						}
+
 						break;
 					}
 				}

src/vs/workbench/contrib/inlineChat/common/inlineChat.ts

@@ -20,6 +20,7 @@ export const enum InlineChatConfigKeys {
 	LineEmptyHint = 'inlineChat.lineEmptyHint',
 	LineNLHint = 'inlineChat.lineNaturalLanguageHint',
 	EnableV2 = 'inlineChat.enableV2',
+	notebookAgent = 'inlineChat.notebookAgent',
 	HideOnRequest = 'inlineChat.hideOnRequest'
 }
 
@@ -77,6 +78,15 @@ Registry.as<IConfigurationRegistry>(Extensions.Configuration).registerConfigurat
 				mode: 'startup'
 			}
 		},
+		[InlineChatConfigKeys.notebookAgent]: {
+			markdownDescription: localize('notebookAgent', "Enable agent-like behavior for inline chat widget in notebooks. Depends on the `#inlineChat.enableV2#` setting being enabled."),
+			default: false,
+			type: 'boolean',
+			tags: ['experimental'],
+			experiment: {
+				mode: 'startup'
+			}
+		}
 	}
 });