2020from cycode .cli .apps .scan .prompt .response_builders import get_response_builder
2121from cycode .cli .apps .scan .prompt .types import AiHookEventType
2222from cycode .cli .apps .scan .prompt .utils import output_json , safe_json_parse
23- from cycode .cli .utils .get_api_client import get_ai_security_manager_client
23+ from cycode .cli .utils .get_api_client import get_ai_security_manager_client , get_scan_cycode_client
2424from cycode .cli .utils .sentry import add_breadcrumb
2525from cycode .logger import get_logger
2626
@@ -52,17 +52,11 @@ def prompt_command(
5252 """
5353 add_breadcrumb ('prompt' )
5454
55- # Initialize AI Security Manager client
56- ai_security_client = get_ai_security_manager_client (ctx )
57- ctx .obj ['ai_security_client' ] = ai_security_client
58-
5955 # Read JSON payload from stdin
6056 stdin_data = sys .stdin .read ().strip ()
6157 payload = safe_json_parse (stdin_data )
6258
6359 tool = ide .lower ()
64-
65- # Get response builder for this IDE
6660 response_builder = get_response_builder (tool )
6761
6862 if not payload :
@@ -77,6 +71,12 @@ def prompt_command(
7771 event_name = unified_payload .event_name
7872 logger .debug ('Processing AI guardrails hook' , extra = {'event_name' : event_name , 'tool' : tool })
7973
74+ scan_client = get_scan_cycode_client (ctx )
75+ ctx .obj ['client' ] = scan_client
76+
77+ ai_security_client = get_ai_security_manager_client (ctx )
78+ ctx .obj ['ai_security_client' ] = ai_security_client
79+
8080 # Load policy (merges defaults <- user config <- repo config)
8181 # Extract first workspace root from payload if available
8282 workspace_roots = payload .get ('workspace_roots' , ['.' ])
0 commit comments