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

Finalize ChatRequest#model (#231811)

Browse Source 
* Finalize ChatRequest#model
Fix #230844

* Register model to fix tests
This commit is contained in:
Rob Lourens
2024-10-21 16:42:23 -07:00
committed by GitHub
parent fb42702b12
commit d9f379b135
6 changed files with 57 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
extensions/vscode-api-tests/src/singlefolder-tests/chat.test.ts
View File

@@ -5,7 +5,7 @@
import * as assert from 'assert';
import 'mocha';
import { ChatContext, ChatRequest, ChatResult, ChatVariableLevel, Disposable, Event, EventEmitter, chat, commands } from 'vscode';
import { ChatContext, ChatRequest, ChatResult, ChatVariableLevel, Disposable, Event, EventEmitter, chat, commands, lm } from 'vscode';
import { DeferredPromise, asPromise, assertNoRpc, closeAllEditors, delay, disposeAll } from '../utils';
suite('chat', () => {
@@ -13,6 +13,25 @@ suite('chat', () => {
let disposables: Disposable[] = [];
setup(() => {
disposables = [];
// Register a dummy default model which is required for a participant request to go through
disposables.push(lm.registerChatModelProvider('test-lm', {
async provideLanguageModelResponse(_messages, _options, _extensionId, _progress, _token) {
return undefined;
},
async provideTokenCount(_text, _token) {
return 1;
},
}, {
name: 'test-lm',
version: '1.0.0',
family: 'test',
vendor: 'test-lm-vendor',
maxInputTokens: 100,
maxOutputTokens: 100,
isDefault: true,
isUserSelectable: true
}));
});
teardown(async function () {