feat: 工作台聊天室功能

This commit is contained in:
2026-05-31 02:37:23 +08:00
parent 83cf9eab94
commit f83f434863
33 changed files with 2520 additions and 265 deletions

View File

@@ -0,0 +1,29 @@
import type Database from "bun:sqlite";
import type { CreateConversationRequest, RuntimeMode } from "../../../shared/api";
import { createConversation } from "../../db/conversations";
import { createApiError, jsonResponse } from "../../helpers";
import { validateIdParam } from "../../middleware";
export async function handleCreateConversation(req: Request, db: Database, mode: RuntimeMode): Promise<Response> {
const url = new URL(req.url);
const projectId = url.pathname.split("/")[3];
const validated = validateIdParam(projectId ?? "", mode);
if (validated instanceof Response) return validated;
let body: CreateConversationRequest = {};
try {
body = (await req.json()) as CreateConversationRequest;
} catch {
// empty body is ok, defaults will be used
}
const result = createConversation(db, validated.id, body.modelId);
if ("error" in result) {
return jsonResponse(createApiError(result.error, result.status), { mode, status: result.status });
}
return jsonResponse({ conversation: result.conversation }, { mode, status: 201 });
}

View File

@@ -0,0 +1,35 @@
import type Database from "bun:sqlite";
import type { RuntimeMode } from "../../../shared/api";
import { deleteConversation, getConversation } from "../../db/conversations";
import { createApiError, jsonResponse } from "../../helpers";
import { validateIdParam } from "../../middleware";
export function handleDeleteConversation(req: Request, db: Database, mode: RuntimeMode): Response {
const parts = new URL(req.url).pathname.split("/");
const projectId = parts[3];
const conversationId = parts[5];
const validatedProject = validateIdParam(projectId ?? "", mode);
if (validatedProject instanceof Response) return validatedProject;
const validatedConv = validateIdParam(conversationId ?? "", mode);
if (validatedConv instanceof Response) return validatedConv;
const convResult = getConversation(db, validatedConv.id);
if ("error" in convResult) {
return jsonResponse(createApiError(convResult.error, convResult.status), { mode, status: convResult.status });
}
if (convResult.conversation.projectId !== validatedProject.id) {
return jsonResponse(createApiError("会话不属于该项目", 403), { mode, status: 403 });
}
const result = deleteConversation(db, validatedConv.id);
if ("error" in result) {
return jsonResponse(createApiError(result.error, result.status), { mode, status: result.status });
}
return jsonResponse({ success: true }, { mode });
}

View File

@@ -0,0 +1,30 @@
import type Database from "bun:sqlite";
import type { RuntimeMode } from "../../../shared/api";
import { getConversation } from "../../db/conversations";
import { createApiError, jsonResponse } from "../../helpers";
import { validateIdParam } from "../../middleware";
export function handleGetConversation(req: Request, db: Database, mode: RuntimeMode): Response {
const parts = new URL(req.url).pathname.split("/");
const projectId = parts[3];
const conversationId = parts[5];
const validatedProject = validateIdParam(projectId ?? "", mode);
if (validatedProject instanceof Response) return validatedProject;
const validatedConv = validateIdParam(conversationId ?? "", mode);
if (validatedConv instanceof Response) return validatedConv;
const result = getConversation(db, validatedConv.id);
if ("error" in result) {
return jsonResponse(createApiError(result.error, result.status), { mode, status: result.status });
}
if (result.conversation.projectId !== validatedProject.id) {
return jsonResponse(createApiError("会话不属于该项目", 403), { mode, status: 403 });
}
return jsonResponse({ conversation: result.conversation }, { mode });
}

View File

@@ -0,0 +1,28 @@
import type Database from "bun:sqlite";
import type { RuntimeMode } from "../../../shared/api";
import { listConversations } from "../../db/conversations";
import { jsonResponse } from "../../helpers";
import { validateIdParam, validatePagination } from "../../middleware";
export function handleListConversations(req: Request, db: Database, mode: RuntimeMode): Response {
const url = new URL(req.url);
const projectId = url.pathname.split("/")[3];
const validated = validateIdParam(projectId ?? "", mode);
if (validated instanceof Response) return validated;
const pageParam = url.searchParams.get("page");
const pageSizeParam = url.searchParams.get("pageSize");
const pagination = validatePagination(pageParam, pageSizeParam, mode);
if (pagination instanceof Response) return pagination;
const result = listConversations(db, validated.id, {
page: pagination.page,
pageSize: pagination.pageSize,
});
return jsonResponse(result, { mode });
}

View File

@@ -0,0 +1,42 @@
import type Database from "bun:sqlite";
import type { RuntimeMode } from "../../../shared/api";
import { getConversation, listMessages } from "../../db/conversations";
import { createApiError, jsonResponse } from "../../helpers";
import { validateIdParam, validatePagination } from "../../middleware";
export function handleListMessages(req: Request, db: Database, mode: RuntimeMode): Response {
const parts = new URL(req.url).pathname.split("/");
const projectId = parts[3];
const conversationId = parts[5];
const validatedProject = validateIdParam(projectId ?? "", mode);
if (validatedProject instanceof Response) return validatedProject;
const validatedConv = validateIdParam(conversationId ?? "", mode);
if (validatedConv instanceof Response) return validatedConv;
const convResult = getConversation(db, validatedConv.id);
if ("error" in convResult) {
return jsonResponse(createApiError(convResult.error, convResult.status), { mode, status: convResult.status });
}
if (convResult.conversation.projectId !== validatedProject.id) {
return jsonResponse(createApiError("会话不属于该项目", 403), { mode, status: 403 });
}
const url = new URL(req.url);
const pageParam = url.searchParams.get("page");
const pageSizeParam = url.searchParams.get("pageSize");
const pagination = validatePagination(pageParam, pageSizeParam, mode);
if (pagination instanceof Response) return pagination;
const result = listMessages(db, validatedConv.id, {
page: pagination.page,
pageSize: pagination.pageSize,
});
return jsonResponse(result, { mode });
}

View File

@@ -0,0 +1,91 @@
import type Database from "bun:sqlite";
import type { RuntimeMode } from "../../../shared/api";
import type { IncomingMessage } from "../../ai/agent-stream";
import { agentStream, extractTextContent } from "../../ai/agent-stream";
import { createMessage, getConversation, updateConversationTimestamp } from "../../db/conversations";
import { createApiError, jsonResponse } from "../../helpers";
import { validateIdParam } from "../../middleware";
export async function handleSendChat(req: Request, db: Database, mode: RuntimeMode): Promise<Response> {
const url = new URL(req.url);
const projectId = url.pathname.split("/")[3];
const validated = validateIdParam(projectId ?? "", mode);
if (validated instanceof Response) return validated;
let body: { conversationId?: string; messages?: IncomingMessage[] };
try {
body = (await req.json()) as typeof body;
} catch {
return jsonResponse(createApiError("Invalid JSON body", 400), { mode, status: 400 });
}
if (!body.conversationId || typeof body.conversationId !== "string") {
return jsonResponse(createApiError("conversationId is required", 400), { mode, status: 400 });
}
if (!Array.isArray(body.messages) || body.messages.length === 0) {
return jsonResponse(createApiError("messages is required and must be a non-empty array", 400), {
mode,
status: 400,
});
}
const conversationResult = getConversation(db, body.conversationId);
if ("error" in conversationResult) {
return jsonResponse(createApiError(conversationResult.error, conversationResult.status), {
mode,
status: conversationResult.status,
});
}
const conversation = conversationResult.conversation;
if (conversation.projectId !== validated.id) {
return jsonResponse(createApiError("会话不属于该项目", 403), { mode, status: 403 });
}
for (const msg of body.messages ?? []) {
createMessage(db, {
content: extractTextContent(msg),
conversationId: conversation.id,
role: (msg.role ?? "user") as "assistant" | "system" | "user",
});
}
updateConversationTimestamp(db, conversation.id);
try {
const result = agentStream({
db,
messages: body.messages,
modelDbId: conversation.modelId,
});
const stream = result.toUIMessageStreamResponse();
const saveReply = async () => {
try {
const fullContent = await result.text;
if (fullContent) {
createMessage(db, {
content: fullContent,
conversationId: conversation.id,
role: "assistant",
});
updateConversationTimestamp(db, conversation.id);
}
} catch {
// stream ended without content, nothing to persist
}
};
void saveReply();
return stream;
} catch (e: unknown) {
const msg = e instanceof Error ? e.message : String(e);
return jsonResponse(createApiError(`AI 调用失败:${msg}`, 500), { mode, status: 500 });
}
}