@@ -338,6 +338,7 @@ class ProcessParams(BaseModel):
338338 source_type : str
339339 index_name : str
340340 authorization : Optional [str ] = None
341+ model_id : Optional [int ] = None
341342
342343
343344class OpinionRequest (BaseModel ):
@@ -679,6 +680,8 @@ class MCPUpdateRequest(BaseModel):
679680 new_mcp_url : str = Field (..., description = "New MCP server URL" )
680681 new_authorization_token : Optional [str ] = Field (
681682 None , description = "New authorization token for MCP server authentication (e.g., Bearer token)" )
683+ custom_headers : Optional [Dict [str , Any ]] = Field (
684+ None , description = "Custom HTTP headers as JSON object" )
682685
683686
684687# Tenant Management Data Models
@@ -1118,6 +1121,7 @@ class AddMcpServiceRequest(BaseModel):
11181121 source : MCPSourceType = Field (default = MCPSourceType .LOCAL , description = "MCP source type" )
11191122 tags : List [str ] = Field (default_factory = list , description = "MCP tags" )
11201123 authorization_token : Optional [str ] = Field (None , description = "Authorization token for MCP server" )
1124+ custom_headers : Optional [Dict [str , Any ]] = Field (None , description = "Custom HTTP headers as JSON object" )
11211125 container_config : Optional [Dict [str , Any ]] = Field (None , description = "Container configuration" )
11221126 registry_json : Optional [Dict [str , Any ]] = Field (None , description = "Registry metadata JSON" )
11231127 enabled : Optional [bool ] = Field (default = False , description = "Whether the MCP is enabled after creation" )
@@ -1157,6 +1161,7 @@ class UpdateMcpServiceRequest(BaseModel):
11571161 server_url : str = Field (..., min_length = 1 , description = "New MCP server URL" )
11581162 tags : List [str ] = Field (default_factory = list , description = "MCP tags" )
11591163 authorization_token : Optional [str ] = Field (None , description = "Authorization token for MCP server" )
1164+ custom_headers : Optional [Dict [str , Any ]] = Field (None , description = "Custom HTTP headers as JSON object" )
11601165
11611166 @field_validator ("name" , "server_url" , "description" , "authorization_token" , mode = "before" )
11621167 @classmethod
0 commit comments