alita-sdk 0.3.257__py3-none-any.whl → 0.3.562__py3-none-any.whl

alita_sdk/configurations/__init__.py

@@ -41,8 +41,19 @@ _safe_import_configuration('bigquery', 'bigquery', 'BigQueryConfiguration')
 _safe_import_configuration('xray', 'xray', 'XrayConfiguration')
 _safe_import_configuration('zephyr', 'zephyr', 'ZephyrConfiguration')
 _safe_import_configuration('zephyr_enterprise', 'zephyr_enterprise', 'ZephyrEnterpriseConfiguration')
+_safe_import_configuration('zephyr_essential', 'zephyr_essential', 'ZephyrEssentialConfiguration')
 _safe_import_configuration('figma', 'figma', 'FigmaConfiguration')
 _safe_import_configuration('rally', 'rally', 'RallyConfiguration')
+_safe_import_configuration('sonar', 'sonar', 'SonarConfiguration')
+_safe_import_configuration('sql', 'sql', 'SqlConfiguration')
+_safe_import_configuration('google_places', 'google_places', 'GooglePlacesConfiguration')
+_safe_import_configuration('salesforce', 'salesforce', 'SalesforceConfiguration')
+_safe_import_configuration('browser', 'browser', 'BrowserConfiguration')
+_safe_import_configuration('sharepoint', 'sharepoint', 'SharepointConfiguration')
+_safe_import_configuration('carrier', 'carrier', 'CarrierConfiguration')
+_safe_import_configuration('report_portal', 'report_portal', 'ReportPortalConfiguration')
+_safe_import_configuration('testio', 'testio', 'TestIOConfiguration')
+_safe_import_configuration('openapi', 'openapi', 'OpenApiConfiguration')
 
 # Log import summary
 available_count = len(AVAILABLE_CONFIGURATIONS)

alita_sdk/configurations/ado.py

@@ -1,5 +1,8 @@
+import re
 from typing import Optional
+from urllib.parse import quote
 
+import requests
 from pydantic import BaseModel, ConfigDict, Field, SecretStr
 
 
@@ -10,7 +13,8 @@ class AdoConfiguration(BaseModel):
                 "label": "Ado",
                 "icon_url": None,
                 "section": "credentials",
-                "type": "ado"
+                "type": "ado",
+                "categories": ["project management"],
             }
         }
     )
@@ -18,6 +22,147 @@ class AdoConfiguration(BaseModel):
     project: str = Field(description="ADO project")
     token: Optional[SecretStr] = Field(description="ADO Token")
 
+    @staticmethod
+    def check_connection(settings: dict) -> str | None:
+        """
+        Test the connection to Azure DevOps API.
+
+        Args:
+            settings: Dictionary containing 'organization_url', 'project', and optionally 'token'
+
+        Returns:
+            None if connection is successful, error message string otherwise
+        """
+        organization_url = settings.get("organization_url")
+        if organization_url is None or organization_url == "":
+            if organization_url == "":
+                return "Organization URL cannot be empty"
+            return "Organization URL is required"
+
+        # Validate organization URL format
+        if not isinstance(organization_url, str):
+            return "Organization URL must be a string"
+
+        organization_url = organization_url.strip()
+        if not organization_url:
+            return "Organization URL cannot be empty"
+
+        if not organization_url.startswith(("http://", "https://")):
+            return "Organization URL must start with http:// or https://"
+
+        # Remove trailing slash for consistency
+        organization_url = organization_url.rstrip("/")
+
+        project = settings.get("project")
+        if project is None or project == "":
+            if project == "":
+                return "Project cannot be empty"
+            return "Project is required"
+
+        # Validate project format
+        if not isinstance(project, str):
+            return "Project must be a string"
+
+        project = project.strip()
+        if not project:
+            return "Project cannot be empty"
+
+        token = settings.get("token")
+
+        # Extract secret value if it's a SecretStr
+        if token is not None and hasattr(token, "get_secret_value"):
+            token = token.get_secret_value()
+
+        # Validate token if provided
+        if token is not None and (not token or not token.strip()):
+            return "Token cannot be empty if provided"
+
+        # NOTE on verification strategy:
+        # - Project endpoints can work anonymously for public projects.
+        #   That makes them a weak signal for detecting a bad/expired token.
+        # - If a token is provided, first validate it against a profile endpoint
+        #   that requires authentication, then check project access.
+
+        # Strictly require a canonical organization URL so we can build reliable API URLs.
+        # Supported formats:
+        # - https://dev.azure.com/<org>
+        # - https://<org>.visualstudio.com
+        org_name: str | None = None
+        org_url_kind: str | None = None  # 'dev.azure.com' | '*.visualstudio.com'
+        m = re.match(r"^https?://dev\.azure\.com/(?P<org>[^/]+)$", organization_url, flags=re.IGNORECASE)
+        if m:
+            org_name = m.group('org')
+            org_url_kind = 'dev.azure.com'
+        else:
+            m = re.match(r"^https?://(?P<org>[^/.]+)\.visualstudio\.com$", organization_url, flags=re.IGNORECASE)
+            if m:
+                org_name = m.group('org')
+                org_url_kind = '*.visualstudio.com'
+
+        if org_name is None:
+            return (
+                "Organization URL format is invalid. Use 'https://dev.azure.com/<org>' "
+                "(recommended) or 'https://<org>.visualstudio.com'."
+            )
+
+        project_encoded = quote(project, safe="")
+        project_url = f"{organization_url}/_apis/projects/{project_encoded}?api-version=7.0"
+        # Auth-required endpoint to validate PAT (works regardless of project visibility)
+        if org_url_kind == 'dev.azure.com':
+            profile_url = f"https://vssps.dev.azure.com/{org_name}/_apis/profile/profiles/me?api-version=7.1-preview.3"
+        else:
+            # For legacy org URLs, use the matching vssps host
+            profile_url = f"https://{org_name}.vssps.visualstudio.com/_apis/profile/profiles/me?api-version=7.1-preview.3"
+
+        try:
+            headers = {}
+            if token:
+                # Use Basic Auth with PAT token (username can be empty)
+                from requests.auth import HTTPBasicAuth
+                auth = HTTPBasicAuth("", token)
+
+                # 1) Validate token first (strong signal)
+                profile_resp = requests.get(profile_url, auth=auth, timeout=10)
+                if profile_resp.status_code == 200:
+                    pass
+                elif profile_resp.status_code == 401:
+                    return "Invalid or expired token (PAT). Please generate a new token and try again."
+                elif profile_resp.status_code == 403:
+                    return "Token is valid but lacks permission to access profile. Check PAT scopes/permissions."
+                elif profile_resp.status_code == 404:
+                    return "Organization not found. Verify the Organization URL."
+                else:
+                    return f"Token validation failed (HTTP {profile_resp.status_code})."
+
+                # 2) Validate project access
+                response = requests.get(project_url, auth=auth, timeout=10)
+            else:
+                # Try without authentication (works for public projects)
+                response = requests.get(project_url, headers=headers, timeout=10)
+
+            if response.status_code == 200:
+                return None  # Connection successful
+            elif response.status_code == 401:
+                if token:
+                    return "Not authorized. Token may be invalid for this organization or expired."
+                else:
+                    return "Authentication required - project may be private"
+            elif response.status_code == 403:
+                return "Access forbidden - token may lack required permissions for this project"
+            elif response.status_code == 404:
+                return f"Project '{project}' not found or not accessible. Check project name and organization URL."
+            else:
+                return f"Connection failed (HTTP {response.status_code})."
+
+        except requests.exceptions.Timeout:
+            return "Connection timeout - Azure DevOps did not respond within 10 seconds"
+        except requests.exceptions.ConnectionError:
+            return "Connection error - unable to reach Azure DevOps. Check the Organization URL and your network."
+        except requests.exceptions.RequestException as e:
+            return f"Request failed: {str(e)}"
+        except Exception:
+            return "Unexpected error during Azure DevOps connection check"
+
 
 class AdoReposConfiguration(BaseModel):
     model_config = ConfigDict(
@@ -26,7 +171,8 @@ class AdoReposConfiguration(BaseModel):
                 "label": "ADO repos",
                 "icon_url": "ado-repos-icon.svg",
                 "section": "credentials",
-                "type": "ado_repos"
+                "type": "ado_repos",
+                "categories": ["code repositories"],
             }
         }
     )

alita_sdk/configurations/azure_search.py

@@ -12,7 +12,7 @@ class AzureSearchConfiguration(BaseModel):
                 "hidden": True,
                 "section": "credentials",
                 "type": "azure_search",
-                "categories": ["search"],
+                "categories": ["other"],
                 "extra_categories": ["azure", "cognitive search", "vector database", "knowledge base"],
             }
         }

alita_sdk/configurations/bigquery.py

@@ -12,7 +12,7 @@ class BigQueryConfiguration(BaseModel):
                 "hidden": True,
                 "section": "credentials",
                 "type": "bigquery",
-                "categories": ["database"],
+                "categories": ["other"],
                 "extra_categories": ["google", "gcp", "data warehouse", "analytics"],
             }
         }

alita_sdk/configurations/bitbucket.py

@@ -1,5 +1,3 @@
-from typing import Optional
-
 from pydantic import BaseModel, ConfigDict, Field, SecretStr
 
 
@@ -30,3 +28,97 @@ class BitbucketConfiguration(BaseModel):
     url: str = Field(description="Bitbucket URL")
     username: str = Field(description="Bitbucket Username")
     password: SecretStr = Field(description="Bitbucket Password/App Password")
+
+    @staticmethod
+    def check_connection(settings: dict) -> str | None:
+        """
+        Check the connection to Bitbucket.
+        
+        Args:
+            settings: Dictionary containing Bitbucket configuration
+                - url: Bitbucket instance URL (required)
+                - username: Bitbucket username (required)
+                - password: Password or App Password (required)
+        
+        Returns:
+            None if connection successful, error message string if failed
+        """
+        import requests
+        from requests.auth import HTTPBasicAuth
+        
+        # Validate url
+        url = settings.get("url", "").strip()
+        if not url:
+            return "Bitbucket URL is required"
+        
+        # Normalize URL - remove trailing slashes
+        url = url.rstrip("/")
+        
+        # Basic URL validation
+        if not url.startswith(("http://", "https://")):
+            return "Bitbucket URL must start with http:// or https://"
+        
+        # Validate username
+        username = settings.get("username", "").strip()
+        if not username:
+            return "Bitbucket username is required"
+        
+        # Validate password
+        password = settings.get("password")
+        if not password:
+            return "Bitbucket password is required"
+        
+        # Extract password value if it's a SecretStr
+        password_value = password.get_secret_value() if hasattr(password, 'get_secret_value') else password
+        
+        if not password_value or not str(password_value).strip():
+            return "Bitbucket password cannot be empty"
+        
+        # Detect if this is Bitbucket Cloud or Server/Data Center
+        is_cloud = "bitbucket.org" in url.lower() or "api.bitbucket.org" in url.lower()
+        is_correct_bitbucket_domain = "bitbucket" in url.lower()
+        
+        if is_cloud:
+            # Bitbucket Cloud: Use API v2.0
+            # Endpoint: /2.0/user - returns current authenticated user
+            test_url = f"{url}/2.0/user"
+        else:
+            # Bitbucket Server/Data Center: Use API v1.0
+            # Endpoint: /rest/api/1.0/users/{username}
+            test_url = f"{url}/rest/api/1.0/users/{username}"
+        
+        try:
+            response = requests.get(
+                test_url,
+                auth=HTTPBasicAuth(username, str(password_value).strip()),
+                timeout=10
+            )
+            
+            # Check response status
+            if response.status_code == 200:
+                # Successfully connected and authenticated
+                return None
+            elif response.status_code == 401:
+                return "Authentication failed: invalid username or password"
+            elif response.status_code == 403:
+                return "Access forbidden: check user permissions"
+            elif response.status_code == 404:
+                if not is_correct_bitbucket_domain:
+                    return f"Url you provided is incorrect. Please provide correct server or cloud bitbucket url."
+                if is_cloud:
+                    return "Bitbucket API endpoint not found: please provide the correct bitbucket cloud URL"
+                else:
+                    return "Bitbucket API endpoint not found: please provide the correct bitbucket server URL"
+            else:
+                return f"Bitbucket API returned status code {response.status_code}"
+
+        except requests.exceptions.SSLError as e:
+            return f"SSL certificate verification failed: {str(e)}"
+        except requests.exceptions.ConnectionError:
+            return f"Cannot connect to Bitbucket at {url if not is_cloud else 'api.bitbucket.org'}: connection refused"
+        except requests.exceptions.Timeout:
+            return f"Connection to Bitbucket at {url if not is_cloud else 'api.bitbucket.org'} timed out"
+        except requests.exceptions.RequestException as e:
+            return f"Error connecting to Bitbucket: {str(e)}"
+        except Exception as e:
+            return f"Unexpected error: {str(e)}"

alita_sdk/configurations/browser.py

@@ -0,0 +1,18 @@
+from pydantic import BaseModel, ConfigDict, Field, SecretStr
+
+
+class BrowserConfiguration(BaseModel):
+    model_config = ConfigDict(
+        json_schema_extra={
+            "metadata": {
+                "label": "Browser",
+                "icon_url": "browser.svg",
+                "section": "credentials",
+                "type": "browser",
+                "categories": ["testing"],
+                "extra_categories": ["browser", "google", "search", "web"],
+            }
+        }
+    )
+    google_cse_id: str = Field(description="Google CSE id", default=None)
+    google_api_key: SecretStr = Field(description="Google API key", json_schema_extra={'secret': True})

alita_sdk/configurations/carrier.py

@@ -0,0 +1,19 @@
+from pydantic import BaseModel, ConfigDict, Field, SecretStr
+
+
+class CarrierConfiguration(BaseModel):
+    model_config = ConfigDict(
+        json_schema_extra={
+            "metadata": {
+                "label": "Carrier",
+                "icon_url": "carrier.svg",
+                "section": "credentials",
+                "type": "carrier",
+                "categories": ["testing"],
+                "extra_categories": ["carrier", "security", "testing", "vulnerability"],
+            }
+        }
+    )
+    organization: str = Field(description="Carrier Organization")
+    url: str = Field(description="Carrier URL")
+    private_token: SecretStr = Field(description="Carrier Private Token")

alita_sdk/configurations/confluence.py

@@ -1,5 +1,7 @@
 from typing import Optional
+from urllib.parse import urlparse, urlunparse
 
+from atlassian import Confluence
 from pydantic import BaseModel, ConfigDict, Field, SecretStr
 
 
@@ -34,4 +36,131 @@ class ConfluenceConfiguration(BaseModel):
     base_url: str = Field(description="Confluence URL")
     username: Optional[str] = Field(description="Confluence Username", default=None)
     api_key: Optional[SecretStr] = Field(description="Confluence API Key", default=None)
-    token: Optional[SecretStr] = Field(description="Confluence Token", default=None)
+    token: Optional[SecretStr] = Field(description="Confluence Token", default=None)
+
+    @staticmethod
+    def check_connection(settings: dict) -> str | None:
+        """
+        Check the connection to Confluence.
+        
+        Args:
+            settings: Dictionary containing Confluence configuration
+                - base_url: Confluence instance URL (required)
+                - username: Username for Basic Auth (optional)
+                - api_key: API key/password for Basic Auth (optional)
+                - token: Bearer token for authentication (optional)
+        
+        Returns:
+            None if connection successful, error message string if failed
+        """
+        import requests
+        from requests.auth import HTTPBasicAuth
+        
+        # Validate base_url
+        base_url_input = settings.get("base_url", "")
+        base_url = base_url_input.strip() if isinstance(base_url_input, str) else ""
+        if not base_url:
+            return "Confluence URL is required"
+        
+        # Basic URL validation
+        if not base_url.startswith(("http://", "https://")):
+            return "Confluence URL must start with http:// or https://"
+
+        # Normalize URL - remove trailing slashes
+        base_url = base_url.rstrip("/")
+
+        # Build candidate base URLs.
+        # Confluence Cloud REST API is typically under /wiki. Users often paste
+        # https://<site>.atlassian.net and shouldn't be forced to know about /wiki.
+        parsed = urlparse(base_url)
+        host = (parsed.hostname or "").lower()
+        path = parsed.path or ""
+
+        def with_wiki_path(url: str) -> str:
+            p = urlparse(url)
+            # Keep existing path if it already starts with /wiki
+            if (p.path or "").startswith("/wiki"):
+                return url
+            # Append /wiki, preserving any existing path (rare but safe)
+            new_path = (p.path or "") + "/wiki"
+            return urlunparse(p._replace(path=new_path.rstrip("/")))
+
+        candidate_base_urls: list[str] = []
+        if host.endswith(".atlassian.net"):
+            # For Atlassian Cloud, prefer the /wiki variant first
+            candidate_base_urls.append(with_wiki_path(base_url))
+        candidate_base_urls.append(base_url)
+        # De-duplicate while preserving order
+        candidate_base_urls = list(dict.fromkeys(candidate_base_urls))
+        
+        # Check authentication credentials
+        username = settings.get("username")
+        api_key = settings.get("api_key")
+        token = settings.get("token")
+
+        api_key_value = api_key.get_secret_value() if hasattr(api_key, 'get_secret_value') else api_key
+        token_value = token.get_secret_value() if hasattr(token, 'get_secret_value') else token
+
+        # Validate authentication - at least one method must be provided
+        has_basic_auth = bool(username and api_key_value and str(api_key_value).strip())
+        has_token = bool(token_value and str(token_value).strip())
+        
+        # Determine authentication method
+        auth_headers = {}
+        auth = None
+        
+        if has_token:
+            # Bearer token authentication
+            auth_headers["Authorization"] = f"Bearer {token_value}"
+        elif has_basic_auth:
+            # Basic authentication
+            auth = HTTPBasicAuth(username, api_key_value)
+        else:
+            return "Authentication required: provide either token or both username and api_key"
+        
+        try:
+            # Test connection using /rest/api/user/current endpoint
+            # This endpoint returns current user info and validates authentication
+            last_status = None
+            for candidate_base in candidate_base_urls:
+                test_url = f"{candidate_base}/rest/api/user/current"
+                response = requests.get(
+                    test_url,
+                    auth=auth,
+                    headers=auth_headers,
+                    timeout=10
+                )
+                last_status = response.status_code
+
+                if response.status_code == 200:
+                    return None
+
+                # If we get 404 on the first candidate, try the next one
+                if response.status_code == 404:
+                    continue
+
+                if response.status_code == 401:
+                    return "Invalid credentials (401) - check token or username/api_key"
+                if response.status_code == 403:
+                    return "Access forbidden (403) - credentials lack Confluence permissions"
+                if response.status_code == 429:
+                    return "Rate limited (429) - please try again later"
+                if 500 <= response.status_code <= 599:
+                    return f"Confluence service error (HTTP {response.status_code})"
+                return f"Confluence request failed (HTTP {response.status_code})"
+
+            # All candidates returned 404
+            return "Confluence API endpoint not found (404) - verify the Confluence URL"
+
+        except requests.exceptions.SSLError as e:
+            if 'Hostname mismatch' in str(e):
+                return "SSL error - hostname mismatch. Verify the Confluence URL"
+            return "SSL error - certificate verification failed"
+        except requests.exceptions.ConnectionError:
+            return "Connection error - unable to reach Confluence. Check URL and network."
+        except requests.exceptions.Timeout:
+            return "Connection timeout - Confluence did not respond within 10 seconds. Check URL and network."
+        except requests.exceptions.RequestException as e:
+            return f"Request failed: {str(e)}"
+        except Exception:
+            return "Unexpected error during Confluence connection check"

alita_sdk/configurations/delta_lake.py

@@ -12,7 +12,7 @@ class DeltaLakeConfiguration(BaseModel):
                 "hidden": True,
                 "section": "credentials",
                 "type": "delta_lake",
-                "categories": ["database"],
+                "categories": ["other"],
                 "extra_categories": ["aws", "data lake", "analytics", "storage"],
             }
         }

alita_sdk/configurations/figma.py

@@ -1,8 +1,30 @@
+from json import JSONDecodeError
 from typing import Optional
 
+import requests
 from pydantic import BaseModel, ConfigDict, Field, SecretStr
 
 
+def _parse_error_response(response: requests.Response) -> Optional[str]:
+    """
+    Parse error response from Figma API to extract detailed error message.
+
+    Args:
+        response: Response object from requests
+
+    Returns:
+        Detailed error message if found, None otherwise
+    """
+    try:
+        json_response = response.json()
+        error = json_response.get("err") or json_response.get("error")
+        if error and 'Invalid token' in str(error):
+            return "Invalid token. Please verify the Figma token and try again."
+    except (JSONDecodeError, KeyError, AttributeError):
+        pass
+    return None
+
+
 class FigmaConfiguration(BaseModel):
     model_config = ConfigDict(
         json_schema_extra={
@@ -16,10 +38,6 @@ class FigmaConfiguration(BaseModel):
                             {
                                 "name": "Token",
                                 "fields": ["token"]
-                            },
-                            {
-                                "name": "Oath2",
-                                "fields": ["oauth2"]
                             }
                         ]
                     }
@@ -32,4 +50,57 @@ class FigmaConfiguration(BaseModel):
         }
     )
     token: Optional[SecretStr] = Field(description="Figma Token", json_schema_extra={"secret": True}, default=None)
-    oauth2: Optional[SecretStr] = Field(description="OAuth2 Token", json_schema_extra={"secret": True}, default=None)
+
+    @staticmethod
+    def check_connection(settings: dict) -> str | None:
+        """
+        Test the connection to Figma API.
+
+        Args:
+            settings: Dictionary containing 'token' (required)
+
+        Returns:
+            None if connection is successful, error message string otherwise
+        """
+        token = settings.get("token")
+        if token is None:
+            return "Token is required"
+
+        # Extract secret value if it's a SecretStr
+        if hasattr(token, "get_secret_value"):
+            token = token.get_secret_value()
+
+        # Validate token is not empty
+        if not token or not token.strip():
+            return "Token cannot be empty"
+
+        # Figma API endpoint
+        base_url = "https://api.figma.com"
+        endpoint = f"{base_url}/v1/me"
+
+        try:
+            response = requests.get(
+                endpoint,
+                headers={"X-Figma-Token": token},
+                timeout=10,
+            )
+
+            if response.status_code == 200:
+                return None  # Connection successful
+            elif response.status_code == 401:
+                detailed_error = _parse_error_response(response)
+                return detailed_error if detailed_error else "Invalid token"
+            elif response.status_code == 403:
+                detailed_error = _parse_error_response(response)
+                return detailed_error if detailed_error else "Access forbidden - token may lack required permissions"
+            else:
+                return f"Connection failed with status {response.status_code}"
+
+        except requests.exceptions.Timeout:
+            return "Connection timeout - Figma API is not responding"
+        except requests.exceptions.ConnectionError:
+            return "Connection error - unable to reach Figma API"
+        except requests.exceptions.RequestException as e:
+            return f"Request failed: {str(e)}"
+        except Exception as e:
+            return f"Unexpected error: {str(e)}"