1c659ce42f
Security: - Add auth + file type/size validation to upload API - Add admin auth to /api/admin/ endpoints - Add SSRF protection to scrape action - Whitelist fields in PUT /api/notes/[id] to prevent mass assignment - Protect /lab, /agents, /chat, /canvas, /notebooks routes in middleware AI provider fixes: - Add deepseek/openrouter to factory ProviderType (was silently falling back to ollama) - Fix title-suggestion.service.ts to use factory instead of hardcoded OpenAI - Fix getAIProvider→getChatProvider in memory-echo, notebook-summary, agent-executor - Fix getAIProvider→getTagsProvider in notebook-suggestion, title-suggestions, transform-markdown Functional bugs: - Fix ALLOW_REGISTRATION AND→OR logic - Fix note-editor.tsx passing stale props to useAutoTagging instead of local state - Fix stale Note.embedding type (migrated to NoteEmbedding table) - Remove hardcoded SQLite path from prisma.ts Frontend: - Add AbortController to useAutoTagging and useTitleSuggestions hooks - Add error rollback to optimistic UI in note-inline-editor - Remove stale closure over notebookId/language in useAutoTagging Cleanup: - Rename docker-compose from keepnotes→memento - Remove unused unstable_cache import from config.ts - Remove dead useUndoRedo hook - Fix TagSuggestion type (add isNewLabel, reasoning) - Remove dead AIConfig/AIProviderType types - Fix ghost-tags unused isEmpty var and as any cast - Fix note-editor titleSuggestions typed as any[] Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
90 lines
2.3 KiB
TypeScript
90 lines
2.3 KiB
TypeScript
import { useState, useEffect, useRef } from 'react'
|
|
import { useDebounce } from './use-debounce'
|
|
|
|
export interface TitleSuggestion {
|
|
title: string
|
|
confidence: number
|
|
reasoning?: string
|
|
}
|
|
|
|
interface UseTitleSuggestionsProps {
|
|
content: string
|
|
enabled?: boolean
|
|
}
|
|
|
|
export function useTitleSuggestions({ content, enabled = true }: UseTitleSuggestionsProps) {
|
|
const [suggestions, setSuggestions] = useState<TitleSuggestion[]>([])
|
|
const [isAnalyzing, setIsAnalyzing] = useState(false)
|
|
const [error, setError] = useState<string | null>(null)
|
|
const abortRef = useRef<AbortController | null>(null)
|
|
|
|
// Debounce content by 2s to avoid excessive API calls
|
|
const debouncedContent = useDebounce(content, 2000)
|
|
|
|
useEffect(() => {
|
|
if (!enabled || !debouncedContent) {
|
|
setSuggestions([])
|
|
return
|
|
}
|
|
|
|
const wordCount = debouncedContent.split(/\s+/).length
|
|
|
|
// Need at least 10 words
|
|
if (wordCount < 10) {
|
|
setSuggestions([])
|
|
return
|
|
}
|
|
|
|
// Cancel previous request
|
|
abortRef.current?.abort()
|
|
const controller = new AbortController()
|
|
abortRef.current = controller
|
|
|
|
const generateTitles = async () => {
|
|
setIsAnalyzing(true)
|
|
setError(null)
|
|
|
|
try {
|
|
const response = await fetch('/api/ai/title-suggestions', {
|
|
method: 'POST',
|
|
headers: { 'Content-Type': 'application/json' },
|
|
body: JSON.stringify({ content: debouncedContent }),
|
|
signal: controller.signal,
|
|
})
|
|
|
|
if (controller.signal.aborted) return
|
|
|
|
if (!response.ok) {
|
|
const errorData = await response.json()
|
|
throw new Error(errorData.error || 'Error generating title suggestions')
|
|
}
|
|
|
|
const data = await response.json()
|
|
setSuggestions(data.suggestions || [])
|
|
} catch (err: any) {
|
|
if (err.name === 'AbortError') return
|
|
console.error('Title suggestions error:', err)
|
|
setError('Failed to generate title suggestions')
|
|
} finally {
|
|
if (!controller.signal.aborted) {
|
|
setIsAnalyzing(false)
|
|
}
|
|
}
|
|
}
|
|
|
|
generateTitles()
|
|
}, [debouncedContent, enabled])
|
|
|
|
// Cleanup on unmount
|
|
useEffect(() => {
|
|
return () => { abortRef.current?.abort(); };
|
|
}, []);
|
|
|
|
return {
|
|
suggestions,
|
|
isAnalyzing,
|
|
error,
|
|
clearSuggestions: () => setSuggestions([])
|
|
}
|
|
}
|