MoerAI
83 lines
2.7 KiB
TypeScript
83 lines
2.7 KiB
TypeScript
import { describe, expect, it } from "bun:test"
|
|
import type { RuntimeFallbackPluginInput } from "./types"
|
|
import { hasVisibleAssistantResponse } from "./visible-assistant-response"
|
|
import { extractAutoRetrySignal } from "./error-classifier"
|
|
|
|
function createContext(messagesResponse: unknown): RuntimeFallbackPluginInput {
|
|
return {
|
|
client: {
|
|
session: {
|
|
abort: async () => ({}),
|
|
messages: async () => messagesResponse,
|
|
promptAsync: async () => ({}),
|
|
},
|
|
tui: {
|
|
showToast: async () => ({}),
|
|
},
|
|
},
|
|
directory: "/test/dir",
|
|
}
|
|
}
|
|
|
|
describe("hasVisibleAssistantResponse", () => {
|
|
it("#given only an old assistant reply before the latest user turn #when visibility is checked #then the stale reply is ignored", async () => {
|
|
// given
|
|
const checkVisibleResponse = hasVisibleAssistantResponse(() => undefined)
|
|
const ctx = createContext({
|
|
data: [
|
|
{ info: { role: "user" }, parts: [{ type: "text", text: "older question" }] },
|
|
{ info: { role: "assistant" }, parts: [{ type: "text", text: "older answer" }] },
|
|
{ info: { role: "user" }, parts: [{ type: "text", text: "latest question" }] },
|
|
],
|
|
})
|
|
|
|
// when
|
|
const result = await checkVisibleResponse(ctx, "session-old-assistant", undefined)
|
|
|
|
// then
|
|
expect(result).toBe(false)
|
|
})
|
|
|
|
it("#given an assistant reply after the latest user turn #when visibility is checked #then the current reply is treated as visible", async () => {
|
|
// given
|
|
const checkVisibleResponse = hasVisibleAssistantResponse(() => undefined)
|
|
const ctx = createContext({
|
|
data: [
|
|
{ info: { role: "user" }, parts: [{ type: "text", text: "latest question" }] },
|
|
{ info: { role: "assistant" }, parts: [{ type: "text", text: "visible answer" }] },
|
|
],
|
|
})
|
|
|
|
// when
|
|
const result = await checkVisibleResponse(ctx, "session-visible-assistant", undefined)
|
|
|
|
// then
|
|
expect(result).toBe(true)
|
|
})
|
|
|
|
it("#given a too-many-requests assistant reply #when visibility is checked #then it is treated as an auto-retry signal", async () => {
|
|
// given
|
|
const checkVisibleResponse = hasVisibleAssistantResponse(extractAutoRetrySignal)
|
|
const ctx = createContext({
|
|
data: [
|
|
{ info: { role: "user" }, parts: [{ type: "text", text: "latest question" }] },
|
|
{
|
|
info: { role: "assistant" },
|
|
parts: [
|
|
{
|
|
type: "text",
|
|
text: "Too Many Requests: Sorry, you've exhausted this model's rate limit. Please try a different model.",
|
|
},
|
|
],
|
|
},
|
|
],
|
|
})
|
|
|
|
// when
|
|
const result = await checkVisibleResponse(ctx, "session-rate-limit", undefined)
|
|
|
|
// then
|
|
expect(result).toBe(false)
|
|
})
|
|
})
|