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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: blocks-cli
-Version: 0.1.37
+Version: 0.1.39
 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.39}/blocks_cli/commands/create.py

@@ -50,19 +50,57 @@ def create(
             
             # Create main.py with basic template
             with open(automation_dir / 'main.py', 'w') as f:
-                f.write('''from blocks import task, on
-
-@task(name="{name}")
-@on("", repos=[])
-def {function_name}(input):
-    print(input)
+                f.write('''import os
+import slack_sdk
+from enum import Enum
+from blocks import agent, on
+from smolagents import CodeAgent, WebSearchTool, LiteLLMModel
+from pydantic import BaseModel
+
+SLACK_TOKEN = os.getenv("SLACK_TOKEN")
+
+class Models(str, Enum):
+    gpt_5 = "gpt-5"
+    claude4sonnet = "claude-sonnet-4-20250514"
+
+class SmolAgentConfig(BaseModel):
+    model: Models = Models.gpt_5
+
+@agent(name="{name}")
+@on("slack.mention")
+def {function_name}(input, config: SmolAgentConfig):
+    event = input.get("event")
+    text = event.get("text", "")
+    channel = event.get("channel", "")
+    ts = event.get("ts", "")
+
+    model = LiteLLMModel(
+        model_id=config.model,
+    )
+    agent = CodeAgent(tools=[WebSearchTool()], model=model)
+    agent.run(text)
+
+    client = slack_sdk.WebClient(token=SLACK_TOKEN)
+    client.chat_postMessage(channel=channel, text="Agent is thinking...", thread_ts=ts)
+
+    for _, step in enumerate(agent.memory.steps):
+        messages = step.to_messages()
+        for message in messages:
+            for content_message in message.content:
+                final_message = content_message.get("text", "")
+                if final_message and content_message.get("role") != "user":
+                    client.chat_postMessage(channel=channel, text=final_message, thread_ts=ts)
 '''.format(name=name, function_name=function_name))
 
             sdk_version = get_latest_sdk_version()
             latest_version = sdk_version.get("latest_version")
 
             with open(automation_dir / 'requirements.txt', 'w') as f:
-                f.write('''blocks-sdk>={version}'''.format(version=latest_version))
+                f.write('''blocks-sdk>={version}
+smolagents[toolkit]
+litellm>=1.61.16,<=1.74.8
+slack-sdk>=3.19.2
+'''.format(version=latest_version))
 
             console.print(f"Successfully created automation [green]{name}[/green] in [green]{automation_dir.absolute()}[/green]")
             console.print(f"[green]{name}/\n   main.py\n   requirements.txt[/green]")

{blocks-cli-0.1.37 → blocks-cli-0.1.39}/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.39}/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.39}/setup.py

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