45 lines
1.8 KiB
TypeScript
45 lines
1.8 KiB
TypeScript
import { apiFetch, readSSE, type SSESummary } from "@/lib/api";
|
||
|
||
/**
|
||
* 客户端两个 Agent 的会话适配层。
|
||
*
|
||
* 两者都是「打开建会话 → 带 session_id 对话 → 关闭结束会话」的三段式，
|
||
* 后端的会话归属校验会拒绝未创建或不属于自己的 session_id（SessionOwnershipError），
|
||
* 所以 session_id 必须来自 session/create，不能在前端自己造。
|
||
*
|
||
* - customer：匿名客服 Agent（/api/agent/customer/*），游客与登录用户都可用，无需 token
|
||
* - client ：登录客户 Agent（/api/agent/client/*），需 require_customer，
|
||
* 能读到本人持仓 / 交易（匿名版没有 customer_id，数据查询会被引导去登录）
|
||
*/
|
||
export type ChatAgentMode = "customer" | "client";
|
||
|
||
const PREFIX: Record<ChatAgentMode, string> = {
|
||
customer: "/agent/customer",
|
||
client: "/agent/client",
|
||
};
|
||
|
||
export interface AgentSession {
|
||
session_id: string;
|
||
customer_id: number | null;
|
||
}
|
||
|
||
export const chatAgentApi = {
|
||
/** POST /session/create —— 建立会话，返回后端签发的 session_id。 */
|
||
createSession: (mode: ChatAgentMode) =>
|
||
apiFetch<AgentSession>(`${PREFIX[mode]}/session/create`, { method: "POST" }),
|
||
|
||
/** POST /chat —— 响应是 SSE 包装，但内容为 {answer, sources, intent...}。 */
|
||
chat: (
|
||
mode: ChatAgentMode,
|
||
body: { session_id: string; query: string },
|
||
signal?: AbortSignal
|
||
): Promise<SSESummary> => readSSE(`${PREFIX[mode]}/chat`, body, signal),
|
||
|
||
/** POST /session/end —— 关闭会话并清理服务端上下文与限流键。 */
|
||
endSession: (mode: ChatAgentMode, sessionId: string) =>
|
||
apiFetch<{ session_id: string; archived?: boolean }>(`${PREFIX[mode]}/session/end`, {
|
||
method: "POST",
|
||
body: JSON.stringify({ session_id: sessionId }),
|
||
}),
|
||
};
|