diff --git a/axonflow/client.py b/axonflow/client.py index fc56bb8..0193903 100644 --- a/axonflow/client.py +++ b/axonflow/client.py @@ -1082,7 +1082,7 @@ async def mcp_check_input( self, connector_type: str, statement: str, - operation: str = "query", + operation: str = "execute", parameters: dict[str, Any] | None = None, ) -> MCPCheckInputResponse: """Validate an MCP request against configured policies without executing it. @@ -1093,7 +1093,7 @@ async def mcp_check_input( Args: connector_type: Type of MCP connector (e.g., "postgres", "snowflake"). statement: The SQL query or command to validate. - operation: Operation type - "query" (default) or "execute". + operation: Operation type - "query" or "execute" (default). parameters: Optional query parameters. Returns: @@ -5953,7 +5953,7 @@ def mcp_check_input( self, connector_type: str, statement: str, - operation: str = "query", + operation: str = "execute", parameters: dict[str, Any] | None = None, ) -> MCPCheckInputResponse: """Validate an MCP request against configured policies without executing it.""" diff --git a/axonflow/types.py b/axonflow/types.py index b97e0a2..32a8143 100644 --- a/axonflow/types.py +++ b/axonflow/types.py @@ -364,7 +364,7 @@ class MCPCheckInputRequest(BaseModel): connector_type: str statement: str parameters: dict[str, Any] | None = Field(default=None) - operation: str = Field(default="query") + operation: str = Field(default="execute") class MCPCheckInputResponse(BaseModel):