zadam/trilium
1
1
Fork 0
mirror of https://github.com/zadam/trilium.git synced 2025-11-09 21:30:53 +08:00
Code Issues Projects Releases Packages Wiki Activity

chore(react/type_widgets): get ai chat widget to render

Browse source
This commit is contained in:
Elian Doran 2025-10-05 10:21:52 +03:00
parent df9d481a93
commit 7192d40e80
No known key found for this signature in database
3 changed files with 42 additions and 45 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
apps/client/src/widgets/NoteDetail.tsx
View file

@ -37,7 +37,8 @@ const TYPE_MAPPINGS: Record<ExtendedNoteType, () => Promise<{ default: TypeWidge
"render": () => import("./type_widgets/Render"),
"canvas": () => import("./type_widgets/Canvas"),
"relationMap": () => import("./type_widgets/relation_map/RelationMap"),
"noteMap": () => import("./type_widgets/NoteMap")
"noteMap": () => import("./type_widgets/NoteMap"),
"aiChat": () => import("./type_widgets/AiChat")
// TODO: finalize the record.
};

38
apps/client/src/widgets/type_widgets/AiChat.tsx Normal file
View file

@ -0,0 +1,38 @@
import { useEffect, useRef, useState } from "preact/hooks";
import { useEditorSpacedUpdate, useLegacyWidget } from "../react/hooks";
import { type TypeWidgetProps } from "./type_widget";
import LlmChatPanel from "../llm_chat";
export default function AiChat({ note, noteContext }: TypeWidgetProps) {
const dataRef = useRef<string>();
const spacedUpdate = useEditorSpacedUpdate({
note,
getData: async () => dataRef.current,
onContentChange: (newContent) => dataRef.current = newContent
});
const [ ChatWidget, llmChatPanel ] = useLegacyWidget(() => {
return new LlmChatPanel();
}, {
noteContext,
containerClassName: "ai-chat-widget-container",
containerStyle: {
height: "100%"
}
});
useEffect(() => {
llmChatPanel.setDataCallbacks(
async (data) => {
dataRef.current = data;
spacedUpdate.scheduleUpdate();
},
async () => dataRef.current
);
}, []);
useEffect(() => {
llmChatPanel.setCurrentNoteId(note.noteId);
}, [ note ]);
return ChatWidget;
}

46
apps/client/src/widgets/type_widgets_old/ai_chat.ts
View file

@ -10,48 +10,6 @@ export default class AiChatTypeWidget extends TypeWidget {
private isInitialized: boolean = false;
private initPromise: Promise<void> | null = null;
constructor() {
super();
this.llmChatPanel = new LlmChatPanel();
// Connect the data callbacks
this.llmChatPanel.setDataCallbacks(
(data) => this.saveData(data),
() => this.getData()
);
}
static getType() {
return "aiChat";
}
doRender() {
this.$widget = $('<div class="ai-chat-widget-container" style="height: 100%;"></div>');
this.$widget.append(this.llmChatPanel.render());
return this.$widget;
}
// 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.setCurrentNoteId(note.noteId);
}
}
// Continue with regular doRefresh
await this.doRefresh(note);
}
async doRefresh(note: FNote | null | undefined) {
try {
console.log("doRefresh called for note:", note?.noteId);
@ -185,10 +143,10 @@ export default class AiChatTypeWidget extends TypeWidget {
// If we have a noteId in the data, that's the AI Chat note we should save to
// This happens when the chat panel is saving its conversation
const targetNoteId = data.noteId;
// If no noteId in data, use the current note (for new chats)
const noteIdToUse = targetNoteId || this.note?.noteId;
if (!noteIdToUse) {
console.warn("Cannot save AI Chat data: no note ID available");
return;