label message instead of thread

Author: elie222

apps/web/utils/ai/actions.ts

@@ -9,7 +9,7 @@ import { ActionType, type ExecutedRule } from "@prisma/client";
 import {
   archiveThread,
   getOrCreateLabel,
-  labelThread,
+  labelMessage,
   markReadThread,
 } from "@/utils/gmail/label";
 import { markSpam } from "@/utils/gmail/spam";
@@ -99,9 +99,9 @@ const label: ActionFunction<{ label: string } | any> = async (
 
   if (!label.id) throw new Error("Label not found and unable to create label");
 
-  await labelThread({
+  await labelMessage({
     gmail,
-    threadId: email.threadId,
+    messageId: email.id,
     addLabelIds: [label.id],
   });
 };

apps/web/utils/gmail/label.ts

@@ -46,6 +46,8 @@ export async function labelThread(options: {
     return;
   }
 
+  logger.trace("Labeling thread", { threadId, addLabelIds, removeLabelIds });
+
   return withGmailRetry(() =>
     gmail.users.threads.modify({
       userId: "me",

apps/web/utils/llms/model.ts

@@ -9,6 +9,9 @@ import { createOllama } from "ollama-ai-provider";
 import { env } from "@/env";
 import { Model, Provider, supportsOllama } from "@/utils/llms/config";
 import type { UserAIFields } from "@/utils/llms/types";
+import { createScopedLogger } from "@/utils/logger";
+
+const logger = createScopedLogger("llms/model");
 
 export function getModel(
   userAi: UserAIFields,
@@ -18,8 +21,17 @@ export function getModel(
   model: string;
   llmModel: LanguageModelV1;
 } {
-  if (useEconomyModel) return selectEconomyModel(userAi);
-  return selectModel(userAi);
+  const model = useEconomyModel
+    ? selectEconomyModel(userAi)
+    : selectModel(userAi);
+
+  logger.trace("Using model", {
+    useEconomyModel,
+    provider: model.provider,
+    model: model.model,
+  });
+
+  return model;
 }
 
 function selectModel(userAi: UserAIFields): {

apps/web/utils/reply-tracker/outbound.ts

@@ -91,7 +91,7 @@ export async function handleOutboundReply(
       logger,
     });
   } else {
-    logger.trace("No need to reply", { rationale: aiResult.rationale });
+    logger.trace("No need to reply");
   }
 }