From f0debd5678f3a10d7c27212a702c0474756be1d3 Mon Sep 17 00:00:00 2001
From: Harshith Mullapudi <harshithmullapudi@gmail.com>
Date: Thu, 2 Oct 2025 11:26:22 +0530
Subject: [PATCH] fix: for new mcp connections create billing automatically

---
 apps/webapp/app/services/mcp.server.ts | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/apps/webapp/app/services/mcp.server.ts b/apps/webapp/app/services/mcp.server.ts
index 721d3ed..fc1b627 100644
--- a/apps/webapp/app/services/mcp.server.ts
+++ b/apps/webapp/app/services/mcp.server.ts
@@ -14,6 +14,7 @@ import { callMemoryTool, memoryTools } from "~/utils/mcp/memory";
 import { logger } from "~/services/logger.service";
 import { type Response, type Request } from "express";
 import { getWorkspaceByUser } from "~/models/workspace.server";
+import { ensureBillingInitialized } from "./billing.server";
 
 const QueryParams = z.object({
   source: z.string().optional(),
@@ -195,6 +196,8 @@ export const handleMCPRequest = async (
   const workspace = await getWorkspaceByUser(userId);
   const workspaceId = workspace?.id as string;
 
+  await ensureBillingInitialized(workspaceId);
+
   try {
     let transport: StreamableHTTPServerTransport;
     let currentSessionId = sessionId;
@@ -263,6 +266,8 @@ export const handleSessionRequest = async (
       workspace?.id as string,
     ))
   ) {
+    await ensureBillingInitialized(workspace?.id as string);
+
     const sessionData = TransportManager.getSessionInfo(sessionId);
 
     if (sessionData.exists) {