7777)
7878from agentveil_mcp_proxy .passthrough import DownstreamConfig , McpPassthrough , PassthroughError
7979from agentveil_mcp_proxy .runtime_gate import RuntimeGateClient
80- from agentveil_mcp_proxy .workflow_guard_cli import (
81- WorkflowGuardCliError ,
82- dispatch_workflow_guard ,
83- register_workflow_guard_parser ,
84- )
8580
8681
8782DEFAULT_BASE_URL = "https://agentveil.dev"
@@ -2198,8 +2193,6 @@ def build_parser() -> argparse.ArgumentParser:
21982193 )
21992194 _add_json_arg (client_config_print )
22002195
2201- register_workflow_guard_parser (subparsers )
2202-
22032196 return parser
22042197
22052198
@@ -2223,11 +2216,7 @@ def _normalize_downstream_arg_values(argv: list[str]) -> list[str]:
22232216def main (argv : list [str ] | None = None ) -> int :
22242217 parser = build_parser ()
22252218 parse_argv = sys .argv [1 :] if argv is None else argv
2226- if parse_argv [:1 ] == ["workflow-guard" ]:
2227- normalized_argv = parse_argv
2228- else :
2229- normalized_argv = _normalize_downstream_arg_values (parse_argv )
2230- args = parser .parse_args (normalized_argv )
2219+ args = parser .parse_args (_normalize_downstream_arg_values (parse_argv ))
22312220 try :
22322221 if args .command == "init" :
22332222 downstream_config = None
@@ -2420,9 +2409,7 @@ def main(argv: list[str] | None = None) -> int:
24202409 output_json = args .json_output ,
24212410 )
24222411 return 0
2423- if args .command == "workflow-guard" :
2424- return dispatch_workflow_guard (args )
2425- except (ProxyCliError , WorkflowGuardCliError , ApprovalEvidenceError , EvidenceExportError , EvidenceVerificationError ) as exc :
2412+ except (ProxyCliError , ApprovalEvidenceError , EvidenceExportError , EvidenceVerificationError ) as exc :
24262413 if getattr (args , "json_output" , False ):
24272414 _print_json ({
24282415 "ok" : False ,
@@ -2433,9 +2420,7 @@ def main(argv: list[str] | None = None) -> int:
24332420 })
24342421 else :
24352422 print (f"ERROR: { exc } " , file = sys .stderr )
2436- if isinstance (exc , (ProxyCliError , WorkflowGuardCliError )):
2437- return exc .exit_code
2438- return 1
2423+ return exc .exit_code if isinstance (exc , ProxyCliError ) else 1
24392424 raise AssertionError (f"unhandled command: { args .command } " )
24402425
24412426
0 commit comments