diff --git a/apps/backend/src/app/api/latest/ai/query/[mode]/route.ts b/apps/backend/src/app/api/latest/ai/query/[mode]/route.ts index 1c3c1047d..b8ee7e0b1 100644 --- a/apps/backend/src/app/api/latest/ai/query/[mode]/route.ts +++ b/apps/backend/src/app/api/latest/ai/query/[mode]/route.ts @@ -97,7 +97,6 @@ export const POST = createSmartRouteHandler({ userId: fullReq.auth?.user?.id, requestedToolsJson: JSON.stringify(toolNames), messagesJson: JSON.stringify(messages), - mcpCorrelationId: body.mcpCallMetadata ? correlationId : undefined, conversationId: conversationIdForLog, }; const startedAt = performance.now(); diff --git a/apps/backend/src/lib/ai/loggers/ai-proxy-logger.ts b/apps/backend/src/lib/ai/loggers/ai-proxy-logger.ts index c426c9283..3a5176751 100644 --- a/apps/backend/src/lib/ai/loggers/ai-proxy-logger.ts +++ b/apps/backend/src/lib/ai/loggers/ai-proxy-logger.ts @@ -46,7 +46,6 @@ export function buildProxyLogRow(fields: ProxyLogFields): AiQueryLogEntry { stepCount: 0, durationMs, errorMessage: responseStatus >= 400 ? `upstream ${responseStatus}` : undefined, - mcpCorrelationId: undefined, conversationId: undefined, }; } diff --git a/apps/backend/src/lib/ai/loggers/ai-query-logger.ts b/apps/backend/src/lib/ai/loggers/ai-query-logger.ts index 3d7db21ee..ccc8b9caf 100644 --- a/apps/backend/src/lib/ai/loggers/ai-query-logger.ts +++ b/apps/backend/src/lib/ai/loggers/ai-query-logger.ts @@ -73,7 +73,6 @@ export type AiQueryLogEntry = { stepCount: number, durationMs: bigint, errorMessage: string | undefined, - mcpCorrelationId: string | undefined, conversationId: string | undefined, }; @@ -104,7 +103,6 @@ export async function logAiQuery(entry: AiQueryLogEntry): Promise { sanitizeRequiredNumber("stepCount", entry.stepCount), entry.durationMs, opt(entry.errorMessage), - opt(entry.mcpCorrelationId), opt(entry.conversationId), ]); } diff --git a/apps/backend/src/lib/ai/types.ts b/apps/backend/src/lib/ai/types.ts index de63f069f..fe511c9e0 100644 --- a/apps/backend/src/lib/ai/types.ts +++ b/apps/backend/src/lib/ai/types.ts @@ -34,7 +34,6 @@ export type CommonLogFields = { userId: string | undefined, requestedToolsJson: string, messagesJson: string, - mcpCorrelationId: string | undefined, conversationId: string | undefined, }; diff --git a/apps/internal-tool/spacetimedb/src/index.ts b/apps/internal-tool/spacetimedb/src/index.ts index ce6c85713..8265b5a97 100644 --- a/apps/internal-tool/spacetimedb/src/index.ts +++ b/apps/internal-tool/spacetimedb/src/index.ts @@ -71,7 +71,6 @@ const aiQueryLog = table( stepCount: t.u32(), durationMs: t.u64(), errorMessage: t.string().optional(), - mcpCorrelationId: t.string().optional(), conversationId: t.string().optional(), } ); @@ -589,7 +588,6 @@ export const log_ai_query = spacetimedb.reducer( stepCount: t.u32(), durationMs: t.u64(), errorMessage: t.string().optional(), - mcpCorrelationId: t.string().optional(), conversationId: t.string().optional(), }, (ctx, args) => { @@ -623,7 +621,6 @@ export const log_ai_query = spacetimedb.reducer( stepCount: args.stepCount, durationMs: args.durationMs, errorMessage: args.errorMessage, - mcpCorrelationId: args.mcpCorrelationId, conversationId: args.conversationId, } as Parameters[0]); } diff --git a/apps/internal-tool/src/components/Usage.tsx b/apps/internal-tool/src/components/Usage.tsx index e9689cb0e..ae7e7cea9 100644 --- a/apps/internal-tool/src/components/Usage.tsx +++ b/apps/internal-tool/src/components/Usage.tsx @@ -642,7 +642,7 @@ export function Usage({ rows, connectionState, onSelect, selectedId }: Props) { {row.systemPromptId} - {row.mcpCorrelationId != null && ( + {row.conversationId != null && ( MCP )} {!row.isAuthenticated && ( diff --git a/apps/internal-tool/src/components/UsageDetail.tsx b/apps/internal-tool/src/components/UsageDetail.tsx index c508bcc4a..b07617786 100644 --- a/apps/internal-tool/src/components/UsageDetail.tsx +++ b/apps/internal-tool/src/components/UsageDetail.tsx @@ -91,7 +91,7 @@ export function UsageDetail({ row, onClose }: { row: AiQueryLogRow, onClose: () {isError && ( error )} - {row.mcpCorrelationId != null && ( + {row.conversationId != null && ( MCP )} @@ -142,7 +142,6 @@ export function UsageDetail({ row, onClose }: { row: AiQueryLogRow, onClose: () {row.projectId && } {row.userId && } {row.conversationId && } - {row.mcpCorrelationId && } 0 ? requestedTools.join(", ") : "—"} /> diff --git a/apps/internal-tool/src/module_bindings/log_ai_query_reducer.ts b/apps/internal-tool/src/module_bindings/log_ai_query_reducer.ts index e3a2e15b5..e3793d29f 100644 --- a/apps/internal-tool/src/module_bindings/log_ai_query_reducer.ts +++ b/apps/internal-tool/src/module_bindings/log_ai_query_reducer.ts @@ -35,6 +35,5 @@ export default { stepCount: __t.u32(), durationMs: __t.u64(), errorMessage: __t.option(__t.string()), - mcpCorrelationId: __t.option(__t.string()), conversationId: __t.option(__t.string()), }; diff --git a/apps/internal-tool/src/module_bindings/my_visible_ai_query_log_table.ts b/apps/internal-tool/src/module_bindings/my_visible_ai_query_log_table.ts index 5d6c3994f..37460d039 100644 --- a/apps/internal-tool/src/module_bindings/my_visible_ai_query_log_table.ts +++ b/apps/internal-tool/src/module_bindings/my_visible_ai_query_log_table.ts @@ -37,6 +37,5 @@ export default __t.row({ stepCount: __t.u32().name("step_count"), durationMs: __t.u64().name("duration_ms"), errorMessage: __t.option(__t.string()).name("error_message"), - mcpCorrelationId: __t.option(__t.string()).name("mcp_correlation_id"), conversationId: __t.option(__t.string()).name("conversation_id"), }); diff --git a/apps/internal-tool/src/module_bindings/types.ts b/apps/internal-tool/src/module_bindings/types.ts index 3fe92e636..572502ea9 100644 --- a/apps/internal-tool/src/module_bindings/types.ts +++ b/apps/internal-tool/src/module_bindings/types.ts @@ -37,7 +37,6 @@ export const AiQueryLog = __t.object("AiQueryLog", { stepCount: __t.u32(), durationMs: __t.u64(), errorMessage: __t.option(__t.string()), - mcpCorrelationId: __t.option(__t.string()), conversationId: __t.option(__t.string()), }); export type AiQueryLog = __Infer;