Tools/Notes
1
0
Fork 0
mirror of https://github.com/TriliumNext/Notes.git synced 2025-08-31 19:51:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

create get/set for private funcs

Browse Source
This commit is contained in:
perf3ct 2025-03-30 19:56:09 +00:00
parent 997edd8de8
commit 229a29d1cb
No known key found for this signature in database
GPG Key ID: 569C4EEC436F5232
2 changed files with 58 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
src/public/app/widgets/llm_chat_panel.ts
View File

@ -99,6 +99,39 @@ export default class LlmChatPanel extends BasicWidget {
private onGetData: (() => Promise<any>) | null = null;
private messages: Array<{role: string; content: string; timestamp?: Date}> = [];
// Public getters and setters for private properties
public getCurrentNoteId(): string | null {
return this.currentNoteId;
}
public setCurrentNoteId(noteId: string | null): void {
this.currentNoteId = noteId;
}
public getMessages(): Array<{role: string; content: string; timestamp?: Date}> {
return this.messages;
}
public setMessages(messages: Array<{role: string; content: string; timestamp?: Date}>): void {
this.messages = messages;
}
public getSessionId(): string | null {
return this.sessionId;
}
public setSessionId(sessionId: string | null): void {
this.sessionId = sessionId;
}
public getNoteContextChatMessages(): HTMLElement {
return this.noteContextChatMessages;
}
public clearNoteContextChatMessages(): void {
this.noteContextChatMessages.innerHTML = '';
}
doRender() {
this.$widget = $(TPL);

50
src/public/app/widgets/type_widgets/ai_chat.ts
View File

@ -13,7 +13,7 @@ export default class AiChatTypeWidget extends TypeWidget {
constructor() {
super();
this.llmChatPanel = new LlmChatPanel();
// Connect the data callbacks
this.llmChatPanel.setDataCallbacks(
(data) => this.saveData(data),
@ -35,19 +35,19 @@ export default class AiChatTypeWidget extends TypeWidget {
// Override the refreshWithNote method to ensure we get note changes
async refreshWithNote(note: FNote | null | undefined) {
console.log("refreshWithNote called for note:", note?.noteId);
// Always force a refresh when the note changes
if (this.note?.noteId !== note?.noteId) {
console.log(`Note ID changed from ${this.note?.noteId} to ${note?.noteId}, forcing reset`);
this.isInitialized = false;
this.initPromise = null;
// Force refresh the chat panel with the new note
if (note) {
this.llmChatPanel.currentNoteId = note.noteId;
this.llmChatPanel.setCurrentNoteId(note.noteId);
}
}
// Continue with regular doRefresh
await this.doRefresh(note);
}
@ -55,7 +55,7 @@ export default class AiChatTypeWidget extends TypeWidget {
async doRefresh(note: FNote | null | undefined) {
try {
console.log("doRefresh called for note:", note?.noteId);
// If we're already initializing, wait for that to complete
if (this.initPromise) {
await this.initPromise;
@ -91,9 +91,9 @@ export default class AiChatTypeWidget extends TypeWidget {
this.initPromise = (async () => {
try {
// Reset the UI before refreshing
this.llmChatPanel.noteContextChatMessages.innerHTML = '';
this.llmChatPanel.messages = [];
this.llmChatPanel.clearNoteContextChatMessages();
this.llmChatPanel.setMessages([]);
// This will load saved data via the getData callback
await this.llmChatPanel.refresh();
this.isInitialized = true;
@ -118,21 +118,21 @@ export default class AiChatTypeWidget extends TypeWidget {
async noteSwitched() {
console.log("Note switched to:", this.noteId);
// Force a full reset when switching notes
this.isInitialized = false;
this.initPromise = null;
if (this.note) {
// Update the chat panel with the new note ID before refreshing
this.llmChatPanel.currentNoteId = this.note.noteId;
this.llmChatPanel.setCurrentNoteId(this.note.noteId);
// Reset the chat panel UI
this.llmChatPanel.noteContextChatMessages.innerHTML = '';
this.llmChatPanel.messages = [];
this.llmChatPanel.sessionId = null;
this.llmChatPanel.clearNoteContextChatMessages();
this.llmChatPanel.setMessages([]);
this.llmChatPanel.setSessionId(null);
}
// Call the parent method to refresh
await super.noteSwitched();
}
@ -141,25 +141,25 @@ export default class AiChatTypeWidget extends TypeWidget {
if (!this.isActive()) {
return;
}
console.log("Active context changed, refreshing AI Chat Panel");
// Always refresh when we become active - this ensures we load the correct note data
try {
// Reset initialization flag to force a refresh
this.isInitialized = false;
// Make sure the chat panel has the current note ID
if (this.note) {
this.llmChatPanel.currentNoteId = this.note.noteId;
this.llmChatPanel.setCurrentNoteId(this.note.noteId);
}
this.initPromise = (async () => {
try {
// Reset the UI before refreshing
this.llmChatPanel.noteContextChatMessages.innerHTML = '';
this.llmChatPanel.messages = [];
this.llmChatPanel.clearNoteContextChatMessages();
this.llmChatPanel.setMessages([]);
await this.llmChatPanel.refresh();
this.isInitialized = true;
} catch (e) {