- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 -236 -237 -238 -239 -240 -241 -242 -243 -244 -245 -246 -247 -248 -249 -250 -251 -252 -253 -254 -255 -256 -257 -258 -259 -260 -261 -262 -263 -264 -265 -266 -267 -268 -269 -270 -271 -272 -273 -274 -275 -276 -277 -278 -279 -280 -281 -282 -283 -284 -285 -286 -287 -288 -289 -290 -291 -292 -293 -294 -295 -296 -297 -298 -299 -300 -301 -302 -303 -304 -305 -306 -307 -308 -309 -310 -311 -312 -313 -314 -315 -316 -317 -318 -319 -320 -321 -322 -323 -324 -325 -326 -327 -328 -329 -330 -331 -332 -333 -334 -335 -336 -337 -338 -339 -340 -341 -342 -343 -344 -345 -346 -347 -348 -349 -350 -351 -352 -353 -354 -355 -356 -357 -358 -359 -360 -361 -362 -363 -364 -365 -366 -367 -368 -369 -370 -371 -372 -373 -374 -375 -376 -377 -378 -379 -380 -381 -382 -383 -384 -385 -386 -387 -388 -389 -390 -391 -392 -393 -394 -395 -396 -397 -398 -399 -400 -401 -402 -403 -404 -405 -406 -407 -408 -409 -410 -411 -412 -413 -414 -415 -416 -417 -418 -419 -420 -421 -422 -423 -424 -425 -426 -427 -428 -429 -430 -431 -432 -433 -434 -435 -436 -437 -438 -439 -440 -441 -442 -443 -444 -445 -446 -447 -448 -449 -450 -451 -452 -453 -454 -455 -456 -457 -458 -459 -460 -461 -462 -463 -464 -465 -466 -467 -468 -469 -470 -471 -472 -473 -474 -475 -476 -477 -478 | 1x - - - - - - -1x -1x -1x - - - - - - - - -1x -1x -1x -1x - -1x - -36x - - - -36x -36x -36x -36x -36x -36x -36x -36x -36x -36x -36x -36x - - - -2x - - - -4x - - - -2x - - - -36x -36x -36x -36x -36x - - - - - - - - -1x - - - -1x - - - -5x -5x -5x -5x - - - - - - - -3x -3x -3x -3x -3x -3x -3x -3x - - - - -146x -28x - - - - - - - -43x -43x - -43x - - -43x -4x - -43x -4x - - -4x - - -43x - -27x - -27x - - -27x -27x - - - -43x - - -282x - - -43x - -43x -262x -262x -109x - -153x - - -153x -153x - - - - -43x -133x -3661x -24x - - - - -282x -43x -1x - - -1x -1x - - - - - -13x - - -13x -13x -2x -2x -2x - - - -13x -13x -4x -2x -2x -2x - - - - -13x - - -13x - - -13x - - - - -13x -13x -13x -7x -7x - -7x - -13x - - - -8x -2x - -8x -2x - -8x -2x - -8x -8x -8x -8x - - - - -2x -2x -2x -2x - - - -292x -26x -26x - - - - -26x - - - -158x -19x -12x - - -12x - -19x -19x - - - - -21x -21x - -21x - -21x -21x - - -19x -19x - - -19x -19x - - - -19x - - -19x - - - -19x - - - - -19x - - - -120x - - - - - - - - -19x - - - - - - - - - - - - - - - - - - - - -19x - -19x -19x - - -19x - - - - - - -19x - - - - - - - - -19x - -19x -19x - -19x -19x -19x -19x -19x -17x -17x - - - -17x -17x - - -17x -4x - - -17x - - - -13x - - - - - -13x - -4x -4x - - - - -4x -6x - - -4x -4x -6x -2x - - -4x - - - - -4x - -2x - - -2x - - - - - -2x -2x -2x - - -2x - - - - - - -2x - - - - - - - -13x -9x - - - - - -13x -2x - -13x - - -19x - - - -6x - - - - -6x -12x - -6x - -21x -21x - - - - - | import { ItemView, WorkspaceLeaf, Notice } from 'obsidian';
-/// <reference lib="dom" />
-// Use global types from JSDOM setup
-type KeyboardEvent = globalThis.KeyboardEvent;
-type HTMLTextAreaElement = globalThis.HTMLTextAreaElement;
-type HTMLButtonElement = globalThis.HTMLButtonElement;
-
-const DEFAULT_VAULT_SEARCH_LIMIT = 3;
-const MAX_MESSAGE_HISTORY = 50;
-const MAX_STREAM_CHUNKS = 1000;
-import {
- PluginSettings,
- OllamaMessage,
- ChatMessage,
- OllamaTool,
- ToolCall,
- ToolResult,
-} from './types';
-import { OllamaClient } from './ollama-client';
-import { VaultIndexer } from './vault-indexer';
-import { ToolExecutor } from './tool-executor';
-import { ErrorHandler } from './error-handler';
-
-export class ChatView extends ItemView {
- private settings: PluginSettings;
- private messages: ChatMessage[] = [];
- private ollamaClient: OllamaClient;
- private vaultIndexer: VaultIndexer;
- private toolExecutor: ToolExecutor;
- private lastMessageEl: HTMLElement | null = null;
- private newChatButton: HTMLButtonElement | null = null;
- private sendButton: HTMLButtonElement | null = null;
- private inputEl: HTMLTextAreaElement | null = null;
- private chatContainer: HTMLElement | null = null;
- private sendButtonClickHandler: (() => Promise<void>) | null = null;
- private inputKeyDownHandler: ((e: KeyboardEvent) => Promise<void>) | null = null;
- private newChatButtonClickHandler: (() => void) | null = null;
- private sendButtonClickWrapper: (() => void) | null = null;
- private inputKeyDownWrapper: ((e: KeyboardEvent) => void) | null = null;
- private newChatButtonClickWrapper: (() => void) | null = null;
- private listenersAttached = false;
-
- // Getters for testing
- public getSendButtonClickHandler(): (() => Promise<void>) | null {
- return this.sendButtonClickHandler;
- }
-
- public getInputKeyDownHandler(): ((e: KeyboardEvent) => Promise<void>) | null {
- return this.inputKeyDownHandler;
- }
-
- public getNewChatButtonClickHandler(): (() => void) | null {
- return this.newChatButtonClickHandler;
- }
-
- constructor(leaf: WorkspaceLeaf, settings: PluginSettings) {
- super(leaf);
- this.settings = settings;
- this.ollamaClient = new OllamaClient(settings.ollamaUrl, settings.model);
- this.vaultIndexer = new VaultIndexer(this.app.vault);
- this.toolExecutor = new ToolExecutor(this.app.vault, this.app);
- }
-
- public updateSettings(newSettings: PluginSettings): void {
- this.settings = newSettings;
- this.ollamaClient = new OllamaClient(newSettings.ollamaUrl, newSettings.model);
- }
-
- getViewType(): string {
- return 'ollama-chat-view';
- }
-
- getDisplayText(): string {
- return 'Ollama Chat';
- }
-
- onOpen(): Promise<void> {
- this.render();
- this.removeEventListeners(); // Clean up any existing listeners before reattaching
- this.setupEventListeners();
- return Promise.resolve();
- }
-
- public onSettingsChange(newSettings: PluginSettings): void {
- this.updateSettings(newSettings);
- }
-
- onClose(): Promise<void> {
- this.ollamaClient.cancelStream();
- this.removeEventListeners();
- this.cleanupStreamingResources();
- this.lastMessageEl = null;
- this.sendButton = null;
- this.inputEl = null;
- this.chatContainer = null;
- return Promise.resolve();
- }
-
- private cleanupStreamingResources(): void {
- // Only cleanup if there's still an active streaming message
- const streamingMessage = this.messages.find((msg) => msg.isStreaming);
- Iif (streamingMessage && this.lastMessageEl && this.lastMessageEl.parentElement) {
- this.lastMessageEl.parentElement.removeChild(this.lastMessageEl);
- this.lastMessageEl = null;
- }
- }
-
- render() {
- const container =
- this.chatContainer || this.contentEl.createEl('div', { cls: 'ollama-chat-container' });
- this.chatContainer = container;
- const inputContainer =
- this.contentEl.querySelector('.ollama-input-container') ||
- this.contentEl.createEl('div', { cls: 'ollama-input-container' });
-
- if (!this.inputEl) {
- this.inputEl = inputContainer.createEl('textarea', { cls: 'ollama-input' });
- }
- if (!this.sendButton) {
- this.sendButton = inputContainer.createEl('button', {
- cls: 'ollama-send-button',
- });
- this.sendButton.textContent = 'Send';
- }
-
- if (!this.newChatButton) {
- const newChatContainer =
- this.contentEl.querySelector('.ollama-new-chat') ||
- this.contentEl.createEl('div', { cls: 'ollama-new-chat' });
- this.newChatButton = newChatContainer.createEl('button', {
- cls: 'ollama-new-chat-button',
- });
- this.newChatButton.textContent = '🔄 New Chat';
- this.newChatButton.title = 'Start a new conversation';
- }
-
- // Create immutable snapshot for rendering
- const messagesSnapshot = [...this.messages];
-
- // Only render messages that are not currently streaming
- const nonStreamingMessages = messagesSnapshot.filter((msg) => !msg.isStreaming);
-
- // Differential update: only update messages that have changed
- const existingMessages = container.querySelectorAll('.ollama-message');
-
- for (const msg of nonStreamingMessages) {
- const existingEl = container.querySelector(`.ollama-message[data-msg-id="${msg.id}"]`);
- if (existingEl) {
- existingEl.textContent = msg.content;
- } else {
- const messageEl = container.createEl('div', {
- cls: `ollama-message ${msg.role}`,
- }) as HTMLElement;
- messageEl.setAttribute('data-msg-id', msg.id);
- messageEl.textContent = msg.content;
- }
- }
-
- // Remove messages that are no longer in the array
- for (const el of Array.from(existingMessages)) {
- const id = el.getAttribute('data-msg-id');
- if (!id || !nonStreamingMessages.some((m) => m.id === id)) {
- el.remove();
- }
- }
-
- // Re-attach streaming message if it exists
- const streamingMessage = messagesSnapshot.find((msg) => msg.isStreaming);
- if (streamingMessage && this.lastMessageEl) {
- const existingStreamingEl = container.querySelector(
- `.ollama-message[data-msg-id="${streamingMessage.id}"]`
- );
- if (!existingStreamingEl) {
- container.appendChild(this.lastMessageEl);
- }
- }
- }
-
- private setupEventListeners(): void {
- Iif (!this.sendButton || !this.inputEl || this.listenersAttached) return;
-
- // Create handlers if they don't exist
- if (!this.sendButtonClickHandler) {
- this.sendButtonClickHandler = async () => {
- Iif (!this.inputEl) return;
- await this.handleUserInput(this.inputEl.value);
- this.inputEl.value = '';
- };
- }
-
- if (!this.inputKeyDownHandler) {
- this.inputKeyDownHandler = async (e: KeyboardEvent) => {
- if (!this.inputEl || e.key !== 'Enter' || e.shiftKey) return;
- e.preventDefault();
- await this.handleUserInput(this.inputEl.value);
- this.inputEl.value = '';
- };
- }
-
- // Create wrapper functions for event listeners
- this.sendButtonClickWrapper = () => {
- void this.sendButtonClickHandler?.();
- };
- this.inputKeyDownWrapper = (e: KeyboardEvent) => {
- void this.inputKeyDownHandler?.(e);
- };
- this.newChatButtonClickWrapper = () => {
- void this.newChatButtonClickHandler?.();
- };
-
- // Add event listeners using wrappers
- this.sendButton.addEventListener('click', this.sendButtonClickWrapper);
- this.inputEl.addEventListener('keydown', this.inputKeyDownWrapper);
- if (this.newChatButton) {
- if (!this.newChatButtonClickHandler) {
- this.newChatButtonClickHandler = () => this.clearConversation();
- }
- this.newChatButton.addEventListener('click', this.newChatButtonClickWrapper);
- }
- this.listenersAttached = true;
- }
-
- private removeEventListeners(): void {
- if (this.sendButton && this.sendButtonClickWrapper) {
- this.sendButton.removeEventListener('click', this.sendButtonClickWrapper);
- }
- if (this.inputEl && this.inputKeyDownWrapper) {
- this.inputEl.removeEventListener('keydown', this.inputKeyDownWrapper);
- }
- if (this.newChatButton && this.newChatButtonClickWrapper) {
- this.newChatButton.removeEventListener('click', this.newChatButtonClickWrapper);
- }
- this.sendButtonClickWrapper = null;
- this.inputKeyDownWrapper = null;
- this.newChatButtonClickWrapper = null;
- this.listenersAttached = false;
- }
-
- private clearConversation(): void {
- // Create new array to ensure immutability
- this.messages = [];
- this.lastMessageEl = null;
- this.render();
- new Notice('Conversation cleared');
- }
-
- private updateMessageById(id: string, partial: Partial<ChatMessage>): boolean {
- const index = this.messages.findIndex((m) => m.id === id);
- Iif (index < 0) return false;
- this.messages = [
- ...this.messages.slice(0, index),
- { ...this.messages[index], ...partial },
- ...this.messages.slice(index + 1),
- ];
- return true;
- }
-
- private updateLastMessage(content: string) {
- const streamingMessage = this.messages.find((msg) => msg.isStreaming);
- if (streamingMessage && !this.lastMessageEl) {
- this.lastMessageEl = this.contentEl.createEl('div', {
- cls: `ollama-message assistant`,
- }) as HTMLElement;
- this.lastMessageEl.setAttribute('data-msg-id', streamingMessage.id);
- }
- if (this.lastMessageEl) {
- this.lastMessageEl.textContent = content;
- }
- }
-
- private async handleUserInput(content: string) {
- Iif (!this.sendButton || !this.inputEl) return;
- this.sendButton.disabled = true;
-
- try {
- // Guard against empty messages
- const userMessage = content.trim();
- if (!userMessage) return;
-
- // Search vault using user message as query
- const entries = await this.vaultIndexer.searchVault(userMessage, DEFAULT_VAULT_SEARCH_LIMIT);
- let context = entries.map((entry) => `### ${entry.title}\n${entry.content}`).join('\n\n');
-
- // Cap context size to prevent prompt bloat with large vaults
- const MAX_CONTEXT_LENGTH = 4000;
- Iif (context.length > MAX_CONTEXT_LENGTH) {
- context = context.substring(0, MAX_CONTEXT_LENGTH) + '\n\n... (truncated)';
- }
-
- const systemContent = context
- ? `You are a helpful assistant.\n\nRelevant vault context:\n${context}`
- : 'You are a helpful assistant.';
- const systemMessage: OllamaMessage = {
- role: 'system',
- content: systemContent,
- };
- const userMessageWithContext: OllamaMessage = {
- role: 'user',
- content: userMessage,
- };
-
- const messages: OllamaMessage[] = [
- systemMessage,
- ...this.messages.map(
- (m) =>
- ({
- role: m.role,
- content: m.content,
- tool_calls: m.tool_calls,
- }) as OllamaMessage
- ),
- userMessageWithContext,
- ];
-
- const tools: OllamaTool[] = [
- {
- type: 'function',
- function: {
- name: 'create_file',
- description: 'Create a new file in the vault',
- parameters: {
- type: 'object' as const,
- properties: {
- path: {
- type: 'string' as const,
- description: "Relative path within the vault, e.g. 'Notes/todo.md'",
- },
- content: { type: 'string' as const, description: 'Content of the file to create' },
- },
- required: ['path', 'content'],
- },
- },
- },
- ];
-
- const messageId = `${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
-
- const userMessageId = messageId;
- const assistantMessageId = `${messageId}-assistant`;
-
- // Store user message in conversation history
- const userChatMessage: ChatMessage = {
- id: userMessageId,
- role: 'user' as const,
- content: userMessage,
- timestamp: Date.now(),
- };
-
- const assistantMessage: ChatMessage = {
- id: assistantMessageId,
- role: 'assistant' as const,
- content: '',
- timestamp: Date.now(),
- isStreaming: true,
- };
-
- // Update messages immutably
- this.messages = [...this.messages, userChatMessage, assistantMessage];
-
- try {
- this.render();
-
- const stream = this.ollamaClient.streamChat(messages, tools);
- let fullResponse = '';
- let toolCalls: ToolCall[] = [];
- let chunkCount = 0;
- for await (const chunk of stream) {
- chunkCount++;
- Iif (chunkCount > MAX_STREAM_CHUNKS) {
- throw new Error('Response too long, stopped streaming');
- }
-
- if (chunk.content) {
- fullResponse += chunk.content;
- }
-
- if (chunk.tool_calls) {
- toolCalls = toolCalls.concat(chunk.tool_calls);
- }
-
- this.updateLastMessage(fullResponse);
- }
-
- // Update the assistant message with the full response immutably
- this.updateMessageById(assistantMessageId, {
- content: fullResponse,
- tool_calls: toolCalls,
- });
-
- // Process tool calls with proper follow-up context
- if (toolCalls.length > 0) {
- // Validate tool calls before processing
- const MAX_TOOL_CALLS = 10;
- Iif (toolCalls.length > MAX_TOOL_CALLS) {
- throw new Error(`Too many tool calls (max ${MAX_TOOL_CALLS})`);
- }
-
- // Collect all tool results using allSettled to support partial results
- const settledResults = await Promise.allSettled(
- toolCalls.map((call) => this.toolExecutor.handleToolCall(call))
- );
-
- const toolResults: ToolResult[] = [];
- for (const result of settledResults) {
- if (result.status === 'fulfilled') {
- toolResults.push(result.value);
- } else {
- // Use centralized error handler for tool errors
- ErrorHandler.handleError(result.reason, 'ChatView.handleUserInput');
- }
- }
-
- // Only create follow-up when we have tool results
- if (toolResults.length > 0) {
- // Create follow-up messages including the assistant's tool calls and results
- const followUpMessages: OllamaMessage[] = [
- ...messages,
- { role: 'assistant' as const, content: fullResponse, tool_calls: toolCalls },
- ...toolResults.map((result) => ({
- role: 'tool' as const,
- content: JSON.stringify(result),
- })),
- ];
-
- const followUp = await this.ollamaClient.chat(followUpMessages, tools);
- fullResponse += followUp.content;
- this.updateLastMessage(fullResponse);
-
- // Update the assistant message with the final response immutably
- this.updateMessageById(assistantMessageId, {
- content: fullResponse,
- isStreaming: false,
- });
- } else {
- // Even if no tool results were successful, mark streaming as complete
- // to prevent the assistant message from disappearing
- this.updateMessageById(assistantMessageId, {
- content: fullResponse,
- isStreaming: false,
- });
- }
- }
-
- // Update assistant message immutably — only if no tool calls were processed
- if (toolCalls.length === 0) {
- this.updateMessageById(assistantMessageId, {
- isStreaming: false,
- });
- }
-
- // Limit conversation history to prevent memory issues
- if (this.messages.length > MAX_MESSAGE_HISTORY) {
- this.messages = this.messages.slice(-MAX_MESSAGE_HISTORY);
- }
- this.render();
- } finally {
- // Clean up streaming resources regardless of outcome
- this.cleanupStreamingResources();
- }
- } catch (error) {
- // Use centralized error handler
- ErrorHandler.handleError(error, 'ChatView.handleUserInput');
- // Update any streaming messages to non-streaming state to prevent stale messages
- // This ensures that if an error occurs during streaming, the assistant message
- // is still visible (with any partial content received) but won't cause issues
- // in subsequent requests due to stale isStreaming: true flag
- this.messages = this.messages.map((msg) =>
- msg.isStreaming ? { ...msg, isStreaming: false } : msg
- );
- this.cleanupStreamingResources();
- } finally {
- if (this.sendButton) {
- this.sendButton.disabled = false;
- }
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 | - -2x -2x - - - - - - - - - - -2x - -9x -9x - -9x -8x - -8x -8x -8x - - -1x -1x - - - - -15x -7x - - -8x -7x - - -1x - - - -7x - -2x - - - -2x - -1x - -1x - -1x - - - - - - - - -2x -1x -1x - -1x - - - -13x - - -13x -2x - - -11x -2x - - -9x -1x - - -8x -1x - - -7x -1x - - -6x -1x - - -5x - - - - - -1x - - - -1x - - - -1x -1x - - - -1x - - - -1x - - - -1x - - - -1x - - - | // src/error-handler.ts
-
-import { Notice } from 'obsidian';
-import {
- OllamaError,
- ErrorType,
- NetworkError,
- ApiError,
- ValidationError,
- StreamingError,
- ToolExecutionError,
- PathValidationError,
-} from './types';
-
-export class ErrorHandler {
- static handleError(error: unknown, context?: string): void {
- const message = this.getUserFriendlyMessage(error);
- new Notice(message);
-
- if (error instanceof Error) {
- const ctx = context ? ` [${context}]` : '';
- // Use console.error instead of ErrorHandler.error for fatal errors
- console.error(`Ollama Plugin Error${ctx}: ${error.message}`);
- if (error.stack) {
- console.error(error.stack);
- }
- } else {
- const ctx = context ? ` [${context}]` : '';
- console.error(`Ollama Plugin Error${ctx}:`, error);
- }
- }
-
- private static getUserFriendlyMessage(error: unknown): string {
- if (error instanceof OllamaError) {
- return this.getUserFriendlyMessageFromOllamaError(error);
- }
-
- if (error instanceof Error) {
- return this.getUserFriendlyMessageFromError(error);
- }
-
- return 'An unexpected error occurred';
- }
-
- private static getUserFriendlyMessageFromOllamaError(error: OllamaError): string {
- switch (error.type) {
- case ErrorType.NETWORK_ERROR:
- return 'Connection error. Please check if Ollama is running.';
- case ErrorType.API_ERROR:
- return `API error: ${error.message}`;
- case ErrorType.VALIDATION_ERROR:
- return this.getUserFriendlyValidationMessage(error);
- case ErrorType.STREAMING_ERROR:
- return 'Response too long. Please try a shorter request.';
- case ErrorType.TOOL_EXECUTION_ERROR:
- return `Tool error for ${(error as ToolExecutionError).toolName}. ${error.message}`;
- case ErrorType.PATH_VALIDATION_ERROR:
- return `Invalid file path: ${(error as PathValidationError).path}`;
- case ErrorType.UNKNOWN_ERROR:
- return 'An unexpected error occurred';
- default:
- return 'An unexpected error occurred';
- }
- }
-
- private static getUserFriendlyValidationMessage(error: OllamaError): string {
- if (error instanceof ValidationError && error.details?.field) {
- const fieldMsg = error.details.field.charAt(0).toUpperCase() + error.details.field.slice(1);
- return `Invalid ${fieldMsg.toLowerCase()}. ${error.details.message ?? error.message}`;
- }
- return 'Input validation error. Please correct your input.';
- }
-
- private static getUserFriendlyMessageFromError(error: Error): string {
- const msg = error.message.toLowerCase();
-
- // Check timeout BEFORE network (more specific matches first)
- if (msg.includes('timeout') || msg.includes('timed out') || msg.includes('time out')) {
- return 'Request timed out. Please check your Ollama connection.';
- }
-
- if (msg.includes('network') || msg.includes('connection') || msg.includes('fetch')) {
- return 'Connection error. Please check if Ollama is running.';
- }
-
- if (msg.includes('validation') || msg.includes('invalid')) {
- return 'Invalid input. Please correct your input.';
- }
-
- if (msg.includes('stream') || msg.includes('chunk')) {
- return 'Response too long. Please try a shorter request.';
- }
-
- if (msg.includes('tool') || msg.includes('function')) {
- return 'Tool error. Please try again.';
- }
-
- if (msg.includes('path') || msg.includes('file')) {
- return 'Invalid file path. Please check the path and try again.';
- }
-
- return 'An unexpected error occurred';
- }
-
- // -- Factory methods --
-
- static createNetworkError(message: string, statusCode?: number): NetworkError {
- return new NetworkError(message, statusCode);
- }
-
- static createApiError(message: string, statusCode?: number): ApiError {
- return new ApiError(message, statusCode);
- }
-
- static createValidationError(message: string, field?: string): ValidationError {
- const details = field ? { field, message } : undefined;
- return new ValidationError(message, details);
- }
-
- static createStreamingError(message: string): StreamingError {
- return new StreamingError(message);
- }
-
- static createToolExecutionError(message: string, toolName?: string): ToolExecutionError {
- return new ToolExecutionError(message, toolName ?? 'unknown');
- }
-
- static createPathValidationError(message: string, path?: string): PathValidationError {
- return new PathValidationError(message, path ?? '');
- }
-
- static createUnknownError(message: string): OllamaError {
- return new OllamaError(message, ErrorType.UNKNOWN_ERROR);
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| chat-view.ts | -
-
- |
- 92.55% | -199/215 | -85.71% | -60/70 | -82.85% | -29/35 | -94.52% | -190/201 | -
| error-handler.ts | -
-
- |
- 94.23% | -49/52 | -87.5% | -42/48 | -100% | -12/12 | -94.23% | -49/52 | -
| ollama-client.ts | -
-
- |
- 91.8% | -112/122 | -72.72% | -40/55 | -88.23% | -15/17 | -91.37% | -106/116 | -
| tool-executor.ts | -
-
- |
- 91.37% | -53/58 | -78.78% | -26/33 | -100% | -4/4 | -91.37% | -53/58 | -
| types.ts | -
-
- |
- 100% | -35/35 | -100% | -2/2 | -100% | -8/8 | -100% | -35/35 | -
| utils.ts | -
-
- |
- 95.65% | -88/92 | -84.78% | -39/46 | -93.33% | -14/15 | -96.59% | -85/88 | -
| vault-indexer.ts | -
-
- |
- 88.51% | -131/148 | -77.77% | -42/54 | -83.33% | -20/24 | -89.85% | -124/138 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 -236 -237 -238 -239 -240 -241 -242 -243 -244 -245 -246 -247 -248 -249 -250 -251 -252 -253 -254 -255 -256 -257 -258 -259 -260 -261 -262 -263 -264 -265 -266 -267 -268 -269 -270 -271 -272 -273 -274 -275 -276 -277 -278 -279 -280 -281 -282 -283 -284 -285 -286 -287 -288 | - - -2x -2x - - - - - -2x - - - -61x -61x - - -61x -61x -61x - - - -6x -2x -2x - - - - - - - -24x - - - - - - -3x -3x - - - - - - - - - - - -29x -29x - -29x -29x - - - - - - - - - - - - - -27x -8x -5x -5x - - - -5x -4x -4x - - -4x -4x -4x -4x -4x - - -4x - - - - - -4x - - - - - -5x -2x - -3x - - -15x -1x - - -14x -14x -1x - - -13x -13x -13x -13x -13x - -13x -13x -22x -22x - -11x - -11x -11x - -11x -69x - -68x -68x -15x - -14x -14x - - - -14x -14x - -54x -1x - - -53x -53x -1x - - -52x - - - - - - - -11x -1x -1x -1x - -1x -1x -1x - - - - - - - - - - - - -13x - - - -29x - - -11x - - - -5x - - - - - - - - -8x -8x -8x - - - - - - - - - - - - - -8x -4x -3x -3x - - - -3x -2x -2x - - -2x -2x -2x -2x -2x - - -2x - - - - - -2x - - - - - -3x - -1x - - -4x -4x - - - - -8x - - - - -16x -1x - - - - -19x -1x - - -18x -18x - - - - - - - | // src/ollama-client.ts
-
-import type { OllamaMessage, OllamaTool } from './types';
-import { ApiError } from './types';
-import { Logger } from './utils';
-
-interface OllamaChatResponse {
- message?: Partial<OllamaMessage>;
-}
-
-export class OllamaClient {
- private baseURL: string;
- private model: string;
- private fetchFn: typeof fetch;
- private readonly maxRetries: number = 3;
- private currentStreamController: AbortController | null = null;
-
- constructor(baseURL: string, model: string, fetchFn?: typeof fetch) {
- this.baseURL = baseURL;
- this.model = model;
- this.fetchFn = fetchFn ?? fetch;
- }
-
- cancelStream(): void {
- if (this.currentStreamController) {
- this.currentStreamController.abort();
- this.currentStreamController = null;
- }
- }
-
- async *streamChat(
- messages: OllamaMessage[],
- tools: OllamaTool[] = []
- ): AsyncGenerator<OllamaMessage, void, unknown> {
- yield* this.streamChatWithRetry(messages, tools, 0);
- }
-
- async streamChatAsPromise(
- messages: OllamaMessage[],
- tools: OllamaTool[] = []
- ): Promise<OllamaMessage[]> {
- const chunks: OllamaMessage[] = [];
- for await (const chunk of this.streamChat(messages, tools)) {
- chunks.push(chunk);
- }
- return chunks;
- }
-
- private async *streamChatWithRetry(
- messages: OllamaMessage[],
- tools: OllamaTool[] = [],
- attempt: number = 0
- ): AsyncGenerator<OllamaMessage, void, unknown> {
- // Create a local controller for this request instead of using the instance variable
- const controller = new AbortController();
- this.currentStreamController = controller;
-
- try {
- const response = await this.fetchFn(`${this.baseURL}/api/chat`, {
- method: 'POST',
- headers: {
- 'Content-Type': 'application/json',
- },
- body: JSON.stringify({
- model: this.model,
- messages,
- tools,
- stream: true,
- }),
- signal: controller.signal,
- });
-
- if (!response.ok) {
- if (response.status >= 500 && attempt < this.maxRetries) {
- const retryDelay = Math.pow(2, attempt) * 100;
- Logger.warn(
- `Network error (status ${response.status}), retrying in ${retryDelay}ms (attempt ${attempt + 1}/${this.maxRetries})`,
- 'ollama-client'
- );
- if (attempt < this.maxRetries - 1) {
- const retryTimeout = new Promise((resolve) => setTimeout(resolve, retryDelay));
- const abortListener = () => {
- Logger.info('Retry aborted by user', 'ollama-client');
- };
- const signal = controller.signal;
- if (signal) {
- signal.addEventListener('abort', abortListener);
- try {
- await Promise.race([
- retryTimeout,
- new Promise<void>((resolve) => {
- signal.addEventListener('abort', () => resolve(), {
- once: true,
- });
- }),
- ]);
- } finally {
- signal.removeEventListener('abort', abortListener);
- }
- } else E{
- await retryTimeout;
- }
- }
- yield* this.streamChatWithRetry(messages, tools, attempt + 1);
- return;
- }
- throw new ApiError(`Ollama API error: ${response.status}`, response.status);
- }
-
- if (!response.body) {
- throw new Error('No response body');
- }
-
- const contentType = response.headers.get('content-type');
- if (!contentType || (!contentType.includes('ndjson') && !contentType.includes('json'))) {
- throw new Error('Invalid response format');
- }
-
- const reader = response.body.getReader();
- const decoder = new TextDecoder();
- let buffer = '';
- let malformedCount = 0;
- const maxMalformed = 50;
-
- try {
- while (true) {
- const { done, value } = await reader.read();
- if (done) break;
-
- buffer += decoder.decode(value, { stream: true });
-
- const lines = buffer.split('\n');
- buffer = lines.pop() ?? '';
-
- for (const line of lines) {
- if (!line.trim()) continue;
-
- try {
- const parsed = JSON.parse(line) as Record<string, unknown>;
- this.throwIfOllamaError(parsed);
-
- const message = this.toOllamaMessage(parsed.message);
- Iif (!message) {
- continue;
- }
-
- malformedCount = 0;
- yield message;
- } catch (error) {
- if (error instanceof Error && error.message.startsWith('Ollama error:')) {
- throw error;
- }
-
- malformedCount++;
- if (malformedCount > maxMalformed) {
- throw new Error('Too many malformed chunks in stream');
- }
-
- Logger.warn(
- `Skipped malformed chunk: ${line.substring(0, 80)}... - ${(error as Error).message}`,
- 'ollama-client'
- );
- }
- }
- }
-
- if (buffer.trim()) {
- try {
- const parsed = JSON.parse(buffer) as Record<string, unknown>;
- this.throwIfOllamaError(parsed);
-
- const message = this.toOllamaMessage(parsed.message);
- if (message) {
- yield message;
- }
- } catch (error) {
- Iif (error instanceof Error && error.message.startsWith('Ollama error:')) {
- throw error;
- }
- Logger.warn(
- `Failed to parse final chunk: ${buffer.substring(0, 80)}...`,
- 'ollama-client'
- );
- }
- }
- } finally {
- reader.releaseLock();
- }
- } finally {
- // Abort the local controller
- controller.abort();
- }
- // Clean up the reference
- this.currentStreamController = null;
- }
-
- async chat(messages: OllamaMessage[], tools: OllamaTool[] = []): Promise<OllamaMessage> {
- return this.chatWithRetry(messages, tools, 0);
- }
-
- private async chatWithRetry(
- messages: OllamaMessage[],
- tools: OllamaTool[] = [],
- attempt: number = 0
- ): Promise<OllamaMessage> {
- // Create a local controller for this request instead of using the instance variable
- const controller = new AbortController();
- try {
- const response = await this.fetchFn(`${this.baseURL}/api/chat`, {
- method: 'POST',
- headers: {
- 'Content-Type': 'application/json',
- },
- body: JSON.stringify({
- model: this.model,
- messages,
- tools,
- stream: false,
- }),
- signal: controller.signal,
- });
-
- if (!response.ok) {
- if (response.status >= 500 && attempt < this.maxRetries) {
- const retryDelay = Math.pow(2, attempt) * 100;
- Logger.warn(
- `Network error (status ${response.status}), retrying in ${retryDelay}ms (attempt ${attempt + 1}/${this.maxRetries})`,
- 'ollama-client'
- );
- if (attempt < this.maxRetries - 1) {
- const retryTimeout = new Promise((resolve) => setTimeout(resolve, retryDelay));
- const abortListener = () => {
- Logger.info('Retry aborted by user', 'ollama-client');
- };
- const signal = controller.signal;
- if (signal) {
- signal.addEventListener('abort', abortListener);
- try {
- await Promise.race([
- retryTimeout,
- new Promise<void>((resolve) => {
- signal.addEventListener('abort', () => resolve(), {
- once: true,
- });
- }),
- ]);
- } finally {
- signal.removeEventListener('abort', abortListener);
- }
- } else E{
- await retryTimeout;
- }
- }
- return this.chatWithRetry(messages, tools, attempt + 1);
- }
- throw new ApiError(`Ollama API error: ${response.status}`, response.status);
- }
-
- const data = (await response.json()) as OllamaChatResponse;
- return (
- this.toOllamaMessage(data.message) ?? { role: 'assistant', content: '', tool_calls: [] }
- );
- } finally {
- // Abort the local controller
- controller.abort();
- }
- }
-
- private throwIfOllamaError(parsed: Record<string, unknown>): void {
- if (parsed.error) {
- throw new Error(`Ollama error: ${String(parsed.error)}`);
- }
- }
-
- private toOllamaMessage(value: unknown): OllamaMessage | null {
- if (!value || typeof value !== 'object') {
- return null;
- }
-
- const record = value as Partial<OllamaMessage>;
- return {
- role: record.role ?? 'assistant',
- content: typeof record.content === 'string' ? record.content : '',
- tool_calls: record.tool_calls ?? [],
- };
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 | - - - -2x - - -2x -2x -2x - -2x - - - - -63x -63x - - - - -23x -1x - - - -22x -1x - - - -21x -8x - - - -13x -3x - - - -10x - - - - -10x -1x - - - -9x -9x -1x - - - -8x -16x - - -16x - - - - -8x - - - -28x -28x -28x - -28x - - - - - -28x -27x -27x - -1x - -1x -1x - - - - - -27x - -26x - -1x - - -20x -20x - - - - -26x -26x - -26x -2x - - -24x -1x - - -23x -15x - - -8x -8x -7x - -1x -1x - - - - | // src/tool-executor.ts
-
-import { Vault, App } from 'obsidian';
-import type { ToolCall, ToolResult } from './types';
-import { safeParseJson } from './utils';
-
-// Disallow characters that are invalid in file paths
-const INVALID_PATH_CHARS = /[<>:"|?*~]/;
-const MAX_PATH_LENGTH = 200;
-const FORBIDDEN_DIRS = ['.obsidian', '.git'];
-
-export class ToolExecutor {
- private vault: Vault;
- private app: App;
-
- constructor(vault: Vault, app: App) {
- this.vault = vault;
- this.app = app;
- }
-
- private isSafePath(path: string): boolean {
- // Reject empty paths
- if (!path || path.trim().length === 0) {
- return false;
- }
-
- // Reject paths that are too long
- if (path.length > MAX_PATH_LENGTH) {
- return false;
- }
-
- // Reject paths with invalid characters
- if (INVALID_PATH_CHARS.test(path)) {
- return false;
- }
-
- // Reject absolute paths
- if (path.startsWith('/') || path.startsWith('\\')) {
- return false;
- }
-
- // Reject Windows drive letters (e.g., C:)
- Iif (/^[a-zA-Z]:/.test(path)) {
- return false;
- }
-
- // Reject paths containing backslashes (Windows-style path separators)
- if (path.includes('\\')) {
- return false;
- }
-
- // Reject paths that traverse to parent directories
- const normalized = path.replace(/^(\.\/)+/, '');
- if (normalized.split('/').includes('..')) {
- return false;
- }
-
- // Reject forbidden directories
- for (const dir of FORBIDDEN_DIRS) {
- Iif (normalized.startsWith(`${dir}/`) || normalized.startsWith(`${dir}\\`)) {
- return false;
- }
- Iif (normalized.includes(`/${dir}/`) || normalized.includes(`\\${dir}\\`)) {
- return false;
- }
- }
-
- return true;
- }
-
- async handleToolCall(toolCall: ToolCall): Promise<ToolResult> {
- try {
- const toolName = toolCall.function?.name;
- const rawArgs = toolCall.function?.arguments;
-
- Iif (!toolName) {
- throw new Error('Tool name is required');
- }
-
- // Parse arguments whether they're a string or object
- let parsedArgs: Record<string, unknown>;
- if (typeof rawArgs === 'string') {
- try {
- parsedArgs = safeParseJson(rawArgs) as Record<string, unknown>;
- } catch {
- throw new Error('Invalid JSON arguments');
- }
- } else if (rawArgs && typeof rawArgs === 'object') {
- parsedArgs = rawArgs;
- } else E{
- throw new Error('Arguments must be an object or JSON string');
- }
-
- // Process the tool call based on its type
- switch (toolName) {
- case 'create_file':
- return await this.handleCreateFile(parsedArgs);
- default:
- return { success: false, message: `Unknown tool: ${toolName}` };
- }
- } catch (error) {
- const errorMessage = error instanceof Error ? error.message : String(error);
- throw new Error(errorMessage);
- }
- }
-
- private async handleCreateFile(args: Record<string, unknown>): Promise<ToolResult> {
- const path = args.path;
- const content = args.content;
-
- if (typeof path !== 'string') {
- throw new Error('Path must be a string');
- }
-
- if (typeof content !== 'string') {
- throw new Error('Content must be a string');
- }
-
- if (!this.isSafePath(path)) {
- throw new Error('Invalid file path detected');
- }
-
- try {
- await this.vault.create(path, content);
- return { success: true, message: 'File created successfully' };
- } catch (error) {
- const errorMessage = error instanceof Error ? error.message : String(error);
- throw new Error(errorMessage);
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 | - - - - - -3x -3x -3x -3x -3x -3x -3x -3x - - -3x - - - -19x -19x -19x - - - -3x - - - -4x -4x -4x - - - -3x - - - -5x -5x -5x - - - - - - - - -3x - - - -3x -3x -3x - - - -3x - -2x -2x - - - -3x - - - -2x -2x -2x - - - -3x - - - -2x -2x -2x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | // src/types.ts
-
-// ============================================================
-// Error Type Hierarchy
-// ============================================================
-
-export enum ErrorType {
- NETWORK_ERROR = 'network_error',
- API_ERROR = 'api_error',
- VALIDATION_ERROR = 'validation_error',
- STREAMING_ERROR = 'streaming_error',
- TOOL_EXECUTION_ERROR = 'tool_execution_error',
- PATH_VALIDATION_ERROR = 'path_validation_error',
- UNKNOWN_ERROR = 'unknown_error',
-}
-
-export class OllamaError extends Error {
- public readonly type: ErrorType;
-
- constructor(message: string, type: ErrorType) {
- super(message);
- this.type = type;
- Object.setPrototypeOf(this, OllamaError.prototype);
- }
-}
-
-export class NetworkError extends OllamaError {
- public readonly statusCode?: number;
-
- constructor(message: string, statusCode?: number) {
- super(message, ErrorType.NETWORK_ERROR);
- this.statusCode = statusCode;
- Object.setPrototypeOf(this, NetworkError.prototype);
- }
-}
-
-export class ApiError extends OllamaError {
- public readonly statusCode?: number;
-
- constructor(message: string, statusCode?: number) {
- super(message, ErrorType.API_ERROR);
- this.statusCode = statusCode;
- Object.setPrototypeOf(this, ApiError.prototype);
- }
-}
-
-export interface ValidationFieldDetails {
- field?: string;
- message?: string;
-}
-
-export class ValidationError extends OllamaError {
- public readonly details?: ValidationFieldDetails;
-
- constructor(message: string, details?: ValidationFieldDetails) {
- super(message, ErrorType.VALIDATION_ERROR);
- this.details = details;
- Object.setPrototypeOf(this, ValidationError.prototype);
- }
-}
-
-export class StreamingError extends OllamaError {
- constructor(message: string) {
- super(message, ErrorType.STREAMING_ERROR);
- Object.setPrototypeOf(this, StreamingError.prototype);
- }
-}
-
-export class ToolExecutionError extends OllamaError {
- public readonly toolName: string;
-
- constructor(message: string, toolName: string) {
- super(message, ErrorType.TOOL_EXECUTION_ERROR);
- this.toolName = toolName;
- Object.setPrototypeOf(this, ToolExecutionError.prototype);
- }
-}
-
-export class PathValidationError extends OllamaError {
- public readonly path: string;
-
- constructor(message: string, path: string) {
- super(message, ErrorType.PATH_VALIDATION_ERROR);
- this.path = path;
- Object.setPrototypeOf(this, PathValidationError.prototype);
- }
-}
-
-// ============================================================
-// Plugin Configuration
-// ============================================================
-
-export interface PluginSettings {
- ollamaUrl: string;
- model: string;
- vaultSearchLimit: number;
- maxMessageHistory: number;
- lastIndexTime: number;
-}
-
-// ============================================================
-// Ollama Protocol Types
-// ============================================================
-
-export interface OllamaTool {
- type: 'function';
- function: {
- name: string;
- description: string;
- parameters: {
- type: 'object';
- properties: Record<string, unknown>;
- required: string[];
- };
- };
-}
-
-export interface OllamaToolCall {
- id: string;
- type: 'function';
- function: {
- name: string;
- arguments: string | Record<string, unknown>;
- };
-}
-
-export interface OllamaMessage {
- role: 'system' | 'user' | 'assistant' | 'tool';
- content: string;
- tool_calls?: OllamaToolCall[];
-}
-
-// ============================================================
-// Tool Execution Types
-// ============================================================
-
-export interface ToolCall {
- id: string;
- type: 'function';
- function: {
- name: string;
- arguments: string | Record<string, unknown>;
- };
-}
-
-export interface ToolResult {
- success: boolean;
- message: string;
-}
-
-export interface ExecutionResult {
- success: boolean;
- output: string;
-}
-
-// ============================================================
-// Chat Message Types
-// ============================================================
-
-export interface ChatMessage {
- id: string;
- role: 'user' | 'assistant';
- content: string;
- timestamp: number;
- isStreaming?: boolean;
- tool_calls?: ToolCall[];
-}
-
-// ============================================================
-// Vault Index Types
-// ============================================================
-
-export interface VaultIndexEntry {
- path: string;
- title: string;
- content: string;
- score: number;
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 | 6x -6x -6x -6x -6x - - -6x - - - - - - -6x -6x - - -23x -11x -11x - -12x - - - - -4x -1x - - - - -4x -2x - - - - -64x -63x - - - - -2x -2x - - - - - - -6x -12x -2x - - -10x - -10x -1x - - -9x -9x -6x - - -6x - -3x - - - -6x -15x -1x - - -14x - - -14x -1x - - -13x -1x - - -12x -1x - - -11x -3x - - - - - -8x - - -6x -4x - -4x -4x -2x - - -4x -4x -2x - - -4x - - - - -6x -6x - - -117x -1x - -116x - - -116x -58x -58x -83x - -58x - - -6x -41x -1x - - -40x -1x - - - -39x -39x - -2x - - - -37x -121x -59x - - -62x -183x -3x - - - -59x -84x - - - - -59x - - -37x -3x - - - -34x -1x - - -33x - - - - | export enum LogLevel {
- DEBUG = 0,
- INFO = 1,
- WARN = 2,
- ERROR = 3,
-}
-
-const SEVERITY_ORDER: Record<string, number> = {
- debug: LogLevel.DEBUG,
- info: LogLevel.INFO,
- warn: LogLevel.WARN,
- error: LogLevel.ERROR,
-};
-
-export class Logger {
- private static minLevel: LogLevel = LogLevel.DEBUG;
-
- static setLevel(level: string | LogLevel): void {
- if (typeof level === 'string') {
- const lowerLevel = level.toLowerCase();
- Logger.minLevel = SEVERITY_ORDER[lowerLevel] ?? LogLevel.DEBUG;
- } else {
- Logger.minLevel = level;
- }
- }
-
- static debug(message: string, category: string = 'general'): void {
- if (LogLevel.DEBUG >= Logger.minLevel) {
- console.debug(`[${category}] DEBUG: ${message}`);
- }
- }
-
- static info(message: string, category: string = 'general'): void {
- if (LogLevel.INFO >= Logger.minLevel) {
- console.info(`[${category}] INFO: ${message}`);
- }
- }
-
- static warn(message: string, category: string = 'general'): void {
- if (LogLevel.WARN >= Logger.minLevel) {
- console.warn(`[${category}] WARN: ${message}`);
- }
- }
-
- static error(message: string, category: string = 'general'): void {
- if (LogLevel.ERROR >= Logger.minLevel) {
- console.error(`[${category}] ERROR: ${message}`);
- }
- }
-}
-
-// ==================== URL & Model Validation ====================
-
-export function validateOllamaUrl(url: string): { valid: boolean; error?: string } {
- if (typeof url !== 'string' || !url.trim()) {
- return { valid: false, error: 'URL cannot be empty' };
- }
-
- const trimmedUrl = url.trim();
-
- if (trimmedUrl.endsWith('/')) {
- return { valid: false, error: 'URL should not end with a slash' };
- }
-
- try {
- const parsed = new URL(trimmedUrl);
- Iif (parsed.protocol !== 'http:' && parsed.protocol !== 'https:') {
- return { valid: false, error: 'Must be a valid HTTP or HTTPS URL' };
- }
- return { valid: true };
- } catch {
- return { valid: false, error: 'Must be a valid HTTP or HTTPS URL' };
- }
-}
-
-export function validateModelName(model: string): { valid: boolean; error?: string } {
- if (typeof model !== 'string') {
- return { valid: false, error: 'Model name must be a string' };
- }
-
- const trimmedModel = model.trim();
-
- // Explicit check for empty string after trimming
- if (!trimmedModel || trimmedModel.length === 0) {
- return { valid: false, error: 'Model name cannot be empty' };
- }
-
- if (trimmedModel.length < 2) {
- return { valid: false, error: 'Model name must be at least 2 characters long' };
- }
-
- if (trimmedModel.length > 100) {
- return { valid: false, error: 'Model name must be less than 100 characters long' };
- }
-
- if (!/^[a-zA-Z0-9._:-]+$/.test(trimmedModel)) {
- return {
- valid: false,
- error: 'Model name can only contain letters, numbers, dots, dashes, underscores, and colons',
- };
- }
-
- return { valid: true };
-}
-
-export function validatePluginSettings(settings: { ollamaUrl: string; model: string }): string[] {
- const errors: string[] = [];
-
- const urlValidation = validateOllamaUrl(settings.ollamaUrl);
- if (!urlValidation.valid) {
- errors.push(`Invalid Ollama URL: ${urlValidation.error}`);
- }
-
- const modelValidation = validateModelName(settings.model);
- if (!modelValidation.valid) {
- errors.push(`Invalid Model Name: ${modelValidation.error}`);
- }
-
- return errors;
-}
-
-// ==================== Safe JSON Parsing ====================
-
-const MAX_JSON_SIZE = 1_000_000;
-const MAX_JSON_NESTING = 24;
-
-function countNestingDepth(value: unknown, depth: number = 0): number {
- if (depth > MAX_JSON_NESTING) {
- return depth;
- }
- Iif (Array.isArray(value)) {
- return Math.max(...value.map((item) => countNestingDepth(item, depth + 1)), depth);
- }
- if (value !== null && typeof value === 'object') {
- const entries = Object.values(value as Record<string, unknown>);
- if (entries.length === 0) return depth;
- return Math.max(...entries.map((item) => countNestingDepth(item, depth + 1)), depth);
- }
- return depth;
-}
-
-export function safeParseJson(jsonString: string): unknown {
- if (typeof jsonString !== 'string') {
- throw new Error('Input must be a string');
- }
-
- if (jsonString.length > MAX_JSON_SIZE) {
- throw new Error('JSON input too large');
- }
-
- let parsed: unknown;
- try {
- parsed = JSON.parse(jsonString);
- } catch {
- throw new Error('Invalid JSON');
- }
-
- // Check for dangerous prototype pollution patterns in object keys only
- const checkDangerousPatterns = (obj: unknown): boolean => {
- if (typeof obj !== 'object' || obj === null) {
- return false;
- }
-
- const dangerousKeys = ['constructor', 'prototype', '__proto__'];
- if (dangerousKeys.some((key) => Object.keys(obj).includes(key))) {
- return true;
- }
-
- // Recursively check nested objects
- for (const key in obj as Record<string, unknown>) {
- Iif (checkDangerousPatterns((obj as Record<string, unknown>)[key])) {
- return true;
- }
- }
-
- return false;
- };
-
- if (checkDangerousPatterns(parsed)) {
- throw new Error('dangerous code pattern detected');
- }
-
- // Check nesting depth
- if (countNestingDepth(parsed) > MAX_JSON_NESTING) {
- throw new Error('JSON nesting too deep');
- }
-
- return parsed;
-}
-
-// ==================== Markdown Utilities ====================
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 -236 -237 -238 -239 -240 -241 -242 -243 -244 -245 -246 -247 -248 -249 -250 -251 -252 -253 -254 -255 -256 -257 -258 -259 -260 -261 -262 -263 -264 -265 -266 -267 -268 -269 -270 -271 -272 -273 -274 -275 -276 -277 -278 -279 -280 -281 -282 -283 -284 -285 -286 -287 -288 -289 -290 -291 -292 -293 -294 -295 -296 -297 -298 -299 -300 -301 -302 -303 -304 -305 -306 -307 -308 -309 -310 -311 -312 -313 -314 -315 -316 -317 -318 -319 -320 -321 -322 -323 -324 -325 -326 -327 -328 -329 -330 -331 -332 -333 -334 -335 -336 -337 -338 -339 -340 -341 -342 -343 -344 -345 -346 -347 -348 -349 -350 -351 -352 -353 -354 -355 -356 -357 -358 -359 -360 -361 -362 -363 -364 -365 -366 -367 -368 -369 -370 -371 -372 | - - -3x - - - - - - - - - - - -1x - - - -2x - - - -1x -1x - - - - - - - - - -34x - - - - - - -18x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -71x - - - -71x -71x - - - -38x -3x - - -35x - - - -35x -35x - -3x -3x - - - - -3x -1x -1x -1x - - - - - - -34x -34x -34x -34x - -34x -53x -38x - - -34x -2x -2x - - - - - - - - -34x - - - - - - - -34x -34x -34x -34x - - - -34x - - - - - -34x -34x -18x - - - -18x -18x - -63x -63x -62x -62x -62x -53x - - - - - -53x -53x -53x - - - -9x - -1x - - - -1x - - - - -18x -63x - -18x - - - - - - - -34x - - -34x - - - -112x - - - - - - - - - - - - - - - - - - - - - - - - - - -112x - - -469x - - - -69x -69x -69x - - -69x -69x -2x -2x -2x -2x -3x -3x -3x -3x -1x -1x - -2x -2x -2x - - - - - - - - -69x -69x -12x - - -69x -69x -69x - - -69x - - - - - -69x - -69x - - - - - - - -67x -67x - -67x -73x -73x -73x - -73x - - - - - -73x - - - - -1x -1x - - -73x -1x -1x - - -73x -4x -4x - - -73x -58x -58x - - -73x -56x -56x - - -73x -63x -63x - - - -67x - - - - - - - -145x -145x -144x -144x -122x - - - -72x -72x - - - -3x - - -3x - - - | // src/vault-indexer.ts
-
-import { VaultIndexEntry } from './types';
-import { Logger } from './utils';
-
-interface Cache {
- get(key: string): Promise<string | null>;
- put(key: string, value: string): Promise<void>;
- clear(): Promise<void>;
-}
-
-class InMemoryCache implements Cache {
- private store: Map<string, string>;
-
- constructor() {
- this.store = new Map();
- }
-
- get(key: string): Promise<string | null> {
- return Promise.resolve(this.store.get(key) || null);
- }
-
- put(key: string, value: string): Promise<void> {
- this.store.set(key, value);
- return Promise.resolve();
- }
-
- clear(): Promise<void> {
- this.store.clear();
- return Promise.resolve();
- }
-}
-
-class CancellationToken {
- private cancelled = false;
-
- cancel(): void {
- this.cancelled = true;
- }
-
- get isCancelled(): boolean {
- return this.cancelled;
- }
-}
-
-interface Frontmatter {
- title?: string;
- tags?: string;
-}
-
-interface VaultFile {
- basename: string;
- path: string;
-}
-
-interface VaultLike {
- getMarkdownFiles(): VaultFile[];
- read(file: VaultFile): Promise<string>;
-}
-
-interface TokenizedContent {
- tokens: string[];
- headings: string[];
- frontmatter: Frontmatter;
- firstParagraph?: string;
-}
-
-interface ScoreResult {
- score: number;
- matchedFields: string[];
-}
-
-class VaultIndexer {
- private vault: VaultLike | null = null;
- private cache?: Cache;
-
- constructor(vault: VaultLike, cache?: Cache) {
- this.vault = vault;
- this.cache = cache;
- }
-
- async searchVault(query: string, limit: number = 5): Promise<VaultIndexEntry[]> {
- if (!query || !query.trim()) {
- return [];
- }
-
- Iif (!this.vault) {
- throw new Error('Vault-like object not provided to VaultIndexer');
- }
-
- const cacheKey = `query:${query.trim()}:limit:${limit}`;
- if (this.cache) {
- let cachedResults;
- try {
- cachedResults = await this.cache.get(cacheKey);
- } catch {
- // Ignore cache retrieval errors and continue with normal processing
- cachedResults = null;
- }
- if (cachedResults) {
- try {
- const parsedResults = JSON.parse(cachedResults) as VaultIndexEntry[];
- return parsedResults.slice(0, limit);
- } catch {
- // Ignore cache parse errors and continue with normal processing
- }
- }
- }
-
- const queryTokens = this.tokenize(query.trim());
- const vault = this.vault;
- const allFiles = vault.getMarkdownFiles();
- const results = await this.processFilesInBatches(vault, allFiles, queryTokens);
-
- const filteredResults = results
- .filter((result): result is NonNullable<typeof result> => result !== null)
- .sort((a, b) => b.score - a.score)
- .slice(0, limit);
-
- if (this.cache) {
- try {
- await this.cache.put(cacheKey, JSON.stringify(filteredResults));
- } catch (error) {
- Logger.warn(
- `Failed to cache results for query "${query}": ${error instanceof Error ? error.message : String(error)}`,
- 'vault-indexer'
- );
- }
- }
-
- return filteredResults;
- }
-
- private async processFilesInBatches(
- vault: VaultLike,
- files: VaultFile[],
- queryTokens: string[]
- ): Promise<Array<VaultIndexEntry | null>> {
- const batchSize = 10;
- const results: VaultIndexEntry[] = [];
- const seenPaths = new Set<string>();
- const cancellationToken = new CancellationToken();
- // Removed unused processedCount variable
-
- // Set up a check for cancellation every 100 files
- const checkInterval = setInterval(() => {
- Iif (cancellationToken.isCancelled) {
- clearInterval(checkInterval);
- }
- }, 100);
-
- try {
- for (let i = 0; i < files.length; i += batchSize) {
- Iif (cancellationToken.isCancelled) {
- break;
- }
-
- const batch = files.slice(i, i + batchSize);
- const batchResults = await Promise.all(
- batch.map(async (file) => {
- try {
- const content = await vault.read(file);
- const tokenized = this.tokenizeContent(content);
- const scoreResult = this.calculateWeightedScore(tokenized, queryTokens, file);
- if (scoreResult.score > 0) {
- const entry: VaultIndexEntry = {
- path: file.path,
- title: file.basename.replace(/\.md$/, ''),
- content: content.substring(0, 500),
- score: scoreResult.score,
- };
- if (!seenPaths.has(entry.path)) {
- seenPaths.add(entry.path);
- return entry;
- }
- return null;
- }
- return null;
- } catch (error) {
- Logger.warn(
- `Failed to process ${file.path}: ${error instanceof Error ? error.message : String(error)}`,
- 'vault-indexer'
- );
- return null;
- }
- })
- );
-
- const validResults = batchResults.filter(
- (result): result is NonNullable<typeof result> => result !== null
- );
- results.push(...validResults);
-
- // Continue processing all files to ensure we don't miss higher-scoring results
- // even if we've already found some matches
-
- // Removed processedCount increment
- }
- } finally {
- clearInterval(checkInterval);
- }
-
- return results;
- }
-
- private tokenize(text: string): string[] {
- const stopWords = new Set([
- 'the',
- 'a',
- 'an',
- 'and',
- 'or',
- 'but',
- 'is',
- 'are',
- 'was',
- 'were',
- 'in',
- 'on',
- 'at',
- 'to',
- 'of',
- 'for',
- 'with',
- 'as',
- 'by',
- 'it',
- 'its',
- 'that',
- 'this',
- 'these',
- 'those',
- ]);
- return text
- .toLowerCase()
- .split(/\W+/)
- .filter((token) => token.length > 1 && !stopWords.has(token));
- }
-
- private tokenizeContent(content: string): TokenizedContent {
- const tokens: string[] = [];
- const headings: string[] = [];
- const frontmatter: Frontmatter = {};
- let firstParagraph: string | undefined;
-
- const frontmatterMatch = content.match(/^---(.*?)---/s);
- if (frontmatterMatch) {
- try {
- const frontmatterContent = frontmatterMatch[1];
- const lines = frontmatterContent.trim().split('\n');
- for (const line of lines) {
- const [key, ...valueParts] = line.split(':');
- Iif (!key) continue;
- const value = valueParts.join(':').trim();
- if (key.trim() === 'title') {
- if (value) {
- frontmatter.title = value;
- }
- } else if (key.trim() === 'tags') {
- if (value) {
- frontmatter.tags = value;
- }
- }
- }
- } catch {
- Logger.warn('Failed to parse frontmatter', 'vault-indexer');
- }
- }
-
- const headingMatches = content.match(/^#{1,6} (.*?)$/gm);
- if (headingMatches) {
- headings.push(...headingMatches.map((h: string) => h.replace(/^#{1,6} /, '')));
- }
-
- const paragraphMatch = content.match(/^([^#]*?)(?=#|\n\n|$)/s);
- if (paragraphMatch) {
- firstParagraph = paragraphMatch[1].trim();
- }
-
- const allText = content
- .replace(/^---.*?---/s, '')
- .replace(/^#.*?$/gm, '')
- .replace(/```.*?```/gs, '')
- .replace(/`.*?`/g, '')
- .replace(/\[.*?\]\(.*?\)/g, '');
- tokens.push(...this.tokenize(allText));
-
- return { tokens, headings, frontmatter, firstParagraph };
- }
-
- private calculateWeightedScore(
- tokenized: TokenizedContent,
- queryTokens: string[],
- file?: VaultFile
- ): ScoreResult {
- let totalScore = 0;
- const matchedTokens: Set<string> = new Set<string>();
-
- for (const queryToken of queryTokens) {
- let tokenScore = 0;
- const stemmed = this.stemToken(queryToken);
- let matched = false;
-
- Iif (
- tokenized.frontmatter?.title &&
- this.exactMatch(tokenized.frontmatter.title, queryToken)
- ) {
- tokenScore += 3;
- matched = true;
- } else if (
- file &&
- file.basename &&
- this.exactMatch(file.basename.replace(/\.md$/, ''), queryToken)
- ) {
- tokenScore += 3;
- matched = true;
- }
-
- if (tokenized.frontmatter?.tags && this.exactMatch(tokenized.frontmatter.tags, queryToken)) {
- tokenScore += 2.5;
- matched = true;
- }
-
- if (tokenized.headings.some((heading: string) => heading.toLowerCase().includes(stemmed))) {
- tokenScore += 5;
- matched = true;
- }
-
- if (tokenized.firstParagraph && tokenized.firstParagraph.toLowerCase().includes(stemmed)) {
- tokenScore += 1.5;
- matched = true;
- }
-
- if (tokenized.tokens.includes(stemmed)) {
- tokenScore += 1;
- matched = true;
- }
-
- if (matched) {
- totalScore += tokenScore;
- matchedTokens.add(queryToken);
- }
- }
-
- return {
- score: totalScore,
- matchedFields: Array.from(matchedTokens),
- };
- }
-
- private stemToken(token: string): string {
- // Improved stemmer that handles edge cases
- Iif (token.length <= 3) return token; // Don't stem very short tokens
- if (token.endsWith('s')) return token.slice(0, -1);
- Iif (token.endsWith('ed') && token.length > 4) return token.slice(0, -2); // Don't stem 3-letter words ending in ed
- if (token.endsWith('ing') && token.length > 5) return token.slice(0, -3); // Don't stem 4-letter words ending in ing
- return token;
- }
-
- private exactMatch(content: string, token: string): boolean {
- const stemmedToken = this.stemToken(token);
- return content.toLowerCase().includes(stemmedToken);
- }
-}
-
-export { VaultIndexer, Cache, InMemoryCache, CancellationToken };
-
-// Convenience method to create a VaultIndexer with an in-memory cache
-export function createVaultIndexerWithCache(vault: VaultLike): VaultIndexer {
- return new VaultIndexer(vault, new InMemoryCache());
-}
- |