fix: relocate and fix counter when branch

Author: NamesMT

apps/backend-convex/convex/_counters.ts

@@ -8,7 +8,6 @@
  * @module
  */
 
-import type * as _counters from "../_counters.js";
 import type * as authInfo from "../authInfo.js";
 import type * as crons from "../crons.js";
 import type * as http_ai from "../http/ai.js";
@@ -32,7 +31,6 @@ import type {
  * ```
  */
 declare const fullApi: ApiFromModules<{
-  _counters: typeof _counters;
   authInfo: typeof authInfo;
   crons: typeof crons;
   "http/ai": typeof http_ai;

apps/backend-convex/convex/_generated/api.d.ts

@@ -1,6 +1,6 @@
 import { objectPick } from '@namesmt/utils'
 import { ConvexError, v } from 'convex/values'
-import { messagesInThreadCounter } from './_counters'
+import { singleShardCounter } from '../utils/counters'
 import { internalMutation, internalQuery, query } from './_generated/server'
 import { assertThreadAccess } from './threads'
 
@@ -37,7 +37,7 @@ export const countByThread = query({
 
     await assertThreadAccess(ctx, { thread, lockerKey: args.lockerKey })
 
-    return await messagesInThreadCounter.count(ctx, args.threadId)
+    return await singleShardCounter.count(ctx, `messages-in-thread_${args.threadId}`)
   },
 })
 
@@ -77,7 +77,7 @@ export const internalAdd = internalMutation({
     lockerKey: v.optional(v.string()),
   },
   handler: async (ctx, args) => {
-    await messagesInThreadCounter.inc(ctx, args.threadId)
+    await singleShardCounter.inc(ctx, `messages-in-thread_${args.threadId}`)
 
     return await ctx.db.insert('messages', {
       ...objectPick(args, ['threadId', 'role', 'content', 'context', 'isStreaming', 'streamId', 'provider', 'model']),

apps/backend-convex/convex/messages.ts

@@ -6,7 +6,7 @@ import { clearUndefined } from '@namesmt/utils'
 import { openrouter } from '@openrouter/ai-sdk-provider'
 import { generateText } from 'ai'
 import { ConvexError, v } from 'convex/values'
-import { messagesInThreadCounter } from './_counters'
+import { singleShardCounter } from '../utils/counters'
 import { api, internal } from './_generated/api'
 import { action, internalMutation, mutation, query } from './_generated/server'
 
@@ -139,7 +139,7 @@ export const branchThreadFromMessage = mutation({
       })
     }))
 
-    await messagesInThreadCounter.add(ctx, thread._id, messages.length)
+    await singleShardCounter.add(ctx, `messages-in-thread_${newThreadId}`, messages.length)
 
     return newThreadId
   },

apps/backend-convex/convex/threads.ts

@@ -0,0 +1,4 @@
+import { ShardedCounter } from '@convex-dev/sharded-counter'
+import { components } from '../convex/_generated/api'
+
+export const singleShardCounter = new ShardedCounter(components.shardedCounter, { defaultShards: 1 })