auto-git:
[add] README.md [add] backend/libraries/punk/library.json [add] backend/libraries/punk/stage/19f1e5d2ceaab5fd1f1dc58ff07422388f156610d16dfdea2bdb35a5b9e70813--GeorgeJordac-TheVoiceOfHumanJustice.pdf [add] backend/libraries/punk/stage/85fce554ff7685f7bccb136aff5768e54b9ba8361672fe45dbce599598c4be4b--4_Strings_-_Take_Me_Away_Into_The_Night_Vocal_Radio_Mix_.mp3 [add] backend/libraries/punk/stage/e816ca61aebd84159747d248fedd6d5ff318c471c36bcc31b1ac6bf9aebcd3c1--The_Evolution_of_Cooperation_Robert_Axelrod_liber3.pdf [add] backend/local_rag.py [add] backend/rag/__init__.py [add] backend/rag/corpus_builder.py [add] backend/rag/corpus_enricher.py [add] backend/rag/index_builder.py [add] backend/rag/unified_rag.py [add] dist/assets/index-Cc0DLWqA.css [add] dist/assets/index-DKAz6gtp.js [add] dist/index.html [add] src/LibraryManager.jsx [add] wheelcheck2117/pydantic-2.11.7-py3-none-any.whl [add] wheelcheck274/pydantic-2.7.4-py3-none-any.whl [change] backend/main.py [change] backend/requirements.txt [change] backend/schemas.py [change] electron/main.cjs [change] electron/preload.cjs [change] package.json [change] run.sh [change] src/App.jsx [change] src/InterfaceSettings.jsx [change] src/colorSchemes.js [change] src/main.jsx [change] src/styles.css
This commit is contained in:
274
src/LibraryManager.jsx
Normal file
274
src/LibraryManager.jsx
Normal file
@@ -0,0 +1,274 @@
|
||||
import React, { useEffect, useState } from 'react'
|
||||
|
||||
function statusLabel(job) {
|
||||
if (!job) return null
|
||||
const progress = typeof job.progress === 'number' ? `${job.progress.toFixed(0)}%` : null
|
||||
const detail = job.detail ? ` ${job.detail}` : ''
|
||||
return `${job.type} · ${job.status}${progress ? ` · ${progress}` : ''}${detail}`
|
||||
}
|
||||
|
||||
export default function LibraryManager({
|
||||
apiBase,
|
||||
library,
|
||||
jobs,
|
||||
chatLibrarySlug,
|
||||
onRefresh,
|
||||
onToggleChatLibrary,
|
||||
onDeleted
|
||||
}) {
|
||||
const [busy, setBusy] = useState(false)
|
||||
const [isRenaming, setIsRenaming] = useState(false)
|
||||
const [renameValue, setRenameValue] = useState('')
|
||||
const [confirmDelete, setConfirmDelete] = useState(false)
|
||||
const [errorMessage, setErrorMessage] = useState('')
|
||||
|
||||
useEffect(() => {
|
||||
setIsRenaming(false)
|
||||
setRenameValue(library?.name || '')
|
||||
setConfirmDelete(false)
|
||||
setErrorMessage('')
|
||||
}, [library?.slug, library?.name])
|
||||
|
||||
async function expectOk(response) {
|
||||
if (response.ok) return response
|
||||
const detail = await response.text()
|
||||
throw new Error(detail || `HTTP ${response.status}`)
|
||||
}
|
||||
|
||||
async function runAction(fn) {
|
||||
setBusy(true)
|
||||
try {
|
||||
setErrorMessage('')
|
||||
await fn()
|
||||
setConfirmDelete(false)
|
||||
} finally {
|
||||
setBusy(false)
|
||||
await onRefresh()
|
||||
}
|
||||
}
|
||||
|
||||
async function addPaths() {
|
||||
if (!library) return
|
||||
const paths = await window.electronAPI?.pickPaths?.()
|
||||
if (!Array.isArray(paths) || paths.length === 0) return
|
||||
try {
|
||||
await runAction(async () => {
|
||||
const response = await fetch(`${apiBase}/libraries/${library.slug}/files/register`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ paths })
|
||||
})
|
||||
await expectOk(response)
|
||||
})
|
||||
} catch (error) {
|
||||
setErrorMessage(String(error?.message || error))
|
||||
}
|
||||
}
|
||||
|
||||
async function removeFile(rel) {
|
||||
if (!library) return
|
||||
try {
|
||||
await runAction(async () => {
|
||||
const response = await fetch(`${apiBase}/libraries/${library.slug}/files`, {
|
||||
method: 'DELETE',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ rel })
|
||||
})
|
||||
await expectOk(response)
|
||||
})
|
||||
} catch (error) {
|
||||
setErrorMessage(String(error?.message || error))
|
||||
}
|
||||
}
|
||||
|
||||
async function renameLibrary() {
|
||||
if (!library) return
|
||||
const name = renameValue.trim()
|
||||
if (!name || name === library.name) {
|
||||
setIsRenaming(false)
|
||||
setRenameValue(library.name || '')
|
||||
return
|
||||
}
|
||||
await runAction(async () => {
|
||||
const response = await fetch(`${apiBase}/libraries/${library.slug}`, {
|
||||
method: 'PATCH',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ name })
|
||||
})
|
||||
await expectOk(response)
|
||||
})
|
||||
setIsRenaming(false)
|
||||
}
|
||||
|
||||
async function deleteLibrary() {
|
||||
if (!library) return
|
||||
await runAction(async () => {
|
||||
const response = await fetch(`${apiBase}/libraries/${library.slug}`, { method: 'DELETE' })
|
||||
await expectOk(response)
|
||||
})
|
||||
onDeleted?.(library.slug)
|
||||
}
|
||||
|
||||
async function startJob(kind) {
|
||||
if (!library) return
|
||||
try {
|
||||
await runAction(async () => {
|
||||
const endpoint = `${apiBase}/libraries/${library.slug}/jobs/${kind}`
|
||||
const options = {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' }
|
||||
}
|
||||
if (kind === 'embed') {
|
||||
options.body = JSON.stringify({})
|
||||
}
|
||||
const response = await fetch(endpoint, options)
|
||||
await expectOk(response)
|
||||
})
|
||||
} catch (error) {
|
||||
setErrorMessage(String(error?.message || error))
|
||||
}
|
||||
}
|
||||
|
||||
if (!library) {
|
||||
return (
|
||||
<div className="placeholder-view">
|
||||
<p>Create a database, add files or folders, then build and index it for local RAG.</p>
|
||||
</div>
|
||||
)
|
||||
}
|
||||
|
||||
const activeJobs = (jobs || []).filter(job => job.slug === library.slug && (job.status === 'queued' || job.status === 'running'))
|
||||
const usingInChat = chatLibrarySlug === library.slug
|
||||
const canStartRename = () => {
|
||||
setRenameValue(library.name || '')
|
||||
setErrorMessage('')
|
||||
setIsRenaming(true)
|
||||
setConfirmDelete(false)
|
||||
}
|
||||
|
||||
return (
|
||||
<div className="library-panel">
|
||||
{isRenaming && (
|
||||
<div className="library-inline-form">
|
||||
<input
|
||||
type="text"
|
||||
className="rename-input"
|
||||
value={renameValue}
|
||||
onChange={(e) => setRenameValue(e.target.value)}
|
||||
onKeyDown={(e) => {
|
||||
if (e.key === 'Enter') {
|
||||
renameLibrary().catch((error) => setErrorMessage(String(error?.message || error)))
|
||||
} else if (e.key === 'Escape') {
|
||||
setIsRenaming(false)
|
||||
setRenameValue(library.name || '')
|
||||
}
|
||||
}}
|
||||
autoFocus
|
||||
/>
|
||||
<div className="new-db-actions">
|
||||
<button
|
||||
className="button"
|
||||
disabled={busy}
|
||||
onClick={() => renameLibrary().catch((error) => setErrorMessage(String(error?.message || error)))}
|
||||
>
|
||||
Save
|
||||
</button>
|
||||
<button
|
||||
className="button ghost"
|
||||
onClick={() => {
|
||||
setIsRenaming(false)
|
||||
setRenameValue(library.name || '')
|
||||
}}
|
||||
>
|
||||
Cancel
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{confirmDelete && (
|
||||
<div className="library-inline-form danger-zone">
|
||||
<div className="muted-copy">Delete "{library.name}"? This removes the local index and metadata for this database.</div>
|
||||
<div className="new-db-actions">
|
||||
<button
|
||||
className="button danger"
|
||||
disabled={busy}
|
||||
onClick={() => deleteLibrary().catch((error) => setErrorMessage(String(error?.message || error)))}
|
||||
>
|
||||
Confirm Delete
|
||||
</button>
|
||||
<button className="button ghost" onClick={() => setConfirmDelete(false)}>Cancel</button>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{errorMessage && <div className="form-error">{errorMessage}</div>}
|
||||
|
||||
<div className="library-toolbar">
|
||||
<button className="button" disabled={busy} onClick={addPaths}>Add Files</button>
|
||||
<button className="button" disabled={busy || !library.files?.length} onClick={() => startJob('build')}>Build Corpus</button>
|
||||
<button className="button" disabled={busy || !library.states?.has_corpus} onClick={() => startJob('enrich')}>Enrich</button>
|
||||
<button className="button" disabled={busy || !library.states?.has_corpus} onClick={() => startJob('embed')}>Index</button>
|
||||
<button className="button" onClick={canStartRename}>Rename</button>
|
||||
<button className="button" onClick={() => onToggleChatLibrary(usingInChat ? null : library.slug)}>
|
||||
{usingInChat ? 'Stop Using In Chat' : 'Use In Chat'}
|
||||
</button>
|
||||
<button
|
||||
className="button danger"
|
||||
onClick={() => {
|
||||
setConfirmDelete(true)
|
||||
setIsRenaming(false)
|
||||
setErrorMessage('')
|
||||
}}
|
||||
>
|
||||
Delete
|
||||
</button>
|
||||
</div>
|
||||
|
||||
<div className="library-states">
|
||||
<div className={`state-pill ${library.states?.has_files ? 'ready' : ''}`}>Files: {library.files?.length || 0}</div>
|
||||
<div className={`state-pill ${library.states?.has_corpus ? 'ready' : ''}`}>Corpus: {library.artifacts?.corpus_records || 0}</div>
|
||||
<div className={`state-pill ${library.states?.is_enriched ? 'ready' : ''}`}>Enriched: {library.artifacts?.enhanced_records || 0}</div>
|
||||
<div className={`state-pill ${library.states?.is_indexed ? 'ready' : ''}`}>Indexed</div>
|
||||
</div>
|
||||
|
||||
{usingInChat && (
|
||||
<div className="library-chat-note">
|
||||
This database will be queried before each chat request and its context will be appended to the prompt.
|
||||
</div>
|
||||
)}
|
||||
|
||||
{activeJobs.length > 0 && (
|
||||
<div className="library-jobs">
|
||||
{activeJobs.map(job => (
|
||||
<div key={job.id} className={`job-card ${job.status}`}>
|
||||
{statusLabel(job)}
|
||||
</div>
|
||||
))}
|
||||
</div>
|
||||
)}
|
||||
|
||||
<div className="library-files">
|
||||
<h2>Files</h2>
|
||||
{library.files?.length ? (
|
||||
<div className="library-file-list">
|
||||
{library.files.map(file => (
|
||||
<div key={file.sha256 || file.rel} className="library-file-row">
|
||||
<div className="library-file-meta">
|
||||
<div className="library-file-name">{file.name || file.path}</div>
|
||||
<div className="library-file-path">{file.path}</div>
|
||||
</div>
|
||||
<div className="library-file-actions">
|
||||
<button className="button ghost" onClick={() => window.electronAPI?.openPath?.(file.path)}>Open</button>
|
||||
<button className="button ghost" onClick={() => removeFile(file.rel)}>Remove</button>
|
||||
</div>
|
||||
</div>
|
||||
))}
|
||||
</div>
|
||||
) : (
|
||||
<p className="muted-copy">No files registered yet.</p>
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
)
|
||||
}
|
||||
Reference in New Issue
Block a user