weni-cli 3.7.1__tar.gz → 3.7.2__tar.gz

{weni_cli-3.7.1 → weni_cli-3.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: weni-cli
-Version: 3.7.1
+Version: 3.7.2
 Summary: 
 Author: Paulo Bernardo
 Author-email: paulo.bernardo@weni.ai

{weni_cli-3.7.1 → weni_cli-3.7.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "weni-cli"
-version = "3.7.1"
+version = "3.7.2"
 description = ""
 authors = ["Paulo Bernardo <paulo.bernardo@weni.ai>", "Matheus Leal <matheus.cardoso@vtex.com>"]
 readme = "README.md"

{weni_cli-3.7.1 → weni_cli-3.7.2}/weni_cli/cli.py

@@ -101,7 +101,9 @@ def current_project():
 @project.command("push")
 @click.argument("definition", required=True, type=click.Path(exists=True, dir_okay=False))
 @click.option("--force-update", is_flag=True, help="Force update to the project")
-def push_project(definition, force_update):
+@click.option("--use-apm", is_flag=True, help="Enable Elastic APM instrumentation for passive agent tool lambdas")
+@click.option("--remove-apm", is_flag=True, help="Remove Elastic APM instrumentation from passive agent tool lambdas")
+def push_project(definition, force_update, use_apm, remove_apm):
     """Push an Agent definition to the current project
 
     DEFINITION: The path to the YAML agent definition file
@@ -109,7 +111,12 @@ def push_project(definition, force_update):
     from weni_cli.commands.project_push import ProjectPushHandler
 
     try:
-        ProjectPushHandler().execute(definition=definition, force_update=force_update)
+        ProjectPushHandler().execute(
+            definition=definition,
+            force_update=force_update,
+            use_apm=use_apm,
+            remove_apm=remove_apm,
+        )
     except Exception as e:
         click.echo(f"Error: {e}")
 

{weni_cli-3.7.1 → weni_cli-3.7.2}/weni_cli/clients/cli_client.py

@@ -50,15 +50,22 @@ def get_cli_version() -> str:
     return importlib.metadata.version("weni-cli")
 
 
-def create_default_payload(project_uuid: str, definition: Dict, agent_type: str) -> Dict[str, str]:
+def create_default_payload(
+    project_uuid: str, definition: Dict, agent_type: str, apm_instrumentation: Optional[str] = None
+) -> Dict[str, str]:
     """Create a default payload for API requests."""
-    return {
+    payload = {
         "project_uuid": project_uuid,
         "definition": json.dumps(definition, ensure_ascii=False),
         "toolkit_version": get_toolkit_version(),
         "type": agent_type,
     }
 
+    if apm_instrumentation:
+        payload["apm_instrumentation"] = apm_instrumentation
+
+    return payload
+
 
 class CLIClient:
     """Client for interacting with the Weni CLI API."""
@@ -179,10 +186,15 @@ class CLIClient:
             raise RequestError(f"Failed to check project permission: {e.message}")
 
     def push_agents(
-        self, project_uuid: str, agents_definition: Dict, resources_folder: Dict[str, BinaryIO], agent_type: str
+        self,
+        project_uuid: str,
+        agents_definition: Dict,
+        resources_folder: Dict[str, BinaryIO],
+        agent_type: str,
+        apm_instrumentation: Optional[str] = None,
     ) -> None:
         """Push agents to the API."""
-        data = create_default_payload(project_uuid, agents_definition, agent_type)
+        data = create_default_payload(project_uuid, agents_definition, agent_type, apm_instrumentation)
 
         with spinner():
             try:

{weni_cli-3.7.1 → weni_cli-3.7.2}/weni_cli/commands/project_push.py

@@ -19,12 +19,36 @@ from weni_cli.validators.agent_definition import (
 
 CONTACT_FIELD_NAME_REGEX = r"^[a-z][a-z0-9_]*$"
 
+APM_OBSERVABILITY_WARNING = (
+    "Use Elastic APM instrumentation only for observability—for example, while debugging or "
+    "investigating an issue. APM adds Lambda layers and environment variables that increase cold "
+    "start time and runtime overhead.\n\n"
+    "When you no longer need it, disable instrumentation with:\n"
+    "  weni project push <definition> --remove-apm"
+)
+
 
 class ProjectPushHandler(Handler):
     def execute(self, **kwargs):
         force_update = self.load_param(kwargs, "force_update", False)
+        use_apm = self.load_param(kwargs, "use_apm", False)
+        remove_apm = self.load_param(kwargs, "remove_apm", False)
         definition_path = self.load_param(kwargs, "definition", None, True)
 
+        if use_apm and remove_apm:
+            formatter = Formatter()
+            formatter.print_error_panel(
+                "Cannot use --use-apm and --remove-apm together. Choose one option.",
+                title="Invalid APM options",
+            )
+            return
+
+        apm_instrumentation = None
+        if use_apm:
+            apm_instrumentation = "enabled"
+        elif remove_apm:
+            apm_instrumentation = "disabled"
+
         store = Store()
         project_uuid = store.get(STORE_PROJECT_UUID_KEY)
 
@@ -55,11 +79,11 @@ class ProjectPushHandler(Handler):
             return
 
         if agent_type == "passive":
-            self.push_passive_agent(force_update, project_uuid, definition_data)
+            self.push_passive_agent(force_update, project_uuid, definition_data, apm_instrumentation)
         elif agent_type == "active":
-            self.push_active_agent(force_update, project_uuid, definition_data)
+            self.push_active_agent(force_update, project_uuid, definition_data, apm_instrumentation)
 
-    def push_passive_agent(self, force_update, project_uuid, definition):
+    def push_passive_agent(self, force_update, project_uuid, definition, apm_instrumentation=None):
         formatter = Formatter()
         error = validate_agent_definition_schema(definition)
         if error:
@@ -74,9 +98,9 @@ class ProjectPushHandler(Handler):
             return
 
         definition = format_definition(definition)
-        self.push_definition(force_update, "passive", project_uuid, definition, tools_folders_map)
+        self.push_definition(force_update, "passive", project_uuid, definition, tools_folders_map, apm_instrumentation)
 
-    def push_active_agent(self, force_update, project_uuid, definition):
+    def push_active_agent(self, force_update, project_uuid, definition, apm_instrumentation=None):
         formatter = Formatter()
         error = validate_active_agent_definition_schema(definition)
         if error:
@@ -98,7 +122,7 @@ class ProjectPushHandler(Handler):
         rules_folders_map.update(preprocessing_folders_map)
         resources_folders_map = rules_folders_map
         definition = format_definition(definition)
-        self.push_definition(force_update, "active", project_uuid, definition, resources_folders_map)
+        self.push_definition(force_update, "active", project_uuid, definition, resources_folders_map, None)
 
     def load_param(self, params, key, default=None, required=False):
         value = params.get(key, default)
@@ -115,13 +139,19 @@ class ProjectPushHandler(Handler):
     def load_preprocessing_folder(self, definition) -> tuple[Optional[dict], Optional[str]]:
         return _load_preprocessing_folder(definition)
 
-    def push_definition(self, force_update, agent_type, project_uuid, definition, resources_folder_map):
+    def push_definition(
+        self, force_update, agent_type, project_uuid, definition, resources_folder_map, apm_instrumentation=None
+    ):
         client = CLIClient()
+        formatter = Formatter()
 
         try:
-            client.push_agents(project_uuid, definition, resources_folder_map, agent_type)
+            client.push_agents(
+                project_uuid, definition, resources_folder_map, agent_type, apm_instrumentation=apm_instrumentation
+            )
         except Exception as e:
-            formatter = Formatter()
             formatter.print_error_panel(f"Failed to push definition: {e}")
         else:
             click.echo("Definition pushed successfully")
+            if apm_instrumentation == "enabled":
+                formatter.print_warning_panel(APM_OBSERVABILITY_WARNING, title="APM instrumentation enabled")

{weni_cli-3.7.1 → weni_cli-3.7.2}/weni_cli/formatter/formatter.py

@@ -32,3 +32,14 @@ class Formatter:
             expand=False,
         )
         print(success_panel)
+
+    def print_warning_panel(self, message, title="Warning"):
+        warning_panel = Panel(
+            self._to_renderable(message),
+            title=f"[bold yellow]{title}[/bold yellow]",
+            title_align="left",
+            style="bold yellow",
+            expand=False,
+            padding=(1, 1),
+        )
+        print(warning_panel)

{weni_cli-3.7.1 → weni_cli-3.7.2}/weni_cli/formatter/tests/test_formatter.py

@@ -53,6 +53,24 @@ def test_print_success_panel(formatter, mocker):
     assert panel_arg.expand is False
 
 
+def test_print_warning_panel(formatter, mocker):
+    """Test the warning panel creation and printing."""
+    mock_print = mocker.patch("weni_cli.formatter.formatter.print")
+
+    warning_message = "This is a warning message"
+
+    formatter.print_warning_panel(warning_message, title="Custom warning")
+
+    mock_print.assert_called_once()
+
+    panel_arg = mock_print.call_args[0][0]
+    assert isinstance(panel_arg, Panel)
+    assert panel_arg.title == "[bold yellow]Custom warning[/bold yellow]"
+    assert panel_arg.renderable == warning_message
+    assert panel_arg.style == "bold yellow"
+    assert panel_arg.expand is False
+
+
 def test_print_error_panel_with_exception(formatter, mocker):
     """Test printing an error panel with an exception object."""
     mock_print = mocker.patch("weni_cli.formatter.formatter.print")