Merge pull request #8 from TarsLab/fix-issue-7

fix issue 7

manifest.json

@@ -1,7 +1,7 @@
 {
 	"id": "ai-zhipu",
 	"name": "AI Zhipu",
-	"version": "2.0.0",
+	"version": "2.0.1",
 	"minAppVersion": "1.0.0",
 	"description": "Generate text using the ZhipuAI API.",
 	"author": "Tarslab",

package.json

@@ -1,6 +1,6 @@
 {
 	"name": "obsidian-ai-zhipu",
-	"version": "2.0.0",
+	"version": "2.0.1",
 	"description": "Generate text using the ZhipuAI API.",
 	"main": "main.js",
 	"scripts": {

src/main.ts

@@ -173,15 +173,15 @@ export default class AIZhipuPlugin extends Plugin {
 		return client
 	}
 
-	createBlockFromSelection(editor: Editor): UserMsgBlock {
+	createBlockFromSelection(editor: Editor, selection: string): UserMsgBlock {
 		const block: UserMsgBlock = {
 			_tag: 'user',
-			content: origin,
+			content: selection,
 			start: editor.getCursor('from').line + 1,
 			end: editor.getCursor('to').line + 3
 		}
 		editor.replaceSelection(
-			LINE_BREAK + USER_MARK_START + LINE_BREAK + origin.trimEnd() + LINE_BREAK + USER_MARK_END + LINE_BREAK
+			LINE_BREAK + USER_MARK_START + LINE_BREAK + selection.trimEnd() + LINE_BREAK + USER_MARK_END + LINE_BREAK
 		)
 		return block
 	}
@@ -208,8 +208,9 @@ export default class AIZhipuPlugin extends Plugin {
 		let block = findBlockByCurrentLine(editor.getValue().split('\n'), editor.getCursor('to').line)
 
 		if (!block) {
-			if (editor.getSelection().trim().length > 0) {
-				block = this.createBlockFromSelection(editor)
+			const selection = editor.getSelection()
+			if (selection.trim().length > 0) {
+				block = this.createBlockFromSelection(editor, selection)
 			} else {
 				const current = editor.getCursor('to').line
 				if (editor.getLine(current).trim().length === 0) {

versions.json

@@ -3,5 +3,6 @@
 	"1.0.2": "1.0.0",
 	"1.0.3": "1.0.0",
 	"1.0.4": "1.0.0",
-	"2.0.0": "1.0.0"
+	"2.0.0": "1.0.0",
+	"2.0.1": "1.0.0"
 }