Update extension to support other languages

README.md

@@ -10,7 +10,7 @@ AI writing assistant (prototype).
 
 3. Install the extension in VSCode
 
-    code --install-extension extension/rht-text-generator/rht-text-generator-0.0.1.vsix
+    code --install-extension extension/rht-text-generator/rht-text-generator-0.0.2.vsix
 
 ## Retrain the model
 

extension/rht-text-generator/package.json

@@ -2,16 +2,20 @@
 	"name": "rht-text-generator",
 	"displayName": "rht-text-generator",
 	"publisher": "red-hat-training",
-	"description": "",
-	"version": "0.0.1",
+	"description": "VSCode RHT extension for writting suggestions",
+	"repository": "https://github.com/RedHatTraining/ai-text-generation",
+	"version": "0.0.2",
 	"engines": {
 		"vscode": "^1.55.0"
 	},
 	"categories": [
 		"Other"
 	],
 	"activationEvents": [
-		"*"
+		"onLanguage:asciidoc",
+		"onLanguage:latex",
+		"onLanguage:markdown",
+		"onLanguage:plaintext"
 	],
 	"main": "./out/extension.js",
 	"scripts": {

extension/rht-text-generator/src/extension.ts

@@ -17,7 +17,7 @@ export function activate(context: vscode.ExtensionContext) {
 
 	context.subscriptions.push(
 		vscode.languages.registerCompletionItemProvider(
-			[{ language: "asciidoc"}],
+			[{ language: "asciidoc" }, { language: "latex" }, { language: "markdown" }, { language: "plaintext" }],
 			provider,
 			...COMPLETION_TRIGGERS
 		)
@@ -60,7 +60,7 @@ async function getCompletionsListItemsFor(
 /**
  * Get the text from the current position back to a specific number of lines
  */
-function getText(document: vscode.TextDocument, position: vscode.Position): string  {
+function getText(document: vscode.TextDocument, position: vscode.Position): string {
 	const config = vscode.workspace.getConfiguration("rht-text-generator");
 	const MAX_LINES: number = config.get("lines") || 3;
 	const currentLineNumber = position.line;