File size: 5,077 Bytes
283b69e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
"""
Installation helper utilities for ARF Demo
"""
import subprocess
import sys
import logging
from typing import Dict, Any, List, Optional

logger = logging.getLogger(__name__)


class InstallationHelper:
    """Helper class for ARF package installation"""
    
    @staticmethod
    def install_arf_oss() -> Dict[str, Any]:
        """Install ARF OSS package"""
        try:
            logger.info("Installing ARF OSS v3.3.7...")
            result = subprocess.run(
                [sys.executable, "-m", "pip", "install", "agentic-reliability-framework==3.3.7"],
                capture_output=True,
                text=True,
                check=True
            )
            
            return {
                "success": True,
                "message": "βœ… ARF OSS v3.3.7 installed successfully",
                "output": result.stdout
            }
            
        except subprocess.CalledProcessError as e:
            return {
                "success": False,
                "message": "❌ Failed to install ARF OSS",
                "error": e.stderr
            }
        except Exception as e:
            return {
                "success": False,
                "message": f"❌ Installation error: {str(e)}",
                "error": str(e)
            }
    
    @staticmethod
    def check_installation() -> Dict[str, Any]:
        """Check ARF package installation"""
        try:
            import agentic_reliability_framework as arf_oss
            oss_version = getattr(arf_oss, '__version__', 'unknown')
            oss_installed = True
        except ImportError:
            oss_installed = False
            oss_version = None
        
        try:
            import arf_enterprise
            enterprise_version = getattr(arf_enterprise, '__version__', 'unknown')
            enterprise_installed = True
        except ImportError:
            enterprise_installed = False
            enterprise_version = None
        
        return {
            "oss_installed": oss_installed,
            "oss_version": oss_version,
            "enterprise_installed": enterprise_installed,
            "enterprise_version": enterprise_version,
            "recommendations": InstallationHelper.get_recommendations(oss_installed, enterprise_installed)
        }
    
    @staticmethod
    def get_recommendations(oss_installed: bool, enterprise_installed: bool) -> List[str]:
        """Get installation recommendations"""
        recommendations = []
        
        if not oss_installed:
            recommendations.append(
                "Install ARF OSS: `pip install agentic-reliability-framework==3.3.7`"
            )
        
        if not enterprise_installed:
            recommendations.append(
                "Install ARF Enterprise: `pip install agentic-reliability-enterprise` (requires license)"
            )
        
        return recommendations
    
    @staticmethod
    def get_installation_html() -> str:
        """Get HTML for installation status display"""
        status = InstallationHelper.check_installation()
        
        if status["oss_installed"]:
            oss_html = f"""
            <div style="padding: 10px; background: #10b981; color: white; border-radius: 8px; margin: 5px 0;">
                βœ… ARF OSS v{status['oss_version']} installed
            </div>
            """
        else:
            oss_html = """
            <div style="padding: 10px; background: #f59e0b; color: white; border-radius: 8px; margin: 5px 0;">
                ⚠️ ARF OSS not installed
            </div>
            """
        
        if status["enterprise_installed"]:
            enterprise_html = f"""
            <div style="padding: 10px; background: #8b5cf6; color: white; border-radius: 8px; margin: 5px 0;">
                πŸš€ ARF Enterprise v{status['enterprise_version']} installed
            </div>
            """
        else:
            enterprise_html = """
            <div style="padding: 10px; background: #64748b; color: white; border-radius: 8px; margin: 5px 0;">
                πŸ”’ ARF Enterprise not installed
            </div>
            """
        
        recommendations_html = ""
        if status["recommendations"]:
            rec_list = "\n".join([f"<li>{rec}</li>" for rec in status["recommendations"]])
            recommendations_html = f"""
            <div style="margin-top: 15px; padding: 10px; background: #f1f5f9; border-radius: 8px;">
                <strong>πŸ’‘ Recommendations:</strong>
                <ul style="margin: 5px 0 0 0; padding-left: 20px;">
                    {rec_list}
                </ul>
            </div>
            """
        
        return f"""
        <div style="border: 1px solid #e2e8f0; border-radius: 12px; padding: 15px; background: white;">
            <h4 style="margin: 0 0 10px 0; color: #1e293b;">πŸ“¦ ARF Installation Status</h4>
            {oss_html}
            {enterprise_html}
            {recommendations_html}
        </div>
        """


# Export singleton
installation_helper = InstallationHelper()