@@ -26,6 +26,7 @@ import { ModelMessage } from 'ai';
2626import { FastifyReply } from 'fastify' ;
2727import { StatusCodes } from 'http-status-codes' ;
2828import removeMarkdown from 'markdown-to-text' ;
29+ import { getProjectScopedRoutePolicy } from '../../core/security/route-policies/route-security-policy-factory' ;
2930import { extractUiToolResultsFromMessage } from '../mcp/tool-utils' ;
3031import {
3132 createChatContext ,
@@ -470,6 +471,9 @@ export const aiMCPChatController: FastifyPluginAsyncTypebox = async (app) => {
470471const OpenChatOptions = {
471472 config : {
472473 allowedPrincipals : [ PrincipalType . USER ] ,
474+ security : getProjectScopedRoutePolicy ( {
475+ allowedPrincipals : [ PrincipalType . USER ] ,
476+ } ) ,
473477 } ,
474478 schema : {
475479 tags : [ 'ai' , 'ai-chat-mcp' ] ,
@@ -482,6 +486,9 @@ const OpenChatOptions = {
482486const NewMessageOptions = {
483487 config : {
484488 allowedPrincipals : [ PrincipalType . USER ] ,
489+ security : getProjectScopedRoutePolicy ( {
490+ allowedPrincipals : [ PrincipalType . USER ] ,
491+ } ) ,
485492 } ,
486493 schema : {
487494 tags : [ 'ai' , 'ai-chat-mcp' ] ,
@@ -494,6 +501,9 @@ const NewMessageOptions = {
494501const ChatNameOptions = {
495502 config : {
496503 allowedPrincipals : [ PrincipalType . USER ] ,
504+ security : getProjectScopedRoutePolicy ( {
505+ allowedPrincipals : [ PrincipalType . USER ] ,
506+ } ) ,
497507 } ,
498508 schema : {
499509 tags : [ 'ai' , 'ai-chat-mcp' ] ,
@@ -505,6 +515,9 @@ const ChatNameOptions = {
505515const CodeGenerationOptions = {
506516 config : {
507517 allowedPrincipals : [ PrincipalType . USER ] ,
518+ security : getProjectScopedRoutePolicy ( {
519+ allowedPrincipals : [ PrincipalType . USER ] ,
520+ } ) ,
508521 } ,
509522 schema : {
510523 tags : [ 'ai' , 'ai-chat' ] ,
@@ -517,6 +530,9 @@ const CodeGenerationOptions = {
517530const RenameChatOptions = {
518531 config : {
519532 allowedPrincipals : [ PrincipalType . USER ] ,
533+ security : getProjectScopedRoutePolicy ( {
534+ allowedPrincipals : [ PrincipalType . USER ] ,
535+ } ) ,
520536 } ,
521537 schema : {
522538 tags : [ 'ai' , 'ai-chat-mcp' ] ,
@@ -529,6 +545,9 @@ const RenameChatOptions = {
529545const DeleteChatOptions = {
530546 config : {
531547 allowedPrincipals : [ PrincipalType . USER ] ,
548+ security : getProjectScopedRoutePolicy ( {
549+ allowedPrincipals : [ PrincipalType . USER ] ,
550+ } ) ,
532551 } ,
533552 schema : {
534553 tags : [ 'ai' , 'ai-chat-mcp' ] ,
@@ -541,6 +560,9 @@ const DeleteChatOptions = {
541560const ListChatsOptions = {
542561 config : {
543562 allowedPrincipals : [ PrincipalType . USER ] ,
563+ security : getProjectScopedRoutePolicy ( {
564+ allowedPrincipals : [ PrincipalType . USER ] ,
565+ } ) ,
544566 } ,
545567 schema : {
546568 tags : [ 'ai' , 'ai-chat-mcp' ] ,
@@ -555,6 +577,9 @@ const ListChatsOptions = {
555577const UpdateChatModelOptions = {
556578 config : {
557579 allowedPrincipals : [ PrincipalType . USER ] ,
580+ security : getProjectScopedRoutePolicy ( {
581+ allowedPrincipals : [ PrincipalType . USER ] ,
582+ } ) ,
558583 } ,
559584 schema : {
560585 tags : [ 'ai' , 'ai-chat-mcp' ] ,
0 commit comments