diff --git a/packages/tools/test/openai-middleware.unit.test.ts b/packages/tools/test/openai-middleware.unit.test.ts index e3249c8d..ce04e1c0 100644 --- a/packages/tools/test/openai-middleware.unit.test.ts +++ b/packages/tools/test/openai-middleware.unit.test.ts @@ -26,7 +26,7 @@ describe("OpenAI middleware memory context", () => { }), }), ) - const originalCreate = vi.fn(() => + const originalCreate = vi.fn((_body?: unknown) => Object.assign(Promise.resolve({ choices: [] }), { asResponse: async () => new Response(), }), @@ -53,8 +53,11 @@ describe("OpenAI middleware memory context", () => { ], }) - const forwarded = originalCreate.mock.calls[0]?.[0] - const content = String(forwarded.messages[0].content) + const forwarded = originalCreate.mock.calls[0]?.[0] as + | { messages: Array<{ role: string; content: unknown }> } + | undefined + expect(forwarded).toBeDefined() + const content = String(forwarded?.messages[0]?.content) expect(content).toContain("Be helpful.") expect(content).toContain("Fresh profile fact") expect(content).not.toContain("Stale profile fact") diff --git a/packages/tools/test/voltagent.unit.test.ts b/packages/tools/test/voltagent.unit.test.ts index 559f1fa1..4c9f8a8f 100644 --- a/packages/tools/test/voltagent.unit.test.ts +++ b/packages/tools/test/voltagent.unit.test.ts @@ -36,10 +36,13 @@ describe("VoltAgent memory context", () => { parts: [], }, ], - } as Parameters>[0] + } as unknown as Parameters>[0] const result = await hooks.onPrepareMessages?.(args) - const content = String(result?.messages?.[0]?.content ?? "") + const firstMessage = result?.messages?.[0] as + | Record + | undefined + const content = String(firstMessage?.content ?? "") expect(content).toContain("Be helpful.") expect(content).toContain("Fresh profile fact") expect(content).not.toContain("Stale profile fact")