mirror of
https://github.com/HKUDS/LightRAG.git
synced 2025-07-04 07:26:17 +00:00
feat: Added webui management, including file upload, text upload, Q&A query, graph database management (can view tags, view knowledge graph based on tags), system status (whether it is good, data storage status, model status, path),request /webui/index.html
This commit is contained in:
hyb
parent
207c0e8299
commit
3dba406644
@ -733,6 +733,8 @@ def create_app(args):
|
|||||||
azure_openai_complete_if_cache,
|
azure_openai_complete_if_cache,
|
||||||
azure_openai_embed,
|
azure_openai_embed,
|
||||||
)
|
)
|
||||||
|
if args.llm_binding_host == "openai-ollama" or args.embedding_binding == "ollama":
|
||||||
|
from lightrag.llm.openai import openai_complete_if_cache, openai_embed
|
||||||
|
|
||||||
async def openai_alike_model_complete(
|
async def openai_alike_model_complete(
|
||||||
prompt,
|
prompt,
|
||||||
@ -1380,7 +1382,16 @@ def create_app(args):
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise HTTPException(status_code=500, detail=str(e))
|
raise HTTPException(status_code=500, detail=str(e))
|
||||||
|
|
||||||
# -------------------------------------------------
|
# query all graph labels
|
||||||
|
@app.get("/graph/label/list")
|
||||||
|
async def get_graph_labels():
|
||||||
|
return await rag.get_graph_labels()
|
||||||
|
|
||||||
|
# query all graph
|
||||||
|
@app.get("/graphs")
|
||||||
|
async def get_graphs(label: str):
|
||||||
|
return await rag.get_graps(nodel_label=label, max_depth=100)
|
||||||
|
|
||||||
# Ollama compatible API endpoints
|
# Ollama compatible API endpoints
|
||||||
# -------------------------------------------------
|
# -------------------------------------------------
|
||||||
@app.get("/api/version")
|
@app.get("/api/version")
|
||||||
@ -1751,6 +1762,7 @@ def create_app(args):
|
|||||||
"working_directory": str(args.working_dir),
|
"working_directory": str(args.working_dir),
|
||||||
"input_directory": str(args.input_dir),
|
"input_directory": str(args.input_dir),
|
||||||
"indexed_files": doc_manager.indexed_files,
|
"indexed_files": doc_manager.indexed_files,
|
||||||
|
"indexed_files_count": len(doc_manager.scan_directory()),
|
||||||
"configuration": {
|
"configuration": {
|
||||||
# LLM configuration binding/host address (if applicable)/model (if applicable)
|
# LLM configuration binding/host address (if applicable)/model (if applicable)
|
||||||
"llm_binding": args.llm_binding,
|
"llm_binding": args.llm_binding,
|
||||||
@ -1761,9 +1773,22 @@ def create_app(args):
|
|||||||
"embedding_binding_host": args.embedding_binding_host,
|
"embedding_binding_host": args.embedding_binding_host,
|
||||||
"embedding_model": args.embedding_model,
|
"embedding_model": args.embedding_model,
|
||||||
"max_tokens": args.max_tokens,
|
"max_tokens": args.max_tokens,
|
||||||
|
"kv_storage": KV_STORAGE,
|
||||||
|
"doc_status_storage": DOC_STATUS_STORAGE,
|
||||||
|
"graph_storage": GRAPH_STORAGE,
|
||||||
|
"vector_storage": VECTOR_STORAGE,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
|
# webui mount /webui/index.html
|
||||||
|
app.mount(
|
||||||
|
"/webui",
|
||||||
|
StaticFiles(
|
||||||
|
directory=Path(__file__).resolve().parent / "webui" / "static", html=True
|
||||||
|
),
|
||||||
|
name="webui_static",
|
||||||
|
)
|
||||||
|
|
||||||
# Serve the static files
|
# Serve the static files
|
||||||
static_dir = Path(__file__).parent / "static"
|
static_dir = Path(__file__).parent / "static"
|
||||||
static_dir.mkdir(exist_ok=True)
|
static_dir.mkdir(exist_ok=True)
|
||||||
|
|||||||
92
lightrag/api/webui/static/css/graph.css
Normal file
92
lightrag/api/webui/static/css/graph.css
Normal file
@ -0,0 +1,92 @@
|
|||||||
|
/* css/lightrag.css */
|
||||||
|
|
||||||
|
/* 模态框样式 */
|
||||||
|
.modal {
|
||||||
|
display: none;
|
||||||
|
position: fixed;
|
||||||
|
top: 0;
|
||||||
|
left: 0;
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
background-color: rgba(0, 0, 0, 0.5);
|
||||||
|
justify-content: center;
|
||||||
|
align-items: center;
|
||||||
|
z-index: 1000;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modal-content {
|
||||||
|
background-color: var(--surface);
|
||||||
|
padding: 1.5rem;
|
||||||
|
border-radius: 8px;
|
||||||
|
width: 80%;
|
||||||
|
max-width: 1200px;
|
||||||
|
box-shadow: var(--shadow);
|
||||||
|
}
|
||||||
|
|
||||||
|
.modal-header {
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
align-items: center;
|
||||||
|
margin-bottom: 1rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modal-body {
|
||||||
|
margin-bottom: 1rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modal-footer {
|
||||||
|
text-align: right;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-close {
|
||||||
|
background: none;
|
||||||
|
border: none;
|
||||||
|
font-size: 1.5rem;
|
||||||
|
cursor: pointer;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-close:hover {
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 图谱节点样式 */
|
||||||
|
.node {
|
||||||
|
cursor: pointer;
|
||||||
|
fill: var(--primary);
|
||||||
|
stroke: var(--surface);
|
||||||
|
stroke-width: 2px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.node:hover {
|
||||||
|
fill: var(--secondary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.link {
|
||||||
|
stroke: var(--text-secondary);
|
||||||
|
stroke-width: 2px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.label {
|
||||||
|
font-size: 12px;
|
||||||
|
fill: var(--text-primary);
|
||||||
|
pointer-events: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 添加边样式 */
|
||||||
|
.link {
|
||||||
|
stroke: #999; /* 连线颜色 */
|
||||||
|
stroke-width: 2px; /* 连线粗细 */
|
||||||
|
stroke-opacity: 0.6;/* 透明度 */
|
||||||
|
}
|
||||||
|
/* 边样式 */
|
||||||
|
.link {
|
||||||
|
stroke: #999; /* 边颜色 */
|
||||||
|
stroke-width: 2px; /* 边粗细 */
|
||||||
|
stroke-opacity: 0.8; /* 边透明度 */
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 箭头颜色匹配边颜色 */
|
||||||
|
#arrow path {
|
||||||
|
fill: #999 !important; /* 覆盖默认颜色 */
|
||||||
|
}
|
||||||
390
lightrag/api/webui/static/css/lightrag.css
Normal file
390
lightrag/api/webui/static/css/lightrag.css
Normal file
@ -0,0 +1,390 @@
|
|||||||
|
|
||||||
|
* {
|
||||||
|
box-sizing: border-box;
|
||||||
|
margin: 0;
|
||||||
|
padding: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
body {
|
||||||
|
font-family: 'Inter', -apple-system, BlinkMacSystemFont, sans-serif;
|
||||||
|
line-height: 1.6;
|
||||||
|
background-color: var(--background);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 主容器 */
|
||||||
|
.app-container {
|
||||||
|
max-width: 1600px;
|
||||||
|
margin: 0 auto;
|
||||||
|
padding: 2rem;
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: 240px 1fr;
|
||||||
|
gap: 2rem;
|
||||||
|
min-height: 100vh;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 导航侧边栏 */
|
||||||
|
.nav-panel {
|
||||||
|
position: sticky;
|
||||||
|
top: 2rem;
|
||||||
|
height: fit-content;
|
||||||
|
background: var(--surface);
|
||||||
|
border-radius: 12px;
|
||||||
|
padding: 1.5rem;
|
||||||
|
box-shadow: var(--shadow);
|
||||||
|
}
|
||||||
|
|
||||||
|
.nav-title {
|
||||||
|
font-size: 1.25rem;
|
||||||
|
font-weight: 600;
|
||||||
|
color: var(--primary);
|
||||||
|
padding-bottom: 1rem;
|
||||||
|
margin-bottom: 1rem;
|
||||||
|
border-bottom: 1px solid var(--border);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 主内容区 */
|
||||||
|
.main-content {
|
||||||
|
display: grid;
|
||||||
|
gap: 1.5rem;
|
||||||
|
align-content: start;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 卡片式模块 */
|
||||||
|
.card {
|
||||||
|
background: var(--surface);
|
||||||
|
border-radius: 12px;
|
||||||
|
padding: 1.5rem;
|
||||||
|
box-shadow: var(--shadow);
|
||||||
|
transition: transform 0.2s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.card:hover {
|
||||||
|
transform: translateY(-2px);
|
||||||
|
}
|
||||||
|
|
||||||
|
.card-title {
|
||||||
|
font-size: 1.125rem;
|
||||||
|
font-weight: 600;
|
||||||
|
margin-bottom: 1rem;
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 文件上传区域 */
|
||||||
|
.file-dropzone {
|
||||||
|
border: 2px dashed var(--border);
|
||||||
|
border-radius: 8px;
|
||||||
|
padding: 2rem;
|
||||||
|
text-align: center;
|
||||||
|
transition: all 0.3s ease;
|
||||||
|
background: rgba(241, 245, 249, 0.5);
|
||||||
|
}
|
||||||
|
|
||||||
|
.file-dropzone.active {
|
||||||
|
border-color: var(--primary);
|
||||||
|
background: rgba(79, 70, 229, 0.05);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 按钮样式 */
|
||||||
|
.btn {
|
||||||
|
display: inline-flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 0.5rem;
|
||||||
|
padding: 0.75rem 1.5rem;
|
||||||
|
border: none;
|
||||||
|
border-radius: 8px;
|
||||||
|
font-weight: 500;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-primary {
|
||||||
|
background: var(--primary);
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-primary:hover {
|
||||||
|
background: #4338CA;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-secondary {
|
||||||
|
background: var(--secondary);
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-secondary:hover {
|
||||||
|
background: #059669;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 输入框样式 */
|
||||||
|
.input-field {
|
||||||
|
width: 100%;
|
||||||
|
padding: 0.875rem;
|
||||||
|
border: 1px solid var(--border);
|
||||||
|
border-radius: 8px;
|
||||||
|
font-size: 1rem;
|
||||||
|
transition: border-color 0.2s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.input-field:focus {
|
||||||
|
outline: none;
|
||||||
|
border-color: var(--primary);
|
||||||
|
box-shadow: 0 0 0 3px rgba(79, 70, 229, 0.1);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 响应式设计 */
|
||||||
|
@media (max-width: 768px) {
|
||||||
|
.app-container {
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
padding: 1rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.nav-panel {
|
||||||
|
position: static;
|
||||||
|
margin-bottom: 1.5rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 添加按钮激活状态样式 */
|
||||||
|
.btn.active {
|
||||||
|
box-shadow: inset 0 2px 4px rgba(0, 0, 0, 0.1);
|
||||||
|
transform: translateY(1px);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.toast {
|
||||||
|
position: fixed;
|
||||||
|
bottom: 20px;
|
||||||
|
right: 20px;
|
||||||
|
padding: 12px 24px;
|
||||||
|
border-radius: 8px;
|
||||||
|
background: var(--surface);
|
||||||
|
box-shadow: var(--shadow);
|
||||||
|
z-index: 1000;
|
||||||
|
}
|
||||||
|
|
||||||
|
.toast.success {
|
||||||
|
background: var(--secondary);
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
|
||||||
|
.toast.error {
|
||||||
|
background: #EF4444;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
|
||||||
|
.loading {
|
||||||
|
position: relative;
|
||||||
|
opacity: 0.6;
|
||||||
|
}
|
||||||
|
|
||||||
|
.loading::after {
|
||||||
|
content: "···";
|
||||||
|
animation: loading 1s infinite;
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes loading {
|
||||||
|
0% {
|
||||||
|
content: "·";
|
||||||
|
}
|
||||||
|
33% {
|
||||||
|
content: "··";
|
||||||
|
}
|
||||||
|
66% {
|
||||||
|
content: "···";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.text-upload-form {
|
||||||
|
display: grid;
|
||||||
|
gap: 1.5rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.form-group label {
|
||||||
|
display: block;
|
||||||
|
margin-bottom: 0.5rem;
|
||||||
|
color: var(--text-primary);
|
||||||
|
font-weight: 500;
|
||||||
|
}
|
||||||
|
|
||||||
|
.action-bar {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 1rem;
|
||||||
|
margin-top: 1rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.status-indicator {
|
||||||
|
padding: 0.5rem 1rem;
|
||||||
|
border-radius: 6px;
|
||||||
|
font-size: 0.9em;
|
||||||
|
transition: all 0.3s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.status-indicator.success {
|
||||||
|
background: var(--secondary);
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
|
||||||
|
.status-indicator.error {
|
||||||
|
background: #ef4444;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 添加以下CSS样式 */
|
||||||
|
.status-grid {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(auto-fit, minmax(280px, 1fr));
|
||||||
|
gap: 1.5rem;
|
||||||
|
margin-top: 1rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.status-card {
|
||||||
|
background: var(--surface);
|
||||||
|
border-radius: 12px;
|
||||||
|
padding: 1.5rem;
|
||||||
|
box-shadow: var(--shadow);
|
||||||
|
}
|
||||||
|
|
||||||
|
.status-header {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 0.75rem;
|
||||||
|
margin-bottom: 1rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.status-icon {
|
||||||
|
font-size: 1.5rem;
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.progress-container {
|
||||||
|
margin-top: 1rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.progress-labels {
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
margin-top: 0.5rem;
|
||||||
|
font-size: 0.875rem;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.config-list {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: max-content 1fr;
|
||||||
|
gap: 0.75rem 1.5rem;
|
||||||
|
margin-top: 1rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.config-list dt {
|
||||||
|
font-weight: 500;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.config-list dd {
|
||||||
|
margin: 0;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.directory-list {
|
||||||
|
margin-top: 1rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.directory-item {
|
||||||
|
display: flex;
|
||||||
|
gap: 0.5rem;
|
||||||
|
margin-bottom: 0.5rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.directory-label {
|
||||||
|
font-weight: 500;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
flex-shrink: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.directory-path {
|
||||||
|
color: var(--text-primary);
|
||||||
|
word-break: break-all;
|
||||||
|
}
|
||||||
|
|
||||||
|
.status-badge {
|
||||||
|
display: inline-block;
|
||||||
|
padding: 0.5rem 1rem;
|
||||||
|
border-radius: 20px;
|
||||||
|
font-size: 0.875rem;
|
||||||
|
font-weight: 500;
|
||||||
|
margin-top: 1rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.health-status .status-badge {
|
||||||
|
background: rgba(16, 185, 129, 0.1);
|
||||||
|
color: #10B981;
|
||||||
|
border: 1px solid rgba(16, 185, 129, 0.2);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 加载状态 */
|
||||||
|
.status-badge.loading {
|
||||||
|
background: rgba(79, 70, 229, 0.1);
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 错误状态 */
|
||||||
|
.status-badge.error {
|
||||||
|
background: rgba(239, 68, 68, 0.1);
|
||||||
|
color: #EF4444;
|
||||||
|
}
|
||||||
|
|
||||||
|
.label-container {
|
||||||
|
max-height: 60vh;
|
||||||
|
overflow-y: auto;
|
||||||
|
border: 1px solid var(--border);
|
||||||
|
border-radius: 8px;
|
||||||
|
padding: 1rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.label-item {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
padding: 0.75rem;
|
||||||
|
margin-bottom: 0.5rem;
|
||||||
|
background: var(--surface);
|
||||||
|
border-radius: 6px;
|
||||||
|
box-shadow: var(--shadow);
|
||||||
|
}
|
||||||
|
|
||||||
|
.label-item:hover {
|
||||||
|
transform: translateX(4px);
|
||||||
|
transition: transform 0.2s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.label-actions {
|
||||||
|
display: flex;
|
||||||
|
gap: 0.5rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.toast {
|
||||||
|
position: fixed;
|
||||||
|
bottom: 2rem;
|
||||||
|
left: 50%;
|
||||||
|
transform: translateX(-50%);
|
||||||
|
padding: 0.75rem 1.5rem;
|
||||||
|
border-radius: 999px;
|
||||||
|
font-size: 0.9rem;
|
||||||
|
z-index: 1000;
|
||||||
|
}
|
||||||
|
|
||||||
|
.toast-success {
|
||||||
|
background: var(--secondary);
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
|
||||||
|
.toast-info {
|
||||||
|
background: var(--primary);
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
|
||||||
|
.toast-error {
|
||||||
|
background: #EF4444;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
292
lightrag/api/webui/static/index.html
Normal file
292
lightrag/api/webui/static/index.html
Normal file
@ -0,0 +1,292 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="zh-CN">
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>LightRAG</title>
|
||||||
|
<style>
|
||||||
|
:root {
|
||||||
|
--primary: #4F46E5;
|
||||||
|
--secondary: #10B981;
|
||||||
|
--background: #F8FAFC;
|
||||||
|
--surface: #FFFFFF;
|
||||||
|
--text-primary: #1E293B;
|
||||||
|
--text-secondary: #64748B;
|
||||||
|
--border: #E2E8F0;
|
||||||
|
--shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1), 0 2px 4px -1px rgba(0, 0, 0, 0.06);
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
<link rel="stylesheet" type="text/css" href="css/lightrag.css">
|
||||||
|
<link rel="stylesheet" type="text/css" href="css/graph.css">
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div class="app-container">
|
||||||
|
<nav class="nav-panel">
|
||||||
|
<h2 class="nav-title">LightRAG</h2>
|
||||||
|
<div class="nav-menu">
|
||||||
|
<button class="btn btn-primary" style="width: 100%; margin-bottom: 1rem;"
|
||||||
|
onclick="showSection('documents')">
|
||||||
|
📁 DOC MANAGE
|
||||||
|
</button>
|
||||||
|
<button class="btn btn-primary" style="width: 100%; margin-bottom: 1rem;"
|
||||||
|
onclick="showSection('text-upload')">
|
||||||
|
📝 TEXT UPLOAD
|
||||||
|
</button>
|
||||||
|
|
||||||
|
<button class="btn btn-primary" style="width: 100%; margin-bottom: 1rem;" onclick="showSection('query')">
|
||||||
|
🔍 GRAPH SEARCH
|
||||||
|
</button>
|
||||||
|
<button class="btn btn-primary" style="width: 100%; margin-bottom: 1rem;"
|
||||||
|
onclick="showSection('label-management'); loadLabels()">
|
||||||
|
🏷️ GRAPH MANAGE
|
||||||
|
</button>
|
||||||
|
<button class="btn btn-secondary" style="width: 100%;" onclick="showSection('status')">
|
||||||
|
📊 SYSTEM STATUS
|
||||||
|
</button>
|
||||||
|
|
||||||
|
</div>
|
||||||
|
</nav>
|
||||||
|
|
||||||
|
<main class="main-content">
|
||||||
|
<section id="documents" class="card">
|
||||||
|
<h3 class="card-title">DOC MANAGE</h3>
|
||||||
|
<div class="file-dropzone" id="dropzone">
|
||||||
|
<input type="file" id="fileInput" hidden multiple>
|
||||||
|
<p>Drap file</p>
|
||||||
|
<button class="btn btn-primary" onclick="document.getElementById('fileInput').click()">
|
||||||
|
📤 select file
|
||||||
|
</button>
|
||||||
|
<div class="file-list" style="margin-top: 1rem;"></div>
|
||||||
|
<div class="status-indicator" id="fileUploadStatus"></div>
|
||||||
|
</div>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
<section id="text-upload" class="card" style="display: none;">
|
||||||
|
<h3 class="card-title">TEXT UPLOAD</h3>
|
||||||
|
<div class="text-upload-form">
|
||||||
|
<div class="form-group">
|
||||||
|
<label for="textDescription">Title（optional）</label>
|
||||||
|
<input
|
||||||
|
type="text"
|
||||||
|
id="textDescription"
|
||||||
|
class="input-field"
|
||||||
|
placeholder="Please enter a text title..."
|
||||||
|
>
|
||||||
|
</div>
|
||||||
|
<div class="form-group">
|
||||||
|
<label for="textContent">TEXT CONTENT</label>
|
||||||
|
<textarea
|
||||||
|
id="textContent"
|
||||||
|
class="input-field"
|
||||||
|
placeholder="lease enter a text content..."
|
||||||
|
rows="8"
|
||||||
|
required
|
||||||
|
></textarea>
|
||||||
|
</div>
|
||||||
|
<div class="action-bar">
|
||||||
|
<button class="btn btn-primary" onclick="handleTextUpload()">
|
||||||
|
📤 submit
|
||||||
|
</button>
|
||||||
|
<div class="status-indicator" id="textUploadStatus"></div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
<section id="label-management" class="card" style="display: none;">
|
||||||
|
<h3 class="card-title">GRAPH MANAGE(neo4j)</h3>
|
||||||
|
<div class="action-bar" style="margin-bottom: 1rem;">
|
||||||
|
<button class="btn btn-secondary" onclick="refreshLabels()">
|
||||||
|
🔄 refresh
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
<div id="label-list" class="label-container">
|
||||||
|
</div>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
<!-- search -->
|
||||||
|
<section id="query" class="card" style="display: none;">
|
||||||
|
<h3 class="card-title">SEARCH API</h3>
|
||||||
|
<select class="input-field" style="margin-bottom: 1rem;">
|
||||||
|
<option value="local">local search</option>
|
||||||
|
<option value="global">global search</option>
|
||||||
|
<option value="native">native search</option>
|
||||||
|
<option value="hybrid">hybrid search</option>
|
||||||
|
<option value="mix">mix search</option>
|
||||||
|
</select>
|
||||||
|
<textarea
|
||||||
|
class="input-field"
|
||||||
|
placeholder="enter your content..."
|
||||||
|
style="height: 120px; resize: vertical;"
|
||||||
|
></textarea>
|
||||||
|
<div style="margin-top: 1rem; display: flex; gap: 0.5rem;">
|
||||||
|
<button class="btn btn-primary">🔍 searching</button>
|
||||||
|
<label class="btn btn-secondary">
|
||||||
|
<input type="checkbox" style="margin-right: 0.5rem;">stream modal
|
||||||
|
</label>
|
||||||
|
</div>
|
||||||
|
<div class="results"
|
||||||
|
style="margin-top: 1rem; padding: 1rem; border: 1px solid var(--border); border-radius: 8px;"></div>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
|
||||||
|
<section id="status" class="card" style="display: none;">
|
||||||
|
<h3 class="card-title">System Status</h3>
|
||||||
|
<div class="status-grid">
|
||||||
|
<div class="status-card health-status">
|
||||||
|
<div class="status-header">
|
||||||
|
<i class="status-icon fas fa-heartbeat"></i>
|
||||||
|
<h4>HEATH STATUS</h4>
|
||||||
|
</div>
|
||||||
|
<div class="status-badge" id="healthStatus">checking...</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- 存储状态 -->
|
||||||
|
<div class="status-card storage-status">
|
||||||
|
<div class="status-header">
|
||||||
|
<i class="status-icon fas fa-database"></i>
|
||||||
|
<h4>Storage status</h4>
|
||||||
|
</div>
|
||||||
|
<div class="progress-container">
|
||||||
|
<progress value="0" max="100" id="storageProgress"></progress>
|
||||||
|
<div class="progress-labels">
|
||||||
|
<span id="indexedFiles">Indexed files：0</span>
|
||||||
|
<span id="storageUsage">Use ratio：0%</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="status-card model-config">
|
||||||
|
<div class="status-header">
|
||||||
|
<i class="status-icon fas fa-brain"></i>
|
||||||
|
<h4>LLM configuration</h4>
|
||||||
|
</div>
|
||||||
|
<dl class="config-list">
|
||||||
|
<dt>LLM MODEL</dt>
|
||||||
|
<dd id="llmModel">Loading...</dd>
|
||||||
|
|
||||||
|
<dt>Embedded model</dt>
|
||||||
|
<dd id="embedModel">Loading...</dd>
|
||||||
|
|
||||||
|
<dt>Max token</dt>
|
||||||
|
<dd id="maxTokens">0</dd>
|
||||||
|
</dl>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="status-card directory-info">
|
||||||
|
<div class="status-header">
|
||||||
|
<i class="status-icon fas fa-folder-open"></i>
|
||||||
|
<h4>Catalog information</h4>
|
||||||
|
</div>
|
||||||
|
<div class="directory-list">
|
||||||
|
<div class="directory-item">
|
||||||
|
<span class="directory-label">Work directory：</span>
|
||||||
|
<span class="directory-path" id="workingDir"></span>
|
||||||
|
</div>
|
||||||
|
<div class="directory-item">
|
||||||
|
<span class="directory-label">Input directory：</span>
|
||||||
|
<span class="directory-path" id="inputDir"></span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="status-card directory-info">
|
||||||
|
<div class="status-header">
|
||||||
|
<i class="status-icon fas fa-folder-open"></i>
|
||||||
|
<h4>Store information</h4>
|
||||||
|
</div>
|
||||||
|
<div class="directory-list">
|
||||||
|
<div class="directory-item">
|
||||||
|
<span class="directory-label">kv_storage：</span>
|
||||||
|
<span class="directory-path" id="kv_storage"></span>
|
||||||
|
</div>
|
||||||
|
<div class="directory-item">
|
||||||
|
<span class="directory-label">doc_status_storage：</span>
|
||||||
|
<span class="directory-path" id="doc_status_storage"></span>
|
||||||
|
</div>
|
||||||
|
<div class="directory-item">
|
||||||
|
<span class="directory-label">graph_storage：</span>
|
||||||
|
<span class="directory-path" id="graph_storage"></span>
|
||||||
|
</div>
|
||||||
|
<div class="directory-item">
|
||||||
|
<span class="directory-label">vector_storage：</span>
|
||||||
|
<span class="directory-path" id="vector_storage"></span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
</div>
|
||||||
|
</section>
|
||||||
|
</main>
|
||||||
|
</div>
|
||||||
|
<div id="graph-modal" class="modal">
|
||||||
|
<div class="modal-content">
|
||||||
|
<div class="modal-header">
|
||||||
|
<h3 id="graph-title">GRAPH MANAGE</h3>
|
||||||
|
<button class="btn btn-close" onclick="closeGraphModal()">×</button>
|
||||||
|
</div>
|
||||||
|
<div class="modal-body">
|
||||||
|
<div id="graph-container">
|
||||||
|
<svg id="graph-svg" width="100%" height="600px">
|
||||||
|
</svg>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div class="modal-footer">
|
||||||
|
<button class="btn btn-secondary" onclick="closeGraphModal()">SHUT DOWN</button>
|
||||||
|
</div>
|
||||||
|
<div class="resize-handle"></div>
|
||||||
|
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<script>
|
||||||
|
const dropzone = document.getElementById('dropzone');
|
||||||
|
|
||||||
|
dropzone.addEventListener('dragover', (e) => {
|
||||||
|
e.preventDefault();
|
||||||
|
dropzone.classList.add('active');
|
||||||
|
});
|
||||||
|
|
||||||
|
dropzone.addEventListener('dragleave', () => {
|
||||||
|
dropzone.classList.remove('active');
|
||||||
|
});
|
||||||
|
|
||||||
|
dropzone.addEventListener('drop', (e) => {
|
||||||
|
e.preventDefault();
|
||||||
|
dropzone.classList.remove('active');
|
||||||
|
const files = e.dataTransfer.files;
|
||||||
|
handleFiles(files);
|
||||||
|
});
|
||||||
|
|
||||||
|
function handleFiles(files) {
|
||||||
|
console.log('selected file:', files);
|
||||||
|
}
|
||||||
|
|
||||||
|
function showSection(sectionId) {
|
||||||
|
// 隐藏所有内容区块 Hide all content blocks
|
||||||
|
const sections = document.querySelectorAll('.main-content .card');
|
||||||
|
sections.forEach(section => {
|
||||||
|
section.style.display = 'none';
|
||||||
|
});
|
||||||
|
|
||||||
|
// 显示目标区块 Display target block
|
||||||
|
const targetSection = document.getElementById(sectionId);
|
||||||
|
if (targetSection) {
|
||||||
|
targetSection.style.display = 'block';
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新按钮激活状态（可选）Update button activation status (optional)
|
||||||
|
const buttons = document.querySelectorAll('.nav-menu button');
|
||||||
|
buttons.forEach(button => {
|
||||||
|
button.classList.remove('active');
|
||||||
|
});
|
||||||
|
event.target.classList.add('active');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 初始化显示文档管理模块 Initialize the display document management module
|
||||||
|
showSection('documents');
|
||||||
|
</script>
|
||||||
|
<!-- 在页面底部添加 -->
|
||||||
|
<script src="js/lightrag.js"></script>
|
||||||
|
<script src="js/graph.js"></script>
|
||||||
|
<script src="https://d3js.org/d3.v7.min.js"></script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
214
lightrag/api/webui/static/js/graph.js
Normal file
214
lightrag/api/webui/static/js/graph.js
Normal file
@ -0,0 +1,214 @@
|
|||||||
|
// js/graph.js
|
||||||
|
function openGraphModal(label) {
|
||||||
|
const modal = document.getElementById("graph-modal");
|
||||||
|
const graphTitle = document.getElementById("graph-title");
|
||||||
|
|
||||||
|
if (!modal || !graphTitle) {
|
||||||
|
console.error("Key element not found");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
graphTitle.textContent = `Knowledge Graph - ${label}`;
|
||||||
|
modal.style.display = "flex";
|
||||||
|
|
||||||
|
renderGraph(label);
|
||||||
|
}
|
||||||
|
|
||||||
|
function closeGraphModal() {
|
||||||
|
const modal = document.getElementById("graph-modal");
|
||||||
|
modal.style.display = "none";
|
||||||
|
clearGraph();
|
||||||
|
}
|
||||||
|
|
||||||
|
function clearGraph() {
|
||||||
|
const svg = document.getElementById("graph-svg");
|
||||||
|
svg.innerHTML = "";
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
async function getGraph(label) {
|
||||||
|
try {
|
||||||
|
const response = await fetch(`/graphs?label=${label}`);
|
||||||
|
const rawData = await response.json();
|
||||||
|
console.log({data: JSON.parse(JSON.stringify(rawData))});
|
||||||
|
|
||||||
|
const nodes = rawData.nodes
|
||||||
|
|
||||||
|
nodes.forEach(node => {
|
||||||
|
node.id = Date.now().toString(36) + Math.random().toString(36).substring(2); // 使用 crypto.randomUUID() 生成唯一 UUID
|
||||||
|
});
|
||||||
|
|
||||||
|
// 严格验证边数据 Strictly verify edge data
|
||||||
|
const edges = (rawData.edges || []).map(edge => {
|
||||||
|
const sourceNode = nodes.find(n => n.labels.includes(edge.source));
|
||||||
|
const targetNode = nodes.find(n => n.labels.includes(edge.target)
|
||||||
|
)
|
||||||
|
;
|
||||||
|
if (!sourceNode || !targetNode) {
|
||||||
|
console.warn("NOT VALID EDGE:", edge);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
return {
|
||||||
|
source: sourceNode,
|
||||||
|
target: targetNode,
|
||||||
|
type: edge.type || ""
|
||||||
|
};
|
||||||
|
}).filter(edge => edge !== null);
|
||||||
|
|
||||||
|
return {nodes, edges};
|
||||||
|
} catch (error) {
|
||||||
|
console.error("Loading graph failed:", error);
|
||||||
|
return {nodes: [], edges: []};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function renderGraph(label) {
|
||||||
|
const data = await getGraph(label);
|
||||||
|
|
||||||
|
|
||||||
|
if (!data.nodes || data.nodes.length === 0) {
|
||||||
|
d3.select("#graph-svg")
|
||||||
|
.html(`<text x="50%" y="50%" text-anchor="middle">No valid nodes</text>`);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
const svg = d3.select("#graph-svg");
|
||||||
|
const width = svg.node().clientWidth;
|
||||||
|
const height = svg.node().clientHeight;
|
||||||
|
|
||||||
|
//
|
||||||
|
svg.selectAll("*").remove();
|
||||||
|
|
||||||
|
// 创建力导向图布局 Create a force oriented diagram layout
|
||||||
|
const simulation = d3.forceSimulation(data.nodes)
|
||||||
|
.force("charge", d3.forceManyBody().strength(-300))
|
||||||
|
.force("center", d3.forceCenter(width / 2, height / 2));
|
||||||
|
|
||||||
|
// 添加连线（如果存在有效边） Add a connection (if there are valid edges)
|
||||||
|
if (data.edges.length > 0) {
|
||||||
|
simulation.force("link",
|
||||||
|
d3.forceLink(data.edges)
|
||||||
|
.id(d => d.id)
|
||||||
|
.distance(100)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
// 绘制节点 Draw nodes
|
||||||
|
const nodes = svg.selectAll(".node")
|
||||||
|
.data(data.nodes)
|
||||||
|
.enter()
|
||||||
|
.append("circle")
|
||||||
|
.attr("class", "node")
|
||||||
|
.attr("r", 10)
|
||||||
|
.call(d3.drag()
|
||||||
|
.on("start", dragStarted)
|
||||||
|
.on("drag", dragged)
|
||||||
|
.on("end", dragEnded)
|
||||||
|
);
|
||||||
|
|
||||||
|
|
||||||
|
svg.append("defs")
|
||||||
|
.append("marker")
|
||||||
|
.attr("id", "arrow-out")
|
||||||
|
.attr("viewBox", "0 0 10 10")
|
||||||
|
.attr("refX", 8)
|
||||||
|
.attr("refY", 5)
|
||||||
|
.attr("markerWidth", 6)
|
||||||
|
.attr("markerHeight", 6)
|
||||||
|
.attr("orient", "auto")
|
||||||
|
.append("path")
|
||||||
|
.attr("d", "M0,0 L10,5 L0,10 Z")
|
||||||
|
.attr("fill", "#999");
|
||||||
|
|
||||||
|
// 绘制边（带箭头） Draw edges (with arrows)
|
||||||
|
const links = svg.selectAll(".link")
|
||||||
|
.data(data.edges)
|
||||||
|
.enter()
|
||||||
|
.append("line")
|
||||||
|
.attr("class", "link")
|
||||||
|
.attr("marker-end", "url(#arrow-out)"); // 始终在 target 侧绘制箭头 Always draw arrows on the target side
|
||||||
|
|
||||||
|
// 边样式配置 Edge style configuration
|
||||||
|
links
|
||||||
|
.attr("stroke", "#999")
|
||||||
|
.attr("stroke-width", 2)
|
||||||
|
.attr("stroke-opacity", 0.8);
|
||||||
|
|
||||||
|
// 绘制标签（带背景框） Draw label (with background box)
|
||||||
|
const labels = svg.selectAll(".label")
|
||||||
|
.data(data.nodes)
|
||||||
|
.enter()
|
||||||
|
.append("text")
|
||||||
|
.attr("class", "label")
|
||||||
|
.text(d => d.labels[0] || "")
|
||||||
|
.attr("text-anchor", "start")
|
||||||
|
.attr("dy", "0.3em")
|
||||||
|
.attr("fill", "#333");
|
||||||
|
|
||||||
|
// 更新位置 Update Location
|
||||||
|
simulation.on("tick", () => {
|
||||||
|
links
|
||||||
|
.attr("x1", d => {
|
||||||
|
// 计算源节点到目标节点的方向向量 Calculate the direction vector from the source node to the target node
|
||||||
|
const dx = d.target.x - d.source.x;
|
||||||
|
const dy = d.target.y - d.source.y;
|
||||||
|
const distance = Math.sqrt(dx * dx + dy * dy);
|
||||||
|
if (distance === 0) return d.source.x; // 避免除以零 Avoid dividing by zero
|
||||||
|
// 根据半径 10 调整起点坐标（源节点边缘）Adjust the starting point coordinates (source node edge) based on radius 10
|
||||||
|
return d.source.x + (dx / distance) * 10;
|
||||||
|
})
|
||||||
|
.attr("y1", d => {
|
||||||
|
const dx = d.target.x - d.source.x;
|
||||||
|
const dy = d.target.y - d.source.y;
|
||||||
|
const distance = Math.sqrt(dx * dx + dy * dy);
|
||||||
|
if (distance === 0) return d.source.y;
|
||||||
|
return d.source.y + (dy / distance) * 10;
|
||||||
|
})
|
||||||
|
.attr("x2", d => {
|
||||||
|
// 根据半径 10 调整终点坐标（目标节点边缘） Adjust the endpoint coordinates (target node edge) based on a radius of 10
|
||||||
|
const dx = d.target.x - d.source.x;
|
||||||
|
const dy = d.target.y - d.source.y;
|
||||||
|
const distance = Math.sqrt(dx * dx + dy * dy);
|
||||||
|
if (distance === 0) return d.target.x;
|
||||||
|
return d.target.x - (dx / distance) * 10;
|
||||||
|
})
|
||||||
|
.attr("y2", d => {
|
||||||
|
const dx = d.target.x - d.source.x;
|
||||||
|
const dy = d.target.y - d.source.y;
|
||||||
|
const distance = Math.sqrt(dx * dx + dy * dy);
|
||||||
|
if (distance === 0) return d.target.y;
|
||||||
|
return d.target.y - (dy / distance) * 10;
|
||||||
|
});
|
||||||
|
|
||||||
|
// 更新节点和标签的位置（保持不变） Update the position of nodes and labels (keep unchanged)
|
||||||
|
nodes
|
||||||
|
.attr("cx", d => d.x)
|
||||||
|
.attr("cy", d => d.y);
|
||||||
|
|
||||||
|
labels
|
||||||
|
.attr("x", d => d.x + 12)
|
||||||
|
.attr("y", d => d.y + 4);
|
||||||
|
});
|
||||||
|
|
||||||
|
// 拖拽逻辑 Drag and drop logic
|
||||||
|
function dragStarted(event, d) {
|
||||||
|
if (!event.active) simulation.alphaTarget(0.3).restart();
|
||||||
|
d.fx = d.x;
|
||||||
|
d.fy = d.y;
|
||||||
|
}
|
||||||
|
|
||||||
|
function dragged(event, d) {
|
||||||
|
d.fx = event.x;
|
||||||
|
d.fy = event.y;
|
||||||
|
simulation.alpha(0.3).restart();
|
||||||
|
}
|
||||||
|
|
||||||
|
function dragEnded(event, d) {
|
||||||
|
if (!event.active) simulation.alphaTarget(0);
|
||||||
|
d.fx = null;
|
||||||
|
d.fy = null;
|
||||||
|
}
|
||||||
|
}
|
||||||
503
lightrag/api/webui/static/js/lightrag.js
Normal file
503
lightrag/api/webui/static/js/lightrag.js
Normal file
@ -0,0 +1,503 @@
|
|||||||
|
// lightrag.js
|
||||||
|
const API_BASE = 'http://localhost:9621'; // 根据实际API地址修改
|
||||||
|
|
||||||
|
// init
|
||||||
|
function initializeApp() {
|
||||||
|
setupFileUpload();
|
||||||
|
setupQueryHandler();
|
||||||
|
setupSectionObserver();
|
||||||
|
updateFileList();
|
||||||
|
// 文本输入框实时字数统计 textarea count
|
||||||
|
const textArea = document.getElementById('textContent');
|
||||||
|
if (textArea) {
|
||||||
|
const charCount = document.createElement('div');
|
||||||
|
charCount.className = 'char-count';
|
||||||
|
textArea.parentNode.appendChild(charCount);
|
||||||
|
|
||||||
|
textArea.addEventListener('input', () => {
|
||||||
|
const count = textArea.value.length;
|
||||||
|
charCount.textContent = `input ${count} character`;
|
||||||
|
charCount.style.color = count > 10000 ? '#ef4444' : 'var (--text-secondary)'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 通用请求方法 api request
|
||||||
|
async function apiRequest(endpoint, method = 'GET', body = null) {
|
||||||
|
const options = {
|
||||||
|
method,
|
||||||
|
headers: {
|
||||||
|
'Content-Type': 'application/json'
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
if (body) {
|
||||||
|
options.body = JSON.stringify(body);
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const response = await fetch(`${API_BASE}${endpoint}`, options);
|
||||||
|
if (!response.ok) {
|
||||||
|
throw new Error(`request failed: ${response.status}`);
|
||||||
|
}
|
||||||
|
return response.json();
|
||||||
|
} catch (error) {
|
||||||
|
console.error('API REQUEST ERROR:', error);
|
||||||
|
showToast(error.message, 'error');
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function handleTextUpload() {
|
||||||
|
const description = document.getElementById('textDescription').value;
|
||||||
|
const content = document.getElementById('textContent').value.trim();
|
||||||
|
const statusDiv = document.getElementById('textUploadStatus');
|
||||||
|
|
||||||
|
// 清空状态提示 clear status tip
|
||||||
|
statusDiv.className = 'status-indicator';
|
||||||
|
statusDiv.textContent = '';
|
||||||
|
|
||||||
|
// 输入验证 input valid
|
||||||
|
if (!content) {
|
||||||
|
showStatus('error', 'TEXT CONTENT NOT NULL', statusDiv);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
showStatus('loading', 'UPLOADING...', statusDiv);
|
||||||
|
|
||||||
|
//
|
||||||
|
const payload = {
|
||||||
|
text: content,
|
||||||
|
...(description && {description})
|
||||||
|
};
|
||||||
|
|
||||||
|
const response = await fetch(`${API_BASE}/documents/text`, {
|
||||||
|
method: 'POST',
|
||||||
|
headers: {
|
||||||
|
'Content-Type': 'application/json'
|
||||||
|
},
|
||||||
|
body: JSON.stringify(payload)
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!response.ok) {
|
||||||
|
const errorData = await response.json();
|
||||||
|
throw new Error(errorData.detail || '上传失败');
|
||||||
|
}
|
||||||
|
|
||||||
|
const result = await response.json();
|
||||||
|
|
||||||
|
showStatus('success', `✅ ${result.message} (文档数: ${result.document_count})`, statusDiv);
|
||||||
|
|
||||||
|
|
||||||
|
document.getElementById('textContent').value = '';
|
||||||
|
|
||||||
|
// 更新文件列表 update file list
|
||||||
|
updateFileList();
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
showStatus('error', `❌ ERROR: ${error.message}`, statusDiv);
|
||||||
|
console.error('FILE UPLOAD FAILED:', error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function showStatus(type, message, container) {
|
||||||
|
container.textContent = message;
|
||||||
|
container.className = `status-indicator ${type}`;
|
||||||
|
|
||||||
|
// 自动清除成功状态 auto clear success status
|
||||||
|
if (type === 'success') {
|
||||||
|
setTimeout(() => {
|
||||||
|
container.textContent = '';
|
||||||
|
container.className = 'status-indicator';
|
||||||
|
}, 5000);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 文件上传处理 upload file
|
||||||
|
function setupFileUpload() {
|
||||||
|
const dropzone = document.getElementById('dropzone');
|
||||||
|
const fileInput = document.getElementById('fileInput');
|
||||||
|
|
||||||
|
|
||||||
|
// 拖放事件处理 Drag and drop event handling
|
||||||
|
dropzone.addEventListener('dragover', (e) => {
|
||||||
|
e.preventDefault();
|
||||||
|
dropzone.classList.add('active');
|
||||||
|
});
|
||||||
|
|
||||||
|
dropzone.addEventListener('dragleave', () => {
|
||||||
|
dropzone.classList.remove('active');
|
||||||
|
});
|
||||||
|
|
||||||
|
dropzone.addEventListener('drop', async (e) => {
|
||||||
|
e.preventDefault();
|
||||||
|
dropzone.classList.remove('active');
|
||||||
|
await handleFiles(e.dataTransfer.files);
|
||||||
|
});
|
||||||
|
|
||||||
|
|
||||||
|
fileInput.addEventListener('change', async (e) => {
|
||||||
|
await handleFiles(e.target.files);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
async function handleFiles(files) {
|
||||||
|
const formData = new FormData();
|
||||||
|
for (const file of files) {
|
||||||
|
formData.append('file', file);
|
||||||
|
}
|
||||||
|
const statusDiv = document.getElementById('fileUploadStatus');
|
||||||
|
|
||||||
|
|
||||||
|
statusDiv.className = 'status-indicator';
|
||||||
|
statusDiv.textContent = '';
|
||||||
|
try {
|
||||||
|
showStatus('loading', 'UPLOADING...', statusDiv);
|
||||||
|
const response = await fetch(`${API_BASE}/documents/upload`, {
|
||||||
|
method: 'POST',
|
||||||
|
body: formData
|
||||||
|
});
|
||||||
|
const result = await response.json();
|
||||||
|
showStatus('success', `✅ ${result.message} `, statusDiv);
|
||||||
|
updateFileList();
|
||||||
|
} catch (error) {
|
||||||
|
showToast(error.message, 'error');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
async function updateFileList() {
|
||||||
|
const fileList = document.querySelector('.file-list');
|
||||||
|
try {
|
||||||
|
const status = await apiRequest('/health');
|
||||||
|
fileList.innerHTML = `
|
||||||
|
<div>INDEXED FILE: ${status.indexed_files_count}</div>
|
||||||
|
`;
|
||||||
|
} catch (error) {
|
||||||
|
fileList.innerHTML = 'UNABLE TO OBTAIN FILE LIST';
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 智能检索处理 Intelligent retrieval processing
|
||||||
|
function setupQueryHandler() {
|
||||||
|
document.querySelector('#query .btn-primary').addEventListener('click', handleQuery);
|
||||||
|
}
|
||||||
|
|
||||||
|
async function handleQuery() {
|
||||||
|
const queryInput = document.querySelector('#query textarea');
|
||||||
|
const modeSelect = document.querySelector('#query select');
|
||||||
|
const streamCheckbox = document.querySelector('#query input[type="checkbox"]');
|
||||||
|
const resultsDiv = document.querySelector('#query .results');
|
||||||
|
|
||||||
|
const payload = {
|
||||||
|
query: queryInput.value,
|
||||||
|
mode: modeSelect.value,
|
||||||
|
stream: streamCheckbox.checked
|
||||||
|
};
|
||||||
|
|
||||||
|
resultsDiv.innerHTML = '<div class="loading">SEARCHING...</div>';
|
||||||
|
|
||||||
|
try {
|
||||||
|
if (payload.stream) {
|
||||||
|
await handleStreamingQuery(payload, resultsDiv);
|
||||||
|
} else {
|
||||||
|
const result = await apiRequest('/query', 'POST', payload);
|
||||||
|
resultsDiv.innerHTML = `<div class="result">${result.response}</div>`;
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
resultsDiv.innerHTML = `<div class="error">SEARCH FAILED: ${error.message}</div>`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 处理流式响应 handle stream api
|
||||||
|
async function handleStreamingQuery(payload, resultsDiv) {
|
||||||
|
try {
|
||||||
|
const response = await fetch(`${API_BASE}/query/stream`, {
|
||||||
|
method: 'POST',
|
||||||
|
headers: {'Content-Type': 'application/json'},
|
||||||
|
body: JSON.stringify(payload)
|
||||||
|
});
|
||||||
|
|
||||||
|
|
||||||
|
const contentType = response.headers.get('Content-Type') || '';
|
||||||
|
const validTypes = ['application/x-ndjson', 'application/json'];
|
||||||
|
if (!validTypes.some(t => contentType.includes(t))) {
|
||||||
|
throw new Error(`INVALID CONTENT TYPE: ${contentType}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
const reader = response.body.getReader();
|
||||||
|
const decoder = new TextDecoder('utf-8');
|
||||||
|
let buffer = '';
|
||||||
|
|
||||||
|
resultsDiv.innerHTML = '';
|
||||||
|
|
||||||
|
while (true) {
|
||||||
|
const {done, value} = await reader.read();
|
||||||
|
if (done) break;
|
||||||
|
|
||||||
|
buffer += decoder.decode(value, {stream: true});
|
||||||
|
|
||||||
|
// 按换行符分割（NDJSON格式要求） Split by line break (NDJSON format requirement)
|
||||||
|
let lineEndIndex;
|
||||||
|
while ((lineEndIndex = buffer.indexOf('\n')) >= 0) {
|
||||||
|
const line = buffer.slice(0, lineEndIndex).trim();
|
||||||
|
buffer = buffer.slice(lineEndIndex + 1);
|
||||||
|
|
||||||
|
if (!line) continue;
|
||||||
|
|
||||||
|
try {
|
||||||
|
const data = JSON.parse(line);
|
||||||
|
|
||||||
|
if (data.response) {
|
||||||
|
resultsDiv.innerHTML += data.response;
|
||||||
|
resultsDiv.scrollTop = resultsDiv.scrollHeight;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (data.error) {
|
||||||
|
resultsDiv.innerHTML += `<div class="error">${data.error}</div>`;
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('JSON PARSING FAILED:', {
|
||||||
|
error,
|
||||||
|
rawLine: line,
|
||||||
|
bufferRemaining: buffer
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 处理剩余数据 Process remaining data
|
||||||
|
if (buffer.trim()) {
|
||||||
|
try {
|
||||||
|
const data = JSON.parse(buffer.trim());
|
||||||
|
if (data.response) {
|
||||||
|
resultsDiv.innerHTML += data.response;
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('TAIL DATA PARSING FAILED:', error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
resultsDiv.innerHTML = `<div class="error">REQUEST FAILED: ${error.message}</div>`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// 知识问答处理 Knowledge Q&A Processing
|
||||||
|
function setupChatHandler() {
|
||||||
|
const sendButton = document.querySelector('#chat button');
|
||||||
|
const chatInput = document.querySelector('#chat input');
|
||||||
|
|
||||||
|
sendButton.addEventListener('click', () => handleChat(chatInput));
|
||||||
|
chatInput.addEventListener('keypress', (e) => {
|
||||||
|
if (e.key === 'Enter') handleChat(chatInput);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
async function handleChat(chatInput) {
|
||||||
|
const chatHistory = document.querySelector('#chat .chat-history');
|
||||||
|
|
||||||
|
|
||||||
|
const userMessage = document.createElement('div');
|
||||||
|
userMessage.className = 'message user';
|
||||||
|
userMessage.textContent = chatInput.value;
|
||||||
|
chatHistory.appendChild(userMessage);
|
||||||
|
|
||||||
|
|
||||||
|
const botMessage = document.createElement('div');
|
||||||
|
botMessage.className = 'message bot loading';
|
||||||
|
botMessage.textContent = 'THINKING...';
|
||||||
|
chatHistory.appendChild(botMessage);
|
||||||
|
chatHistory.scrollTop = chatHistory.scrollHeight;
|
||||||
|
|
||||||
|
try {
|
||||||
|
const response = await fetch(`${API_BASE}/api/chat`, {
|
||||||
|
method: 'POST',
|
||||||
|
headers: {'Content-Type': 'application/json'},
|
||||||
|
body: JSON.stringify({
|
||||||
|
messages: [{role: "user", content: chatInput.value}],
|
||||||
|
stream: true
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
const reader = response.body.getReader();
|
||||||
|
const decoder = new TextDecoder();
|
||||||
|
botMessage.classList.remove('loading');
|
||||||
|
botMessage.textContent = '';
|
||||||
|
|
||||||
|
while (true) {
|
||||||
|
const {done, value} = await reader.read();
|
||||||
|
if (done) break;
|
||||||
|
|
||||||
|
const chunk = decoder.decode(value);
|
||||||
|
const data = JSON.parse(chunk);
|
||||||
|
botMessage.textContent += data.message?.content || '';
|
||||||
|
chatHistory.scrollTop = chatHistory.scrollHeight;
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
botMessage.textContent = `ERROR: ${error.message}`;
|
||||||
|
botMessage.classList.add('error');
|
||||||
|
}
|
||||||
|
|
||||||
|
chatInput.value = '';
|
||||||
|
}
|
||||||
|
|
||||||
|
// 系统状态更新 system status update
|
||||||
|
async function updateSystemStatus() {
|
||||||
|
const statusElements = {
|
||||||
|
health: document.getElementById('healthStatus'),
|
||||||
|
storageProgress: document.getElementById('storageProgress'),
|
||||||
|
indexedFiles: document.getElementById('indexedFiles'),
|
||||||
|
storageUsage: document.getElementById('storageUsage'),
|
||||||
|
llmModel: document.getElementById('llmModel'),
|
||||||
|
embedModel: document.getElementById('embedModel'),
|
||||||
|
maxTokens: document.getElementById('maxTokens'),
|
||||||
|
workingDir: document.getElementById('workingDir'),
|
||||||
|
inputDir: document.getElementById('inputDir'),
|
||||||
|
kv_storage: document.getElementById("kv_storage"),
|
||||||
|
doc_status_storage: document.getElementById("doc_status_storage"),
|
||||||
|
graph_storage: document.getElementById("graph_storage"),
|
||||||
|
vector_storage: document.getElementById("vector_storage")
|
||||||
|
};
|
||||||
|
|
||||||
|
try {
|
||||||
|
const status = await apiRequest('/health');
|
||||||
|
|
||||||
|
// 健康状态 heath status
|
||||||
|
statusElements.health.className = 'status-badge';
|
||||||
|
statusElements.health.textContent = status.status === 'healthy' ?
|
||||||
|
'✅ Healthy operation in progress' : '⚠️ Service exception';
|
||||||
|
|
||||||
|
// 存储状态（示例逻辑，可根据实际需求修改） kv status
|
||||||
|
const progressValue = Math.min(Math.round((status.indexed_files_count / 1000) * 100), 100);
|
||||||
|
statusElements.storageProgress.value = progressValue;
|
||||||
|
statusElements.indexedFiles.textContent = `INDEXED FILES：${status.indexed_files_count}`;
|
||||||
|
statusElements.storageUsage.textContent = `USE PERCENT：${progressValue}%`;
|
||||||
|
|
||||||
|
// 模型配置 model state
|
||||||
|
statusElements.llmModel.textContent = `${status.configuration.llm_model} (${status.configuration.llm_binding})`;
|
||||||
|
statusElements.embedModel.textContent = `${status.configuration.embedding_model} (${status.configuration.embedding_binding})`;
|
||||||
|
statusElements.maxTokens.textContent = status.configuration.max_tokens.toLocaleString();
|
||||||
|
|
||||||
|
// 目录信息 dir msg
|
||||||
|
statusElements.workingDir.textContent = status.working_directory;
|
||||||
|
statusElements.inputDir.textContent = status.input_directory;
|
||||||
|
|
||||||
|
//存储信息 stack msg
|
||||||
|
statusElements.kv_storage.textContent = status.configuration.kv_storage;
|
||||||
|
statusElements.doc_status_storage.textContent = status.configuration.doc_status_storage;
|
||||||
|
statusElements.graph_storage.textContent = status.configuration.graph_storage;
|
||||||
|
statusElements.vector_storage.textContent = status.configuration.vector_storage
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
statusElements.health.className = 'status-badge error';
|
||||||
|
statusElements.health.textContent = '❌GET STATUS FAILED';
|
||||||
|
statusElements.storageProgress.value = 0;
|
||||||
|
statusElements.indexedFiles.textContent = 'INDEXED FILES：GET FAILED';
|
||||||
|
console.error('STATUS UPDATE FAILED:', error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// 区域切换监听 Area switching monitoring
|
||||||
|
function setupSectionObserver() {
|
||||||
|
const observer = new MutationObserver(mutations => {
|
||||||
|
mutations.forEach(mutation => {
|
||||||
|
if (mutation.attributeName === 'style') {
|
||||||
|
const isVisible = mutation.target.style.display !== 'none';
|
||||||
|
if (isVisible && mutation.target.id === 'status') {
|
||||||
|
updateSystemStatus();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
document.querySelectorAll('.card').forEach(section => {
|
||||||
|
observer.observe(section, {attributes: true});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 显示提示信息 Display prompt information
|
||||||
|
function showToast(message, type = 'info') {
|
||||||
|
const toast = document.createElement('div');
|
||||||
|
toast.className = `toast ${type}`;
|
||||||
|
toast.textContent = message;
|
||||||
|
document.body.appendChild(toast);
|
||||||
|
|
||||||
|
setTimeout(() => {
|
||||||
|
toast.remove();
|
||||||
|
}, 3000);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// 动态加载标签列表 Dynamically load tag list
|
||||||
|
async function loadLabels() {
|
||||||
|
try {
|
||||||
|
const response = await fetch('/graph/label/list');
|
||||||
|
const labels = await response.json();
|
||||||
|
renderLabels(labels);
|
||||||
|
} catch (error) {
|
||||||
|
console.error('DYNAMICALLY LOAD TAG LIST FAILED:', error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function loadGraph(label) {
|
||||||
|
try {
|
||||||
|
// 渲染标签列表 render label list
|
||||||
|
openGraphModal(label)
|
||||||
|
} catch (error) {
|
||||||
|
console.error('LOADING LABEL FAILED:', error);
|
||||||
|
|
||||||
|
|
||||||
|
const labelList = document.getElementById("label-list");
|
||||||
|
labelList.innerHTML = `
|
||||||
|
<div class="error-message">
|
||||||
|
LOADING ERROR: ${error.message}
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 渲染标签列表 render graph label list
|
||||||
|
function renderLabels(labels) {
|
||||||
|
const container = document.getElementById('label-list');
|
||||||
|
container.innerHTML = labels.map(label => `
|
||||||
|
<div class="label-item">
|
||||||
|
<span style="font-weight: 500; color: var(--text-primary);">
|
||||||
|
${label}
|
||||||
|
</span>
|
||||||
|
<div class="label-actions">
|
||||||
|
<button class="btn btn-primary"
|
||||||
|
onclick="handleLabelAction('${label}')">
|
||||||
|
📋 graph
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
`).join('');
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
function handleLabelAction(label) {
|
||||||
|
loadGraph(label)
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
function refreshLabels() {
|
||||||
|
showToast('LOADING GRAPH LABELS...', 'info');
|
||||||
|
loadLabels();
|
||||||
|
}
|
||||||
|
|
||||||
|
function showToast(message, type = 'info') {
|
||||||
|
const toast = document.createElement('div');
|
||||||
|
toast.className = `toast toast-${type}`;
|
||||||
|
toast.textContent = message;
|
||||||
|
document.body.appendChild(toast);
|
||||||
|
|
||||||
|
setTimeout(() => {
|
||||||
|
toast.remove();
|
||||||
|
}, 3000);
|
||||||
|
}
|
||||||
|
|
||||||
|
document.addEventListener('DOMContentLoaded', initializeApp);
|
||||||
@ -339,3 +339,175 @@ class Neo4JStorage(BaseGraphStorage):
|
|||||||
|
|
||||||
async def _node2vec_embed(self):
|
async def _node2vec_embed(self):
|
||||||
print("Implemented but never called.")
|
print("Implemented but never called.")
|
||||||
|
|
||||||
|
async def get_knowledge_graph(
|
||||||
|
self, node_label: str, max_depth: int = 5
|
||||||
|
) -> Dict[str, List[Dict]]:
|
||||||
|
"""
|
||||||
|
获取指定节点的完整连通子图（包含起始节点本身）
|
||||||
|
|
||||||
|
修复要点：
|
||||||
|
1. 包含起始节点自身
|
||||||
|
2. 处理多标签节点
|
||||||
|
3. 明确关系方向
|
||||||
|
4. 添加深度控制
|
||||||
|
"""
|
||||||
|
label = node_label.strip('"')
|
||||||
|
result = {"nodes": [], "edges": []}
|
||||||
|
seen_nodes = set()
|
||||||
|
seen_edges = set()
|
||||||
|
|
||||||
|
async with self._driver.session(database=self._DATABASE) as session:
|
||||||
|
try:
|
||||||
|
# 关键调试步骤：先验证起始节点是否存在
|
||||||
|
validate_query = f"MATCH (n:`{label}`) RETURN n LIMIT 1"
|
||||||
|
validate_result = await session.run(validate_query)
|
||||||
|
if not await validate_result.single():
|
||||||
|
logger.warning(f"起始节点 {label} 不存在！")
|
||||||
|
return result
|
||||||
|
|
||||||
|
# 优化后的查询语句（包含方向处理和自循环）
|
||||||
|
main_query = f"""
|
||||||
|
MATCH (start:`{label}`)
|
||||||
|
WITH start
|
||||||
|
CALL apoc.path.subgraphAll(start, {{
|
||||||
|
relationshipFilter: '>',
|
||||||
|
minLevel: 0,
|
||||||
|
maxLevel: {max_depth},
|
||||||
|
bfs: true
|
||||||
|
}})
|
||||||
|
YIELD nodes, relationships
|
||||||
|
RETURN nodes, relationships
|
||||||
|
"""
|
||||||
|
result_set = await session.run(main_query)
|
||||||
|
record = await result_set.single()
|
||||||
|
|
||||||
|
if record:
|
||||||
|
# 处理节点（兼容多标签情况）
|
||||||
|
for node in record["nodes"]:
|
||||||
|
# 使用节点ID + 标签组合作为唯一标识
|
||||||
|
node_id = f"{node.id}_{'_'.join(node.labels)}"
|
||||||
|
if node_id not in seen_nodes:
|
||||||
|
node_data = dict(node)
|
||||||
|
node_data["labels"] = list(node.labels) # 保留所有标签
|
||||||
|
result["nodes"].append(node_data)
|
||||||
|
seen_nodes.add(node_id)
|
||||||
|
|
||||||
|
# 处理关系（包含方向信息）
|
||||||
|
for rel in record["relationships"]:
|
||||||
|
edge_id = f"{rel.id}_{rel.type}"
|
||||||
|
if edge_id not in seen_edges:
|
||||||
|
start = rel.start_node
|
||||||
|
end = rel.end_node
|
||||||
|
edge_data = dict(rel)
|
||||||
|
edge_data.update(
|
||||||
|
{
|
||||||
|
"source": f"{start.id}_{'_'.join(start.labels)}",
|
||||||
|
"target": f"{end.id}_{'_'.join(end.labels)}",
|
||||||
|
"type": rel.type,
|
||||||
|
"direction": rel.element_id.split(
|
||||||
|
"->" if rel.end_node == end else "<-"
|
||||||
|
)[1],
|
||||||
|
}
|
||||||
|
)
|
||||||
|
result["edges"].append(edge_data)
|
||||||
|
seen_edges.add(edge_id)
|
||||||
|
|
||||||
|
logger.info(
|
||||||
|
f"子图查询成功 | 节点数: {len(result['nodes'])} | 边数: {len(result['edges'])}"
|
||||||
|
)
|
||||||
|
|
||||||
|
except neo4jExceptions.ClientError as e:
|
||||||
|
logger.error(f"APOC查询失败: {str(e)}")
|
||||||
|
return await self._robust_fallback(label, max_depth)
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
async def _robust_fallback(
|
||||||
|
self, label: str, max_depth: int
|
||||||
|
) -> Dict[str, List[Dict]]:
|
||||||
|
"""强化版降级查询方案"""
|
||||||
|
result = {"nodes": [], "edges": []}
|
||||||
|
visited_nodes = set()
|
||||||
|
visited_edges = set()
|
||||||
|
|
||||||
|
async def traverse(current_label: str, current_depth: int):
|
||||||
|
if current_depth > max_depth:
|
||||||
|
return
|
||||||
|
|
||||||
|
# 获取当前节点详情
|
||||||
|
node = await self.get_node(current_label)
|
||||||
|
if not node:
|
||||||
|
return
|
||||||
|
|
||||||
|
node_id = f"{current_label}"
|
||||||
|
if node_id in visited_nodes:
|
||||||
|
return
|
||||||
|
visited_nodes.add(node_id)
|
||||||
|
|
||||||
|
# 添加节点数据（带完整标签）
|
||||||
|
node_data = {k: v for k, v in node.items()}
|
||||||
|
node_data["labels"] = [current_label] # 假设get_node方法返回包含标签信息
|
||||||
|
result["nodes"].append(node_data)
|
||||||
|
|
||||||
|
# 获取所有出边和入边
|
||||||
|
query = f"""
|
||||||
|
MATCH (a)-[r]-(b)
|
||||||
|
WHERE a:`{current_label}` OR b:`{current_label}`
|
||||||
|
RETURN a, r, b,
|
||||||
|
CASE WHEN startNode(r) = a THEN 'OUTGOING' ELSE 'INCOMING' END AS direction
|
||||||
|
"""
|
||||||
|
async with self._driver.session(database=self._DATABASE) as session:
|
||||||
|
results = await session.run(query)
|
||||||
|
async for record in results:
|
||||||
|
# 处理边
|
||||||
|
rel = record["r"]
|
||||||
|
edge_id = f"{rel.id}_{rel.type}"
|
||||||
|
if edge_id not in visited_edges:
|
||||||
|
edge_data = dict(rel)
|
||||||
|
edge_data.update(
|
||||||
|
{
|
||||||
|
"source": list(record["a"].labels)[0],
|
||||||
|
"target": list(record["b"].labels)[0],
|
||||||
|
"type": rel.type,
|
||||||
|
"direction": record["direction"],
|
||||||
|
}
|
||||||
|
)
|
||||||
|
result["edges"].append(edge_data)
|
||||||
|
visited_edges.add(edge_id)
|
||||||
|
|
||||||
|
# 递归遍历相邻节点
|
||||||
|
next_label = (
|
||||||
|
list(record["b"].labels)[0]
|
||||||
|
if record["direction"] == "OUTGOING"
|
||||||
|
else list(record["a"].labels)[0]
|
||||||
|
)
|
||||||
|
await traverse(next_label, current_depth + 1)
|
||||||
|
|
||||||
|
await traverse(label, 0)
|
||||||
|
return result
|
||||||
|
|
||||||
|
async def get_all_labels(self) -> List[str]:
|
||||||
|
"""
|
||||||
|
获取数据库中所有存在的节点标签
|
||||||
|
Returns:
|
||||||
|
["Person", "Company", ...] # 按字母排序的标签列表
|
||||||
|
"""
|
||||||
|
async with self._driver.session(database=self._DATABASE) as session:
|
||||||
|
# 方法1：直接查询元数据（Neo4j 4.3+ 可用）
|
||||||
|
# query = "CALL db.labels() YIELD label RETURN label"
|
||||||
|
|
||||||
|
# 方法2：兼容旧版本的查询方式
|
||||||
|
query = """
|
||||||
|
MATCH (n)
|
||||||
|
WITH DISTINCT labels(n) AS node_labels
|
||||||
|
UNWIND node_labels AS label
|
||||||
|
RETURN DISTINCT label
|
||||||
|
ORDER BY label
|
||||||
|
"""
|
||||||
|
|
||||||
|
result = await session.run(query)
|
||||||
|
labels = []
|
||||||
|
async for record in result:
|
||||||
|
labels.append(record["label"])
|
||||||
|
return labels
|
||||||
|
|||||||
@ -292,6 +292,15 @@ class LightRAG:
|
|||||||
embedding_func=None,
|
embedding_func=None,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
async def get_graph_labels(self):
|
||||||
|
text = await self.chunk_entity_relation_graph.get_all_labels()
|
||||||
|
return text
|
||||||
|
|
||||||
|
async def get_graps(self, nodel_label: str, max_depth: int):
|
||||||
|
return await self.chunk_entity_relation_graph.get_knowledge_graph(
|
||||||
|
node_label=nodel_label, max_depth=max_depth
|
||||||
|
)
|
||||||
|
|
||||||
def _get_storage_class(self, storage_name: str) -> dict:
|
def _get_storage_class(self, storage_name: str) -> dict:
|
||||||
import_path = STORAGES[storage_name]
|
import_path = STORAGES[storage_name]
|
||||||
storage_class = lazy_external_import(import_path, storage_name)
|
storage_class = lazy_external_import(import_path, storage_name)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user