diff --git a/common.ts b/common.ts index 7874fb3..f2cba54 100644 --- a/common.ts +++ b/common.ts @@ -26,6 +26,7 @@ export interface LoomSettings { topP: number; frequencyPenalty: number; presencePenalty: number; + prepend: string; bestOf: number; n: number; diff --git a/main.ts b/main.ts index d9c05d0..dd7d153 100644 --- a/main.ts +++ b/main.ts @@ -73,6 +73,7 @@ const DEFAULT_SETTINGS: LoomSettings = { topP: 1, frequencyPenalty: 0, presencePenalty: 0, + prepend: "<|endoftext|>", bestOf: 0, n: 5, @@ -700,24 +701,38 @@ export default class LoomPlugin extends Plugin { ); // update the editor's text - const cursor = this.editor.getCursor(); - const linesBefore = this.editor.getValue().split("\n"); + // const cursor = this.editor.getCursor(); + // const linesBefore = this.editor.getValue().split("\n"); this.editor.setValue(this.fullText(file, id)); - // if the text preceding the cursor has changed, move the cursor to the end of the text - // otherwise, restore the cursor position - const linesAfter = this.editor - .getValue() - .split("\n") - .slice(0, cursor.line + 1); - for (let i = 0; i < cursor.line; i++) - if (linesBefore[i] !== linesAfter[i]) { - const line = this.editor.lineCount() - 1; - const ch = this.editor.getLine(line).length; - this.editor.setCursor({ line, ch }); - return; - } - this.editor.setCursor(cursor); + // always move cursor to the end of the editor + const line = this.editor.lineCount() - 1; + const ch = this.editor.getLine(line).length; + this.editor.setCursor({ line, ch }); + // return; + + // // if the cursor is at the beginning of the editor, move it to the end + // if(cursor.line === 0 && cursor.ch === 0) { + // const line = this.editor.lineCount() - 1; + // const ch = this.editor.getLine(line).length; + // this.editor.setCursor({ line, ch }); + // return; + // } + + // // if the text preceding the cursor has changed, move the cursor to the end of the text + // // otherwise, restore the cursor position + // const linesAfter = this.editor + // .getValue() + // .split("\n") + // .slice(0, cursor.line + 1); + // for (let i = 0; i < cursor.line; i++) + // if (linesBefore[i] !== linesAfter[i]) { + // const line = this.editor.lineCount() - 1; + // const ch = this.editor.getLine(line).length; + // this.editor.setCursor({ line, ch }); + // return; + // } + // this.editor.setCursor(cursor); }) ) ); @@ -1154,7 +1169,9 @@ export default class LoomPlugin extends Plugin { async complete(file: TFile) { const state = this.state[file.path]; - this.breakAtPoint(file); + const [parentNode] = this.breakAtPoint(file); + // switch to the parent node + this.app.workspace.trigger("loom:switch-to", parentNode); await this.generate(file, state.current); } @@ -1174,7 +1191,7 @@ export default class LoomPlugin extends Plugin { // show the "Generating..." indicator in the loom view this.renderLoomViews(); - let prompt = `<|endoftext|>${this.fullText(file, rootNode)}`; + let prompt = `${this.settings.prepend}${this.fullText(file, rootNode)}`; // remove a trailing space if there is one // store whether there was, so it can be added back post-completion diff --git a/package-lock.json b/package-lock.json index 3c75ad9..4063bf6 100644 --- a/package-lock.json +++ b/package-lock.json @@ -1,12 +1,12 @@ { "name": "obsidian-loom", - "version": "1.16.3", + "version": "1.17.0", "lockfileVersion": 3, "requires": true, "packages": { "": { "name": "obsidian-loom", - "version": "1.16.3", + "version": "1.17.0", "license": "MIT", "dependencies": { "@codemirror/state": "^6.2.0", diff --git a/views.ts b/views.ts index 6c1dbd4..21880d1 100644 --- a/views.ts +++ b/views.ts @@ -146,6 +146,12 @@ export class LoomView extends ItemView { this.renderTree(this.tree, state); this.containerEl.scrollTop = scroll; + + // scroll to active node in the tree + const activeNode = this.tree.querySelector(".is-active"); + if (activeNode){ //&& !container.contains(activeNode)){ + activeNode.scrollIntoView({ block: "nearest" }); + } } renderNavButtons(settings: LoomSettings) { @@ -327,6 +333,7 @@ export class LoomView extends ItemView { setting("Top p", "topP", String(settings.topP), "float"); setting("Frequency penalty", "frequencyPenalty", String(settings.frequencyPenalty), "float"); setting("Presence penalty", "presencePenalty", String(settings.presencePenalty), "float"); + setting("Prepend sequence", "prepend", settings.prepend, "string"); } renderBookmarks(container: HTMLElement, state: NoteState) {