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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: blocks-cli
-Version: 0.1.38
+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.38 → 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.38 → blocks-cli-0.1.39}/setup.py

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