Microsoft/vscode
1
0
Fork 0
mirror of https://github.com/microsoft/vscode.git synced 2026-04-24 02:28:34 +01:00
Code Issues Packages Projects Releases Wiki Activity

wip - inline chat via editing session

Browse Source
This commit is contained in:
Johannes
2025-01-22 09:13:23 +01:00
parent 63c3204155
commit 3aece83761
12 changed files with 341 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
src/vs/workbench/contrib/inlineChat/browser/inlineChatSessionServiceImpl.ts
View File

@@ -22,14 +22,15 @@ import { ITelemetryService } from '../../../../platform/telemetry/common/telemet
import { DEFAULT_EDITOR_ASSOCIATION } from '../../../common/editor.js';
import { ChatAgentLocation, IChatAgentService } from '../../chat/common/chatAgents.js';
import { IChatService } from '../../chat/common/chatService.js';
import { CTX_INLINE_CHAT_HAS_AGENT, CTX_INLINE_CHAT_POSSIBLE } from '../common/inlineChat.js';
import { CTX_INLINE_CHAT_HAS_AGENT, CTX_INLINE_CHAT_HAS_AGENT2, CTX_INLINE_CHAT_POSSIBLE } from '../common/inlineChat.js';
import { IEditorService } from '../../../services/editor/common/editorService.js';
import { UntitledTextEditorInput } from '../../../services/untitled/common/untitledTextEditorInput.js';
import { HunkData, Session, SessionWholeRange, StashedSession, TelemetryData, TelemetryDataClassification } from './inlineChatSession.js';
import { IInlineChatSessionEndEvent, IInlineChatSessionEvent, IInlineChatSessionService, ISessionKeyComputer } from './inlineChatSessionService.js';
import { IInlineChatSession2, IInlineChatSessionEndEvent, IInlineChatSessionEvent, IInlineChatSessionService, ISessionKeyComputer } from './inlineChatSessionService.js';
import { isEqual } from '../../../../base/common/resources.js';
import { ILanguageService } from '../../../../editor/common/languages/language.js';
import { ITextFileService } from '../../../services/textfile/common/textfiles.js';
import { IChatEditingService } from '../../chat/common/chatEditingService.js';
type SessionData = {
@@ -79,7 +80,8 @@ export class InlineChatSessionServiceImpl implements IInlineChatSessionService {
@ITextFileService private readonly _textFileService: ITextFileService,
@ILanguageService private readonly _languageService: ILanguageService,
@IChatService private readonly _chatService: IChatService,
@IChatAgentService private readonly _chatAgentService: IChatAgentService
@IChatAgentService private readonly _chatAgentService: IChatAgentService,
@IChatEditingService private readonly _chatEditingService: IChatEditingService,
) { }
dispose() {
@@ -311,6 +313,49 @@ export class InlineChatSessionServiceImpl implements IInlineChatSessionService {
this._keyComputers.set(scheme, value);
return toDisposable(() => this._keyComputers.delete(scheme));
}
// ---- NEW
private readonly _sessions2 = new Map<string, IInlineChatSession2>();
private readonly _onDidChangeSessions = this._store.add(new Emitter<this>());
readonly onDidChangeSessions: Event<this> = this._onDidChangeSessions.event;
async createSession2(editor: ICodeEditor, uri: URI, token: CancellationToken): Promise<IInlineChatSession2> {
const key = this._key(editor, uri);
if (this._sessions2.has(key)) {
throw new Error('Session already exists');
}
this._onWillStartSession.fire(editor as IActiveCodeEditor);
const chatModel = this._chatService.startSession(ChatAgentLocation.EditingSession, token);
const editingSession = await this._chatEditingService.createAdhocEditingSession(chatModel.sessionId);
editingSession.addFileToWorkingSet(uri);
const result: IInlineChatSession2 = {
chatModel,
editingSession,
dispose: () => {
if (this._sessions2.delete(key)) {
editingSession.dispose();
chatModel.dispose();
this._onDidChangeSessions.fire(this);
}
}
};
this._sessions2.set(key, result);
this._onDidChangeSessions.fire(this);
return result;
}
getSession2(editor: ICodeEditor, uri: URI): IInlineChatSession2 | undefined {
const key = this._key(editor, uri);
return this._sessions2.get(key);
}
}
export class InlineChatEnabler {
@@ -318,6 +363,7 @@ export class InlineChatEnabler {
static Id = 'inlineChat.enabler';
private readonly _ctxHasProvider: IContextKey<boolean>;
private readonly _ctxHasProvider2: IContextKey<boolean>;
private readonly _ctxPossible: IContextKey<boolean>;
private readonly _store = new DisposableStore();
@@ -328,11 +374,21 @@ export class InlineChatEnabler {
@IEditorService editorService: IEditorService,
) {
this._ctxHasProvider = CTX_INLINE_CHAT_HAS_AGENT.bindTo(contextKeyService);
this._ctxHasProvider2 = CTX_INLINE_CHAT_HAS_AGENT2.bindTo(contextKeyService);
this._ctxPossible = CTX_INLINE_CHAT_POSSIBLE.bindTo(contextKeyService);
const updateAgent = () => {
const hasEditorAgent = Boolean(chatAgentService.getDefaultAgent(ChatAgentLocation.Editor));
this._ctxHasProvider.set(hasEditorAgent);
const agent = chatAgentService.getDefaultAgent(ChatAgentLocation.Editor);
if (agent?.locations.length === 1) {
this._ctxHasProvider.set(true);
this._ctxHasProvider2.reset();
} else if (agent?.locations.includes(ChatAgentLocation.EditingSession)) {
this._ctxHasProvider.reset();
this._ctxHasProvider2.set(true);
} else {
this._ctxHasProvider.reset();
this._ctxHasProvider2.reset();
}
};
this._store.add(chatAgentService.onDidChangeAgents(updateAgent));