mirror of
https://github.com/supermemoryai/supermemory.git
synced 2026-05-19 16:13:19 +00:00
ea9bf13d31
add openai middleware functionality fix critical type errors and linting issues update readme with middleware documentation
62 lines
No EOL
1.8 KiB
Python
62 lines
No EOL
1.8 KiB
Python
"""Custom exceptions for Supermemory OpenAI middleware."""
|
|
|
|
from typing import Optional
|
|
|
|
|
|
class SupermemoryError(Exception):
|
|
"""Base exception for all Supermemory-related errors."""
|
|
|
|
def __init__(self, message: str, original_error: Optional[Exception] = None):
|
|
super().__init__(message)
|
|
self.message = message
|
|
self.original_error = original_error
|
|
|
|
def __str__(self) -> str:
|
|
if self.original_error:
|
|
return f"{self.message}: {self.original_error}"
|
|
return self.message
|
|
|
|
|
|
class SupermemoryConfigurationError(SupermemoryError):
|
|
"""Raised when there are configuration issues (e.g., missing API key)."""
|
|
pass
|
|
|
|
|
|
class SupermemoryAPIError(SupermemoryError):
|
|
"""Raised when Supermemory API requests fail."""
|
|
|
|
def __init__(
|
|
self,
|
|
message: str,
|
|
status_code: Optional[int] = None,
|
|
response_text: Optional[str] = None,
|
|
original_error: Optional[Exception] = None,
|
|
):
|
|
super().__init__(message, original_error)
|
|
self.status_code = status_code
|
|
self.response_text = response_text
|
|
|
|
def __str__(self) -> str:
|
|
parts = [self.message]
|
|
if self.status_code:
|
|
parts.append(f"Status: {self.status_code}")
|
|
if self.response_text:
|
|
parts.append(f"Response: {self.response_text}")
|
|
if self.original_error:
|
|
parts.append(f"Cause: {self.original_error}")
|
|
return " | ".join(parts)
|
|
|
|
|
|
class SupermemoryMemoryOperationError(SupermemoryError):
|
|
"""Raised when memory operations (search, add) fail."""
|
|
pass
|
|
|
|
|
|
class SupermemoryTimeoutError(SupermemoryError):
|
|
"""Raised when operations timeout."""
|
|
pass
|
|
|
|
|
|
class SupermemoryNetworkError(SupermemoryError):
|
|
"""Raised when network operations fail."""
|
|
pass |