linting and jinja2

Author: prassanna-ravishankar

examples/tutorials/20_behavior_testing/000_basic_sync_testing/test_sync_agent.py

@@ -17,10 +17,10 @@
 """
 
 from agentex.lib.testing import (
+    test_sync_agent,
+    assert_valid_agent_response,
     assert_agent_response_contains,
     assert_conversation_maintains_context,
-    assert_valid_agent_response,
-    test_sync_agent,
 )
 
 # TODO: Replace with your actual sync agent name from 'agentex agents list'

examples/tutorials/20_behavior_testing/010_agentic_testing/test_agentic_agent.py

@@ -18,7 +18,7 @@
 
 import pytest
 
-from agentex.lib.testing import assert_valid_agent_response, test_agentic_agent
+from agentex.lib.testing import test_agentic_agent, assert_valid_agent_response
 
 # TODO: Replace with your actual agent name from 'agentex agents list'
 AGENT_NAME = "ab000-hello-acp"

src/agentex/lib/cli/commands/init.py

@@ -6,8 +6,6 @@
 
 import questionary
 from jinja2 import Environment, FileSystemLoader
-from rich.rule import Rule
-from rich.text import Text
 from rich.panel import Panel
 from rich.table import Table
 from rich.console import Console
@@ -27,18 +25,14 @@ class TemplateType(str, Enum):
     SYNC = "sync"
 
 
-def render_template(
-    template_path: str, context: Dict[str, Any], template_type: TemplateType
-) -> str:
+def render_template(template_path: str, context: Dict[str, Any], template_type: TemplateType) -> str:
     """Render a template with the given context"""
     env = Environment(loader=FileSystemLoader(TEMPLATES_DIR / template_type.value))
     template = env.get_template(template_path)
     return template.render(**context)
 
 
-def create_project_structure(
-    path: Path, context: Dict[str, Any], template_type: TemplateType, use_uv: bool
-):
+def create_project_structure(path: Path, context: Dict[str, Any], template_type: TemplateType, use_uv: bool):
     """Create the project structure from templates"""
     # Create project directory
     project_dir: Path = path / context["project_name"]
@@ -83,6 +77,9 @@ def create_project_structure(
     # Add development notebook for agents
     root_templates["dev.ipynb.j2"] = "dev.ipynb"
 
+    # Add test file
+    root_templates["test_agent.py.j2"] = "test_agent.py"
+
     for template, output in root_templates.items():
         output_path = project_dir / output
         output_path.write_text(render_template(template, context, template_type))
@@ -101,10 +98,7 @@ def get_project_context(answers: Dict[str, Any], project_path: Path, manifest_ro
     return {
         **answers,
         "project_name": project_name,
-        "workflow_class": "".join(
-            word.capitalize() for word in answers["agent_name"].split("-")
-        )
-        + "Workflow",
+        "workflow_class": "".join(word.capitalize() for word in answers["agent_name"].split("-")) + "Workflow",
         "workflow_name": answers["agent_name"],
         "queue_name": project_name + "_queue",
         "project_path_from_build_root": project_path_from_build_root,
@@ -159,9 +153,7 @@ def validate_agent_name(text: str) -> bool | str:
     if not template_type:
         return
 
-    project_path = questionary.path(
-        "Where would you like to create your project?", default="."
-    ).ask()
+    project_path = questionary.path("Where would you like to create your project?", default=".").ask()
     if not project_path:
         return
 
@@ -179,9 +171,7 @@ def validate_agent_name(text: str) -> bool | str:
     if not agent_directory_name:
         return
 
-    description = questionary.text(
-        "Provide a brief description of your agent:", default="An Agentex agent"
-    ).ask()
+    description = questionary.text("Provide a brief description of your agent:", default="An AgentEx agent").ask()
     if not description:
         return
 
@@ -212,159 +202,24 @@ def validate_agent_name(text: str) -> bool | str:
     context["use_uv"] = answers["use_uv"]
 
     # Create project structure
-    create_project_structure(
-        project_path, context, answers["template_type"], answers["use_uv"]
-    )
-
-    # Show success message
-    console.print()
-    success_text = Text("✅ Project created successfully!", style="bold green")
-    success_panel = Panel(
-        success_text,
-        border_style="green",
-        padding=(0, 2),
-        title="[bold white]Status[/bold white]",
-        title_align="left"
-    )
-    console.print(success_panel)
-    
-    # Main header
-    console.print()
-    console.print(Rule("[bold blue]Next Steps[/bold blue]", style="blue"))
-    console.print()
+    create_project_structure(project_path, context, answers["template_type"], answers["use_uv"])
+
+    # Show next steps
+    console.print("\n[bold green]✨ Project created successfully![/bold green]")
+    console.print("\n[bold]Next steps:[/bold]")
+    console.print(f"1. cd {project_path}/{context['project_name']}")
+    console.print("2. Review and customize the generated files")
+    console.print("3. Update the container registry in manifest.yaml")
+
+    if answers["template_type"] == TemplateType.TEMPORAL:
+        console.print("4. Run locally:")
+        console.print("   agentex agents run --manifest manifest.yaml")
+    else:
+        console.print("4. Run locally:")
+        console.print("   agentex agents run --manifest manifest.yaml")
 
-    # Local Development Section
-    local_steps = Text()
-    local_steps.append("1. ", style="bold white")
-    local_steps.append("Navigate to your project directory:\n", style="white")
-    local_steps.append(f"   cd {project_path}/{context['project_name']}\n\n", style="dim cyan")
-    
-    local_steps.append("2. ", style="bold white")
-    local_steps.append("Review the generated files. ", style="white")
-    local_steps.append("project/acp.py", style="yellow")
-    local_steps.append(" is your agent's entrypoint.\n", style="white")
-    local_steps.append("   See ", style="dim white")
-    local_steps.append("https://agentex.sgp.scale.com/docs", style="blue underline")
-    local_steps.append(" for how to customize different agent types", style="dim white")
-    local_steps.append("\n\n", style="white")
-    
-    local_steps.append("3. ", style="bold white")
-    local_steps.append("Set up your environment and test locally ", style="white")
-    local_steps.append("(no deployment needed)", style="dim white")
-    local_steps.append(":\n", style="white")
-    local_steps.append("   uv venv && uv sync && source .venv/bin/activate", style="dim cyan")
-    local_steps.append("\n   agentex agents run --manifest manifest.yaml", style="dim cyan")
-    
-    local_panel = Panel(
-        local_steps,
-        title="[bold blue]Development Setup[/bold blue]",
-        title_align="left",
-        border_style="blue",
-        padding=(1, 2)
-    )
-    console.print(local_panel)
-    console.print()
+    console.print("5. Test your agent:")
+    console.print("   pytest test_agent.py -v")
 
-    # Prerequisites Note
-    prereq_text = Text()
-    prereq_text.append("The above is all you need for local development. Once you're ready for production, read this box and below.\n\n", style="white")
-    
-    prereq_text.append("• ", style="bold white")
-    prereq_text.append("Prerequisites for Production: ", style="bold yellow")
-    prereq_text.append("You need Agentex hosted on a Kubernetes cluster.\n", style="white")
-    prereq_text.append("  See ", style="dim white")
-    prereq_text.append("https://agentex.sgp.scale.com/docs", style="blue underline")
-    prereq_text.append(" for setup instructions. ", style="dim white")
-    prereq_text.append("Scale GenAI Platform (SGP) customers", style="dim cyan")
-    prereq_text.append(" already have this setup as part of their enterprise license.\n\n", style="dim white")
-    
-    prereq_text.append("• ", style="bold white")
-    prereq_text.append("Best Practice: ", style="bold blue")
-    prereq_text.append("Use CI/CD pipelines for production deployments, not manual commands.\n", style="white")
-    prereq_text.append("  Commands below demonstrate Agentex's quick deployment capabilities.", style="dim white")
-    
-    prereq_panel = Panel(
-        prereq_text,
-        border_style="yellow",
-        padding=(1, 2)
-    )
-    console.print(prereq_panel)
-    console.print()
-
-    # Production Setup Section (includes deployment)
-    prod_steps = Text()
-    prod_steps.append("4. ", style="bold white")
-    prod_steps.append("Configure where to push your container image", style="white")
-    prod_steps.append(":\n", style="white")
-    prod_steps.append("   Edit ", style="dim white")
-    prod_steps.append("manifest.yaml", style="dim yellow")
-    prod_steps.append(" → ", style="dim white")
-    prod_steps.append("deployment.image.repository", style="dim yellow")
-    prod_steps.append(" → replace ", style="dim white")
-    prod_steps.append('""', style="dim red")
-    prod_steps.append(" with your registry", style="dim white")
-    prod_steps.append("\n   Examples: ", style="dim white")
-    prod_steps.append("123456789012.dkr.ecr.us-west-2.amazonaws.com/my-agent", style="dim blue")
-    prod_steps.append(", ", style="dim white")
-    prod_steps.append("gcr.io/my-project", style="dim blue")
-    prod_steps.append(", ", style="dim white")
-    prod_steps.append("myregistry.azurecr.io", style="dim blue")
-    prod_steps.append("\n\n", style="white")
-    
-    prod_steps.append("5. ", style="bold white")
-    prod_steps.append("Build your agent as a container and push to registry", style="white")
-    prod_steps.append(":\n", style="white")
-    prod_steps.append("   agentex agents build --manifest manifest.yaml --registry <your-registry> --push", style="dim cyan")
-    prod_steps.append("\n\n", style="white")
-    
-    prod_steps.append("6. ", style="bold white")
-    prod_steps.append("Upload secrets to cluster ", style="white")
-    prod_steps.append("(API keys, credentials your agent needs)", style="dim white")
-    prod_steps.append(":\n", style="white")
-    prod_steps.append("   agentex secrets sync --manifest manifest.yaml --cluster your-cluster", style="dim cyan")
-    prod_steps.append("\n   ", style="white")
-    prod_steps.append("Note: ", style="dim yellow")
-    prod_steps.append("Secrets are ", style="dim white")
-    prod_steps.append("never stored in manifest.yaml", style="dim red")
-    prod_steps.append(". You provide them via ", style="dim white")
-    prod_steps.append("--values file", style="dim blue")
-    prod_steps.append(" or interactive prompts", style="dim white")
-    prod_steps.append("\n\n", style="white")
-    
-    prod_steps.append("7. ", style="bold white")
-    prod_steps.append("Deploy your agent to run on the cluster", style="white")
-    prod_steps.append(":\n", style="white")
-    prod_steps.append("   agentex agents deploy --cluster your-cluster --namespace your-namespace", style="dim cyan")
-    prod_steps.append("\n\n", style="white")
-    prod_steps.append("Note: These commands use Helm charts hosted by Scale to deploy agents.", style="dim italic")
-    
-    prod_panel = Panel(
-        prod_steps,
-        title="[bold magenta]Production Setup & Deployment[/bold magenta]",
-        title_align="left",
-        border_style="magenta",
-        padding=(1, 2)
-    )
-    console.print(prod_panel)
-    
-    # Professional footer with helpful context
-    console.print()
-    console.print(Rule(style="dim white"))
-    
-    # Add helpful context about the workflow
-    help_text = Text()
-    help_text.append("ℹ️  ", style="blue")
-    help_text.append("Quick Start: ", style="bold white")
-    help_text.append("Steps 1-3 for local development. Steps 4-7 require Agentex cluster for production.", style="dim white")
-    console.print("   ", help_text)
-    
-    tip_text = Text()
-    tip_text.append("💡 ", style="yellow")
-    tip_text.append("Need help? ", style="bold white")
-    tip_text.append("Use ", style="dim white")
-    tip_text.append("agentex --help", style="cyan")
-    tip_text.append(" or ", style="dim white")
-    tip_text.append("agentex [command] --help", style="cyan")
-    tip_text.append(" for detailed options", style="dim white")
-    console.print("   ", tip_text)
-    console.print()
+    console.print("6. Deploy your agent:")
+    console.print("   agentex agents deploy --cluster your-cluster --namespace your-namespace")

src/agentex/lib/testing/__init__.py

@@ -46,9 +46,9 @@ async def test_my_agentic_agent():
     assert_conversation_maintains_context,
 )
 from agentex.lib.testing.exceptions import (
+    AgentTimeoutError,
     AgentNotFoundError,
     AgentSelectionError,
-    AgentTimeoutError,
 )
 
 __all__ = [

src/agentex/lib/testing/config.py

@@ -4,8 +4,8 @@
 Centralized configuration management with environment variable support.
 """
 
-import logging
 import os
+import logging
 from dataclasses import dataclass
 
 logger = logging.getLogger(__name__)

src/agentex/lib/testing/exceptions.py

@@ -3,6 +3,7 @@
 
 Provides specific error types for better error handling and debugging.
 """
+from __future__ import annotations
 
 
 class AgentexTestingError(Exception):

src/agentex/lib/testing/retry.py

@@ -9,8 +9,8 @@
 import time
 import asyncio
 import logging
+from typing import TypeVar, Callable, ParamSpec
 from functools import wraps
-from typing import Callable, TypeVar, ParamSpec
 
 from agentex.lib.testing.config import config
 

src/agentex/lib/testing/sessions/agentic.py

@@ -13,15 +13,14 @@
 
 from agentex import AsyncAgentex
 from agentex.types import Task, Agent
+from agentex.lib.testing.retry import with_async_retry
+from agentex.lib.testing.config import config
+from agentex.lib.testing.poller import MessagePoller
 from agentex.types.text_content import TextContent
+from agentex.lib.testing.type_utils import create_user_message
 from agentex.types.agent_rpc_params import ParamsSendEventRequest
-
-from agentex.lib.testing.config import config
-from agentex.lib.testing.agent_selector import AgentSelector
 from agentex.lib.testing.task_manager import TaskManager
-from agentex.lib.testing.poller import MessagePoller
-from agentex.lib.testing.retry import with_async_retry
-from agentex.lib.testing.type_utils import create_user_message, extract_agent_response, extract_task_id_from_response
+from agentex.lib.testing.agent_selector import AgentSelector
 
 logger = logging.getLogger(__name__)
 

src/agentex/lib/testing/sessions/sync.py

@@ -12,14 +12,13 @@
 
 from agentex import Agentex
 from agentex.types import Agent
-from agentex.types.text_content import TextContent
-from agentex.types.agent_rpc_params import ParamsSendMessageRequest
-
-from agentex.lib.testing.config import config
-from agentex.lib.testing.agent_selector import AgentSelector
 from agentex.lib.testing.retry import with_retry
+from agentex.lib.testing.config import config
+from agentex.types.text_content import TextContent
 from agentex.lib.testing.exceptions import AgentResponseError
 from agentex.lib.testing.type_utils import create_user_message, extract_agent_response
+from agentex.types.agent_rpc_params import ParamsSendMessageRequest
+from agentex.lib.testing.agent_selector import AgentSelector
 
 logger = logging.getLogger(__name__)
 

src/agentex/lib/testing/type_utils.py

@@ -10,11 +10,11 @@
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from agentex.types import SendMessageResponse, SendEventResponse
+    from agentex.types import SendEventResponse, SendMessageResponse
     from agentex.types.text_content import TextContent
 
-from agentex.types.text_content_param import TextContentParam
 from agentex.lib.testing.exceptions import AgentResponseError
+from agentex.types.text_content_param import TextContentParam
 
 logger = logging.getLogger(__name__)