Better no rule found handling

Author: elie222

apps/web/utils/ai/choose-rule/ai-choose-rule.ts

@@ -4,11 +4,11 @@ import { chatCompletionObject } from "@/utils/llms";
 import { stringifyEmail } from "@/utils/stringify-email";
 import type { EmailForLLM } from "@/utils/types";
 import { createScopedLogger } from "@/utils/logger";
-import { Braintrust } from "@/utils/braintrust";
+// import { Braintrust } from "@/utils/braintrust";
 
 const logger = createScopedLogger("ai-choose-rule");
 
-const braintrust = new Braintrust("choose-rule-2");
+// const braintrust = new Braintrust("choose-rule-2");
 
 type GetAiResponseOptions = {
  email: EmailForLLM;
@@ -98,7 +98,7 @@ ${emailSection}
  ],
  schema: z.object({
  reason: z.string(),
- ruleName: z.string(),
+ ruleName: z.string().nullish(),
  noMatchFound: z.boolean().nullish(),
  }),
  userEmail: emailAccount.email,
@@ -107,20 +107,20 @@ ${emailSection}
 
  logger.trace("Response", aiResponse.object);
 
- braintrust.insertToDataset({
- id: email.id,
- input: {
- email: emailSection,
- rules: rules.map((rule) => ({
- name: rule.name,
- instructions: rule.instructions,
- })),
- hasAbout: !!emailAccount.about,
- userAbout: emailAccount.about,
- userEmail: emailAccount.email,
- },
- expected: aiResponse.object.ruleName,
- });
+ // braintrust.insertToDataset({
+ //  id: email.id,
+ //  input: {
+ //  email: emailSection,
+ //  rules: rules.map((rule) => ({
+ //  name: rule.name,
+ //  instructions: rule.instructions,
+ //  })),
+ //  hasAbout: !!emailAccount.about,
+ //  userAbout: emailAccount.about,
+ //  userEmail: emailAccount.email,
+ //  },
+ //  expected: aiResponse.object.ruleName,
+ // });
 
  return aiResponse.object;
 }
@@ -149,7 +149,8 @@ export async function aiChooseRule<
 
  const selectedRule = aiResponse.ruleName
  ? rules.find(
- (rule) => rule.name.toLowerCase() === aiResponse.ruleName.toLowerCase(),
+ (rule) =>
+ rule.name.toLowerCase() === aiResponse.ruleName?.toLowerCase(),
  )
  : undefined;
 

apps/web/utils/ai/clean/ai-clean.ts

@@ -5,8 +5,8 @@ import { createScopedLogger } from "@/utils/logger";
 import type { EmailForLLM } from "@/utils/types";
 import { stringifyEmailSimple } from "@/utils/stringify-email";
 import { formatDateForLLM, formatRelativeTimeForLLM } from "@/utils/date";
-import { Braintrust } from "@/utils/braintrust";
 import { preprocessBooleanLike } from "@/utils/zod";
+// import { Braintrust } from "@/utils/braintrust";
 
 const logger = createScopedLogger("ai/clean");
 
@@ -18,7 +18,7 @@ const schema = z.object({
  // reasoning: z.string(),
 });
 
-const braintrust = new Braintrust("cleaner-1");
+// const braintrust = new Braintrust("cleaner-1");
 
 export async function aiClean({
  emailAccount,
@@ -103,11 +103,11 @@ The current date is ${currentDate}.
 
  logger.trace("Result", { response: aiResponse.object });
 
- braintrust.insertToDataset({
- id: messageId,
- input: { message, currentDate },
- expected: aiResponse.object,
- });
+ // braintrust.insertToDataset({
+ //  id: messageId,
+ //  input: { message, currentDate },
+ //  expected: aiResponse.object,
+ // });
 
  return aiResponse.object as { archive: boolean };
 }

version.txt

@@ -1 +1 @@
-v1.3.13
+v1.3.14