feat: add fallback

This commit is contained in:
eric
2026-02-23 15:16:15 +01:00
parent a4650c14ae
commit e26d665bdf
4 changed files with 134 additions and 51 deletions

View File

@@ -43,6 +43,32 @@ export class OpenRouterProvider implements AiProvider {
}
}
private getClassificationModelCandidates(): string[] {
const models = [
config.ai.classificationModel,
...config.ai.classificationFallbackModels,
config.ai.model,
];
return Array.from(new Set(models.map((model) => model.trim()).filter((model) => model.length > 0)));
}
private getErrorStatus(error: unknown): number | undefined {
const err = error as {
status?: number;
code?: number | string;
error?: { code?: number | string };
};
const code = typeof err.code === "number"
? err.code
: typeof err.error?.code === "number"
? err.error.code
: undefined;
return err.status ?? code;
}
async ask(options: AskOptions): Promise<AiResponse> {
const { prompt, systemPrompt, maxTokens, temperature } = options;
const model = config.ai.model;
@@ -237,13 +263,19 @@ The user's Discord ID is: ${context.userId}`;
* Classify a message to determine the appropriate response style
*/
async classifyMessage(message: string): Promise<MessageStyle> {
try {
const classification = await this.client.chat.completions.create({
model: config.ai.classificationModel,
messages: [
{
role: "user",
content: `Classify this message into exactly one category. Only respond with the category name, nothing else.
const models = this.getClassificationModelCandidates();
for (let i = 0; i < models.length; i++) {
const model = models[i];
const hasNextModel = i < models.length - 1;
try {
const classification = await this.client.chat.completions.create({
model,
messages: [
{
role: "user",
content: `Classify this message into exactly one category. Only respond with the category name, nothing else.
Message: "${message}"
@@ -255,44 +287,62 @@ Categories:
- helpful: User has a genuine question or needs actual help
Category:`,
},
],
max_tokens: 10,
temperature: 0.1,
});
},
],
max_tokens: 10,
temperature: 0.1,
});
const result = classification.choices[0]?.message?.content?.toLowerCase().trim() as MessageStyle;
// Validate the result is a valid style
if (STYLE_OPTIONS.includes(result)) {
logger.debug("Message classified", { style: result });
return result;
const result = classification.choices[0]?.message?.content?.toLowerCase().trim() as MessageStyle;
if (STYLE_OPTIONS.includes(result)) {
logger.debug("Message classified", { style: result, model });
return result;
}
logger.debug("Classification returned invalid style, defaulting to snarky", { result, model });
return "snarky";
} catch (error) {
if (hasNextModel) {
logger.warn("Classification model failed, trying fallback", {
method: "classifyMessage",
model,
nextModel: models[i + 1],
status: this.getErrorStatus(error),
});
continue;
}
logger.error("Failed to classify message", {
method: "classifyMessage",
model,
messageLength: message.length,
attempts: models.length,
});
logger.error("Classification error details", error);
}
logger.debug("Classification returned invalid style, defaulting to snarky", { result });
return "snarky";
} catch (error) {
logger.error("Failed to classify message", {
method: "classifyMessage",
model: config.ai.classificationModel,
messageLength: message.length,
});
logger.error("Classification error details", error);
return "snarky"; // Default to snarky on error
}
return "snarky";
}
/**
* Cheap binary classifier to detect if a message is directed at Joel
*/
async classifyJoelDirected(message: string): Promise<boolean> {
try {
const classification = await this.client.chat.completions.create({
model: config.ai.classificationModel,
messages: [
{
role: "user",
content: `Determine if this Discord message is directed at Joel (the bot), or talking about Joel in a way Joel should respond.
const models = this.getClassificationModelCandidates();
for (let i = 0; i < models.length; i++) {
const model = models[i];
const hasNextModel = i < models.length - 1;
try {
const classification = await this.client.chat.completions.create({
model,
messages: [
{
role: "user",
content: `Determine if this Discord message is directed at Joel (the bot), or talking about Joel in a way Joel should respond.
Only respond with one token: YES or NO.
@@ -303,22 +353,35 @@ Guidance:
Message: "${message}"
Answer:`,
},
],
max_tokens: 3,
temperature: 0,
});
},
],
max_tokens: 3,
temperature: 0,
});
const result = classification.choices[0]?.message?.content?.trim().toUpperCase();
return result?.startsWith("YES") ?? false;
} catch (error) {
logger.error("Failed to classify directed message", {
method: "classifyJoelDirected",
model: config.ai.classificationModel,
messageLength: message.length,
});
logger.error("Directed classification error details", error);
return false;
const result = classification.choices[0]?.message?.content?.trim().toUpperCase();
return result?.startsWith("YES") ?? false;
} catch (error) {
if (hasNextModel) {
logger.warn("Directed classification model failed, trying fallback", {
method: "classifyJoelDirected",
model,
nextModel: models[i + 1],
status: this.getErrorStatus(error),
});
continue;
}
logger.error("Failed to classify directed message", {
method: "classifyJoelDirected",
model,
messageLength: message.length,
attempts: models.length,
});
logger.error("Directed classification error details", error);
}
}
return false;
}
}