alita-sdk 0.3.161__tar.gz → 0.3.163__tar.gz

{alita_sdk-0.3.161 → alita_sdk-0.3.163}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.161
+Version: 0.3.163
 Summary: SDK for building langchain agents using resources from Alita
 Author-email: Artem Rozumenko <artyom.rozumenko@gmail.com>, Mikalai Biazruchka <mikalai_biazruchka@epam.com>, Roman Mitusov <roman_mitusov@epam.com>, Ivan Krakhmaliuk <lifedjik@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.161 → alita_sdk-0.3.163}/alita_sdk/tools/__init__.py

@@ -74,6 +74,7 @@ _safe_import_tool('carrier', 'carrier', 'get_tools', 'AlitaCarrierToolkit')
 _safe_import_tool('ocr', 'ocr', 'get_tools', 'OCRToolkit')
 _safe_import_tool('pptx', 'pptx', 'get_tools', 'PPTXToolkit')
 _safe_import_tool('postman', 'postman', 'get_tools', 'PostmanToolkit')
+_safe_import_tool('zephyr_squad', 'zephyr_squad', 'get_tools', 'ZephyrSquadToolkit')
 
 # Log import summary
 available_count = len(AVAILABLE_TOOLS)

{alita_sdk-0.3.161 → alita_sdk-0.3.163}/alita_sdk/tools/carrier/api_wrapper.py

@@ -67,6 +67,10 @@ class CarrierAPIWrapper(BaseModel):
     def run_test(self, test_id: str, json_body):
         return self._client.run_test(test_id, json_body)
 
+    def run_ui_test(self, test_id: str, json_body):
+        """Run a UI test with the given test ID and JSON body."""
+        return self._client.run_ui_test(test_id, json_body)
+
     def get_engagements_list(self) -> List[Dict[str, Any]]:
         return self._client.get_engagements_list()
 
@@ -81,12 +85,16 @@ class CarrierAPIWrapper(BaseModel):
 
     def get_ui_reports_list(self) -> List[Dict[str, Any]]:
         """Get list of UI test reports from the Carrier platform."""
-        return self._client.get_ui_tests_list()
+        return self._client.get_ui_reports_list()
 
     def get_ui_tests_list(self) -> List[Dict[str, Any]]:
         """Get list of UI tests from the Carrier platform."""
         return self._client.get_ui_tests_list()
 
+    def get_locations(self) -> Dict[str, Any]:
+        """Get list of available locations/cloud settings from the Carrier platform."""
+        return self._client.get_locations()
+
     def get_ui_report_links(self, uid: str) -> list:
         """Get all unique file_names for a given UI report UID, ending with .html, without #index=*, and only unique values."""
         endpoint = f"api/v1/ui_performance/results/{self.project_id}/{uid}?sort=loop&order=asc"
@@ -120,4 +128,68 @@ class CarrierAPIWrapper(BaseModel):
             return [prefix + name for name in sorted_names]
         except Exception as e:
             logger.error(f"Failed to fetch UI report links: {e}")
-            return []
+            return []
+
+    def update_ui_test(self, test_id: str, json_body) -> Dict[str, Any]:
+        """Update UI test configuration and schedule."""
+        return self._client.update_ui_test(test_id, json_body)
+
+    def get_ui_test_details(self, test_id: str) -> Dict[str, Any]:
+        """Get detailed UI test configuration by test ID."""
+        return self._client.get_ui_test_details(test_id)
+
+    def create_ui_test(self, json_body: Dict[str, Any]) -> Dict[str, Any]:
+        """Create a new UI test."""
+        return self._client.create_ui_test(json_body)
+
+    def cancel_ui_test(self, test_id: str) -> Dict[str, Any]:
+        """Cancel a UI test by setting its status to Canceled."""
+        return self._client.cancel_ui_test(test_id)
+
+    def get_ui_report_json_files(self, uid: str) -> list:
+        """Get all JSON file names for a given UI report UID for Excel processing."""
+        endpoint = f"api/v1/ui_performance/results/{self.project_id}/{uid}?sort=loop&order=asc"
+        try:
+            response = self._client.request('get', endpoint)
+            file_names = set()
+            
+            def clean_file_name(file_name):
+                # Extract JSON files only and clean the names
+                if file_name.endswith('.json'):
+                    return file_name
+                return None
+            
+            # If the response is a dict with lists as values, flatten all file_names from all values
+            if isinstance(response, dict):
+                for value in response.values():
+                    if isinstance(value, list):
+                        for item in value:
+                            file_name = item.get("file_name")
+                            if file_name:
+                                clean_name = clean_file_name(file_name)
+                                if clean_name:
+                                    file_names.add(clean_name)
+            elif isinstance(response, list):
+                for item in response:
+                    file_name = item.get("file_name")
+                    if file_name:
+                        clean_name = clean_file_name(file_name)
+                        if clean_name:
+                            file_names.add(clean_name)
+            
+            sorted_names = sorted(file_names)
+            prefix = f"https://platform.getcarrier.io/api/v1/artifacts/artifact/default/{self.project_id}/reports/"
+            return [prefix + name for name in sorted_names]
+        except Exception as e:
+            logger.error(f"Failed to fetch UI report JSON files: {e}")
+            return []
+
+    def download_ui_report_json(self, bucket: str, file_name: str) -> str:
+        """Download UI report JSON file content."""
+        endpoint = f"api/v1/artifacts/artifact/default/{self.project_id}/{bucket}/{file_name}"
+        try:
+            response = self._client.request('get', endpoint)
+            return response
+        except Exception as e:
+            logger.error(f"Failed to download UI report JSON: {e}")
+            return None

alita_sdk-0.3.163/alita_sdk/tools/carrier/cancel_ui_test_tool.py

@@ -0,0 +1,178 @@
+import logging
+import re
+import traceback
+from typing import Type
+from langchain_core.tools import BaseTool, ToolException
+from pydantic.fields import Field
+from pydantic import create_model, BaseModel
+from .api_wrapper import CarrierAPIWrapper
+
+logger = logging.getLogger(__name__)
+
+
+class CancelUITestTool(BaseTool):
+    api_wrapper: CarrierAPIWrapper = Field(..., description="Carrier API Wrapper instance")
+    name: str = "cancel_ui_test"
+    description: str = "Cancel a UI test or show available tests to cancel in the Carrier platform."
+    args_schema: Type[BaseModel] = create_model(
+        "CancelUITestInput",
+        **{
+            "message": (str, Field(description="User input message (e.g., 'Cancel UI test' or 'Cancel UI test 12345')")),
+        }
+    )
+
+    def _run(self, message: str):
+        try:
+            # Parse the message to extract test ID if provided
+            test_id = self._extract_test_id(message)
+            
+            if test_id:
+                # User provided a specific test ID to cancel
+                return self._cancel_specific_test(test_id)
+            else:
+                # User didn't provide ID, show available tests to cancel
+                return self._show_cancelable_tests()
+                
+        except Exception as e:
+            logger.error(f"Error in cancel UI test: {e}")
+            raise ToolException(f"Failed to process cancel UI test request: {str(e)}")
+
+    def _extract_test_id(self, message: str) -> str:
+        """Extract test ID from user message if present."""
+        # Look for patterns like "Cancel UI test 12345" or "cancel ui test 12345"
+        match = re.search(r'cancel\s+ui\s+test\s+(\d+)', message.lower())
+        if match:
+            return match.group(1)
+        return ""
+
+    def _show_cancelable_tests(self) -> str:
+        """Show list of tests that can be canceled."""
+        try:
+            # Get all UI reports/tests
+            reports = self.api_wrapper.get_ui_reports_list()
+            
+            if not reports:
+                return "❌ **No UI tests found.**"
+            
+            # Filter tests that can be canceled (not in final states)
+            final_states = {"Canceled", "Finished", "Failed"}
+            cancelable_tests = []
+            
+            for report in reports:
+                test_status = report.get("test_status", {})
+                status = test_status.get("status", "Unknown")
+                
+                # Check if status is not in final states
+                if status not in final_states:
+                    cancelable_tests.append({
+                        "id": report.get("id"),
+                        "name": report.get("name", "Unknown"),
+                        "status": status,
+                        "percentage": test_status.get("percentage", 0),
+                        "description": test_status.get("description", "")
+                    })
+            
+            if not cancelable_tests:
+                return """# ℹ️ No Tests Available for Cancellation
+
+All UI tests are already in final states (Canceled, Finished, or Failed).
+
+## 🔍 To cancel a specific test:
+Use the command: `Cancel UI test <test_id>`
+
+Example: `Cancel UI test 12345`"""
+            
+            # Build the response message
+            response = """# 🚫 UI Tests Available for Cancellation
+
+The following tests are currently running and can be canceled:
+
+## 📋 Active Tests:
+"""
+            
+            for test in cancelable_tests:
+                response += f"""
+### 🔸 Test ID: `{test['id']}`
+- **Name:** `{test['name']}`
+- **Status:** `{test['status']}`
+- **Progress:** {test['percentage']}%
+- **Description:** {test['description']}
+"""
+            
+            response += """
+## 🚫 To cancel a specific test:
+Use the command: `Cancel UI test <test_id>`
+
+Example: `Cancel UI test 12345`"""
+            
+            return response
+            
+        except Exception as e:
+            logger.error(f"Error fetching cancelable tests: {e}")
+            return f"❌ **Error fetching tests:** {str(e)}"
+
+    def _cancel_specific_test(self, test_id: str) -> str:
+        """Cancel a specific UI test by ID."""
+        try:
+            # First, get the current status of the test
+            reports = self.api_wrapper.get_ui_reports_list()
+            target_test = None
+            
+            for report in reports:
+                if str(report.get("id")) == test_id:
+                    target_test = report
+                    break
+            
+            if not target_test:
+                return f"❌ **Test with ID `{test_id}` not found.**"
+            
+            # Check if test can be canceled
+            test_status = target_test.get("test_status", {})
+            current_status = test_status.get("status", "Unknown")
+            final_states = {"Canceled", "Finished", "Failed"}
+            
+            if current_status in final_states:
+                return f"""# ❌ Cannot Cancel Test
+
+## Test Information:
+- **Test ID:** `{test_id}`
+- **Name:** `{target_test.get('name', 'Unknown')}`
+- **Current Status:** `{current_status}`
+
+## 🚫 Reason:
+This test cannot be canceled because it is already in a final state (`{current_status}`).
+
+Only tests with status **not** in `Canceled`, `Finished`, or `Failed` can be canceled."""
+            
+            # Attempt to cancel the test
+            try:
+                cancel_response = self.api_wrapper.cancel_ui_test(test_id)
+                
+                return f"""# ✅ UI Test Canceled Successfully!
+
+## Test Information:
+- **Test ID:** `{test_id}`
+- **Name:** `{target_test.get('name', 'Unknown')}`
+- **Previous Status:** `{current_status}`
+- **New Status:** `Canceled`
+
+## 🎯 Result:
+The test has been successfully canceled and will stop executing."""
+                
+            except Exception as cancel_error:
+                logger.error(f"Error canceling test {test_id}: {cancel_error}")
+                return f"""# ❌ Failed to Cancel Test
+
+## Test Information:
+- **Test ID:** `{test_id}`
+- **Name:** `{target_test.get('name', 'Unknown')}`
+- **Current Status:** `{current_status}`
+
+## 🚫 Error:
+{str(cancel_error)}
+
+Please check the test ID and try again."""
+                
+        except Exception as e:
+            logger.error(f"Error canceling specific test {test_id}: {e}")
+            return f"❌ **Error processing cancellation for test `{test_id}`: {str(e)}**"

{alita_sdk-0.3.161 → alita_sdk-0.3.163}/alita_sdk/tools/carrier/carrier_sdk.py

@@ -85,6 +85,11 @@ class CarrierClient(BaseModel):
         endpoint = f"api/v1/backend_performance/test/{self.credentials.project_id}/{test_id}"
         return self.request('post', endpoint, json=json_body).get("result_id", "")
 
+    def run_ui_test(self, test_id: str, json_body):
+        """Run a UI test with the given test ID and JSON body."""
+        endpoint = f"api/v1/ui_performance/test/{self.credentials.project_id}/{test_id}"
+        return self.request('post', endpoint, json=json_body).get("result_id", "")
+
     def get_engagements_list(self) -> List[Dict[str, Any]]:
         endpoint = f"api/v1/engagements/engagements/{self.credentials.project_id}"
         return self.request('get', endpoint).get("items", [])
@@ -204,6 +209,69 @@ class CarrierClient(BaseModel):
         return self.request('get', endpoint).get("rows", [])
 
     def get_ui_reports_list(self) -> List[Dict[str, Any]]:
-            """Get list of UI test reports from the Carrier platform."""
-            endpoint = f"api/v1/ui_performance/reports/{self.credentials.project_id}"
-            return self.request('get', endpoint).get("rows", [])
+        """Get list of UI test reports from the Carrier platform."""
+        endpoint = f"api/v1/ui_performance/reports/{self.credentials.project_id}"
+        return self.request('get', endpoint).get("rows", [])
+
+    def get_locations(self) -> Dict[str, Any]:
+        """Get list of available locations/cloud settings from the Carrier platform."""
+        endpoint = f"api/v1/shared/locations/{self.credentials.project_id}"
+        return self.request('get', endpoint)
+
+    def update_ui_test(self, test_id: str, json_body) -> Dict[str, Any]:
+        """Update UI test configuration and schedule."""
+        endpoint = f"api/v1/ui_performance/test/{self.credentials.project_id}/{test_id}"
+        return self.request('put', endpoint, json=json_body)
+
+    def get_ui_test_details(self, test_id: str) -> Dict[str, Any]:
+        """Get detailed UI test configuration by test ID."""
+        endpoint = f"api/v1/ui_performance/test/{self.credentials.project_id}/{test_id}"
+        return self.request('get', endpoint)
+
+    def create_ui_test(self, json_body: Dict[str, Any]) -> Dict[str, Any]:
+        """Create a new UI test."""
+        endpoint = f"api/v1/ui_performance/tests/{self.credentials.project_id}"
+        
+        # Print full JSON POST body for debugging
+        print("=" * 60)
+        print("DEBUG: Full JSON POST body for create_ui_test:")
+        print("=" * 60)
+        print(json.dumps(json_body, indent=2))
+        print("=" * 60)
+        
+        # Use multipart/form-data with data field containing the JSON body
+        form_data = {'data': json.dumps(json_body)}
+        
+        # Temporarily remove Content-Type header to let requests set it for multipart
+        original_headers = self.session.headers.copy()
+        if 'Content-Type' in self.session.headers:
+            del self.session.headers['Content-Type']
+        
+        try:
+            full_url = f"{self.credentials.url.rstrip('/')}/{endpoint.lstrip('/')}"
+            response = self.session.post(full_url, data=form_data)
+            response.raise_for_status()
+            return response.json()
+        except requests.HTTPError as http_err:
+            logger.error(f"HTTP {response.status_code} error: {response.text[:500]}")
+            raise CarrierAPIError(f"Request to {full_url} failed with status {response.status_code}")
+        except json.JSONDecodeError:
+            logger.error(f"Response was not valid JSON. Body:\n{response.text[:500]}")
+            raise CarrierAPIError("Server returned non-JSON response")
+        finally:
+            # Restore original headers
+            self.session.headers.update(original_headers)
+
+    def cancel_ui_test(self, test_id: str) -> Dict[str, Any]:
+        """Cancel a UI test by setting its status to Canceled."""
+        endpoint = f"api/v1/ui_performance/report_status/{self.credentials.project_id}/{test_id}"
+        
+        cancel_body = {
+            "test_status": {
+                "status": "Canceled",
+                "percentage": 100,
+                "description": "Test was canceled"
+            }
+        }
+        
+        return self.request('put', endpoint, json=cancel_body)