elias/core
1
0
Fork 0
mirror of https://github.com/eliasstepanik/core.git synced 2026-01-24 19:48:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: for new mcp connections create billing automatically

Browse Source
This commit is contained in:
Harshith Mullapudi 2025-10-02 11:26:22 +05:30
parent 27762262d2
commit f0debd5678
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/webapp/app/services/mcp.server.ts
View File

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