PR updates

Author: jorgebaralt

apps/web/src/app/api/ai/journl-agent/route.ts

@@ -16,8 +16,10 @@ export async function POST(req: Request) {
 
     const result = await journlAgent.stream(messages, {
       onFinish: async (result) => {
-        const model = JSON.parse(result.request.body || "{}").model;
-        const provider = Object.keys(result.providerMetadata ?? {})[0] || "";
+        const modelData = await journlAgent.getModel();
+
+        const provider = modelData.provider;
+        const model = modelData.modelId;
 
         if (result.usage && session.user?.id) {
           await api.usage.trackAiModelUsage({

apps/web/src/app/api/supabase/embed-document/route.ts

@@ -108,26 +108,16 @@ export const POST = handler(zDocumentEmbeddingTask, async (payload) => {
       values: mDocument.map((chunk) => chunk.text),
     });
 
-    if (usage) {
-      try {
-        await api.usage.trackAiModelUsage({
-          metadata: {
-            document_id: document.id,
-            model_version: model.specificationVersion,
-          },
-          metrics: [{ quantity: usage.tokens, unit: "tokens" }],
-          model_id: model.modelId,
-          model_provider: model.provider,
-          user_id: document.user_id,
-        });
-      } catch (error) {
-        console.error(
-          "Failed to track AI usage for document embedding:",
-          error,
-        );
-        // Continue with embedding even if usage tracking fails
-      }
-    }
+    await api.usage.trackAiModelUsage({
+      metadata: {
+        document_id: document.id,
+        model_version: model.specificationVersion,
+      },
+      metrics: [{ quantity: usage.tokens, unit: "tokens" }],
+      model_id: model.modelId,
+      model_provider: model.provider,
+      user_id: document.user_id,
+    });
 
     const insertions: z.infer<typeof zInsertDocumentEmbedding>[] = [];
 

packages/db/src/index.ts

@@ -1,4 +1,2 @@
 export { alias } from "drizzle-orm/pg-core";
 export * from "drizzle-orm/sql";
-export * from "./client.js";
-export * from "./schema.js";

packages/db/src/usage/usage-event.schema.ts

@@ -8,7 +8,7 @@ type UsageEventMetrics = {
 
 export const UsageEventStatus = pgEnum("usage_event_status", [
   "pending",
-  "completed",
+  "processed",
   "failed",
 ]);