blocks-cli 0.1.37__tar.gz → 0.1.38__tar.gz

{blocks-cli-0.1.37 → blocks-cli-0.1.38}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: blocks-cli
-Version: 0.1.37
+Version: 0.1.38
 Summary: CLI tool for Blocks, a platform for writing custom AI-enabled codebase automations in Python. Leverage a full codebase-aware API. Automatically trigger automations from Github, Slack, and other providers.
 Home-page: https://github.com/BlocksOrg/sdk
 Author: BlocksOrg

{blocks-cli-0.1.37 → blocks-cli-0.1.38}/blocks_cli/commands/push.py

@@ -183,6 +183,10 @@ def push(
                     "task_kwargs": {},
                 }
 
+                # Extract is_agent flag if present
+                is_agent = task_kwargs.get("is_agent", False)
+                automation_config["is_agent"] = is_agent
+
                 # Add any additional args that weren't explicitly handled
                 additional_task_kwargs = {
                     k: v
@@ -198,7 +202,8 @@ def push(
                         "config_class", 
                         "plugins", 
                         "config_schema",
-                        "required_env_vars"
+                        "required_env_vars",
+                        "is_agent"
                     ]
                 }
                 automation_config["trigger_kwargs"] = trigger_kwargs

{blocks-cli-0.1.37 → blocks-cli-0.1.38}/blocks_cli/registration.py

@@ -21,12 +21,15 @@ def get_blocks_state_and_module_from_file(file: Path):
     automation_module = get_module_from_file(file)
     has_blocks_import = hasattr(automation_module, "blocks")
     has_task_import = hasattr(automation_module, "task")
+    has_agent_import = hasattr(automation_module, "agent")
 
-    if not any([has_blocks_import, has_task_import]):
-        raise Exception("No blocks or task import found in the specified file, likely not an automation file.")
+    if not any([has_blocks_import, has_task_import, has_agent_import]):
+        raise Exception("No blocks, task, or agent import found in the specified file, likely not an automation file.")
     if has_blocks_import:
         state = automation_module.state.dag
     elif has_task_import:
         state = automation_module.task.blocks_state
+    elif has_agent_import:
+        state = automation_module.agent.blocks_state
     
     return state, automation_module

{blocks-cli-0.1.37 → blocks-cli-0.1.38}/setup.py

@@ -6,7 +6,7 @@ with open("requirements.txt") as f:
 
 setup(
     name="blocks-cli",  
-    version="0.1.37",
+    version="0.1.38",
     packages=find_packages(),
     include_package_data=True, 
     install_requires=requirements,