mirror of
https://github.com/cpacker/MemGPT.git
synced 2025-06-03 04:30:22 +00:00
97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
import json
|
|
from typing import TYPE_CHECKING, List, Optional, Union
|
|
|
|
# Avoid circular imports
|
|
if TYPE_CHECKING:
|
|
from letta.schemas.message import Message
|
|
|
|
|
|
class LettaError(Exception):
|
|
"""Base class for all Letta related errors."""
|
|
|
|
|
|
class LettaToolCreateError(LettaError):
|
|
"""Error raised when a tool cannot be created."""
|
|
|
|
default_error_message = "Error creating tool."
|
|
|
|
def __init__(self, message=None):
|
|
if message is None:
|
|
message = self.default_error_message
|
|
self.message = message
|
|
super().__init__(self.message)
|
|
|
|
|
|
class LLMError(LettaError):
|
|
pass
|
|
|
|
|
|
class LLMJSONParsingError(LettaError):
|
|
"""Exception raised for errors in the JSON parsing process."""
|
|
|
|
def __init__(self, message="Error parsing JSON generated by LLM"):
|
|
self.message = message
|
|
super().__init__(self.message)
|
|
|
|
|
|
class LocalLLMError(LettaError):
|
|
"""Generic catch-all error for local LLM problems"""
|
|
|
|
def __init__(self, message="Encountered an error while running local LLM"):
|
|
self.message = message
|
|
super().__init__(self.message)
|
|
|
|
|
|
class LocalLLMConnectionError(LettaError):
|
|
"""Error for when local LLM cannot be reached with provided IP/port"""
|
|
|
|
def __init__(self, message="Could not connect to local LLM"):
|
|
self.message = message
|
|
super().__init__(self.message)
|
|
|
|
|
|
class LettaMessageError(LettaError):
|
|
"""Base error class for handling message-related errors."""
|
|
|
|
messages: List[Union["Message", "LettaMessage"]]
|
|
default_error_message: str = "An error occurred with the message."
|
|
|
|
def __init__(self, *, messages: List[Union["Message", "LettaMessage"]], explanation: Optional[str] = None) -> None:
|
|
error_msg = self.construct_error_message(messages, self.default_error_message, explanation)
|
|
super().__init__(error_msg)
|
|
self.messages = messages
|
|
|
|
@staticmethod
|
|
def construct_error_message(messages: List[Union["Message", "LettaMessage"]], error_msg: str, explanation: Optional[str] = None) -> str:
|
|
"""Helper method to construct a clean and formatted error message."""
|
|
if explanation:
|
|
error_msg += f" (Explanation: {explanation})"
|
|
|
|
# Pretty print out message JSON
|
|
message_json = json.dumps([message.model_dump() for message in messages], indent=4)
|
|
return f"{error_msg}\n\n{message_json}"
|
|
|
|
|
|
class MissingFunctionCallError(LettaMessageError):
|
|
"""Error raised when a message is missing a function call."""
|
|
|
|
default_error_message = "The message is missing a function call."
|
|
|
|
|
|
class InvalidFunctionCallError(LettaMessageError):
|
|
"""Error raised when a message uses an invalid function call."""
|
|
|
|
default_error_message = "The message uses an invalid function call or has improper usage of a function call."
|
|
|
|
|
|
class MissingInnerMonologueError(LettaMessageError):
|
|
"""Error raised when a message is missing an inner monologue."""
|
|
|
|
default_error_message = "The message is missing an inner monologue."
|
|
|
|
|
|
class InvalidInnerMonologueError(LettaMessageError):
|
|
"""Error raised when a message has a malformed inner monologue."""
|
|
|
|
default_error_message = "The message has a malformed inner monologue."
|