lyceum-cli 1.0.25__py3-none-any.whl → 1.0.27__py3-none-any.whl

lyceum/external/storage/files.py

@@ -1,273 +0,0 @@
-"""Storage file management commands"""
-
-from datetime import datetime
-from pathlib import Path
-
-import httpx
-import typer
-from rich.console import Console
-from rich.table import Table
-
-from ...shared.config import config
-
-console = Console()
-
-storage_app = typer.Typer(name="storage", help="File storage commands")
-
-
-def format_size(size_bytes: int) -> str:
-    """Format bytes into human-readable size."""
-    for unit in ["B", "KB", "MB", "GB"]:
-        if size_bytes < 1024:
-            return f"{size_bytes:.1f} {unit}"
-        size_bytes /= 1024
-    return f"{size_bytes:.1f} TB"
-
-
-@storage_app.command("list")
-def list_files(
-    prefix: str = typer.Option("", "--prefix", "-p", help="Filter by prefix/folder"),
-    limit: int = typer.Option(100, "--limit", "-n", help="Maximum files to list"),
-):
-    """List files in your storage bucket."""
-    try:
-        config.get_client()
-
-        params = {"prefix": prefix, "max_files": limit}
-
-        response = httpx.get(
-            f"{config.base_url}/api/v2/external/storage/list-files",
-            headers={"Authorization": f"Bearer {config.api_key}"},
-            params=params,
-            timeout=30.0,
-        )
-
-        if response.status_code != 200:
-            console.print(f"[red]Error: HTTP {response.status_code}[/red]")
-            if response.status_code == 401:
-                console.print("[yellow]Run 'lyceum auth login' to re-authenticate.[/yellow]")
-            else:
-                console.print(f"[red]{response.content.decode()}[/red]")
-            raise typer.Exit(1)
-
-        files = response.json()
-
-        if not files:
-            console.print("[dim]No files found.[/dim]")
-            if prefix:
-                console.print(f"[dim]Prefix filter: {prefix}[/dim]")
-            return
-
-        table = Table(title="Storage Files")
-        table.add_column("Key", style="cyan")
-        table.add_column("Size", style="green", justify="right")
-        table.add_column("Last Modified", style="dim")
-
-        for f in files:
-            last_mod = f.get("last_modified", "")
-            if last_mod:
-                try:
-                    dt = datetime.fromisoformat(last_mod.replace("Z", "+00:00"))
-                    last_mod = dt.strftime("%Y-%m-%d %H:%M")
-                except Exception:
-                    pass
-
-            table.add_row(
-                f.get("key", ""),
-                format_size(f.get("size", 0)),
-                last_mod,
-            )
-
-        console.print(table)
-        console.print(f"\n[dim]Total: {len(files)} file(s)[/dim]")
-
-    except typer.Exit:
-        raise
-    except Exception as e:
-        console.print(f"[red]Error: {e}[/red]")
-        raise typer.Exit(1)
-
-
-@storage_app.command("upload")
-def upload_file(
-    file_path: Path = typer.Argument(..., help="Local file to upload"),
-    dest: str = typer.Option(None, "--dest", "-d", help="Destination path in storage (defaults to filename)"),
-):
-    """Upload a file to your storage bucket."""
-    try:
-        config.get_client()
-
-        if not file_path.exists():
-            console.print(f"[red]Error: File not found: {file_path}[/red]")
-            raise typer.Exit(1)
-
-        if not file_path.is_file():
-            console.print(f"[red]Error: Not a file: {file_path}[/red]")
-            raise typer.Exit(1)
-
-        remote_path = dest or file_path.name
-        file_size = file_path.stat().st_size
-
-        console.print(f"[dim]Uploading {file_path.name} ({format_size(file_size)})...[/dim]")
-
-        with open(file_path, "rb") as f:
-            files = {"file": (file_path.name, f)}
-            data = {"key": remote_path} if dest else {}
-
-            response = httpx.post(
-                f"{config.base_url}/api/v2/external/storage/upload",
-                headers={"Authorization": f"Bearer {config.api_key}"},
-                files=files,
-                data=data,
-                timeout=300.0,  # 5 min timeout for large files
-            )
-
-        if response.status_code != 200:
-            console.print(f"[red]Error: HTTP {response.status_code}[/red]")
-            if response.status_code == 401:
-                console.print("[yellow]Run 'lyceum auth login' to re-authenticate.[/yellow]")
-            else:
-                console.print(f"[red]{response.content.decode()}[/red]")
-            raise typer.Exit(1)
-
-        result = response.json()
-        console.print(f"[green]Uploaded successfully![/green]")
-        console.print(f"[dim]Key: {result.get('key')}[/dim]")
-        console.print(f"[dim]Size: {format_size(result.get('size', 0))}[/dim]")
-
-    except typer.Exit:
-        raise
-    except Exception as e:
-        console.print(f"[red]Error: {e}[/red]")
-        raise typer.Exit(1)
-
-
-@storage_app.command("download")
-def download_file(
-    path: str = typer.Argument(..., help="File path in storage to download"),
-    output: Path = typer.Option(None, "--output", "-o", help="Local output path (defaults to filename)"),
-):
-    """Download a file from your storage bucket."""
-    try:
-        config.get_client()
-
-        # Default output to the filename part of the path
-        output_path = output or Path(path.split("/")[-1])
-
-        console.print(f"[dim]Downloading {path}...[/dim]")
-
-        response = httpx.get(
-            f"{config.base_url}/api/v2/external/storage/download/{path}",
-            headers={"Authorization": f"Bearer {config.api_key}"},
-            timeout=300.0,
-        )
-
-        if response.status_code == 404:
-            console.print(f"[red]Error: File not found: {path}[/red]")
-            raise typer.Exit(1)
-
-        if response.status_code != 200:
-            console.print(f"[red]Error: HTTP {response.status_code}[/red]")
-            if response.status_code == 401:
-                console.print("[yellow]Run 'lyceum auth login' to re-authenticate.[/yellow]")
-            else:
-                console.print(f"[red]{response.content.decode()}[/red]")
-            raise typer.Exit(1)
-
-        with open(output_path, "wb") as f:
-            f.write(response.content)
-
-        console.print(f"[green]Downloaded successfully![/green]")
-        console.print(f"[dim]Saved to: {output_path}[/dim]")
-        console.print(f"[dim]Size: {format_size(len(response.content))}[/dim]")
-
-    except typer.Exit:
-        raise
-    except Exception as e:
-        console.print(f"[red]Error: {e}[/red]")
-        raise typer.Exit(1)
-
-
-@storage_app.command("delete")
-def delete_file(
-    path: str = typer.Argument(..., help="File path in storage to delete"),
-    force: bool = typer.Option(False, "--force", "-f", help="Skip confirmation"),
-):
-    """Delete a file from your storage bucket."""
-    try:
-        config.get_client()
-
-        if not force:
-            confirm = typer.confirm(f"Delete '{path}'?")
-            if not confirm:
-                console.print("[dim]Cancelled.[/dim]")
-                raise typer.Exit(0)
-
-        response = httpx.delete(
-            f"{config.base_url}/api/v2/external/storage/delete/{path}",
-            headers={"Authorization": f"Bearer {config.api_key}"},
-            timeout=30.0,
-        )
-
-        if response.status_code == 404:
-            console.print(f"[red]Error: File not found: {path}[/red]")
-            raise typer.Exit(1)
-
-        if response.status_code != 200:
-            console.print(f"[red]Error: HTTP {response.status_code}[/red]")
-            if response.status_code == 401:
-                console.print("[yellow]Run 'lyceum auth login' to re-authenticate.[/yellow]")
-            else:
-                console.print(f"[red]{response.content.decode()}[/red]")
-            raise typer.Exit(1)
-
-        console.print(f"[green]Deleted: {path}[/green]")
-
-    except typer.Exit:
-        raise
-    except Exception as e:
-        console.print(f"[red]Error: {e}[/red]")
-        raise typer.Exit(1)
-
-
-@storage_app.command("delete-folder")
-def delete_folder(
-    prefix: str = typer.Argument(..., help="Folder prefix to delete"),
-    force: bool = typer.Option(False, "--force", "-f", help="Skip confirmation"),
-):
-    """Delete all files in a folder (by prefix)."""
-    try:
-        config.get_client()
-
-        if not force:
-            confirm = typer.confirm(f"Delete all files in '{prefix}/'?")
-            if not confirm:
-                console.print("[dim]Cancelled.[/dim]")
-                raise typer.Exit(0)
-
-        response = httpx.delete(
-            f"{config.base_url}/api/v2/external/storage/delete-folder/{prefix}",
-            headers={"Authorization": f"Bearer {config.api_key}"},
-            timeout=60.0,
-        )
-
-        if response.status_code == 404:
-            console.print(f"[yellow]No files found in folder: {prefix}/[/yellow]")
-            raise typer.Exit(0)
-
-        if response.status_code != 200:
-            console.print(f"[red]Error: HTTP {response.status_code}[/red]")
-            if response.status_code == 401:
-                console.print("[yellow]Run 'lyceum auth login' to re-authenticate.[/yellow]")
-            else:
-                console.print(f"[red]{response.content.decode()}[/red]")
-            raise typer.Exit(1)
-
-        result = response.json()
-        console.print(f"[green]{result.get('message')}[/green]")
-
-    except typer.Exit:
-        raise
-    except Exception as e:
-        console.print(f"[red]Error: {e}[/red]")
-        raise typer.Exit(1)

lyceum_cli-1.0.25.dist-info/RECORD

@@ -1,46 +0,0 @@
-lyceum/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-lyceum/main.py,sha256=jE4P4SvHTeguLKL9wRBfOBxGHzDKxH_PcM46ouY-xnY,921
-lyceum/external/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-lyceum/external/auth/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-lyceum/external/auth/login.py,sha256=T-V6pzi3s0ZynpUUeLnN2y-cxJI4msj5z26S3_wJ1AE,23526
-lyceum/external/compute/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-lyceum/external/compute/execution/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-lyceum/external/compute/execution/config.py,sha256=6JJgLJnDPTwevEaNdB1nEICih_qbBmws5u5_S9gj7k0,8866
-lyceum/external/compute/execution/docker.py,sha256=lATtJzsmsu9iCUH-ynXZsfrIg-_aCkYE5r2l3DolisQ,9603
-lyceum/external/compute/execution/docker_config.py,sha256=tIiEVlp3yYZxLqhRsBqri7wV9_K0fPm6bvTRhRnKEd4,4071
-lyceum/external/compute/execution/notebook.py,sha256=fbN546-j1Cw88j7jbDKkup8J1xaws0eBYZJgt1NTJVg,7544
-lyceum/external/compute/execution/python.py,sha256=1ekNuF8_j-YG0-oNFP3C6Jjyw7IUlIngvL06C1usyu8,12864
-lyceum/external/compute/execution/workloads.py,sha256=4fsRWbYGmsQMGPPIN1jUG8cG5NPG9yV26ANJ-DtaXqc,5844
-lyceum/external/compute/inference/__init__.py,sha256=4YLoUKDEzitexynJv_Q5O0w1lty8CJ6uyRxuc1LiaBw,89
-lyceum/external/compute/inference/batch.py,sha256=HWwS6XHDZOG7QftNTNGvzao8Y6DzkEOdpOAZAEZyav4,11438
-lyceum/external/compute/inference/chat.py,sha256=hITj_UGLaxCJQskU-YbeaEerM5Xt_eJpEsYrTJoUpk4,8485
-lyceum/external/compute/inference/models.py,sha256=BkCEdvyliezGOUulj557e-Eoif0_HKR3CxqpEhdAZaA,10339
-lyceum/external/general/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-lyceum/external/storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lyceum/external/storage/files.py,sha256=WdSp6t90jLzChOkg2uI4rZ5pq6bmHDCSo7CDUDVXWyk,9311
-lyceum/shared/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-lyceum/shared/config.py,sha256=mZXbdwKxbkRA6RCnTPruaNWtLOaheYO4MnCxXTFNLu8,5548
-lyceum/shared/display.py,sha256=n9QQy-JVvoAgzss9D9cGfdw7GUY6t9qTYWr81v9q1fk,4855
-lyceum/shared/imports.py,sha256=wEG4wfVTIqJ6MBWDRAN96iGmVCb9ST2aOqSjkbvajug,11768
-lyceum/shared/streaming.py,sha256=wFb7w7fra63y8WWaIA8_E1Z6Sx_6G-0J53Zh010eZgk,9355
-lyceum_cloud_execution_api_client/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-lyceum_cloud_execution_api_client/api/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-lyceum_cloud_execution_api_client/models/__init__.py,sha256=AMlb9R9O9aNC9hvKz_8TFpEfOolYC3VtFS5JX17kYks,4888
-tests/__init__.py,sha256=yuZY5OKj5n7_jxuEuXtUjl77XCIopOxfiz7s_leAYPc,24
-tests/conftest.py,sha256=TwnnolWULDaAyCygQLj595TKreS1p2y-ezeN0b2WWX8,5052
-tests/unit/__init__.py,sha256=_w8WnsodwDTzGEx1Xtj1xjihh_VCUevi4LA4tOdW_Uw,33
-tests/unit/external/__init__.py,sha256=7lGFpul2lNMPJVPV1Jd27wzpsgb9EZwwa6q8fixUFLY,39
-tests/unit/external/compute/__init__.py,sha256=hf16EmvKpT4P0iw292Z0xCRswKqcU8PteT4Zr1sgRLo,38
-tests/unit/external/compute/execution/__init__.py,sha256=eUpROWmXCsK9ZaImYMw9iIDiQUIxj5HCrd06SyjE0Vg,40
-tests/unit/external/compute/execution/test_data.py,sha256=49p19xquOlmm7jKwCr6NQLooG9MpK8GkGQ3PwHrVuj0,856
-tests/unit/external/compute/execution/test_dependency_resolver.py,sha256=2X0Lan8C4av7QDeHpjZxk_RlTB1C6GzU1Gv5G0WYnSg,9479
-tests/unit/external/compute/execution/test_python_helpers.py,sha256=KyeTX0ps7L0x7xBSslwiQHiGa3oriX_vcqgkcJvbcm4,13421
-tests/unit/external/compute/execution/test_python_run.py,sha256=fKuWyTVshC3iGvoWfQmeNMi0N-46iCC_Zm2-lq-n-Ro,10353
-tests/unit/shared/__init__.py,sha256=ouZueovCsCdgfuKoMP1Uh69E3fmn0U8ltllHWhXhpgg,39
-tests/unit/shared/test_config.py,sha256=rokbpv_S9J-aUtg41eddn0ZuVWyTYZNXO9VtCZdXstU,11315
-tests/unit/shared/test_streaming.py,sha256=sFjkla5SvaWWPi42lQ6vHSTor_htLMMbBttnCyGsB98,8706
-lyceum_cli-1.0.25.dist-info/METADATA,sha256=mC4h00m-5ihgHqd9gdNoIpET5Ok0EQv5cAqLKaVLc2c,1482
-lyceum_cli-1.0.25.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lyceum_cli-1.0.25.dist-info/entry_points.txt,sha256=Oq-9wDkxVd6MHgNiUTYwXI9SGhvR3VkD7Mvk0xhiUZo,43
-lyceum_cli-1.0.25.dist-info/top_level.txt,sha256=-546wowhLvi8w6Gef9vwO0T1i0ME2PanTrgC0YX29y4,47
-lyceum_cli-1.0.25.dist-info/RECORD,,

tests/__init__.py

@@ -1 +0,0 @@
-"""Lyceum CLI tests."""

tests/conftest.py

@@ -1,200 +0,0 @@
-"""Shared pytest fixtures for CLI tests."""
-
-import json
-import os
-from pathlib import Path
-from unittest.mock import AsyncMock, MagicMock, Mock, patch
-
-import pytest
-
-
-# --- Basic Fixtures ---
-
-
-@pytest.fixture
-def mock_execution_id() -> str:
-    """Consistent execution ID for testing."""
-    return "550e8400-e29b-41d4-a716-446655440000"
-
-
-@pytest.fixture
-def mock_streaming_url(mock_execution_id) -> str:
-    """Mock streaming URL."""
-    return f"https://api.lyceum.dev/stream/{mock_execution_id}"
-
-
-# --- Config Fixtures ---
-
-
-@pytest.fixture
-def mock_config_data() -> dict:
-    """Mock config file data."""
-    return {
-        "access_token": "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJleHAiOjk5OTk5OTk5OTl9.test",
-        "refresh_token": "test-refresh-token",
-        "base_url": "https://api.lyceum.dev",
-    }
-
-
-@pytest.fixture
-def setup_config_file(tmp_path, mock_config_data):
-    """Setup mock config file in tmp directory."""
-
-    def _setup(config_data: dict = None):
-        config_dir = tmp_path / ".lyceum"
-        config_dir.mkdir(parents=True, exist_ok=True)
-        config_file = config_dir / "config.json"
-        config_file.write_text(json.dumps(config_data or mock_config_data))
-        return config_file
-
-    return _setup
-
-
-# --- HTTP Response Fixtures (following PR #502 pattern) ---
-
-
-@pytest.fixture
-def setup_httpx_response():
-    """Setup mock for httpx response."""
-
-    def _setup(status_code: int = 200, json_data: dict = None, text: str = "OK"):
-        mock_response = MagicMock()
-        mock_response.status_code = status_code
-        mock_response.json.return_value = json_data or {}
-        mock_response.text = text
-        mock_response.content = text.encode()
-        return mock_response
-
-    return _setup
-
-
-@pytest.fixture
-def setup_httpx_post(setup_httpx_response):
-    """Setup mock for httpx.post calls."""
-
-    def _setup(status_code: int = 200, execution_id: str = "test-exec-id"):
-        response = setup_httpx_response(
-            status_code=status_code,
-            json_data={
-                "execution_id": execution_id,
-                "streaming_url": f"https://api.lyceum.dev/stream/{execution_id}",
-                "status": "queued",
-            },
-        )
-        return response
-
-    return _setup
-
-
-@pytest.fixture
-def setup_httpx_stream():
-    """Setup mock for httpx.stream SSE responses."""
-
-    def _setup(events: list[dict], status_code: int = 200):
-        mock_response = MagicMock()
-        mock_response.status_code = status_code
-        mock_response.iter_lines.return_value = [
-            f"data: {json.dumps(event)}" for event in events
-        ]
-
-        mock_context = MagicMock()
-        mock_context.__enter__.return_value = mock_response
-        mock_context.__exit__.return_value = None
-
-        return mock_context
-
-    return _setup
-
-
-# --- Machine/Quota Fixtures ---
-
-
-@pytest.fixture
-def setup_available_machines():
-    """Setup mock for available machines API response."""
-
-    def _setup(machines: list[str] = None):
-        machines = machines or ["cpu", "a100", "h100"]
-        mock_response = MagicMock()
-        mock_response.status_code = 200
-        mock_response.json.return_value = {
-            "hardware_profiles": [{"hardware_profile": m} for m in machines]
-        }
-        return mock_response
-
-    return _setup
-
-
-# --- File System Fixtures ---
-
-
-@pytest.fixture
-def sample_workspace(tmp_path):
-    """Create a sample workspace with Python files and packages."""
-    workspace = tmp_path / "workspace"
-    workspace.mkdir()
-
-    # Main file
-    main_py = workspace / "main.py"
-    main_py.write_text(
-        '''"""Main script"""
-import numpy as np
-from mypackage import helper_function
-from standalone import standalone_func
-
-def main():
-    print(helper_function())
-    print(standalone_func())
-
-if __name__ == "__main__":
-    main()
-'''
-    )
-
-    # Package
-    pkg = workspace / "mypackage"
-    pkg.mkdir()
-    (pkg / "__init__.py").write_text(
-        '''"""My package"""
-from .utils import helper_function
-'''
-    )
-    (pkg / "utils.py").write_text(
-        '''"""Utils module"""
-def helper_function():
-    return "Hello from utils"
-'''
-    )
-
-    # Standalone module
-    (workspace / "standalone.py").write_text(
-        '''"""Standalone module"""
-def standalone_func():
-    return "Hello from standalone"
-'''
-    )
-
-    # Requirements file
-    (workspace / "requirements.txt").write_text("numpy==1.24.0\n")
-
-    return workspace
-
-
-@pytest.fixture
-def sample_workspace_config(sample_workspace):
-    """Create .lyceum/config.json in sample workspace."""
-
-    def _setup(config_data: dict = None):
-        lyceum_dir = sample_workspace / ".lyceum"
-        lyceum_dir.mkdir()
-        config_file = lyceum_dir / "config.json"
-
-        default_config = {
-            "workspace": str(sample_workspace),
-            "dependencies": {"merged": ["numpy==1.24.0"]},
-            "local_packages": ["mypackage"],
-        }
-        config_file.write_text(json.dumps(config_data or default_config))
-        return config_file
-
-    return _setup

tests/unit/__init__.py

@@ -1 +0,0 @@
-"""Unit tests for Lyceum CLI."""

tests/unit/external/__init__.py

@@ -1 +0,0 @@
-"""Unit tests for external modules."""

tests/unit/external/compute/__init__.py

@@ -1 +0,0 @@
-"""Unit tests for compute modules."""

tests/unit/external/compute/execution/__init__.py

@@ -1 +0,0 @@
-"""Unit tests for execution modules."""

tests/unit/external/compute/execution/test_data.py

@@ -1,33 +0,0 @@
-"""Shared test data for CLI tests."""
-
-VALID_EXECUTION_PAYLOAD = {
-    "code": "print('hello')",
-    "execution_type": "cpu",
-    "timeout": 60,
-}
-
-VALID_EXECUTION_WITH_REQUIREMENTS = {
-    **VALID_EXECUTION_PAYLOAD,
-    "requirements_content": "numpy==1.24.0",
-}
-
-VALID_EXECUTION_WITH_IMPORTS = {
-    **VALID_EXECUTION_PAYLOAD,
-    "import_files": '{"mypackage/__init__.py": "# init", "mypackage/utils.py": "def foo(): pass"}',
-}
-
-SAMPLE_SSE_OUTPUT_EVENT = {"output": {"content": "Hello, World!\n"}}
-
-SAMPLE_SSE_JOB_FINISHED = {
-    "jobFinished": {"job": {"result": {"returnCode": "0"}}}
-}
-
-SAMPLE_SSE_JOB_FAILED = {
-    "jobFinished": {"job": {"result": {"returnCode": "1"}}}
-}
-
-SAMPLE_WORKSPACE_CONFIG = {
-    "workspace": "/path/to/workspace",
-    "dependencies": {"merged": ["numpy==1.24.0", "pandas>=2.0.0"]},
-    "local_packages": ["mypackage"],
-}