@@ -40,6 +40,9 @@ func telemetryAISubCommand(telemetryAIWrapper wrappers.TelemetryWrapper) *cobra.
4040 telemetryAICmd .PersistentFlags ().String (params .SubTypeFlag , "" , "Sub Type" )
4141 telemetryAICmd .PersistentFlags ().String (params .EngineFlag , "" , "Engine" )
4242 telemetryAICmd .PersistentFlags ().String (params .AgentFlag , "" , "Agent" )
43+ telemetryAICmd .PersistentFlags ().String (params .ScanTypeFlag , "" , "Scan Type" )
44+ telemetryAICmd .PersistentFlags ().String (params .StatusFlag , "" , "Status" )
45+ telemetryAICmd .PersistentFlags ().Int (params .TotalCountFlag , 0 , "Total Count" )
4346
4447 return telemetryAICmd
4548}
@@ -52,6 +55,9 @@ func runTelemetryAI(telemetryWrapper wrappers.TelemetryWrapper) func(*cobra.Comm
5255 subType , _ := cmd .Flags ().GetString ("sub-type" )
5356 agent , _ := cmd .Flags ().GetString ("agent" )
5457 engine , _ := cmd .Flags ().GetString ("engine" )
58+ scanType , _ := cmd .Flags ().GetString ("scan-type" )
59+ status , _ := cmd .Flags ().GetString ("status" )
60+ totalCount , _ := cmd .Flags ().GetInt ("total-count" )
5561
5662 err := telemetryWrapper .SendAIDataToLog (& wrappers.DataForAITelemetry {
5763 AIProvider : aiProvider ,
@@ -60,6 +66,9 @@ func runTelemetryAI(telemetryWrapper wrappers.TelemetryWrapper) func(*cobra.Comm
6066 SubType : subType ,
6167 Agent : agent ,
6268 Engine : engine ,
69+ ScanType : scanType ,
70+ Status : status ,
71+ TotalCount : totalCount ,
6372 })
6473
6574 if err != nil {
0 commit comments