262 lines
7.8 KiB
JavaScript
262 lines
7.8 KiB
JavaScript
import { apiGet, apiPost } from './base'
|
||
|
||
/**
|
||
* 图数据库API模块
|
||
* 包含LightRAG图知识库和Neo4j图数据库两种接口
|
||
* 采用命名空间分组模式，清晰区分接口类型
|
||
*/
|
||
|
||
// =============================================================================
|
||
// === 统一图谱接口 (Unified Graph API) ===
|
||
// =============================================================================
|
||
|
||
export const unifiedApi = {
|
||
/**
|
||
* 获取所有可用的知识图谱列表
|
||
* @returns {Promise} - 图谱列表
|
||
*/
|
||
getGraphs: async () => {
|
||
return await apiGet('/api/graph/list', {}, true)
|
||
},
|
||
|
||
/**
|
||
* 获取子图数据 (统一接口)
|
||
* @param {Object} params - 查询参数
|
||
* @param {string} params.db_id - 图谱ID
|
||
* @param {string} params.node_label - 节点标签/关键词
|
||
* @param {number} params.max_depth - 最大深度
|
||
* @param {number} params.max_nodes - 最大节点数
|
||
* @returns {Promise} - 子图数据
|
||
*/
|
||
getSubgraph: async (params) => {
|
||
const { db_id, node_label = '*', max_depth = 2, max_nodes = 100 } = params
|
||
|
||
if (!db_id) {
|
||
throw new Error('db_id is required')
|
||
}
|
||
|
||
const queryParams = new URLSearchParams({
|
||
db_id: db_id,
|
||
node_label: node_label,
|
||
max_depth: max_depth.toString(),
|
||
max_nodes: max_nodes.toString()
|
||
})
|
||
|
||
return await apiGet(`/api/graph/subgraph?${queryParams.toString()}`, {}, true)
|
||
},
|
||
|
||
/**
|
||
* 获取图谱统计信息 (统一接口)
|
||
* @param {string} db_id - 图谱ID
|
||
* @returns {Promise} - 统计信息
|
||
*/
|
||
getStats: async (db_id) => {
|
||
if (!db_id) {
|
||
throw new Error('db_id is required')
|
||
}
|
||
|
||
const queryParams = new URLSearchParams({
|
||
db_id: db_id
|
||
})
|
||
|
||
return await apiGet(`/api/graph/stats?${queryParams.toString()}`, {}, true)
|
||
},
|
||
|
||
/**
|
||
* 获取图谱标签列表 (统一接口)
|
||
* @param {string} db_id - 图谱ID
|
||
* @returns {Promise} - 标签列表
|
||
*/
|
||
getLabels: async (db_id) => {
|
||
if (!db_id) {
|
||
throw new Error('db_id is required')
|
||
}
|
||
|
||
const queryParams = new URLSearchParams({
|
||
db_id: db_id
|
||
})
|
||
|
||
return await apiGet(`/api/graph/labels?${queryParams.toString()}`, {}, true)
|
||
}
|
||
}
|
||
|
||
// =============================================================================
|
||
// === Neo4j图数据库接口分组 ===
|
||
// =============================================================================
|
||
|
||
export const neo4jApi = {
|
||
/**
|
||
* 获取Neo4j图数据库样例节点
|
||
* @param {string} kgdb_name - Neo4j数据库名称（默认为'neo4j'）
|
||
* @param {number} num - 节点数量
|
||
* @returns {Promise} - 样例节点数据
|
||
*/
|
||
getSampleNodes: async (kgdb_name = 'neo4j', num = 100) => {
|
||
const queryParams = new URLSearchParams({
|
||
kgdb_name: kgdb_name,
|
||
num: num.toString()
|
||
})
|
||
|
||
return await apiGet(`/api/graph/neo4j/nodes?${queryParams.toString()}`, {}, true)
|
||
},
|
||
|
||
/**
|
||
* 根据实体名称查询Neo4j图节点
|
||
* @param {string} entity_name - 实体名称
|
||
* @returns {Promise} - 节点数据
|
||
*/
|
||
queryNode: async (entity_name) => {
|
||
if (!entity_name) {
|
||
throw new Error('entity_name is required')
|
||
}
|
||
|
||
const queryParams = new URLSearchParams({
|
||
entity_name: entity_name
|
||
})
|
||
|
||
return await apiGet(`/api/graph/neo4j/node?${queryParams.toString()}`, {}, true)
|
||
},
|
||
|
||
/**
|
||
* 通过JSONL文件添加图谱实体到Neo4j
|
||
* @param {string} file_path - JSONL文件路径
|
||
* @param {string} kgdb_name - Neo4j数据库名称（默认为'neo4j'）
|
||
* @param {string} embed_model_name - 嵌入模型名称 (可选)
|
||
* @param {number} batch_size - 批处理大小 (可选)
|
||
* @returns {Promise} - 添加结果
|
||
*/
|
||
addEntities: async (
|
||
file_path,
|
||
kgdb_name = 'neo4j',
|
||
embed_model_name = null,
|
||
batch_size = null
|
||
) => {
|
||
return await apiPost(
|
||
'/api/graph/neo4j/add-entities',
|
||
{
|
||
file_path: file_path,
|
||
kgdb_name: kgdb_name,
|
||
embed_model_name: embed_model_name,
|
||
batch_size: batch_size
|
||
},
|
||
{},
|
||
true
|
||
)
|
||
},
|
||
|
||
/**
|
||
* 为Neo4j图谱节点添加嵌入向量索引
|
||
* @param {string} kgdb_name - Neo4j数据库名称（默认为'neo4j'）
|
||
* @returns {Promise} - 索引结果
|
||
*/
|
||
indexEntities: async (kgdb_name = 'neo4j') => {
|
||
return await apiPost(
|
||
'/api/graph/neo4j/index-entities',
|
||
{
|
||
kgdb_name: kgdb_name
|
||
},
|
||
{},
|
||
true
|
||
)
|
||
},
|
||
|
||
/**
|
||
* 获取Neo4j图数据库信息
|
||
* @returns {Promise} - 图数据库信息
|
||
*/
|
||
getInfo: async () => {
|
||
return await apiGet('/api/graph/neo4j/info', {}, true)
|
||
}
|
||
}
|
||
|
||
// =============================================================================
|
||
// === 工具函数分组 ===
|
||
// =============================================================================
|
||
|
||
/**
|
||
* 根据实体类型获取颜色
|
||
* @param {string} entityType - 实体类型
|
||
* @returns {string} - 颜色值
|
||
*/
|
||
export const getEntityTypeColor = (entityType) => {
|
||
const colorMap = {
|
||
person: '#FF6B6B', // 红色 - 人物
|
||
organization: '#4ECDC4', // 青色 - 组织
|
||
location: '#45B7D1', // 蓝色 - 地点
|
||
geo: '#45B7D1', // 蓝色 - 地理位置
|
||
event: '#96CEB4', // 绿色 - 事件
|
||
category: '#FFEAA7', // 黄色 - 分类
|
||
equipment: '#DDA0DD', // 紫色 - 设备
|
||
athlete: '#FF7675', // 红色 - 运动员
|
||
record: '#FD79A8', // 粉色 - 记录
|
||
year: '#FDCB6E', // 橙色 - 年份
|
||
UNKNOWN: '#B2BEC3', // 灰色 - 未知
|
||
unknown: '#B2BEC3' // 灰色 - 未知
|
||
}
|
||
|
||
return colorMap[entityType] || colorMap['unknown']
|
||
}
|
||
|
||
/**
|
||
* 根据权重计算边的粗细
|
||
* @param {number} weight - 权重值
|
||
* @param {number} minWeight - 最小权重
|
||
* @param {number} maxWeight - 最大权重
|
||
* @returns {number} - 边的粗细
|
||
*/
|
||
export const calculateEdgeWidth = (weight, minWeight = 1, maxWeight = 10) => {
|
||
const minWidth = 1
|
||
const maxWidth = 5
|
||
const normalizedWeight = (weight - minWeight) / (maxWeight - minWeight)
|
||
return minWidth + normalizedWeight * (maxWidth - minWidth)
|
||
}
|
||
|
||
// =============================================================================
|
||
// === 兼容性导出（可选，用于平滑迁移）===
|
||
// =============================================================================
|
||
|
||
// 保持向后兼容的导出，后续可以移除
|
||
export const getGraphNodes = async (params = {}) => {
|
||
console.warn('getGraphNodes is deprecated, use neo4jApi.getSampleNodes instead')
|
||
return neo4jApi.getSampleNodes(params.kgdb_name || 'neo4j', params.num || 100)
|
||
}
|
||
|
||
export const getGraphNode = async (params = {}) => {
|
||
console.warn('getGraphNode is deprecated, use neo4jApi.queryNode instead')
|
||
return neo4jApi.queryNode(params.entity_name)
|
||
}
|
||
|
||
export const addByJsonl = async (file_path, kgdb_name = 'neo4j') => {
|
||
console.warn('addByJsonl is deprecated, use neo4jApi.addEntities instead')
|
||
return neo4jApi.addEntities(file_path, kgdb_name)
|
||
}
|
||
|
||
export const indexNodes = async (kgdb_name = 'neo4j') => {
|
||
console.warn('indexNodes is deprecated, use neo4jApi.indexEntities instead')
|
||
return neo4jApi.indexEntities(kgdb_name)
|
||
}
|
||
|
||
export const getGraphStats = async () => {
|
||
console.warn('getGraphStats is deprecated, use neo4jApi.getInfo instead')
|
||
return neo4jApi.getInfo()
|
||
}
|
||
|
||
// 兼容性导出 - 使用统一接口替代旧有的 graphApi
|
||
export const graphApi = {
|
||
// 使用统一接口替代 LightRAG 接口
|
||
getSubgraph: unifiedApi.getSubgraph,
|
||
getDatabases: async () => {
|
||
// 使用统一接口获取所有图谱，然后过滤出 LightRAG 类型的
|
||
const response = await unifiedApi.getGraphs()
|
||
if (response.success) {
|
||
const lightragDbs = response.data.filter((graph) => graph.type === 'lightrag')
|
||
return { success: true, data: { databases: lightragDbs } }
|
||
}
|
||
return response
|
||
},
|
||
getLabels: unifiedApi.getLabels,
|
||
getStats: unifiedApi.getStats,
|
||
// 保留 Neo4j 接口
|
||
...neo4jApi
|
||
}
|