import px for agent status

Summary: TSIA

Test Plan: tested manually.

Reviewers: michelle, #engineering

Reviewed By: michelle, #engineering

Differential Revision: https://phab.corp.pixielabs.ai/D4195

GitOrigin-RevId: 9ce2432e66f33562ba0e2eaba6f757d166d90054

src/utils/pixie_cli/cmd/get.go

@@ -32,7 +32,7 @@ var GetPEMsCmd = &cobra.Command{
 		format, _ := cmd.Flags().GetString("output")
 		format = strings.ToLower(format)
 		v := mustConnectDefaultVizier(cloudAddr)
-		q := `px.display(px.GetAgentStatus())`
+		q := "import px\npx.display(px.GetAgentStatus())"
 
 		ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
 		defer cancel()