mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 00:28:31 +00:00
c3ad5887ef
- Update to Python 3.10+ typing across LLM, callbacks, storage, and errors - Complete typing updates for crew_chat and hitl - Add stop attr to mock LLM, suppress test warnings - Add type-ignore for aisuite import
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
"""Error message definitions for CrewAI database operations.
|
|
|
|
This module provides standardized error classes and message templates
|
|
for database operations and agent repository handling.
|
|
"""
|
|
|
|
from typing import Final
|
|
|
|
|
|
class DatabaseOperationError(Exception):
|
|
"""Base exception class for database operation errors."""
|
|
|
|
def __init__(self, message: str, original_error: Exception | None = None) -> None:
|
|
"""Initialize the database operation error.
|
|
|
|
Args:
|
|
message: The error message to display
|
|
original_error: The original exception that caused this error, if any
|
|
"""
|
|
super().__init__(message)
|
|
self.original_error = original_error
|
|
|
|
|
|
class DatabaseError:
|
|
"""Standardized error message templates for database operations.
|
|
|
|
Provides consistent error message formatting for various database
|
|
operation failures.
|
|
"""
|
|
|
|
INIT_ERROR: Final[str] = "Database initialization error: {}"
|
|
SAVE_ERROR: Final[str] = "Error saving task outputs: {}"
|
|
UPDATE_ERROR: Final[str] = "Error updating task outputs: {}"
|
|
LOAD_ERROR: Final[str] = "Error loading task outputs: {}"
|
|
DELETE_ERROR: Final[str] = "Error deleting task outputs: {}"
|
|
|
|
@classmethod
|
|
def format_error(cls, template: str, error: Exception) -> str:
|
|
"""Format an error message with the given template and error.
|
|
|
|
Args:
|
|
template: The error message template to use
|
|
error: The exception to format into the template
|
|
|
|
Returns:
|
|
The formatted error message
|
|
"""
|
|
return template.format(str(error))
|
|
|
|
|
|
class AgentRepositoryError(Exception):
|
|
"""Exception raised when an agent repository is not found."""
|