agent-dev-cli 0.0.1b251223__py3-none-any.whl

agentdev/cli.py

@@ -0,0 +1,165 @@
+"""
+agentdev CLI - Agent Dev CLI Command Line Interface.
+
+Provides a CLI wrapper for running Python agent scripts with automatic
+agentdev instrumentation, eliminating the need for users to modify their code.
+
+Usage:
+    agentdev run my_agent.py
+    agentdev run workflow.py --port 9000
+    agentdev run my_agent.py -- --custom-arg value
+"""
+
+import os
+import sys
+import subprocess
+from importlib.metadata import version as get_version
+from typing import Optional
+
+import click
+
+
+def _get_agentdev_version() -> str:
+    """Get the agentdev version from package metadata."""
+    try:
+        return get_version("agent-dev-cli")
+    except Exception:
+        return "unknown"
+
+
+@click.group()
+@click.version_option(version=_get_agentdev_version(), prog_name="agentdev")
+def cli():
+    """agentdev - Agent Dev CLI for agent debugging and visualization.
+    
+    Run your Python agent scripts with automatic agentdev instrumentation
+    for workflow visualization and debugging in VS Code.
+    
+    Examples:
+    
+        agentdev run my_agent.py
+        
+        agentdev run workflow.py --port 9000
+    """
+    pass
+
+
+@cli.command()
+@click.argument('script', type=click.Path(exists=True))
+@click.argument('args', nargs=-1, type=click.UNPROCESSED)
+@click.option(
+    '--port', '-p',
+    default=8088,
+    type=int,
+    help='Agent server port (default: 8088)'
+)
+@click.option(
+    '--verbose', '-v',
+    is_flag=True,
+    default=False,
+    help='Enable verbose output'
+)
+def run(
+    script: str,
+    args: tuple,
+    port: int,
+    verbose: bool
+):
+    """Run a Python agent script with agentdev instrumentation.
+    
+    SCRIPT is the path to the Python agent script to run.
+    
+    Any additional arguments after the script path (or after --)
+    will be passed to the script.
+    
+    Examples:
+    
+        agentdev run my_agent.py
+        
+        agentdev run workflow.py --port 9000
+        
+        agentdev run my_agent.py -- --model gpt-4
+    """
+    # Resolve script to absolute path
+    script_path = os.path.abspath(script)
+    
+    if verbose:
+        click.echo(f"agentdev: Running {script_path} with instrumentation")
+        click.echo(f"agentdev: Port={port}")
+    
+    # Set up environment for instrumentation
+    env = os.environ.copy()
+    env['AGENTDEV_ENABLED'] = '1'
+    env['AGENTDEV_PORT'] = str(port)
+    # azure-ai-agentserver SDK starts the HTTP server and listens on DEFAULT_AD_PORT
+    # https://github.com/Azure/azure-sdk-for-python/blob/e18d002a1bf56706e83596a4720ec5f21488bab6/sdk/agentserver/azure-ai-agentserver-core/azure/ai/agentserver/core/server/base.py#L233
+    env['DEFAULT_AD_PORT'] = str(port)
+
+    if verbose:
+        env['AGENTDEV_VERBOSE'] = '1'
+    
+    # Build command to run the bootstrap module
+    cmd = [
+        sys.executable,
+        '-m', 'agentdev._bootstrap',
+        script_path,
+        *args
+    ]
+    
+    if verbose:
+        click.echo(f"agentdev: Executing: {' '.join(cmd)}")
+    
+    try:
+        # Run the user's script with instrumentation
+        result = subprocess.run(cmd, env=env)
+        sys.exit(result.returncode)
+    except KeyboardInterrupt:
+        click.echo("\nagentdev: Interrupted by user")
+        sys.exit(130)
+    except Exception as e:
+        click.echo(f"agentdev: Error running script: {e}", err=True)
+        sys.exit(1)
+
+
+@cli.command()
+def info():
+    """Show agentdev configuration and status information."""
+    click.echo("agentdev - Agent Dev CLI")
+    click.echo("=" * 40)
+    click.echo(f"Version: {_get_agentdev_version()}")
+    click.echo(f"Python: {sys.version}")
+    click.echo(f"Platform: {sys.platform}")
+    click.echo()
+    
+    # Check for VS Code
+    term_program = os.environ.get("TERM_PROGRAM", "Unknown")
+    term_version = os.environ.get("TERM_PROGRAM_VERSION", "Unknown")
+    click.echo(f"Terminal Program: {term_program}")
+    click.echo(f"Terminal Version: {term_version}")
+    click.echo()
+    
+    # Check installed dependencies
+    click.echo("Dependencies:")
+    deps = [
+        "agent_framework",
+        "agent_framework_azure_ai", 
+        "azure.ai.agentserver.agentframework",
+        "starlette",
+    ]
+    for dep in deps:
+        try:
+            # Convert package name to importable module name (hyphens to underscores)
+            module_name = dep.replace("-", "_")
+            __import__(module_name)
+            click.echo(f"  ✓ {dep}")
+        except ImportError:
+            click.echo(f"  ✗ {dep} (not installed)")
+
+
+def main():
+    """Main entry point for the CLI."""
+    cli()
+
+
+if __name__ == '__main__':
+    main()

agentdev/localdebug.py

@@ -0,0 +1,65 @@
+"""
+Workflow visualization setup module.
+
+This module provides functionality to set up workflow or agent visualization
+"""
+
+from agent_framework import WorkflowAgent
+import os
+import sys
+import time
+
+
+def setup_test_tool(agent_server):
+    """
+    Set up workflow or agent visualization for an agent server.
+    
+    This function configures a health check endpoint and starts a visualization
+    server for workflow agents on port 8090.
+    
+    Args:
+        agent_server: The agent server instance to set up visualization for.
+                     Should have 'app' (Starlette) and 'agent' attributes.
+    
+    Example:
+        >>> from azure.ai.agentserver.agentframework import from_agent_framework
+        >>> agent_server = from_agent_framework(agent)
+        >>> setup_test_tool(agent_server)
+        >>> await agent_server.run_async()
+    """
+    from starlette.applications import Starlette
+    from starlette.responses import JSONResponse
+    from starlette.routing import Route
+    from threading import Thread
+    from agentdev.backend.server import TestToolServer
+    
+    async def health_check(request):
+        """Health check endpoint handler."""
+        return JSONResponse({"status": "ok"}, status_code=200)
+    
+    app = agent_server.app
+    agent = agent_server.agent
+
+    # Mount health check endpoint
+    app.mount(
+        "/agentdev/health",
+        Starlette(routes=[Route("/", health_check)])
+    )
+
+    # Prepare entities for visualization
+    entities = []
+    if type(agent) == WorkflowAgent:
+        entities.append(agent.workflow)
+    else:
+        entities.append(agent)
+
+    test_tool_server = TestToolServer(entities)
+    test_tool_server.mount_backend(app)
+    def show_endspattern():
+        time.sleep(2)
+        print("agentdev: Application startup complete")
+    thread = Thread(target=show_endspattern)
+    thread.daemon = True
+    thread.start()
+    
+    print(agent)