diff --git a/src/vs/workbench/contrib/inlineChat/browser/inlineChatSessionServiceImpl.ts b/src/vs/workbench/contrib/inlineChat/browser/inlineChatSessionServiceImpl.ts index bc59dd6d902..3844c320ab0 100644 --- a/src/vs/workbench/contrib/inlineChat/browser/inlineChatSessionServiceImpl.ts +++ b/src/vs/workbench/contrib/inlineChat/browser/inlineChatSessionServiceImpl.ts @@ -229,6 +229,7 @@ export class InlineChatError extends Error { const _bridgeAgentId = 'brigde.editor'; const _inlineChatContext = '_inlineChatContext'; +const _inlineChatDocument = '_inlineChatDocument'; class InlineChatContext { @@ -392,6 +393,17 @@ export class InlineChatSessionServiceImpl implements IInlineChatSessionService { return undefined; } )); + this._store.add(chatVariableService.registerVariable( + { name: _inlineChatDocument, description: '', hidden: true }, + async (_message, _arg, model) => { + for (const [, data] of this._sessions) { + if (data.session.chatModel === model) { + return [{ level: 'full', value: data.session.textModelN.uri }]; + } + } + return undefined; + } + )); }