mirror of
https://github.com/microsoft/vscode-extension-samples.git
synced 2026-06-13 07:10:26 +08:00
Adopt prompt-tsx for chat sample (#995)
* Adopt prompt-tsx for chat sample * Rename * Update prompt-tsx to published version * Update for latest * Fix up * Small fix * Latest prompt-tsx * Format
This commit is contained in:
Rob Lourens
20
chat-sample/package-lock.json
generated
20
chat-sample/package-lock.json
generated
@ -7,8 +7,10 @@
|
||||
"": {
|
||||
"name": "chat-sample",
|
||||
"version": "0.1.0",
|
||||
"hasInstallScript": true,
|
||||
"devDependencies": {
|
||||
"@types/node": "^20.5.9",
|
||||
"@vscode/prompt-tsx": "^0.1.10-alpha",
|
||||
"eslint": "^7.22.0",
|
||||
"tslint": "^6.1.3",
|
||||
"typescript": "^4.0.3"
|
||||
@ -200,6 +202,15 @@
|
||||
"integrity": "sha512-ZnQMnLV4e7hDlUvw8H+U8ASL02SS2Gn6+9Ac3wGGLIe7+je2AeAOxPY+izIPJDfFDb7eDjev0Us8MO1iFRN8hA==",
|
||||
"dev": true
|
||||
},
|
||||
"node_modules/@microsoft/tiktokenizer": {
|
||||
"version": "1.0.6",
|
||||
"resolved": "https://registry.npmjs.org/@microsoft/tiktokenizer/-/tiktokenizer-1.0.6.tgz",
|
||||
"integrity": "sha512-Vi12N9vsU9jP3L95Ji1ZNMZyEbA9zIhLQd3oUPufoPqQZsnH2UGV2e2PDp2auJr7HvuDejGznWzHPj0pIjgBWg==",
|
||||
"dev": true,
|
||||
"engines": {
|
||||
"node": ">=18.0.0"
|
||||
}
|
||||
},
|
||||
"node_modules/@types/node": {
|
||||
"version": "20.9.0",
|
||||
"resolved": "https://registry.npmjs.org/@types/node/-/node-20.9.0.tgz",
|
||||
@ -209,6 +220,15 @@
|
||||
"undici-types": "~5.26.4"
|
||||
}
|
||||
},
|
||||
"node_modules/@vscode/prompt-tsx": {
|
||||
"version": "0.1.10-alpha",
|
||||
"resolved": "https://registry.npmjs.org/@vscode/prompt-tsx/-/prompt-tsx-0.1.10-alpha.tgz",
|
||||
"integrity": "sha512-iqe1XlYsmxGGle7hmMnKd4DOebjFVmQ2odgoSzCYutOzHcqMzlRPac2sNYDQLpYlvutFi0vXl+VMPg0D3uOvVA==",
|
||||
"dev": true,
|
||||
"dependencies": {
|
||||
"@microsoft/tiktokenizer": "^1.0.6"
|
||||
}
|
||||
},
|
||||
"node_modules/acorn": {
|
||||
"version": "8.11.2",
|
||||
"resolved": "https://registry.npmjs.org/acorn/-/acorn-8.11.2.tgz",
|
||||
|
||||
@ -15,15 +15,15 @@
|
||||
"AI",
|
||||
"Chat"
|
||||
],
|
||||
"activationEvents": [ ],
|
||||
"activationEvents": [],
|
||||
"enabledApiProposals": [
|
||||
"chatVariableResolver"
|
||||
],
|
||||
"contributes": {
|
||||
"chatParticipants": [
|
||||
{
|
||||
"id": "chat-sample.cat",
|
||||
"fullName": "Cat",
|
||||
"id": "chat-sample.cat",
|
||||
"fullName": "Cat",
|
||||
"name": "cat",
|
||||
"description": "Meow! What can I teach you?",
|
||||
"isSticky": true,
|
||||
@ -50,10 +50,13 @@
|
||||
"scripts": {
|
||||
"vscode:prepublish": "npm run compile",
|
||||
"compile": "tsc -p ./",
|
||||
"watch": "tsc -watch -p ./"
|
||||
"watch": "tsc -watch -p ./",
|
||||
"setupTiktoken": "cp node_modules/@vscode/prompt-tsx/dist/base/tokenizer/cl100k_base.tiktoken out/",
|
||||
"postinstall": "npm run setupTiktoken"
|
||||
},
|
||||
"devDependencies": {
|
||||
"@types/node": "^20.5.9",
|
||||
"@vscode/prompt-tsx": "^0.1.10-alpha",
|
||||
"eslint": "^7.22.0",
|
||||
"tslint": "^6.1.3",
|
||||
"typescript": "^4.0.3"
|
||||
|
||||
@ -1,4 +1,6 @@
|
||||
import { renderPrompt, Cl100KBaseTokenizer } from '@vscode/prompt-tsx';
|
||||
import * as vscode from 'vscode';
|
||||
import { PlayPrompt } from './play';
|
||||
|
||||
const CAT_NAMES_COMMAND_ID = 'cat.namesInEditor';
|
||||
const CAT_PARTICIPANT_ID = 'chat-sample.cat';
|
||||
@ -41,12 +43,14 @@ export function activate(context: vscode.ExtensionContext) {
|
||||
return { metadata: { command: 'teach' } };
|
||||
} else if (request.command == 'play') {
|
||||
stream.progress('Throwing away the computer science books and preparing to play with some Python code...');
|
||||
const messages = [
|
||||
vscode.LanguageModelChatMessage.User('You are a cat! Reply in the voice of a cat, using cat analogies when appropriate. Be concise to prepare for cat play time.'),
|
||||
vscode.LanguageModelChatMessage.User('Give a small random python code samples (that have cat names for variables). ' + request.prompt)
|
||||
];
|
||||
const [model] = await vscode.lm.selectChatModels(MODEL_SELECTOR);
|
||||
if (model) {
|
||||
// Here's an example of how to use the prompt-tsx library to build a prompt
|
||||
const { messages } = await renderPrompt(
|
||||
PlayPrompt,
|
||||
{ userQuery: request.prompt },
|
||||
{ modelMaxPromptTokens: model.maxInputTokens },
|
||||
new Cl100KBaseTokenizer());
|
||||
const chatResponse = await model.sendRequest(messages, {}, token);
|
||||
for await (const fragment of chatResponse.text) {
|
||||
stream.markdown(fragment);
|
||||
|
||||
25
chat-sample/src/play.tsx
Normal file
25
chat-sample/src/play.tsx
Normal file
@ -0,0 +1,25 @@
|
||||
import {
|
||||
BasePromptElementProps,
|
||||
PromptElement,
|
||||
PromptSizing,
|
||||
UserMessage
|
||||
} from '@vscode/prompt-tsx';
|
||||
|
||||
export interface PromptProps extends BasePromptElementProps {
|
||||
userQuery: string;
|
||||
}
|
||||
|
||||
export class PlayPrompt extends PromptElement<PromptProps, void> {
|
||||
render(state: void, sizing: PromptSizing) {
|
||||
return (
|
||||
<>
|
||||
<UserMessage>
|
||||
You are a cat! Reply in the voice of a cat, using cat analogies when
|
||||
appropriate. Be concise to prepare for cat play time. Give a small random
|
||||
python code sample (that has cat names for variables).
|
||||
</UserMessage>
|
||||
<UserMessage>{this.props.userQuery}</UserMessage>
|
||||
</>
|
||||
);
|
||||
}
|
||||
}
|
||||
@ -8,7 +8,10 @@
|
||||
"sourceMap": true,
|
||||
"rootDir": "src",
|
||||
"strict": true, /* enable all strict type-checking options */
|
||||
"outDir": "out"
|
||||
"outDir": "out",
|
||||
"jsx": "react",
|
||||
"jsxFactory": "vscpp",
|
||||
"jsxFragmentFactory": "vscppf"
|
||||
/* Additional Checks */
|
||||
// "noImplicitReturns": true, /* Report error when not all code paths in function return a value. */
|
||||
// "noFallthroughCasesInSwitch": true, /* Report errors for fallthrough cases in switch statement. */
|
||||
|
||||
Reference in New Issue
Block a user