715 lines
22 KiB
TypeScript
715 lines
22 KiB
TypeScript
/// <reference types="vite/client" />
|
|
|
|
import { convexTest } from "convex-test";
|
|
import { generateText } from "ai";
|
|
import { beforeEach, describe, expect, test, vi } from "vitest";
|
|
import { api, internal } from "./_generated/api";
|
|
import type { Id } from "./_generated/dataModel";
|
|
import { buildToolTraceFromSteps } from "./savingsChat";
|
|
import schema from "./schema";
|
|
|
|
vi.mock("ai", async (importOriginal) => {
|
|
const actual = await importOriginal<typeof import("ai")>();
|
|
return {
|
|
...actual,
|
|
generateText: vi.fn(async (options: {
|
|
tools: {
|
|
get_transactions: { execute: (input: unknown) => Promise<unknown> };
|
|
summarize_spending: { execute: (input: unknown) => Promise<unknown> };
|
|
};
|
|
}) => {
|
|
const transactionInput = { from: "2026-02-01", to: "2026-02-28", limit: 2 };
|
|
const summaryInput = { from: "2026-02-01", to: "2026-02-28" };
|
|
const transactionOutput = await options.tools.get_transactions.execute(transactionInput);
|
|
const summaryOutput = await options.tools.summarize_spending.execute(summaryInput);
|
|
|
|
return {
|
|
text: "Agenten-Antwort",
|
|
steps: [
|
|
{
|
|
toolResults: [
|
|
{
|
|
toolName: "get_transactions",
|
|
input: transactionInput,
|
|
output: transactionOutput,
|
|
},
|
|
{
|
|
toolName: "summarize_spending",
|
|
input: summaryInput,
|
|
output: summaryOutput,
|
|
},
|
|
],
|
|
},
|
|
],
|
|
};
|
|
}),
|
|
};
|
|
});
|
|
|
|
const modules = import.meta.glob("./**/*.ts");
|
|
delete modules["./savingsChat.test.ts"];
|
|
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
});
|
|
|
|
describe("savingsChat.getContext", () => {
|
|
test("counts and sums every matching transaction before applying prompt limits", async () => {
|
|
const t = convexTest(schema, modules);
|
|
|
|
const seeded = await t.run(async (ctx) => {
|
|
const userId = await ctx.db.insert("users", {
|
|
name: "Test User",
|
|
email: "test@example.com",
|
|
});
|
|
const giroAccountId = await ctx.db.insert("accounts", {
|
|
userId,
|
|
name: "Girokonto",
|
|
type: "checking",
|
|
openingBalance: 0,
|
|
currency: "EUR",
|
|
isArchived: false,
|
|
});
|
|
const otherAccountId = await ctx.db.insert("accounts", {
|
|
userId,
|
|
name: "Tagesgeld",
|
|
type: "savings",
|
|
openingBalance: 0,
|
|
currency: "EUR",
|
|
isArchived: false,
|
|
});
|
|
|
|
const amounts: number[] = [];
|
|
const months = ["2025-12", "2026-01", "2026-02", "2026-03", "2026-04", "2026-05", "2026-06"];
|
|
for (let index = 0; index < 450; index++) {
|
|
const month = months[index % months.length];
|
|
const day = String((index % 27) + 1).padStart(2, "0");
|
|
const bookingDate = `${month}-${day}`;
|
|
const amount = index % 3 === 0 ? 100 : -25;
|
|
amounts.push(amount);
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId: giroAccountId,
|
|
bookingDate,
|
|
valueDate: bookingDate,
|
|
description: `Giro transaction ${index}`,
|
|
counterparty: "Counterparty",
|
|
amount,
|
|
isPending: false,
|
|
effectiveMonth: index % 10 === 0 ? undefined : bookingDate.slice(0, 7),
|
|
});
|
|
}
|
|
|
|
for (let index = 0; index < 50; index++) {
|
|
const bookingDate = `2026-06-${String((index % 27) + 1).padStart(2, "0")}`;
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId: otherAccountId,
|
|
bookingDate,
|
|
valueDate: bookingDate,
|
|
description: `Other account transaction ${index}`,
|
|
amount: 999,
|
|
isPending: false,
|
|
effectiveMonth: bookingDate.slice(0, 7),
|
|
});
|
|
}
|
|
|
|
return {
|
|
userId,
|
|
giroAccountId,
|
|
expectedIncome: amounts.filter((amount) => amount > 0).reduce((sum, amount) => sum + amount, 0),
|
|
expectedExpenses: amounts.filter((amount) => amount < 0).reduce((sum, amount) => sum + amount, 0),
|
|
expectedBalance: amounts.reduce((sum, amount) => sum + amount, 0),
|
|
};
|
|
});
|
|
|
|
const asUser = t.withIdentity({
|
|
subject: `${seeded.userId}|test-session`,
|
|
tokenIdentifier: `test:${seeded.userId}`,
|
|
});
|
|
|
|
const context = await asUser.query(api.savingsChat.getContext, {
|
|
from: "2025-12-01",
|
|
to: "2026-06-30",
|
|
accountId: seeded.giroAccountId as Id<"accounts">,
|
|
basis: "effective",
|
|
});
|
|
|
|
expect(context.accountName).toBe("Girokonto");
|
|
expect(context.isComplete).toBe(true);
|
|
expect(context.totals.transactionCount).toBe(450);
|
|
expect(context.totals.income).toBe(seeded.expectedIncome);
|
|
expect(context.totals.expenses).toBe(seeded.expectedExpenses);
|
|
expect(context.totals.balance).toBe(seeded.expectedBalance);
|
|
});
|
|
|
|
test("builds complete prompt lines for every matching transaction", async () => {
|
|
const t = convexTest(schema, modules);
|
|
|
|
const seeded = await t.run(async (ctx) => {
|
|
const userId = await ctx.db.insert("users", {
|
|
name: "Prompt User",
|
|
email: "prompt@example.com",
|
|
});
|
|
const categoryId = await ctx.db.insert("categories", {
|
|
userId,
|
|
name: "Lebensmittel",
|
|
kind: "ausgabe",
|
|
block: "variabel",
|
|
color: "#22c55e",
|
|
sortOrder: 1,
|
|
isSystem: false,
|
|
});
|
|
const giroAccountId = await ctx.db.insert("accounts", {
|
|
userId,
|
|
name: "Girokonto",
|
|
type: "checking",
|
|
openingBalance: 0,
|
|
currency: "EUR",
|
|
isArchived: false,
|
|
});
|
|
const otherAccountId = await ctx.db.insert("accounts", {
|
|
userId,
|
|
name: "Depot",
|
|
type: "investment",
|
|
openingBalance: 0,
|
|
currency: "EUR",
|
|
isArchived: false,
|
|
});
|
|
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId: giroAccountId,
|
|
categoryId,
|
|
bookingDate: "2026-02-14",
|
|
valueDate: "2026-02-14",
|
|
description: "Supermarkt",
|
|
counterparty: "Markt GmbH",
|
|
amount: -42.5,
|
|
isPending: false,
|
|
});
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId: giroAccountId,
|
|
bookingDate: "2026-02-15",
|
|
valueDate: "2026-02-15",
|
|
description: "Gehalt",
|
|
counterparty: "Arbeitgeber",
|
|
amount: 2500,
|
|
isPending: false,
|
|
effectiveMonth: "2026-02",
|
|
});
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId: otherAccountId,
|
|
bookingDate: "2026-02-16",
|
|
valueDate: "2026-02-16",
|
|
description: "Other account should not appear",
|
|
amount: 999,
|
|
isPending: false,
|
|
effectiveMonth: "2026-02",
|
|
});
|
|
|
|
return { userId, giroAccountId };
|
|
});
|
|
|
|
const asUser = t.withIdentity({
|
|
subject: `${seeded.userId}|test-session`,
|
|
tokenIdentifier: `test:${seeded.userId}`,
|
|
});
|
|
|
|
const context = await asUser.query(internal.savingsChat.getPromptContext, {
|
|
from: "2026-02-01",
|
|
to: "2026-02-28",
|
|
accountId: seeded.giroAccountId as Id<"accounts">,
|
|
basis: "effective",
|
|
});
|
|
|
|
expect(context.totals.transactionCount).toBe(2);
|
|
expect(context.transactionLines).toHaveLength(2);
|
|
expect(context.transactionLines.join("\n")).toContain(
|
|
"2026-02-14 | Supermarkt (Markt GmbH) | -42.50€ | Lebensmittel | Girokonto",
|
|
);
|
|
expect(context.transactionLines.join("\n")).toContain(
|
|
"2026-02-15 | Gehalt (Arbeitgeber) | 2500.00€ | Ohne Kategorie | Girokonto",
|
|
);
|
|
expect(context.transactionLines.join("\n")).not.toContain("Other account should not appear");
|
|
});
|
|
});
|
|
|
|
describe("savingsChat read-only agent tools", () => {
|
|
test("getTransactionsTool applies account scope, exact totals, limits, and sanitizes rows", async () => {
|
|
const t = convexTest(schema, modules);
|
|
|
|
const seeded = await t.run(async (ctx) => {
|
|
const userId = await ctx.db.insert("users", {
|
|
name: "Tool User",
|
|
email: "tool@example.com",
|
|
});
|
|
const categoryId = await ctx.db.insert("categories", {
|
|
userId,
|
|
name: "Lebensmittel",
|
|
kind: "ausgabe",
|
|
block: "variabel",
|
|
color: "#22c55e",
|
|
sortOrder: 1,
|
|
isSystem: false,
|
|
});
|
|
const giroAccountId = await ctx.db.insert("accounts", {
|
|
userId,
|
|
name: "Girokonto",
|
|
type: "checking",
|
|
openingBalance: 0,
|
|
currency: "EUR",
|
|
isArchived: false,
|
|
});
|
|
const savingsAccountId = await ctx.db.insert("accounts", {
|
|
userId,
|
|
name: "Tagesgeld",
|
|
type: "savings",
|
|
openingBalance: 0,
|
|
currency: "EUR",
|
|
isArchived: false,
|
|
});
|
|
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId: giroAccountId,
|
|
categoryId,
|
|
bookingDate: "2026-01-05",
|
|
valueDate: "2026-01-05",
|
|
description: "Supermarkt",
|
|
counterparty: "Markt GmbH",
|
|
amount: -100,
|
|
isPending: false,
|
|
rawText: "RAW BANK PAYLOAD",
|
|
notes: "private note",
|
|
dedupHash: "secret-hash",
|
|
externalRef: "external-ref",
|
|
effectiveMonth: "2026-01",
|
|
});
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId: giroAccountId,
|
|
bookingDate: "2026-01-07",
|
|
valueDate: "2026-01-07",
|
|
description: "Baecker",
|
|
amount: -15,
|
|
isPending: false,
|
|
effectiveMonth: "2026-01",
|
|
});
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId: giroAccountId,
|
|
bookingDate: "2026-01-03",
|
|
valueDate: "2026-01-03",
|
|
description: "Kaffee",
|
|
amount: -5,
|
|
isPending: false,
|
|
effectiveMonth: "2026-01",
|
|
});
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId: savingsAccountId,
|
|
bookingDate: "2026-01-08",
|
|
valueDate: "2026-01-08",
|
|
description: "Other account",
|
|
amount: 999,
|
|
isPending: false,
|
|
effectiveMonth: "2026-01",
|
|
});
|
|
|
|
return { userId, giroAccountId };
|
|
});
|
|
|
|
const asUser = t.withIdentity({
|
|
subject: `${seeded.userId}|test-session`,
|
|
tokenIdentifier: `test:${seeded.userId}`,
|
|
});
|
|
|
|
const result = await asUser.query(internal.savingsChat.getTransactionsTool, {
|
|
scope: {
|
|
from: "2026-01-01",
|
|
to: "2026-01-31",
|
|
accountId: seeded.giroAccountId as Id<"accounts">,
|
|
basis: "booking",
|
|
},
|
|
limit: 2,
|
|
});
|
|
|
|
expect(result.totalCount).toBe(3);
|
|
expect(result.hasMore).toBe(true);
|
|
expect(result.rows).toHaveLength(2);
|
|
expect(result.totals).toEqual({
|
|
transactionCount: 3,
|
|
income: 0,
|
|
expenses: -120,
|
|
balance: -120,
|
|
});
|
|
expect(result.rows[0].accountName).toBe("Girokonto");
|
|
expect(result.rows.find((row) => row.description === "Supermarkt")?.categoryName).toBe(
|
|
"Lebensmittel",
|
|
);
|
|
expect(Object.keys(result.rows[0])).not.toContain("rawText");
|
|
expect(Object.keys(result.rows[0])).not.toContain("notes");
|
|
expect(Object.keys(result.rows[0])).not.toContain("dedupHash");
|
|
expect(Object.keys(result.rows[0])).not.toContain("externalRef");
|
|
expect(Object.keys(result.rows[0])).not.toContain("userId");
|
|
});
|
|
|
|
test("getTransactionsTool includes legacy effective-month rows via booking fallback", async () => {
|
|
const t = convexTest(schema, modules);
|
|
|
|
const seeded = await t.run(async (ctx) => {
|
|
const userId = await ctx.db.insert("users", {
|
|
name: "Legacy User",
|
|
email: "legacy@example.com",
|
|
});
|
|
const accountId = await ctx.db.insert("accounts", {
|
|
userId,
|
|
name: "Girokonto",
|
|
type: "checking",
|
|
openingBalance: 0,
|
|
currency: "EUR",
|
|
isArchived: false,
|
|
});
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId,
|
|
bookingDate: "2026-03-12",
|
|
valueDate: "2026-03-12",
|
|
description: "Legacy no effective month",
|
|
amount: -30,
|
|
isPending: false,
|
|
});
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId,
|
|
bookingDate: "2026-03-20",
|
|
valueDate: "2026-03-20",
|
|
description: "Modern effective month",
|
|
amount: -20,
|
|
isPending: false,
|
|
effectiveMonth: "2026-03",
|
|
});
|
|
return { userId, accountId };
|
|
});
|
|
|
|
const asUser = t.withIdentity({
|
|
subject: `${seeded.userId}|test-session`,
|
|
tokenIdentifier: `test:${seeded.userId}`,
|
|
});
|
|
|
|
const result = await asUser.query(internal.savingsChat.getTransactionsTool, {
|
|
scope: {
|
|
from: "2026-03-01",
|
|
to: "2026-03-31",
|
|
accountId: seeded.accountId as Id<"accounts">,
|
|
basis: "effective",
|
|
},
|
|
limit: 10,
|
|
});
|
|
|
|
expect(result.totalCount).toBe(2);
|
|
expect(result.totals.expenses).toBe(-50);
|
|
expect(result.rows.map((row) => row.description)).toContain("Legacy no effective month");
|
|
});
|
|
|
|
test("summarizeSpendingTool computes exact monthly and category aggregates", async () => {
|
|
const t = convexTest(schema, modules);
|
|
|
|
const seeded = await t.run(async (ctx) => {
|
|
const userId = await ctx.db.insert("users", {
|
|
name: "Summary User",
|
|
email: "summary@example.com",
|
|
});
|
|
const accountId = await ctx.db.insert("accounts", {
|
|
userId,
|
|
name: "Girokonto",
|
|
type: "checking",
|
|
openingBalance: 0,
|
|
currency: "EUR",
|
|
isArchived: false,
|
|
});
|
|
const rentId = await ctx.db.insert("categories", {
|
|
userId,
|
|
name: "Miete",
|
|
kind: "ausgabe",
|
|
block: "wiederkehrend",
|
|
color: "#64748b",
|
|
sortOrder: 1,
|
|
isSystem: false,
|
|
});
|
|
const foodId = await ctx.db.insert("categories", {
|
|
userId,
|
|
name: "Lebensmittel",
|
|
kind: "ausgabe",
|
|
block: "variabel",
|
|
color: "#22c55e",
|
|
sortOrder: 2,
|
|
isSystem: false,
|
|
});
|
|
const salaryId = await ctx.db.insert("categories", {
|
|
userId,
|
|
name: "Gehalt",
|
|
kind: "einnahme",
|
|
color: "#0ea5e9",
|
|
sortOrder: 3,
|
|
isSystem: false,
|
|
});
|
|
|
|
for (const tx of [
|
|
{ date: "2026-01-01", description: "Gehalt Januar", amount: 3000, categoryId: salaryId },
|
|
{ date: "2026-01-02", description: "Miete Januar", amount: -1000, categoryId: rentId },
|
|
{ date: "2026-01-10", description: "Supermarkt Januar", amount: -200, categoryId: foodId },
|
|
{ date: "2026-02-01", description: "Gehalt Februar", amount: 3000, categoryId: salaryId },
|
|
{ date: "2026-02-02", description: "Sonstiges Februar", amount: -50, categoryId: undefined },
|
|
]) {
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId,
|
|
categoryId: tx.categoryId,
|
|
bookingDate: tx.date,
|
|
valueDate: tx.date,
|
|
description: tx.description,
|
|
amount: tx.amount,
|
|
isPending: false,
|
|
effectiveMonth: tx.date.slice(0, 7),
|
|
});
|
|
}
|
|
|
|
return { userId, accountId };
|
|
});
|
|
|
|
const asUser = t.withIdentity({
|
|
subject: `${seeded.userId}|test-session`,
|
|
tokenIdentifier: `test:${seeded.userId}`,
|
|
});
|
|
|
|
const result = await asUser.query(internal.savingsChat.summarizeSpendingTool, {
|
|
scope: {
|
|
from: "2026-01-01",
|
|
to: "2026-02-28",
|
|
accountId: seeded.accountId as Id<"accounts">,
|
|
basis: "effective",
|
|
},
|
|
});
|
|
|
|
expect(result.totals).toEqual({
|
|
transactionCount: 5,
|
|
income: 6000,
|
|
expenses: -1250,
|
|
balance: 4750,
|
|
});
|
|
expect(result.fixedCosts).toBe(-1000);
|
|
expect(result.variableCosts).toBe(-200);
|
|
expect(result.monthlyTrend).toEqual([
|
|
{ month: "2026-01", income: 3000, expenses: -1200, balance: 1800 },
|
|
{ month: "2026-02", income: 3000, expenses: -50, balance: 2950 },
|
|
]);
|
|
expect(result.categoryBreakdown.map((entry) => [entry.name, entry.amount])).toEqual([
|
|
["Miete", -1000],
|
|
["Lebensmittel", -200],
|
|
["Ohne Kategorie", -50],
|
|
]);
|
|
});
|
|
|
|
test("forecastCashflowTool excludes partial current month from the baseline", async () => {
|
|
const t = convexTest(schema, modules);
|
|
|
|
const seeded = await t.run(async (ctx) => {
|
|
const userId = await ctx.db.insert("users", {
|
|
name: "Forecast User",
|
|
email: "forecast@example.com",
|
|
});
|
|
const accountId = await ctx.db.insert("accounts", {
|
|
userId,
|
|
name: "Girokonto",
|
|
type: "checking",
|
|
openingBalance: 0,
|
|
currency: "EUR",
|
|
isArchived: false,
|
|
});
|
|
|
|
for (const tx of [
|
|
{ date: "2026-04-01", description: "Gehalt April", amount: 3000 },
|
|
{ date: "2026-04-10", description: "Kosten April", amount: -2000 },
|
|
{ date: "2026-05-01", description: "Gehalt Mai", amount: 3200 },
|
|
{ date: "2026-05-10", description: "Kosten Mai", amount: -2200 },
|
|
{ date: "2026-06-01", description: "Gehalt Juni", amount: 3000 },
|
|
{ date: "2026-06-10", description: "Teilkosten Juni", amount: -500 },
|
|
]) {
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId,
|
|
bookingDate: tx.date,
|
|
valueDate: tx.date,
|
|
description: tx.description,
|
|
amount: tx.amount,
|
|
isPending: false,
|
|
effectiveMonth: tx.date.slice(0, 7),
|
|
});
|
|
}
|
|
|
|
return { userId, accountId };
|
|
});
|
|
|
|
const asUser = t.withIdentity({
|
|
subject: `${seeded.userId}|test-session`,
|
|
tokenIdentifier: `test:${seeded.userId}`,
|
|
});
|
|
|
|
const result = await asUser.query(internal.savingsChat.forecastCashflowTool, {
|
|
scope: {
|
|
from: "2026-04-01",
|
|
to: "2026-06-30",
|
|
accountId: seeded.accountId as Id<"accounts">,
|
|
basis: "effective",
|
|
},
|
|
horizonMonths: 3,
|
|
asOf: "2026-06-15",
|
|
});
|
|
|
|
expect(result.baselineMonths).toEqual(["2026-04", "2026-05"]);
|
|
expect(result.excludedPartialMonth).toBe("2026-06");
|
|
expect(result.monthlyAverage).toEqual({ income: 3100, expenses: -2100, balance: 1000 });
|
|
expect(result.projection).toEqual([
|
|
{ month: "2026-07", income: 3100, expenses: -2100, balance: 1000 },
|
|
{ month: "2026-08", income: 3100, expenses: -2100, balance: 1000 },
|
|
{ month: "2026-09", income: 3100, expenses: -2100, balance: 1000 },
|
|
]);
|
|
});
|
|
|
|
test("buildToolTraceFromSteps returns compact summaries without raw tool output", () => {
|
|
const trace = buildToolTraceFromSteps([
|
|
{
|
|
toolResults: [
|
|
{
|
|
toolName: "get_transactions",
|
|
input: { from: "2026-01-01", to: "2026-01-31", limit: 50 },
|
|
output: {
|
|
totalCount: 2,
|
|
hasMore: false,
|
|
totals: { income: 0, expenses: -115, balance: -115, transactionCount: 2 },
|
|
rows: [
|
|
{ description: "Supermarkt", amount: -100 },
|
|
{ description: "Baecker", amount: -15 },
|
|
],
|
|
},
|
|
},
|
|
],
|
|
},
|
|
]);
|
|
|
|
expect(trace).toEqual([
|
|
{
|
|
name: "get_transactions",
|
|
inputSummary: "2026-01-01 bis 2026-01-31, Limit 50",
|
|
resultSummary: "2 Umsätze, Saldo -115.00€, vollständig",
|
|
},
|
|
]);
|
|
});
|
|
|
|
test("ask returns compact tool traces from executed read-only tools", async () => {
|
|
const t = convexTest(schema, modules);
|
|
const previousKey = process.env.OPENAI_API_KEY;
|
|
const previousModel = process.env.SAVINGS_CHAT_MODEL;
|
|
|
|
const seeded = await t.run(async (ctx) => {
|
|
const userId = await ctx.db.insert("users", {
|
|
name: "Ask User",
|
|
email: "ask@example.com",
|
|
});
|
|
const accountId = await ctx.db.insert("accounts", {
|
|
userId,
|
|
name: "Girokonto",
|
|
type: "checking",
|
|
openingBalance: 0,
|
|
currency: "EUR",
|
|
isArchived: false,
|
|
});
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId,
|
|
bookingDate: "2026-02-01",
|
|
valueDate: "2026-02-01",
|
|
description: "Gehalt",
|
|
amount: 3000,
|
|
isPending: false,
|
|
effectiveMonth: "2026-02",
|
|
rawText: "RAW PAYLOAD SHOULD NOT LEAK",
|
|
});
|
|
await ctx.db.insert("transactions", {
|
|
userId,
|
|
accountId,
|
|
bookingDate: "2026-02-10",
|
|
valueDate: "2026-02-10",
|
|
description: "Supermarkt",
|
|
amount: -120,
|
|
isPending: false,
|
|
effectiveMonth: "2026-02",
|
|
notes: "private note should not leak",
|
|
});
|
|
return { userId, accountId };
|
|
});
|
|
|
|
try {
|
|
process.env.OPENAI_API_KEY = "test-key";
|
|
delete process.env.SAVINGS_CHAT_MODEL;
|
|
const asUser = t.withIdentity({
|
|
subject: `${seeded.userId}|test-session`,
|
|
tokenIdentifier: `test:${seeded.userId}`,
|
|
});
|
|
|
|
const result = await asUser.action(api.savingsChat.ask, {
|
|
messages: [{ role: "user", content: "Wie sieht Februar aus?" }],
|
|
from: "2026-02-01",
|
|
to: "2026-02-28",
|
|
accountId: seeded.accountId as Id<"accounts">,
|
|
basis: "effective",
|
|
});
|
|
|
|
expect(result.answer).toBe("Agenten-Antwort");
|
|
expect(result.model).toBe("gpt-5.4-mini");
|
|
expect(result.usedTransactions).toBe(2);
|
|
expect(result.usedBalance).toEqual({ income: 3000, expenses: -120, balance: 2880 });
|
|
expect(result.toolTrace).toEqual([
|
|
{
|
|
name: "get_transactions",
|
|
inputSummary: "2026-02-01 bis 2026-02-28, Limit 2",
|
|
resultSummary: "2 Umsätze, Saldo 2880.00€, vollständig",
|
|
},
|
|
{
|
|
name: "summarize_spending",
|
|
inputSummary: "2026-02-01 bis 2026-02-28",
|
|
resultSummary: "2 Umsätze, Saldo 2880.00€, 1 Kategorien",
|
|
},
|
|
]);
|
|
expect(JSON.stringify(result.toolTrace)).not.toContain("RAW PAYLOAD");
|
|
expect(JSON.stringify(result.toolTrace)).not.toContain("private note");
|
|
|
|
expect(generateText).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
tools: expect.objectContaining({
|
|
get_transactions: expect.any(Object),
|
|
summarize_spending: expect.any(Object),
|
|
forecast_cashflow: expect.any(Object),
|
|
}),
|
|
stopWhen: expect.any(Function),
|
|
}),
|
|
);
|
|
} finally {
|
|
if (previousKey === undefined) {
|
|
delete process.env.OPENAI_API_KEY;
|
|
} else {
|
|
process.env.OPENAI_API_KEY = previousKey;
|
|
}
|
|
if (previousModel === undefined) {
|
|
delete process.env.SAVINGS_CHAT_MODEL;
|
|
} else {
|
|
process.env.SAVINGS_CHAT_MODEL = previousModel;
|
|
}
|
|
}
|
|
});
|
|
});
|