ai: new flow, crack consumption trackers 2

Author: OmarMcAdam

apps/backend/api/ai/chat.ts

@@ -7,6 +7,7 @@ import { PROMPTS } from "@potential/consts";
 import { desc, eq, trackableLogs, trackables } from "@potential/db";
 
 import type { AppContext } from "../index";
+import { createConsumptionTrackables } from "./functions";
 
 const ai = new Hono<AppContext>();
 
@@ -109,6 +110,7 @@ ai.post("/chat", async (c: Context<AppContext>) => {
               PROMPTS.TOOLS.GENERATE_CONSUMPTION_TRACKABLES.PARAMETERS,
             execute: async () => {
               console.log("🍕 GENERATE CONSUMPTION TRACKABLES");
+              await createConsumptionTrackables({ userId: user!.id });
               return { completed: true };
             },
           },

apps/backend/api/ai/functions.ts

@@ -0,0 +1,49 @@
+import type { InferInsertModel } from "@potential/db";
+import type { CloudTypeId } from "@potential/utils";
+import { and, db, eq, trackables } from "@potential/db";
+import { TRACKABLE_TEMPLATES } from "@potential/templates";
+import { cloudTypeIdGenerator } from "@potential/utils";
+
+export async function createConsumptionTrackables({
+  userId,
+}: {
+  userId: CloudTypeId<"user">;
+}) {
+  console.log("🍕 CREATING CONSUMPTION TRACKABLES FUNCTION");
+  const userTrackables = await db.query.trackables.findMany({
+    where: and(
+      eq(trackables.ownerId, userId),
+      eq(trackables.type, "consumption"),
+    ),
+    columns: {
+      id: true,
+    },
+  });
+  if (userTrackables.length > 0) {
+    return;
+  }
+
+  const templates = TRACKABLE_TEMPLATES.consumption;
+
+  const newTrackablesToInsertArray: InferInsertModel<typeof trackables>[] = [];
+  for (const template of Object.values(templates)) {
+    newTrackablesToInsertArray.push({
+      id: cloudTypeIdGenerator("trackable"),
+      ownerId: userId,
+      name: template.name,
+      description: template.description,
+      color: null,
+      configType: template.defaultConfig.type,
+      public: false,
+      type: "consumption",
+      subType: template.subType,
+      customConfig: template.defaultConfig,
+    });
+  }
+  console.log("🍕 INSERTING CONSUMPTION TRACKABLES", {
+    newTrackablesToInsertArray,
+  });
+  await db.insert(trackables).values(newTrackablesToInsertArray);
+
+  return;
+}

apps/backend/package.json

@@ -20,6 +20,7 @@
     "@potential/db": "workspace:*",
     "@potential/env": "workspace:*",
     "@potential/storage": "workspace:*",
+    "@potential/templates": "workspace:*",
     "@potential/trpc": "workspace:*",
     "@potential/utils": "workspace:*",
     "@react-email/components": "^0.0.38",

apps/mobile/src/app/(home)/new.tsx

@@ -17,7 +17,7 @@ import { Button } from "~/components/ui/button";
 import { Card } from "~/components/ui/card";
 import { LongText } from "~/components/ui/long-text";
 import { Text as UIText } from "~/components/ui/text";
-import { getAuthHeaders } from "~/utils/api";
+import { getAuthHeaders, queryClient, trpc } from "~/utils/api";
 import { getApiUrl } from "~/utils/base-url";
 import { iconColor } from "~/utils/ui";
 
@@ -243,7 +243,7 @@ function ToolInvocationComponent({
             </Card>
           );
         case "call": {
-          const args = toolInvocation.args as
+          const _args = toolInvocation.args as
             | { description?: string }
             | undefined;
           return (
@@ -353,10 +353,40 @@ export default function NewTrackableScreen() {
     },
   );
 
+  const processedToolCallIds = useRef(new Set<string>());
+
   useEffect(() => {
     if (scrollViewRef.current) {
       scrollViewRef.current.scrollToEnd({ animated: true });
     }
+
+    const lastMessage = messages[messages.length - 1];
+    if (lastMessage) {
+      for (const part of lastMessage.parts) {
+        if (
+          part.type === "tool-invocation" &&
+          part.toolInvocation.toolName === "generateConsumptionTrackables" &&
+          part.toolInvocation.state === "result" &&
+          !processedToolCallIds.current.has(part.toolInvocation.toolCallId)
+        ) {
+          queryClient
+            .invalidateQueries({
+              queryKey: trpc.trackables.getTrackablesForParentType.queryKey({
+                trackableParentType: "consumption",
+              }),
+            })
+            .then(() => {
+              processedToolCallIds.current.add(part.toolInvocation.toolCallId);
+            })
+            .catch((error: Error) => {
+              console.error(
+                "Failed to invalidate consumption trackables queries:",
+                error,
+              );
+            });
+        }
+      }
+    }
   }, [messages]);
 
   const onActualSubmit = () => {

packages/db/src/index.ts

@@ -1,3 +1,4 @@
+export type { InferInsertModel, InferSelectModel } from "drizzle-orm";
 export { alias } from "drizzle-orm/mysql-core";
 export * from "drizzle-orm/sql";
 export * from "./client";

packages/templates/src/templates/index.ts

@@ -1,12 +1,16 @@
 // packages/templates/src/templates/index.ts
 import { bodyTemplates } from "./body";
+import { consumptionTemplates } from "./consumption";
 import { mindTemplates } from "./mind";
 import { TRACKABLE_TEMPLATES } from "./templates-registry";
 
 // Populate the registry
 TRACKABLE_TEMPLATES.body = bodyTemplates;
 TRACKABLE_TEMPLATES.mind = mindTemplates;
+TRACKABLE_TEMPLATES.consumption = consumptionTemplates;
 
 export * from "./body";
+export * from "./consumption";
 export * from "./mind";
+
 export { TRACKABLE_TEMPLATES };