Well the AI chat note type "kinda" works...

This commit is contained in:
perf3ct 2025-03-28 20:01:39 +00:00
parent 5456ac32ef
commit aaa3ee2697
No known key found for this signature in database
GPG Key ID: 569C4EEC436F5232
16 changed files with 192 additions and 97 deletions

View File

@ -84,6 +84,7 @@ export type CommandMappings = {
showLaunchBarSubtree: CommandData;
showRevisions: CommandData;
showLlmChat: CommandData;
createAiChat: CommandData;
showOptions: CommandData & {
section: string;
};

View File

@ -369,11 +369,6 @@ class NoteContext extends Component implements EventListener<"entitiesReloaded">
const { note, viewScope } = this;
// For llmChat viewMode, show a custom title
if (viewScope?.viewMode === "llmChat") {
return "Chat with Notes";
}
const isNormalView = (viewScope?.viewMode === "default" || viewScope?.viewMode === "contextual-help");
let title = (isNormalView ? note.title : `${note.title}: ${viewScope?.viewMode}`);

View File

@ -9,10 +9,9 @@ import froca from "../services/froca.js";
import utils from "../services/utils.js";
import LlmChatPanel from "../widgets/llm_chat_panel.js";
import toastService from "../services/toast.js";
import noteCreateService from "../services/note_create.js";
export default class RootCommandExecutor extends Component {
private llmChatPanel: any = null;
editReadOnlyNoteCommand() {
const noteContext = appContext.tabManager.getActiveContext();
if (noteContext?.viewScope) {
@ -231,22 +230,39 @@ export default class RootCommandExecutor extends Component {
}
}
async showLlmChatCommand() {
console.log("showLlmChatCommand triggered");
toastService.showMessage("Opening LLM Chat...");
async createAiChatCommand() {
try {
// We'll use the Note Map approach - open a known note ID that corresponds to the LLM chat panel
await appContext.tabManager.openTabWithNoteWithHoisting("_globalNoteMap", {
activate: true,
viewScope: {
viewMode: "llmChat" // We'll need to handle this custom view mode elsewhere
}
// Create a new AI Chat note
const parentNoteId = appContext.tabManager.getActiveContextNotePath();
if (!parentNoteId) {
toastService.showError("No active note to create AI Chat under");
return;
}
const result = await noteCreateService.createNote(parentNoteId, {
title: "New AI Chat",
type: "aiChat",
content: JSON.stringify({
messages: [],
title: "New AI Chat"
})
});
if (!result.note) {
toastService.showError("Failed to create AI Chat note");
return;
}
await appContext.tabManager.openTabWithNoteWithHoisting(result.note.noteId, {
activate: true
});
toastService.showMessage("Created new AI Chat note");
}
catch (e) {
console.error("Error opening LLM Chat:", e);
toastService.showError("Failed to open LLM Chat: " + (e as Error).message);
console.error("Error creating AI Chat note:", e);
toastService.showError("Failed to create AI Chat note: " + (e as Error).message);
}
}
}

View File

@ -28,7 +28,8 @@ const NOTE_TYPE_ICONS = {
doc: "bx bxs-file-doc",
contentWidget: "bx bxs-widget",
mindMap: "bx bx-sitemap",
geoMap: "bx bx-map-alt"
geoMap: "bx bx-map-alt",
aiChat: "bx bx-bot"
};
/**
@ -36,7 +37,7 @@ const NOTE_TYPE_ICONS = {
* end user. Those types should be used only for checking against, they are
* not for direct use.
*/
export type NoteType = "file" | "image" | "search" | "noteMap" | "launcher" | "doc" | "contentWidget" | "text" | "relationMap" | "render" | "canvas" | "mermaid" | "book" | "webView" | "code" | "mindMap" | "geoMap";
export type NoteType = "file" | "image" | "search" | "noteMap" | "launcher" | "doc" | "contentWidget" | "text" | "relationMap" | "render" | "canvas" | "mermaid" | "book" | "webView" | "code" | "mindMap" | "geoMap" | "aiChat";
export interface NotePathRecord {
isArchived: boolean;

View File

@ -3,12 +3,12 @@ import type FNote from "../../entities/fnote.js";
import options from "../../services/options.js";
import CommandButtonWidget from "./command_button.js";
export default class LlmChatButton extends CommandButtonWidget {
export default class AiChatButton extends CommandButtonWidget {
constructor(note: FNote) {
super();
this.command("showLlmChat")
this.command("createAiChat")
.title(() => note.title)
.icon(() => note.getIcon())
.class("launcher-button");
@ -23,5 +23,4 @@ export default class LlmChatButton extends CommandButtonWidget {
this.refresh();
}
}
}

View File

@ -0,0 +1,27 @@
import { t } from "../../services/i18n.js";
import options from "../../services/options.js";
import CommandButtonWidget from "./command_button.js";
export default class CreateAiChatButton extends CommandButtonWidget {
constructor() {
super();
this.icon("bx bx-bot")
.title(t("ai.create_new_ai_chat"))
.titlePlacement("bottom")
.command("createAiChat")
.class("icon-action");
}
isEnabled() {
return options.get("aiEnabled") === "true";
}
async refreshWithNote() {
if (this.isEnabled()) {
this.$widget.show();
} else {
this.$widget.hide();
}
}
}

View File

@ -13,7 +13,7 @@ import HistoryNavigationButton from "../buttons/history_navigation.js";
import QuickSearchLauncherWidget from "../quick_search_launcher.js";
import type FNote from "../../entities/fnote.js";
import type { CommandNames } from "../../components/app_context.js";
import LlmChatButton from "../buttons/llm_chat_button.js";
import AiChatButton from "../buttons/ai_chat_button.js";
interface InnerWidget extends BasicWidget {
settings?: {
@ -124,8 +124,8 @@ export default class LauncherWidget extends BasicWidget {
return new TodayLauncher(note);
case "quickSearch":
return new QuickSearchLauncherWidget(this.isHorizontalLayout);
case "llmChatLauncher":
return new LlmChatButton(note);
case "aiChatLauncher":
return new AiChatButton(note);
default:
throw new Error(`Unrecognized builtin widget ${builtinWidget} for launcher ${note.noteId} "${note.title}"`);
}

View File

@ -28,7 +28,8 @@ export const byNoteType: Record<Exclude<NoteType, "book">, string | null> = {
render: null,
search: null,
text: null,
webView: null
webView: null,
aiChat: null
};
export const byBookType: Record<ViewTypeOptions, string | null> = {

View File

@ -35,8 +35,8 @@ import GeoMapTypeWidget from "./type_widgets/geo_map.js";
import utils from "../services/utils.js";
import type { NoteType } from "../entities/fnote.js";
import type TypeWidget from "./type_widgets/type_widget.js";
import LlmChatTypeWidget from "./type_widgets/llm_chat.js";
import { MermaidTypeWidget } from "./type_widgets/mermaid.js";
import AiChatTypeWidget from "./type_widgets/ai_chat.js";
const TPL = `
<div class="note-detail">
@ -75,7 +75,7 @@ const typeWidgetClasses = {
attachmentList: AttachmentListTypeWidget,
mindMap: MindMapWidget,
geoMap: GeoMapTypeWidget,
llmChat: LlmChatTypeWidget,
aiChat: AiChatTypeWidget,
// Split type editors
mermaid: MermaidTypeWidget
@ -95,7 +95,7 @@ type ExtendedNoteType =
| "attachmentDetail"
| "attachmentList"
| "protectedSession"
| "llmChat";
| "aiChat";
export default class NoteDetailWidget extends NoteContextAwareWidget {
@ -228,9 +228,6 @@ export default class NoteDetailWidget extends NoteContextAwareWidget {
if (viewScope?.viewMode === "source") {
resultingType = "readOnlyCode";
} else if (viewScope?.viewMode === "llmChat") {
// Special handling for our LLM Chat view mode
resultingType = "llmChat"; // This will need to be added to the ExtendedNoteType
} else if (viewScope && viewScope.viewMode === "attachments") {
resultingType = viewScope.attachmentId ? "attachmentDetail" : "attachmentList";
} else if (type === "text" && (await this.noteContext?.isReadOnly())) {

View File

@ -38,6 +38,7 @@ const NOTE_TYPES: NoteTypeMapping[] = [
// Misc note types
{ type: "render", mime: "", title: t("note_types.render-note"), selectable: true },
{ type: "webView", mime: "", title: t("note_types.web-view"), selectable: true },
{ type: "aiChat", mime: "application/json", title: t("note_types.ai-chat"), selectable: true },
// Code notes
{ type: "code", mime: "text/plain", title: t("note_types.code"), selectable: true },

View File

@ -0,0 +1,99 @@
import TypeWidget from "./type_widget.js";
import LlmChatPanel from "../llm_chat_panel.js";
import { type EventData } from "../../components/app_context.js";
import type FNote from "../../entities/fnote.js";
import server from "../../services/server.js";
export default class AiChatTypeWidget extends TypeWidget {
private llmChatPanel: LlmChatPanel;
private isInitialized: boolean = false;
constructor() {
super();
this.llmChatPanel = new LlmChatPanel();
}
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;
}
async doRefresh(note: FNote | null | undefined) {
// Initialize the chat panel if not already done
if (!this.isInitialized) {
console.log("Initializing AI Chat Panel for note:", note?.noteId);
await this.llmChatPanel.refresh();
this.isInitialized = true;
}
// If this is a newly created note, we can initialize the content
if (note) {
try {
const content = await note.getContent();
// Check if content is empty
if (!content || content === '{}') {
// Initialize with empty chat history
await this.saveData({
messages: [],
title: note.title
});
}
} catch (e) {
console.error("Error initializing AI Chat note content:", e);
}
}
}
async entitiesReloadedEvent(data: EventData<"entitiesReloaded">) {
// We don't need to refresh on entities reloaded for the chat
}
async activeContextChangedEvent(data: EventData<"activeContextChanged">) {
// Only refresh when this becomes active and we're not initialized yet
if (this.isActive() && !this.isInitialized) {
await this.llmChatPanel.refresh();
this.isInitialized = true;
}
}
// Save chat data to the note
async saveData(data: any) {
if (!this.note) {
return;
}
try {
await server.put(`notes/${this.note.noteId}/content`, {
content: JSON.stringify(data, null, 2)
});
} catch (e) {
console.error("Error saving AI Chat data:", e);
}
}
// Get data from the note
async getData() {
if (!this.note) {
return null;
}
try {
const content = await this.note.getContent();
if (!content) {
return null;
}
return JSON.parse(content as string);
} catch (e) {
console.error("Error loading AI Chat data:", e);
return null;
}
}
}

View File

@ -1,51 +0,0 @@
import TypeWidget from "./type_widget.js";
import LlmChatPanel from "../llm_chat_panel.js";
import { type EventData } from "../../components/app_context.js";
import type FNote from "../../entities/fnote.js";
export default class LlmChatTypeWidget extends TypeWidget {
private llmChatPanel: LlmChatPanel;
private isInitialized: boolean = false;
constructor() {
super();
this.llmChatPanel = new LlmChatPanel();
}
static getType() {
return "llmChat";
}
doRender() {
this.$widget = $('<div class="llm-chat-widget-container" style="height: 100%;"></div>');
this.$widget.append(this.llmChatPanel.render());
return this.$widget;
}
async doRefresh(note: FNote | null | undefined) {
// Initialize only once
if (!this.isInitialized) {
console.log("Initializing LLM Chat Panel");
await this.llmChatPanel.refresh();
this.isInitialized = true;
}
}
async entitiesReloadedEvent(data: EventData<"entitiesReloaded">) {
// We don't need to refresh on entities reloaded for the chat
}
async activeContextChangedEvent(data: EventData<"activeContextChanged">) {
// Only refresh when this becomes active and we're not initialized yet
if (this.isActive() && !this.isInitialized) {
await this.llmChatPanel.refresh();
this.isInitialized = true;
}
}
// Handle data saving - we don't need to save anything
getData() {
return {};
}
}

View File

@ -1550,7 +1550,8 @@
"widget": "Widget",
"confirm-change": "It is not recommended to change note type when note content is not empty. Do you want to continue anyway?",
"geo-map": "Geo Map",
"beta-feature": "Beta"
"beta-feature": "Beta",
"ai-chat": "AI Chat"
},
"protect_note": {
"toggle-on": "Protect the note",
@ -1826,7 +1827,12 @@
"confirm_delete_embeddings": "Are you sure you want to delete all AI embeddings? This will remove all semantic search capabilities until notes are reindexed, which can take a significant amount of time.",
"empty_key_warning": "Warning: Empty API key. You need to configure your API key in settings.",
"enable_ai": "Enable AI Features",
"enhanced_context_description": "Uses semantic search to find relevant information across your notes",
"name": "AI",
"openai": "OpenAI",
"use_enhanced_context": "Use enhanced context",
"enhanced_context_description": "Provides the AI with more context from the note and its related notes for better responses",
"show_thinking": "Show thinking",
"show_thinking_description": "Show the AI's chain of thought process",
"enter_message": "Enter your message...",
"error_contacting_provider": "Error contacting AI provider. Please check your settings and internet connection.",
"error_generating_response": "Error generating AI response",
@ -1842,15 +1848,17 @@
"notes_indexed": "{{ count }} note indexed",
"notes_indexed_plural": "{{ count }} notes indexed",
"reset_embeddings": "Reset Embeddings",
"show_thinking": "Show Thinking",
"show_thinking_description": "Reveals the reasoning process used to generate responses",
"sources": "Sources",
"start_indexing": "Start Indexing",
"use_advanced_context": "Use Advanced Context",
"use_enhanced_context": "Use Enhanced Note Context",
"processing": {
"common": "Processing..."
}
"common": "Processing...",
"thinking": "Thinking...",
"loading": "Loading...",
"generating": "Generating..."
},
"create_new_ai_chat": "Create new AI Chat"
},
"switch_layout_button": {
"title_vertical": "Move editing pane to the bottom",

View File

@ -53,7 +53,7 @@ export interface HiddenSubtreeItem {
| "protectedSession"
| "calendar"
| "quickSearch"
| "llmChatLauncher";
| "aiChatLauncher";
command?: keyof typeof Command;
}
@ -64,7 +64,7 @@ enum Command {
createNoteIntoInbox,
showRecentChanges,
showOptions,
showLlmChat
createAiChat
}
/*

View File

@ -70,10 +70,10 @@ export default function buildLaunchBarConfig() {
{ id: "_lbNoteMap", title: t("hidden-subtree.note-map-title"), type: "launcher", targetNoteId: "_globalNoteMap", icon: "bx bxs-network-chart" },
{
id: "_lbLlmChat",
title: t("hidden-subtree.llm-chat-title"),
title: t("hidden-subtree.ai-chat-title"),
type: "launcher",
icon: "bx bx-bot",
builtinWidget: "llmChatLauncher",
builtinWidget: "aiChatLauncher",
attributes: [
{ type: "label", name: "desktopOnly" }
]

View File

@ -15,7 +15,8 @@ const noteTypes = [
{ type: "doc", defaultMime: "" },
{ type: "contentWidget", defaultMime: "" },
{ type: "mindMap", defaultMime: "application/json" },
{ type: "geoMap", defaultMime: "application/json" }
{ type: "geoMap", defaultMime: "application/json" },
{ type: "aiChat", defaultMime: "application/json" }
];
function getDefaultMimeForNoteType(typeName: string) {