wizelit-sdk 0.1.31__py3-none-any.whl → 0.1.32__py3-none-any.whl

wizelit_sdk/cli.py

@@ -0,0 +1,337 @@
+"""Command-line utilities for Wizelit SDK."""
+
+from __future__ import annotations
+
+import re
+from pathlib import Path
+from textwrap import dedent
+
+import click
+
+
+TRANSPORT_CHOICES = ["streamable-http", "sse", "stdio", "http"]
+
+
+def _slugify(value: str) -> str:
+    value = value.strip().lower()
+    value = re.sub(r"[^a-z0-9]+", "-", value)
+    value = value.strip("-")
+    if not value:
+        raise click.ClickException("Agent name must contain letters or numbers.")
+    return value
+
+
+def _write_file(path: Path, content: str, force: bool) -> None:
+    if path.exists() and not force:
+        raise click.ClickException(
+            f"File already exists: {path}. Use --force to overwrite."
+        )
+    path.parent.mkdir(parents=True, exist_ok=True)
+    path.write_text(content, encoding="utf-8")
+
+
+def _render_main_py(
+    agent_name: str, transport: str, host: str, port: int
+) -> str:
+    return dedent(
+        f'''"""Wizelit Agent Scaffold
+        Generated by wizelit-agent CLI.
+        """
+
+        from wizelit_sdk.agent_wrapper import WizelitAgent
+
+        mcp = WizelitAgent(
+            "{agent_name}",
+            transport="{transport}",
+            host="{host}",
+            port={port},
+        )
+
+
+        @mcp.ingest(
+            is_long_running=False,
+            description="Describe your tool here.",
+            response_handling={{"mode": "json"}},
+        )
+        def example_tool(text: str) -> dict:
+            """Example tool."""
+            return {{"echo": text}}
+
+
+        if __name__ == "__main__":
+            mcp.run()
+        '''
+    ).lstrip()
+
+
+def _render_readme(agent_name: str) -> str:
+    return dedent(
+        f"""
+        # {agent_name}
+
+        Generated Wizelit agent scaffold.
+
+        ## Run
+
+        ```bash
+        python main.py
+        ```
+        """
+    ).lstrip()
+
+
+def _render_env_example() -> str:
+    return dedent(
+        """
+        # Example environment variables
+        REDIS_URL=redis://localhost:6379
+        ENABLE_LOG_STREAMING=true
+
+        # AWS (if using Bedrock)
+        AWS_ACCESS_KEY_ID=
+        AWS_SECRET_ACCESS_KEY=
+        AWS_REGION=us-east-1
+        """
+    ).lstrip()
+
+
+def _render_requirements(template: str) -> str:
+    # Always include the SDK and click for CLI-based scaffolds
+    base = ["wizelit-sdk", "click>=8.1.0"]
+    if template == "fast":
+        extras = []
+    elif template == "slow":
+        extras = ["redis>=4.5.0"]
+    else:  # hybrid
+        extras = ["redis>=4.5.0"]
+    return "\n".join(base + extras) + "\n"
+
+
+def _render_main_for_template(agent_name: str, template: str) -> str:
+    # Fast agent: synchronous/simple
+    if template == "fast":
+        return dedent(
+            f'''
+            """Fast Wizelit Agent
+            Generated by wizelit CLI
+            """
+
+            from wizelit_sdk.agent_wrapper import WizelitAgent
+
+            mcp = WizelitAgent(
+                "{agent_name}",
+                transport="streamable-http",
+            )
+
+
+            @mcp.ingest(is_long_running=False, description="Fast sync tool")
+            def analyze(text: str) -> dict:
+                """A fast synchronous tool example."""
+                return {{"echo": text}}
+
+
+            if __name__ == "__main__":
+                mcp.run()
+            '''
+        ).lstrip()
+
+    # Slow agent: long-running async example
+    if template == "slow":
+        return dedent(
+            f'''
+            """Slow Wizelit Agent (long-running)
+            Generated by wizelit CLI
+            """
+
+            import asyncio
+            from wizelit_sdk.agent_wrapper import WizelitAgent, Job
+
+            mcp = WizelitAgent(
+                "{agent_name}",
+                transport="streamable-http",
+                enable_streaming=True,
+            )
+
+
+            @mcp.ingest(is_long_running=True, description="Long-running refactor")
+            async def long_task(code: str, job: Job) -> dict:
+                job.logger.info("Starting long-running task")
+                # Simulate work and progress
+                for i in range(3):
+                    await asyncio.sleep(1)
+                    job.logger.info(f"Progress: {{i+1}}/3")
+                return {{"result": "done"}}
+
+
+            if __name__ == "__main__":
+                mcp.run()
+            '''
+        ).lstrip()
+
+    # Hybrid: both examples present
+    return dedent(
+        f'''
+        """Hybrid Wizelit Agent
+        Generated by wizelit CLI
+        """
+
+        import asyncio
+        from wizelit_sdk.agent_wrapper import WizelitAgent, Job
+
+        mcp = WizelitAgent(
+            "{agent_name}",
+            transport="streamable-http",
+            enable_streaming=True,
+        )
+
+
+        @mcp.ingest(is_long_running=False, description="Fast helper")
+        def quick(text: str) -> dict:
+            return {{"echo": text}}
+
+
+        @mcp.ingest(is_long_running=True, description="Slow worker")
+        async def worker(payload: str, job: Job) -> dict:
+            job.logger.info("Hybrid worker starting")
+            await asyncio.sleep(1)
+            return {{"status": "ok"}}
+
+
+        if __name__ == "__main__":
+            mcp.run()
+        '''
+    ).lstrip()
+
+
+@click.group()
+def cli() -> None:
+    """Wizelit SDK command-line tools."""
+
+
+@cli.command("scaffold")
+@click.argument("name")
+@click.option(
+    "--output-dir",
+    "-o",
+    type=click.Path(file_okay=False, dir_okay=True, path_type=Path),
+    default=Path.cwd(),
+    show_default="current directory",
+)
+@click.option(
+    "--transport",
+    type=click.Choice(TRANSPORT_CHOICES),
+    default="streamable-http",
+    show_default=True,
+)
+@click.option("--host", default="0.0.0.0", show_default=True)
+@click.option("--port", default=8080, show_default=True, type=int)
+@click.option("--force", is_flag=True, help="Overwrite existing files.")
+def scaffold(
+    name: str,
+    output_dir: Path,
+    transport: str,
+    host: str,
+    port: int,
+    force: bool,
+) -> None:
+    """Scaffold a new Wizelit agent project."""
+    slug = _slugify(name)
+    project_dir = output_dir / slug
+
+    project_dir.mkdir(parents=True, exist_ok=True)
+
+    _write_file(project_dir / "main.py", _render_main_py(name, transport, host, port), force)
+    _write_file(project_dir / "README.md", _render_readme(name), force)
+    # Ensure generated scaffold includes click as a runtime dependency
+    _write_file(
+        project_dir / "requirements.txt",
+        "wizelit-sdk\nclick>=8.1.0\n",
+        force,
+    )
+
+    click.echo(f"✅ Scaffold created at: {project_dir}")
+    click.echo("Next steps:")
+    click.echo("  1) Install dependencies: pip install -r requirements.txt")
+    click.echo("  2) Run the agent: python main.py")
+
+
+@cli.command("init")
+@click.argument("name")
+@click.option("--template", type=click.Choice(["fast", "slow", "hybrid"]), default="fast")
+@click.option(
+    "--output-dir",
+    "-o",
+    type=click.Path(file_okay=False, dir_okay=True, path_type=Path),
+    default=Path.cwd(),
+    show_default="current directory",
+)
+@click.option("--force", is_flag=True, help="Overwrite existing files.")
+def init(name: str, template: str, output_dir: Path, force: bool) -> None:
+    """Initialize a new Wizelit agent project using a template."""
+    slug = _slugify(name)
+    project_dir = output_dir / slug
+    project_dir.mkdir(parents=True, exist_ok=True)
+
+    # Write main.py for the selected template
+    _write_file(project_dir / "main.py", _render_main_for_template(name, template), force)
+    _write_file(project_dir / "README.md", _render_readme(name), force)
+    _write_file(project_dir / "requirements.txt", _render_requirements(template), force)
+    _write_file(project_dir / ".env.example", _render_env_example(), force)
+
+    click.echo(f"✅ Project initialized at: {project_dir}")
+    click.echo("Next steps:")
+    click.echo("  1) Install dependencies: pip install -r requirements.txt")
+    click.echo("  2) Run the agent: python main.py")
+
+
+@cli.command("validate")
+@click.argument("path", type=click.Path(file_okay=True, dir_okay=True, path_type=Path), default=Path.cwd())
+def validate(path: Path) -> None:
+    """Validate a Wizelit agent project structure."""
+    if path.is_file():
+        project_dir = path.parent
+    else:
+        project_dir = path
+
+    required = ["main.py", "requirements.txt", "README.md"]
+    missing = [p for p in required if not (project_dir / p).exists()]
+    if missing:
+        click.echo(f"❌ Missing files: {missing}")
+        raise click.Abort()
+    click.echo("✅ Project looks valid")
+
+
+@cli.command("list-tools")
+@click.argument("path", type=click.Path(file_okay=True, dir_okay=True, path_type=Path), default=Path.cwd())
+def list_tools(path: Path) -> None:
+    """List tools (functions decorated with @mcp.ingest) in a project."""
+    if path.is_file():
+        project_dir = path.parent
+    else:
+        project_dir = path
+
+    py_files = list(project_dir.glob("*.py"))
+    tools = []
+    for f in py_files:
+        text = f.read_text(encoding="utf-8")
+        if "@mcp.ingest" in text:
+            # naive extraction: look for def <name>( after decorator
+            for line in text.splitlines():
+                if line.strip().startswith("def "):
+                    name = line.strip().split()[1].split("(")[0]
+                    tools.append((f.name, name))
+
+    if not tools:
+        click.echo("No tools found (no @mcp.ingest decorators detected).")
+        return
+
+    for file, name in tools:
+        click.echo(f"- {name} (in {file})")
+
+
+def main() -> None:
+    cli()
+
+
+if __name__ == "__main__":
+    main()

wizelit_sdk-0.1.32.dist-info/METADATA

@@ -0,0 +1,222 @@
+Metadata-Version: 2.4
+Name: wizelit-sdk
+Version: 0.1.32
+Summary: Wizelit Agent Wrapper - Internal utility package
+Author-email: Your Name <your.email@company.com>
+Requires-Python: >=3.10
+Requires-Dist: asyncpg>=0.26.0
+Requires-Dist: click>=8.1.0
+Requires-Dist: fastmcp>=0.1.0
+Requires-Dist: redis>=4.5.0
+Requires-Dist: sqlalchemy>=1.4
+Requires-Dist: typeguard>=4.3.0
+Requires-Dist: typing-extensions>=4.0.0
+Provides-Extra: dev
+Requires-Dist: black>=22.0.0; extra == 'dev'
+Requires-Dist: build>=1.0.0; extra == 'dev'
+Requires-Dist: pytest>=7.0.0; extra == 'dev'
+Requires-Dist: ruff>=0.1.0; extra == 'dev'
+Provides-Extra: streaming
+Requires-Dist: redis>=4.5.0; extra == 'streaming'
+Description-Content-Type: text/markdown
+
+# wizelit-sdk
+
+Internal utility package for Wizelit Agent operations.
+
+> **New to Wizelit?** Start with the [**Quick Start Guide**](./QUICKSTART.md) to build your first agent in under 30 minutes!
+
+## Installation
+
+### Install from PyPI
+
+```bash
+uv pip install wizelit-sdk
+```
+
+### Add to pyproject.toml
+
+```toml
+[project]
+dependencies = [
+    "wizelit-sdk"
+]
+```
+
+## Quickstart
+
+1. Install the package (see above).
+2. Configure environment variables (see Configuration).
+3. Import and use the SDK from your app.
+
+### CLI
+
+The package also installs a small command-line tool named `wizelit-sdk` which helps scaffold and manage agent projects.
+
+Install and run the CLI:
+
+```bash
+# Install package (global or in virtualenv)
+pip install wizelit-sdk
+
+# Create a new project using a template (fast|slow|hybrid)
+wizelit-sdk init "My Agent" --template hybrid
+
+# Alternative (without wrapper):
+python -m wizelit_sdk.cli init "My Agent" --template hybrid
+```
+
+Useful commands:
+
+- `wizelit-sdk init <name> [--template fast|slow|hybrid]` — scaffold a new agent project
+- `wizelit-sdk scaffold <name>` — basic scaffold (older command)
+- `wizelit-sdk validate [path]` — validate project structure
+- `wizelit-sdk list-tools [path]` — list functions decorated with `@mcp.ingest`
+
+If you are developing locally, install editable and use the wrapper from your virtualenv:
+
+```bash
+# from repo root
+pip install -e .
+source .venv/bin/activate
+wizelit-sdk init "My Agent"
+```
+
+## Usage
+
+```python
+from wizelit_agent_wrapper import your_module
+
+# Use the wrapper
+result = your_module.function()
+```
+
+## SDK Guide
+
+### Basic import patterns
+
+```python
+from wizelit_sdk import database, exceptions
+from wizelit_sdk.agent_wrapper import agent_wrapper
+```
+
+### Initialize and call
+
+```python
+# Example: create a wrapper and call a method
+wrapper = agent_wrapper.WizelitAgentWrapper()
+result = wrapper.run()
+```
+
+### Error handling
+
+```python
+from wizelit_sdk.exceptions import WizelitError
+
+try:
+    wrapper.run()
+except WizelitError as exc:
+    # handle SDK-specific errors
+    print(exc)
+```
+
+## Configuration
+
+The SDK reads database configuration from the hosting application's environment. Provide these variables in the consuming project (e.g., via your app's .env or deployment secrets):
+
+- POSTGRES_USER
+- POSTGRES_PASSWORD
+- POSTGRES_HOST
+- POSTGRES_PORT
+- POSTGRES_DB
+
+You can also supply a full connection string via `DATABASE_URL` (overrides the individual fields). If using streaming/logging with Redis, set `REDIS_HOST`, `REDIS_PORT`, and optionally `REDIS_PASSWORD`.
+
+## Development
+
+### Setup Development Environment
+
+```bash
+# Clone repository
+git clone https://github.com/your-org/wizelit-sdk.git
+cd wizelit-sdk
+
+# Set up environment
+make setup
+
+# Activate virtual environment
+source .venv/bin/activate  # Windows: .venv\Scripts\activate
+
+# Install in development mode
+make install-dev
+```
+
+### Available Make Commands
+
+```bash
+make setup                  # Set up development environment
+make install                # Install package (production mode)
+make install-dev            # Install in development mode
+make test                   # Run tests
+make lint                   # Run code linting
+make format                 # Format code with black
+make check                  # Run tests and linting
+make clean                  # Clean build artifacts
+make build                  # Build package
+make release x.x.x          # Create new release (updates version, tags, pushes)
+make tag VERSION=x.x.x      # Create and push git tag
+make push                   # Push code and tags to remote
+make publish                # Publish package (run checks, build, push to remote)
+make publish-pypi           # Publish package to public PyPI
+make publish-artifactory    # Publish package to private Artifactory/PyPI
+make version                # Show current version
+make versions               # List all available versions
+```
+
+### Deploy to PyPI
+
+Use the built-in Makefile target (recommended):
+
+```bash
+make publish-pypi
+```
+
+This target will:
+
+1. Verify the git working tree is clean.
+2. Run tests and linting (`make check`).
+3. Build the package (`make build`).
+4. Upload the artifacts in `dist/` to PyPI via `twine`.
+
+Optional release/tag flow (before publishing):
+
+```bash
+# Interactive release flow (updates version, tags, pushes)
+make release x.x.x
+
+# Or tag a specific version
+make tag VERSION=x.x.x
+```
+
+Notes:
+
+- Ensure your PyPI credentials are configured locally (e.g., via $HOME/.pypirc or your preferred environment variables).
+- The package version must be unique on PyPI; if a version already exists, bump it and rebuild.
+
+## Contributing
+
+1. Create a feature branch
+2. Make your changes
+3. Run tests and linting: `make check`
+4. Commit your changes
+5. Push and create a pull request
+
+## Versioning
+
+We use [Semantic Versioning](https://semver.org/):
+
+- **MAJOR** version for incompatible API changes
+- **MINOR** version for new functionality (backward compatible)
+- **PATCH** version for bug fixes
+
+See [CHANGELOG.md](CHANGELOG.md) for version history.

{wizelit_sdk-0.1.31.dist-info → wizelit_sdk-0.1.32.dist-info}/RECORD

@@ -1,4 +1,5 @@
 wizelit_sdk/__init__.py,sha256=GRV3STJaNUg2KelKmvU60nhpYMdQ5lX0_FQyV_e5g1I,1168
+wizelit_sdk/cli.py,sha256=9leN1OHMj9IszbEtmldJDd09epg5KCji6RiQ6xjzraI,9541
 wizelit_sdk/database.py,sha256=HzvMY-KZQBkGs-Ey_wKAIPOJEb5BqGjEi98Tip7CQBo,4830
 wizelit_sdk/exceptions.py,sha256=KHkexTbD5t05nD-t9RnrJIFcqOsATBOUX9y5uHVfQeE,8557
 wizelit_sdk/agent_wrapper/__init__.py,sha256=srVcHZfvpf3q8m2nzIST6Ofl2V8Vv5GRtcNXudOLiD8,333
@@ -9,6 +10,7 @@ wizelit_sdk/agent_wrapper/streaming.py,sha256=l7SbvPuQYMdGO38t1upyI5wTVdd5eB-2BS
 wizelit_sdk/models/__init__.py,sha256=UB7nfHoE6hGcjfzy7w8AmuKVmYrTg9wIgGjG8GWziJ0,143
 wizelit_sdk/models/base.py,sha256=11XyxLUG0VjdiNxe4zmzGDUjKEOsjSvUjA8yaQgfCa0,758
 wizelit_sdk/models/job.py,sha256=HD8ohVqzakC8J7VAwmh8UCi_BLLiqx85TbR0FjO2rlg,2494
-wizelit_sdk-0.1.31.dist-info/METADATA,sha256=fOk9ur63IBthIIyjAMPW-cjahQVWycOAgPsy2ORfqOY,3297
-wizelit_sdk-0.1.31.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-wizelit_sdk-0.1.31.dist-info/RECORD,,
+wizelit_sdk-0.1.32.dist-info/METADATA,sha256=94NCVGl52cRNM7Z7z93l-jwh-OAdmMx-uBx_dPymBAg,5787
+wizelit_sdk-0.1.32.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+wizelit_sdk-0.1.32.dist-info/entry_points.txt,sha256=XRtYie5Jhxb5xv0Kda2OwA1k6fuf4utEttjo29sJ2FM,53
+wizelit_sdk-0.1.32.dist-info/RECORD,,

wizelit_sdk-0.1.32.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+wizelit-sdk = wizelit_sdk.cli:main

wizelit_sdk-0.1.31.dist-info/METADATA

@@ -1,124 +0,0 @@
-Metadata-Version: 2.4
-Name: wizelit-sdk
-Version: 0.1.31
-Summary: Wizelit Agent Wrapper - Internal utility package
-Author-email: Your Name <your.email@company.com>
-Requires-Python: >=3.10
-Requires-Dist: asyncpg>=0.26.0
-Requires-Dist: fastmcp>=0.1.0
-Requires-Dist: redis>=4.5.0
-Requires-Dist: sqlalchemy>=1.4
-Requires-Dist: typeguard>=4.3.0
-Requires-Dist: typing-extensions>=4.0.0
-Provides-Extra: dev
-Requires-Dist: black>=22.0.0; extra == 'dev'
-Requires-Dist: build>=1.0.0; extra == 'dev'
-Requires-Dist: pytest>=7.0.0; extra == 'dev'
-Requires-Dist: ruff>=0.1.0; extra == 'dev'
-Provides-Extra: streaming
-Requires-Dist: redis>=4.5.0; extra == 'streaming'
-Description-Content-Type: text/markdown
-
-# wizelit-sdk
-
-Internal utility package for Wizelit Agent operations.
-
-## Installation
-
-### Install from Git Repository
-
-```bash
-# Install latest version from main branch
-uv pip install git+https://github.com/your-org/wizelit-sdk.git
-
-# Install specific version
-uv pip install git+https://github.com/your-org/wizelit-sdk.git@v0.1.0
-
-# Install with SSH (Recommended for Private Repos)
-uv pip install git+ssh://git@github.com/your-org/wizelit-sdk.git@v0.1.0
-```
-
-### Add to pyproject.toml
-
-```toml
-[project]
-dependencies = [
-    "wizelit-sdk @ git+ssh://git@github.com/your-org/wizelit-sdk.git@v0.1.0"
-]
-```
-
-## Usage
-
-```python
-from wizelit_agent_wrapper import your_module
-
-# Use the wrapper
-result = your_module.function()
-```
-
-## Configuration
-
-The SDK reads database configuration from the hosting application's environment. Provide these variables in the consuming project (e.g., via your app's .env or deployment secrets):
-
-- POSTGRES_USER
-- POSTGRES_PASSWORD
-- POSTGRES_HOST
-- POSTGRES_PORT
-- POSTGRES_DB
-
-You can also supply a full connection string via `DATABASE_URL` (overrides the individual fields). If using streaming/logging with Redis, set `REDIS_HOST`, `REDIS_PORT`, and optionally `REDIS_PASSWORD`.
-
-## Development
-
-### Setup Development Environment
-
-```bash
-# Clone repository
-git clone https://github.com/your-org/wizelit-sdk.git
-cd wizelit-sdk
-
-# Set up environment
-make setup
-
-# Activate virtual environment
-source .venv/bin/activate  # Windows: .venv\Scripts\activate
-
-# Install in development mode
-make install-dev
-```
-
-### Available Make Commands
-
-```bash
-make setup          # Set up development environment
-make install-dev    # Install in development mode
-make test           # Run tests
-make format         # Format code with black
-make lint           # Lint code with ruff
-make check          # Run tests and linting
-make clean          # Clean build artifacts
-make build          # Build package
-make release        # Create new release (interactive)
-make tag VERSION=x.x.x  # Create specific version tag
-make push           # Push code and tags
-make version        # Show current version
-make versions       # List all available versions
-```
-
-## Contributing
-
-1. Create a feature branch
-2. Make your changes
-3. Run tests and linting: `make check`
-4. Commit your changes
-5. Push and create a pull request
-
-## Versioning
-
-We use [Semantic Versioning](https://semver.org/):
-
-- **MAJOR** version for incompatible API changes
-- **MINOR** version for new functionality (backward compatible)
-- **PATCH** version for bug fixes
-
-See [CHANGELOG.md](CHANGELOG.md) for version history.