blocks-cli 0.1.28__tar.gz → 0.1.30__tar.gz

{blocks-cli-0.1.28 → blocks-cli-0.1.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: blocks-cli
-Version: 0.1.28
+Version: 0.1.30
 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.30/blocks_cli/builds.py

@@ -0,0 +1,23 @@
+import time
+
+from blocks_cli.api import api_client
+from blocks_cli.config.config import config
+
+def poll_build_status(image_id: str, build_id: str):
+    build_completed = False
+    while not build_completed:
+        try:
+            time.sleep(0.5)
+            res = api_client.get(f"{config.clients.orchestrator_url}/v1/images/{image_id}/builds/{build_id}")
+            build_status_response = res.json()
+                
+            is_completed = build_status_response.get("is_completed")
+            is_succeeded = build_status_response.get("is_succeeded")
+
+            build_completed = is_completed
+
+            if is_completed and not is_succeeded:
+                raise Exception("Build failed")
+        except Exception:
+            raise Exception("Failed to retrieve build status")
+        

{blocks-cli-0.1.28 → blocks-cli-0.1.30}/blocks_cli/commands/push.py

@@ -119,6 +119,9 @@ def push(file: Path = typer.Argument(..., help="Name of blocks file to push.")):
                 memory = task_kwargs.get("memory")
                 gpu_count = task_kwargs.get("gpu_count")
                 gpu_type = task_kwargs.get("gpu_type")
+                runtime = task_kwargs.get("runtime") # e.x. "python3.10"
+                plugins = task_kwargs.get("plugins", [])
+
                 repos: list = trigger_kwargs.get("repos", [])
 
                 if len(repos) == 0 and git_remote_url:
@@ -126,7 +129,7 @@ def push(file: Path = typer.Argument(..., help="Name of blocks file to push.")):
 
                 function_name = automation.get("function_name")
                 function_source_code = automation.get("function_source_code")
-                config_schema = automation.get("config_schema")
+                config_schema = automation.get("config_class").model_json_schema() if automation.get("config_class") else None
                 trigger_alias = automation.get("trigger_alias")
 
                 # Extract known fields
@@ -134,6 +137,8 @@ def push(file: Path = typer.Argument(..., help="Name of blocks file to push.")):
                     "name": automation_name,
                     "import_path": f"{file.name}:{function_name}",
                     "runner": runner,
+                    "runtime": runtime,
+                    "plugins": plugins,
                     "vcpus": vcpus,
                     "memory": memory,
                     "gpu_count": gpu_count,
@@ -148,10 +153,11 @@ def push(file: Path = typer.Argument(..., help="Name of blocks file to push.")):
                 additional_task_kwargs = {
                     k: v
                     for k, v in task_kwargs.items()
-                    if k not in ["vcpus", "memory_mib", "gpu_count", "gpu_type", "name"]
+                    if k not in ["vcpus", "memory_mib", "gpu_count", "gpu_type", "name", "runtime", "runner", "config_class", "plugins", "config_schema"]
                 }
                 automation_config["trigger_kwargs"] = trigger_kwargs
                 automation_config["task_kwargs"] = additional_task_kwargs
+                
                 registration_payload["automations"].append(automation_config)
 
         # Register automation

{blocks-cli-0.1.28 → blocks-cli-0.1.30}/blocks_cli/registration.py

@@ -1,5 +1,6 @@
 import sys
 import importlib.util
+
 from pathlib import Path
 
 def get_module_from_file(file: Path):

{blocks-cli-0.1.28 → blocks-cli-0.1.30}/setup.py

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

blocks-cli-0.1.28/blocks_cli/builds.py

@@ -1,20 +0,0 @@
-import time
-
-from blocks_cli.api import api_client
-from blocks_cli.config.config import config
-
-def poll_build_status(image_id: str, build_id: str):
-    build_completed = False
-    while not build_completed:
-        time.sleep(0.5)
-        res = api_client.get(f"{config.clients.orchestrator_url}/v1/images/{image_id}/builds/{build_id}")
-        build_status_response = res.json()
-        
-        is_completed = build_status_response.get("is_completed")
-        is_succeeded = build_status_response.get("is_succeeded")
-
-        build_completed = is_completed
-
-        if is_completed and not is_succeeded:
-                raise Exception("Build failed")
-