MCP Code Analysis Server

"""Custom exceptions for MCP Code Analysis Server.""" from typing import Any class MCPError(Exception): """Base exception for MCP server errors.""" def __init__( self, message: str, code: str | None = None, details: dict[str, Any] | None = None, ) -> None: super().__init__(message) self.message = message self.code = code or self.__class__.__name__ self.details = details or {} class ConfigurationError(MCPError): """Configuration related errors.""" class RepositoryError(MCPError): """Repository operation errors.""" class GitHubError(RepositoryError): """GitHub API related errors.""" def __init__( self, message: str, status_code: int | None = None, github_error: dict[str, Any] | None = None, ) -> None: super().__init__(message) self.status_code = status_code if github_error: self.details["github_error"] = github_error class ParserError(MCPError): """Alias for ParsingError for backward compatibility.""" class ParsingError(MCPError): """Code parsing errors.""" def __init__( self, message: str, file_path: str | None = None, line_number: int | None = None, language: str | None = None, ) -> None: super().__init__(message) if file_path: self.details["file_path"] = file_path if line_number: self.details["line_number"] = line_number if language: self.details["language"] = language class EmbeddingError(MCPError): """Embedding generation errors.""" def __init__( self, message: str = "Failed to generate embedding", code: str | None = None, details: dict[str, Any] | None = None, ) -> None: super().__init__(message, code=code, details=details) class OpenAIError(EmbeddingError): """OpenAI API related errors.""" def __init__( self, message: str, status_code: int | None = None, openai_error: dict[str, Any] | None = None, ) -> None: super().__init__(message) self.status_code = status_code if openai_error: self.details["openai_error"] = openai_error class DatabaseError(MCPError): """Database operation errors.""" class QueryError(MCPError): """Query processing errors.""" class VectorSearchError(QueryError): """Vector similarity search errors.""" class RateLimitError(MCPError): """Rate limiting errors.""" def __init__( self, message: str, retry_after: int | None = None, limit: int | None = None, remaining: int | None = None, ) -> None: super().__init__(message) if retry_after: self.details["retry_after"] = retry_after if limit: self.details["limit"] = limit if remaining is not None: self.details["remaining"] = remaining class AuthenticationError(MCPError): """Authentication related errors.""" class AuthorizationError(MCPError): """Authorization related errors.""" class ValidationError(MCPError): """Data validation errors.""" def __init__( self, message: str, field: str | None = None, value: Any | None = None, errors: list | None = None, ) -> None: super().__init__(message) if field: self.details["field"] = field if value is not None: self.details["value"] = value if errors: self.details["errors"] = errors class NotFoundError(MCPError): """Resource not found errors.""" def __init__( self, message: str, resource_type: str | None = None, resource_id: str | None = None, ) -> None: super().__init__(message) if resource_type: self.details["resource_type"] = resource_type if resource_id: self.details["resource_id"] = resource_id class TimeoutError(MCPError): # noqa: A001 """Operation timeout errors.""" def __init__( self, message: str, operation: str | None = None, timeout_seconds: float | None = None, ) -> None: super().__init__(message) if operation: self.details["operation"] = operation if timeout_seconds: self.details["timeout_seconds"] = timeout_seconds class WebhookError(MCPError): """Webhook processing errors.""" def __init__( self, message: str, event_type: str | None = None, delivery_id: str | None = None, ) -> None: super().__init__(message) if event_type: self.details["event_type"] = event_type if delivery_id: self.details["delivery_id"] = delivery_id class DomainError(MCPError): """Domain analysis related errors.""" def __init__( self, message: str, entity_type: str | None = None, entity_id: int | None = None, ) -> None: super().__init__(message) if entity_type: self.details["entity_type"] = entity_type if entity_id: self.details["entity_id"] = entity_id