Files
knowledge-graph-agent/frontend/src/services/graphApi.ts
T

116 lines
3.1 KiB
TypeScript

import axios from 'axios';
import type {
GraphData,
Node,
CreateNodeRequest,
UpdateNodeRequest,
CreateEdgeRequest,
DeleteResponse,
EdgeResponse,
NeighborResponse,
SearchResponse,
SimpleGraphData,
GraphStats,
} from '../types/graph';
import { API_BASE_URL, API_TIMEOUT } from '../config/constants';
/**
* API 客户端实例
* 统一配置了 base URL、超时时间和默认请求头
*/
const apiClient = axios.create({
baseURL: API_BASE_URL,
timeout: API_TIMEOUT,
headers: {
'Content-Type': 'application/json',
},
});
/**
* 知识图谱 API 服务
* 提供与后端交互的所有图谱相关接口
*/
export const graphApi = {
/**
* 获取完整的知识图谱数据
* @returns 节点和边的集合
*/
async getData(): Promise<GraphData> {
try {
const response = await apiClient.get<GraphData>('/graph');
return response.data;
} catch (error) {
console.error('Failed to fetch graph data:', error);
return { nodes: [], edges: [] };
}
},
/**
* 根据 ID 获取节点详情
* @param id 节点 ID
* @returns 节点对象或 null
*/
async getNodeById(id: string): Promise<Node | null> {
try {
const response = await apiClient.get<Node>(`/nodes/${id}`);
return response.data;
} catch (error) {
console.error('Failed to fetch node:', error);
return null;
}
},
async createNode(request: CreateNodeRequest): Promise<Node> {
const response = await apiClient.post<Node>('/nodes', request);
return response.data;
},
async updateNode(id: string, request: UpdateNodeRequest): Promise<Node> {
const response = await apiClient.put<Node>(`/nodes/${id}`, request);
return response.data;
},
async deleteNode(id: string): Promise<DeleteResponse> {
const response = await apiClient.delete<DeleteResponse>(`/nodes/${id}`);
return response.data;
},
async getNodeNeighbors(id: string): Promise<NeighborResponse> {
const response = await apiClient.get<NeighborResponse>(`/nodes/${id}/neighbors`);
return response.data;
},
async createEdge(request: CreateEdgeRequest): Promise<EdgeResponse> {
const response = await apiClient.post<EdgeResponse>('/edges', request);
return response.data;
},
async deleteEdge(id: string): Promise<DeleteResponse> {
const response = await apiClient.delete<DeleteResponse>(`/edges/${id}`);
return response.data;
},
async getGraphStats(): Promise<GraphStats> {
const response = await apiClient.get<GraphStats>('/graph/stats');
return response.data;
},
async getSimpleGraphData(): Promise<SimpleGraphData> {
const response = await apiClient.get<SimpleGraphData>('/graph/simpleJson');
return response.data;
},
async searchNodes(query: string): Promise<SearchResponse> {
const response = await apiClient.get<SearchResponse>('/search', {
params: { q: query },
});
return response.data;
},
async healthCheck(): Promise<{ status: string }> {
const response = await apiClient.get<{ status: string }>('/health');
return response.data;
},
};
export default apiClient;