85 lines
2.5 KiB
TypeScript
85 lines
2.5 KiB
TypeScript
import axios from 'axios';
|
||
// 导入类型
|
||
import type { GenerateRequest, GenerateResponse, Message } from '@/types';
|
||
|
||
// 定义角色列表接口
|
||
// 结构: { "RoleName": ["file1.jsonl", "file2.jsonl"] }
|
||
export interface ChatRoleListResponse {
|
||
[roleName: string]: string[];
|
||
}
|
||
|
||
// 创建 Axios 实例
|
||
const apiClient = axios.create({
|
||
baseURL: '/api', // 配合 Vite 代理，转发到 http://backend:8000
|
||
timeout: 60000,
|
||
headers: {
|
||
'Content-Type': 'application/json',
|
||
},
|
||
});
|
||
|
||
// ==========================================
|
||
// API 接口函数封装
|
||
// ==========================================
|
||
|
||
/**
|
||
* 获取所有角色和聊天记录列表
|
||
* 对应后端: GET /get_all_role_and_chat
|
||
*/
|
||
export const getChatRoleList = async (): Promise<ChatRoleListResponse> => {
|
||
const response = await apiClient.get<ChatRoleListResponse>('/get_all_role_and_chat');
|
||
return response.data;
|
||
};
|
||
|
||
/**
|
||
* 获取特定聊天记录
|
||
* 对应后端: GET /chat_history?file_path=xxx.jsonl
|
||
*
|
||
* @param filePath - 文件路径，如 "RoleA/chat1.jsonl"
|
||
* @returns Message[] - 消息数组
|
||
*/
|
||
export const getChatHistory = async (filePath: string): Promise<Message[]> => {
|
||
const response = await apiClient.get<Message[]>('/chat_history', {
|
||
params: { file_path: filePath }
|
||
});
|
||
return response.data;
|
||
};
|
||
|
||
/**
|
||
* 发送消息并获取回复 (普通 POST，非流式)
|
||
* 对应后端: POST /generate_reply
|
||
*/
|
||
export const generateReply = async (payload: GenerateRequest): Promise<GenerateResponse> => {
|
||
const response = await apiClient.post<GenerateResponse>('/generate_reply', payload);
|
||
return response.data;
|
||
};
|
||
|
||
/**
|
||
* 获取本地图片列表
|
||
* 对应后端: GET /local_images
|
||
*/
|
||
export const getLocalImages = async (path: string): Promise<string[]> => {
|
||
const response = await apiClient.get<string[]>('/local_images', { params: { path } });
|
||
return response.data;
|
||
};
|
||
|
||
/**
|
||
* 上传图片
|
||
* 对应后端: POST /upload_image
|
||
*/
|
||
export const uploadImage = async (file: File): Promise<{ url: string }> => {
|
||
const formData = new FormData();
|
||
formData.append('file', file);
|
||
const response = await apiClient.post<{ url: string }>('/upload_image', formData, {
|
||
headers: { 'Content-Type': 'multipart/form-data' },
|
||
});
|
||
return response.data;
|
||
};
|
||
|
||
// 3. 预留：流式响应处理 (SSE)
|
||
export const generateReplyStream = async (payload: GenerateRequest): Promise<void> => {
|
||
// TODO: 待实现 SSE 逻辑
|
||
console.log('SSE stream logic placeholder', payload);
|
||
};
|
||
|
||
export default apiClient;
|