44 lines
1.8 KiB
TypeScript
44 lines
1.8 KiB
TypeScript
import { afterAll, beforeAll, describe, expect, it } from "vitest";
|
|
import { createAgentModule } from "../../src/modules/agent/index.ts";
|
|
import { createHarness, type Harness, pgAvailable } from "../auth-settings/harness.ts";
|
|
import { DEV_KEY, json } from "../helpers.ts";
|
|
|
|
// 智能体页会话分栏:对话(用户发起)/ 自动巡检(定时、事件触发);子会话随父会话归类
|
|
const hasPg = await pgAvailable();
|
|
|
|
describe.skipIf(!hasPg)("会话列表 group 筛选(真实 PG)", () => {
|
|
let h: Harness;
|
|
beforeAll(async () => {
|
|
h = await createHarness({ extraModules: [createAgentModule()] });
|
|
}, 120_000);
|
|
afterAll(async () => {
|
|
await h?.close();
|
|
});
|
|
|
|
it("auto = schedule / event 及其子会话;conversation = 其余", async () => {
|
|
const run = async (trigger: string, goal: string, parent?: string) =>
|
|
(
|
|
await h.db
|
|
.insertInto("agent_run")
|
|
.values({ trigger, goal, status: "succeeded", mode: "execute", parent_run_id: parent ?? null } as never)
|
|
.returning("id")
|
|
.executeTakeFirstOrThrow()
|
|
).id;
|
|
const patrol = await run("schedule", "巡检");
|
|
await run("subagent", "巡检子会话", patrol);
|
|
await run("event", "事件");
|
|
const chat = await run("user", "对话");
|
|
await run("subagent", "对话子会话", chat);
|
|
const goals = async (group: string) =>
|
|
(
|
|
await json(
|
|
await h.app.request(`/api/v1/agent/runs?group=${group}`, { headers: { authorization: `Bearer ${DEV_KEY}` } }),
|
|
)
|
|
).data.items
|
|
.map((r: { goal: string }) => r.goal)
|
|
.sort();
|
|
expect(await goals("auto")).toEqual(["事件", "巡检", "巡检子会话"].sort());
|
|
expect(await goals("conversation")).toEqual(["对话", "对话子会话"].sort());
|
|
});
|
|
});
|