lyceum-cli 1.0.14__py3-none-any.whl → 1.0.18__py3-none-any.whl

lyceum/shared/streaming.py

@@ -0,0 +1,134 @@
+"""
+Streaming utilities for execution output
+"""
+
+import json
+import re
+import httpx
+from rich.console import Console
+
+from .config import config
+
+console = Console()
+
+
+def strip_ansi_codes(text: str) -> str:
+    """Remove ANSI escape codes from text"""
+    ansi_escape = re.compile(r'\x1b\[[0-9;]*m')
+    return ansi_escape.sub('', text)
+
+
+def stream_execution_output(execution_id: str, streaming_url: str = None) -> bool:
+    """Stream execution output in real-time. Returns True if successful, False if failed."""
+    if not streaming_url:
+        # Fallback to old endpoint if no streaming URL provided
+        stream_url = f"{config.base_url}/api/v2/external/execution/streaming/{execution_id}"
+    else:
+        stream_url = streaming_url
+
+    try:
+        console.print(f"[dim]🔗 Connecting to execution stream...[/dim]")
+        console.print(f"[dim]Stream URL: {stream_url}[/dim]")
+
+        with httpx.stream("GET", stream_url, headers={"Authorization": f"Bearer {config.api_key}"}, timeout=600.0) as response:
+            if response.status_code != 200:
+                console.print(f"[red]❌ Stream failed: HTTP {response.status_code}[/red]")
+                return False
+            
+            console.print("[dim]📡 Streaming output...[/dim]")
+            
+            for line in response.iter_lines():
+                if line.strip():
+                    # Parse Server-Sent Events format
+                    if line.startswith("data: "):
+                        data_json = line[6:]  # Remove "data: " prefix
+                        try:
+                            data = json.loads(data_json)
+                            event_type = data.get("type", "unknown")
+                            
+                            if event_type == "output":
+                                # Print output without extra formatting, stripping ANSI codes
+                                output = data.get("content", "")  # Fixed: server sends "content" not "output"
+                                if output:
+                                    clean_output = strip_ansi_codes(output)
+                                    console.print(clean_output, end="")
+                            
+                            elif event_type == "completed":
+                                status = data.get("status", "unknown")
+                                exec_time = data.get("execution_time", 0)
+                                
+                                if status == "completed":
+                                    console.print(f"\n[green]✅ Execution completed successfully in {exec_time:.1f}s[/green]")
+                                elif status in ["failed_user", "failed_system"]:
+                                    console.print(f"\n[red]❌ Execution failed: {status}[/red]")
+                                    # Show errors if available
+                                    errors = data.get("errors")
+                                    if errors:
+                                        console.print(f"[red]Error: {errors}[/red]")
+                                elif status == "timeout":
+                                    console.print(f"\n[yellow]⏰ Execution timed out after {exec_time:.1f}s[/yellow]")
+                                elif status == "cancelled":
+                                    console.print(f"\n[yellow]🛑 Execution was cancelled[/yellow]")
+                                
+                                return status == "completed"
+                            
+                            elif event_type == "error":
+                                error_msg = data.get("message", "Unknown error")
+                                console.print(f"\n[red]❌ Error: {error_msg}[/red]")
+                                return False
+                                
+                        except json.JSONDecodeError:
+                            # Skip malformed JSON
+                            continue
+            
+            console.print(f"\n[yellow]⚠️ Stream ended without completion signal[/yellow]")
+            # Fallback: poll execution status
+            return check_execution_status(execution_id)
+
+    except Exception as e:
+        console.print(f"\n[red]❌ Streaming error: {e}[/red]")
+        # Fallback: poll execution status
+        return check_execution_status(execution_id)
+
+
+def check_execution_status(execution_id: str) -> bool:
+    """Check execution status as fallback when streaming fails."""
+    import time
+
+    console.print("[dim]⏳ Checking execution status...[/dim]")
+
+    for _ in range(30):  # Poll for up to 30 seconds
+        try:
+            response = httpx.get(
+                f"{config.base_url}/api/v2/external/execution/streaming/{execution_id}/status",
+                headers={"Authorization": f"Bearer {config.api_key}"},
+                timeout=10.0
+            )
+
+            if response.status_code == 200:
+                data = response.json()
+                status = data.get('status', 'unknown')
+
+                if status == 'completed':
+                    console.print(f"[green]✅ Execution completed successfully![/green]")
+                    return True
+                elif status in ['failed_user', 'failed_system', 'failed']:
+                    console.print(f"[red]❌ Execution failed: {status}[/red]")
+                    errors = data.get('errors')
+                    if errors:
+                        console.print(f"[red]Error: {errors}[/red]")
+                    return False
+                elif status in ['timeout', 'cancelled']:
+                    console.print(f"[yellow]⚠️ Execution {status}[/yellow]")
+                    return False
+                elif status in ['running', 'pending', 'queued']:
+                    # Still running, continue polling
+                    time.sleep(1)
+                    continue
+
+        except Exception as e:
+            console.print(f"[red]Error checking status: {e}[/red]")
+            break
+
+    console.print("[yellow]⚠️ Status check timed out[/yellow]")
+    return False

{lyceum_cli-1.0.14.dist-info → lyceum_cli-1.0.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lyceum-cli
-Version: 1.0.14
+Version: 1.0.18
 Summary: Command-line interface for Lyceum Cloud Execution API
 Home-page: https://lyceum.technology
 Author: Lyceum Team

lyceum_cli-1.0.18.dist-info/RECORD

@@ -0,0 +1,25 @@
+lyceum/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lyceum/main.py,sha256=GO__4_w7Xrr2M6sUlgtG5VK6q64Ako-Qm1i5tcuYk2U,934
+lyceum/external/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lyceum/external/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lyceum/external/auth/login.py,sha256=GYhvhQrWG49LYsqxfNdVfs59GQtjKOlRmcytww9PCWk,23289
+lyceum/external/compute/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lyceum/external/compute/execution/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lyceum/external/compute/execution/python.py,sha256=IiQpudxLAD4lMaQsvON6MGPNGYtEW9Zoip9dSv8KI0g,3745
+lyceum/external/compute/inference/__init__.py,sha256=4YLoUKDEzitexynJv_Q5O0w1lty8CJ6uyRxuc1LiaBw,89
+lyceum/external/compute/inference/batch.py,sha256=OHwJndqvsQx0K0-3to8Tc9HB2xA6iq1MNvXvpMxi5Oc,11775
+lyceum/external/compute/inference/chat.py,sha256=IZri9YrgMWHHQkJMj_ausdjBgNmqBAluAZcmRwJ-sIM,8756
+lyceum/external/compute/inference/models.py,sha256=VVHMUhTPVhLKiuvC9Q1zvBD-lM4KDJPUqTmFO7kHpUA,10319
+lyceum/external/general/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lyceum/shared/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lyceum/shared/config.py,sha256=7HAKztOvEv8zrn_2t-9iZlrgkV2n_5sIN-XOK7Mb8pI,5550
+lyceum/shared/display.py,sha256=LeGZjCFn8_8sOgao2pni_qN3OhiFwKJHpEJLjCwftwM,4884
+lyceum/shared/streaming.py,sha256=-_MLgrACwe1oU5n3UG4kzg9TcUQXc3cLNV6WyplEc_k,6073
+lyceum_cloud_execution_api_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lyceum_cloud_execution_api_client/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lyceum_cloud_execution_api_client/models/__init__.py,sha256=AMlb9R9O9aNC9hvKz_8TFpEfOolYC3VtFS5JX17kYks,4888
+lyceum_cli-1.0.18.dist-info/METADATA,sha256=Xr3_q1LQTamNgwcEidQ6QHCdnQcdxK-AY-wyVTwp2f4,1482
+lyceum_cli-1.0.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lyceum_cli-1.0.18.dist-info/entry_points.txt,sha256=Oq-9wDkxVd6MHgNiUTYwXI9SGhvR3VkD7Mvk0xhiUZo,43
+lyceum_cli-1.0.18.dist-info/top_level.txt,sha256=CR7FEMloAXgLsHUR6ti3mWNcpgje27HRHSfq8doIils,41
+lyceum_cli-1.0.18.dist-info/RECORD,,

lyceum_cli-1.0.18.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+lyceum
+lyceum_cloud_execution_api_client

lyceum_cloud_execution_api_client/models/__init__.py

@@ -0,0 +1,105 @@
+"""Contains all the data models used in inputs/outputs"""
+
+from .abort_response import AbortResponse
+from .api_key_create import ApiKeyCreate
+from .api_key_create_response import ApiKeyCreateResponse
+from .api_key_response import ApiKeyResponse
+from .body_upload_bulk_files_api_v2_external_storage_upload_bulk_post import (
+    BodyUploadBulkFilesApiV2ExternalStorageUploadBulkPost,
+)
+from .body_upload_file_api_v1_upload_file_post import BodyUploadFileApiV1UploadFilePost
+from .body_upload_file_api_v2_external_storage_upload_post import BodyUploadFileApiV2ExternalStorageUploadPost
+from .bulk_upload_response import BulkUploadResponse
+from .bulk_upload_result import BulkUploadResult
+from .checkout_session_request import CheckoutSessionRequest
+from .checkout_session_response import CheckoutSessionResponse
+from .cloud_storage_status import CloudStorageStatus
+from .cloud_storage_status_aws_s3 import CloudStorageStatusAwsS3
+from .cloud_storage_status_azure_blob import CloudStorageStatusAzureBlob
+from .cloud_storage_status_gcp_storage import CloudStorageStatusGcpStorage
+from .code_execution import CodeExecution
+from .connect_request import ConnectRequest
+from .connect_request_credentials import ConnectRequestCredentials
+from .create_checkout_session_request import CreateCheckoutSessionRequest
+from .credits_balance import CreditsBalance
+from .delete_user_request import DeleteUserRequest
+from .docker_execution import DockerExecution
+from .docker_execution_docker_env_type_0 import DockerExecutionDockerEnvType0
+from .docker_execution_response import DockerExecutionResponse
+from .execution_response import ExecutionResponse
+from .execution_response_result_files_type_0 import ExecutionResponseResultFilesType0
+from .execution_response_result_files_type_0_additional_property import (
+    ExecutionResponseResultFilesType0AdditionalProperty,
+)
+from .execution_summary import ExecutionSummary
+from .file_info import FileInfo
+from .http_validation_error import HTTPValidationError
+from .login_request import LoginRequest
+from .login_response import LoginResponse
+from .machine_type import MachineType
+from .machine_types_response import MachineTypesResponse
+from .start_execution_api_v1_execution_start_post_response_start_execution_api_v1_execution_start_post import (
+    StartExecutionApiV1ExecutionStartPostResponseStartExecutionApiV1ExecutionStartPost,
+)
+from .start_execution_api_v2_external_compute_execution_run_post_response_start_execution_api_v2_external_compute_execution_run_post import (
+    StartExecutionApiV2ExternalComputeExecutionRunPostResponseStartExecutionApiV2ExternalComputeExecutionRunPost,
+)
+from .start_prebuilt_execution_aws_credentials import StartPrebuiltExecutionAWSCredentials
+from .start_prebuilt_execution_request import StartPrebuiltExecutionRequest
+from .start_prebuilt_execution_request_docker_run_env_type_0 import StartPrebuiltExecutionRequestDockerRunEnvType0
+from .start_prebuilt_execution_response import StartPrebuiltExecutionResponse
+from .storage_credentials import StorageCredentials
+# from .test_credentials_request import TestCredentialsRequest
+# from .test_credentials_request_credentials import TestCredentialsRequestCredentials
+from .upload_response import UploadResponse
+from .user_credits import UserCredits
+from .validation_error import ValidationError
+
+__all__ = (
+    "AbortResponse",
+    "ApiKeyCreate",
+    "ApiKeyCreateResponse",
+    "ApiKeyResponse",
+    "BodyUploadBulkFilesApiV2ExternalStorageUploadBulkPost",
+    "BodyUploadFileApiV1UploadFilePost",
+    "BodyUploadFileApiV2ExternalStorageUploadPost",
+    "BulkUploadResponse",
+    "BulkUploadResult",
+    "CheckoutSessionRequest",
+    "CheckoutSessionResponse",
+    "CloudStorageStatus",
+    "CloudStorageStatusAwsS3",
+    "CloudStorageStatusAzureBlob",
+    "CloudStorageStatusGcpStorage",
+    "CodeExecution",
+    "ConnectRequest",
+    "ConnectRequestCredentials",
+    "CreateCheckoutSessionRequest",
+    "CreditsBalance",
+    "DeleteUserRequest",
+    "DockerExecution",
+    "DockerExecutionDockerEnvType0",
+    "DockerExecutionResponse",
+    "ExecutionResponse",
+    "ExecutionResponseResultFilesType0",
+    "ExecutionResponseResultFilesType0AdditionalProperty",
+    "ExecutionSummary",
+    "FileInfo",
+    "HTTPValidationError",
+    "LoginRequest",
+    "LoginResponse",
+    "MachineType",
+    "MachineTypesResponse",
+    "StartExecutionApiV1ExecutionStartPostResponseStartExecutionApiV1ExecutionStartPost",
+    "StartExecutionApiV2ExternalComputeExecutionRunPostResponseStartExecutionApiV2ExternalComputeExecutionRunPost",
+    "StartPrebuiltExecutionAWSCredentials",
+    "StartPrebuiltExecutionRequest",
+    "StartPrebuiltExecutionRequestDockerRunEnvType0",
+    "StartPrebuiltExecutionResponse",
+    "StorageCredentials",
+    # "TestCredentialsRequest",
+    # "TestCredentialsRequestCredentials",
+    "UploadResponse",
+    "UserCredits",
+    "ValidationError",
+)

lyceum_cli-1.0.14.dist-info/RECORD

@@ -1,5 +0,0 @@
-lyceum_cli-1.0.14.dist-info/METADATA,sha256=GPKsz9WxGtuFqsRrY92c5dKWDhAvYFdzcBPE-5ew1w8,1482
-lyceum_cli-1.0.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lyceum_cli-1.0.14.dist-info/entry_points.txt,sha256=Oq-9wDkxVd6MHgNiUTYwXI9SGhvR3VkD7Mvk0xhiUZo,43
-lyceum_cli-1.0.14.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-lyceum_cli-1.0.14.dist-info/RECORD,,

lyceum_cli-1.0.14.dist-info/top_level.txt

@@ -1 +0,0 @@
-