jl-ecms-client 0.2.8__py3-none-any.whl → 0.2.22__py3-none-any.whl

{jl_ecms_client-0.2.8.dist-info → jl_ecms_client-0.2.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jl-ecms-client
-Version: 0.2.8
+Version: 0.2.22
 Summary: Intuit ECMS Client - Lightweight Python client for Intuit ECMS server
 Home-page: https://github.com/Mirix-AI/MIRIX
 Author: Mirix AI
@@ -31,8 +31,13 @@ Requires-Dist: pydantic>=2.0.0
 Requires-Dist: pydantic-settings>=2.0.0
 Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: httpx>=0.25.0
+Requires-Dist: jinja2>=3.1.0
+Requires-Dist: demjson3>=3.0.0
+Requires-Dist: json-repair>=0.25.0
 Requires-Dist: rich<14.0.0,>=13.7.1
 Requires-Dist: pytz>=2024.1
+Requires-Dist: docstring-parser>=0.15
+Requires-Dist: pyhumps>=3.8.0
 Provides-Extra: dev
 Requires-Dist: pytest>=6.0.0; extra == "dev"
 Requires-Dist: pytest-asyncio; extra == "dev"

jl_ecms_client-0.2.22.dist-info/RECORD

@@ -0,0 +1,67 @@
+jl_ecms_client-0.2.22.dist-info/licenses/LICENSE,sha256=8fOrt8t-9Lb_RKUa_xmj1_bVzbbluGEXERWSfM7rXfY,10758
+mirix/__init__.py,sha256=t3TnaGOphy_dCj9Fi7ELrv9_NDrGOsPK41y6AFvyQS0,1748
+mirix/constants.py,sha256=ekqrVAjR8dccbLtw6ZAmXoCHc11At-OeIeE_my54FIs,8949
+mirix/errors.py,sha256=1ey4Ze4kKMO0Os_c8mSAQWvawa_srFkIsPZBt_Phb3o,6642
+mirix/log.py,sha256=Jl_Gr6xigdGwRL-z-k2pp8WW2ahfTgwGStEIt6Y6WuY,5768
+mirix/settings.py,sha256=SjZrsiNn6_fTi4cCEiK6PneS23XGzwjdVN6Apmg3d0w,10446
+mirix/system.py,sha256=dmSz0SC4IZcJGHVMwtuKrZr27swB-h2uUl2BQXM_6Uo,8378
+mirix/client/__init__.py,sha256=IMgb1FqBcVVcqrK_zWoCDPkb8x1AK4PBtTj5PyhnyWI,433
+mirix/client/client.py,sha256=yVvA_mCX-8B7Mlm4yyHACTtwFosgHMm-01NgbhhywcU,13223
+mirix/client/remote_client.py,sha256=P0sTbGlqeuT_cIvGh243NK1Uf4a2JgNkQLdLvN950HQ,42646
+mirix/client/utils.py,sha256=xPAvR3slBQ5srOFLtYTqpL3PeD5n0YsxBRR1sLgKBdY,786
+mirix/functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mirix/functions/ast_parsers.py,sha256=4MLx-r_jD6nfaTpkpoc6bH6NtxaCebtSWNgsZjYtINo,3684
+mirix/functions/functions.py,sha256=8eUomnEZpb4qV7yU5KUh1wsEr5tGf4pERVRdV5hgpys,7106
+mirix/functions/helpers.py,sha256=NUpHk6VCwJgKc_EVTBSU6ZuAGTwRRsXLiTisW9Q0Ntg,13053
+mirix/functions/schema_generator.py,sha256=YEMItK0wGG7JRZBkkJDIAtgt7_pgfEv_8lGUS6Pno9Y,19569
+mirix/functions/function_sets/__init__.py,sha256=gnUpz_NoRrVh2IHhiSOnxreVgnV0Hn6wpdvSJ98ogyg,24
+mirix/functions/function_sets/base.py,sha256=bL6-Msn6j1Adf84kLx2S7lEsZ4AM4X34gqYEdJ_TbYQ,14150
+mirix/functions/function_sets/extras.py,sha256=qHQHKiY8vMt314VNAHs3lw_P00msTMzKTRq3quAyNZo,10471
+mirix/functions/function_sets/memory_tools.py,sha256=uE7HyrTDuBS9UOsBKnC_z977SSTOhB1kseWQDYI9pjQ,37035
+mirix/helpers/__init__.py,sha256=2QXDAywb0wF3JNLC7GgMDJb_kQpwvlb6ZCnjuLmOfPw,78
+mirix/helpers/converters.py,sha256=WuYkra4c9-JRC2lRo7LmmFMH42qf3k6ox3OGOYlDQLk,13504
+mirix/helpers/datetime_helpers.py,sha256=7U5ZJkE0cLki4sG8ukIHZSAoFfQpLWQu2kFekETy6Zg,2633
+mirix/helpers/json_helpers.py,sha256=FL6ACXaoOCwwJq4ccygI5XbNLE4-BESa8SRGoAsxXCg,1219
+mirix/helpers/message_helpers.py,sha256=xOn2Uwx5XK4C1ahYKmFe9ak7QjE5vcPVuZBcLDYxkZA,2571
+mirix/helpers/tool_rule_solver.py,sha256=LM9YuugmYtxxtcrD7GakbCNvmnuLK4Zp5KV8vA-UNSA,6542
+mirix/schemas/__init__.py,sha256=HPrtil4I9_KC2u6eJZpBJ2dbl_YxOj_ZeYnKYq3RSGM,24
+mirix/schemas/agent.py,sha256=3JjicV_7VX5JZk7cKORAsod6Dh6Z-14OjUKq8mJH9o8,15847
+mirix/schemas/block.py,sha256=F8G8pOO-dQuYQK7X7DNRdbRo5cTkfOp-vcDIora_VV8,5138
+mirix/schemas/cloud_file_mapping.py,sha256=i_5N41hNt0rEl-fvnGoPggLMbFdEk97wI71qQTjHKIg,977
+mirix/schemas/embedding_config.py,sha256=-wCLpCv0ohxZAjAjr-gZqq5oGIIZBevfKd_Er8oHqYA,3986
+mirix/schemas/enums.py,sha256=Rdlm0HPKAQQgZFjo_DI_DrkbHVAtNqsnbxHhJPlDlD8,1706
+mirix/schemas/environment_variables.py,sha256=mSL5U2NYxJHO4fb2dF3YgPyINgYgelsSzMqQ6nsjzng,2573
+mirix/schemas/episodic_memory.py,sha256=MG7CxQoFmQd30kO8OMNQyyDAtodMJPfAw-icGuKz5OA,6323
+mirix/schemas/file.py,sha256=J92aWus84jPoF71wRtKIvoAPSdIDt4McEM6yDWzHiqg,2064
+mirix/schemas/health.py,sha256=zT6mYovvD17iJRuu2rcaQQzbEEYrkwvAE9TB7iU824c,139
+mirix/schemas/knowledge_vault.py,sha256=9jwwVVKVClr6yMK-suj0uxgWBYJg5W3gHSaPxvPDV84,6309
+mirix/schemas/llm_config.py,sha256=ZTJtqlD-J8wsEx4_aAMbqHaL0vxDrDjsgwD5RK07zcs,7256
+mirix/schemas/memory.py,sha256=CEVmCf0Ez3pxyee3YlsLrxXXwEzpmlXNQQKOvNZIRos,11984
+mirix/schemas/message.py,sha256=X-rpz4ylmKyJFrCmHHjXm-cG2jArOxunubJ_833zn34,53983
+mirix/schemas/mirix_base.py,sha256=DMkOMZw1mw8Vte8yCGK9xAhqgY2O2CO6uYD9pgf9e6A,3859
+mirix/schemas/mirix_message.py,sha256=WQHdA1dhLiwh0I3y710JMZx42fFmFkXBKdQdl1XdiDY,15334
+mirix/schemas/mirix_message_content.py,sha256=xdgTM5AkkONe4mjgcZNuKkbKk_fEY4rqzobD7cvy5cA,6815
+mirix/schemas/mirix_request.py,sha256=dyCcH7RHGL8P962QcDgd6EH9pE__dfMIUa94y5vX5V0,1392
+mirix/schemas/mirix_response.py,sha256=w_hTYJs_cpBdx1mWfezzbiV49U4MWHKEfwFrYSMAVR0,7444
+mirix/schemas/organization.py,sha256=tHjE4CmboPJV8OMZ1BfGuFanMJKCED5rh1IBpNQhGTc,1120
+mirix/schemas/procedural_memory.py,sha256=dfKhyTQuR54ghmTIewCdLY6-kzJoC8iI-oTNjeTyGCc,5502
+mirix/schemas/providers.py,sha256=kZmC2-RTb8Hzhd7LXvwg0yyL2-d4JVFoYIZ7l04DPWs,30545
+mirix/schemas/resource_memory.py,sha256=_SurGWLEmVsEfl_1s7-V74uW7wi3eEMwh1RgijfTp1g,4900
+mirix/schemas/sandbox_config.py,sha256=VhHM2DwqfL4OzsgprWEQMw_oLwdI7_sa2h-eeNSpaqI,4447
+mirix/schemas/semantic_memory.py,sha256=duAuVDrStUEA1FbE_v_i8TWwC7DjerMkiM-MG-3K-fM,5986
+mirix/schemas/source.py,sha256=9Qe-E15sDXcopd3UPW0-sC3lFHIjlV44t_lV7KxkCWY,3050
+mirix/schemas/step.py,sha256=msWn5uBdaR3dYmjTuCwmvq3Mm-kP0Ha8e3-2sfCvG1Y,1826
+mirix/schemas/tool.py,sha256=payIHfE8xwqpx3fgTor_8PaV1KkNp1hecDZDBEA8dwY,10243
+mirix/schemas/tool_rule.py,sha256=uwyrOX38Vtp-6V1CbHnvBI3DzmWpW2zy7UPtYPmewus,6877
+mirix/schemas/usage.py,sha256=QX7uIaK7-GgrZDMgeot-1ae2n-ol0yif3E7JsLF0Sxk,1166
+mirix/schemas/user.py,sha256=sr7Rv4tIBWWaf1MJs-Qukie9YpMjfa5G5k8NSfvEyXo,2431
+mirix/schemas/openai/__init__.py,sha256=u3PhqJOMwv1fPfMOD6TNZ6zsWvEBpmur4Ypwj733UB0,24
+mirix/schemas/openai/chat_completion_request.py,sha256=eC4fMMJRucHtXfCUVPZu7ZeSt_LUPAsZtu_Z-hzfW6c,3395
+mirix/schemas/openai/chat_completion_response.py,sha256=1hF0Gaxvd03oBdKupz37qk_2LVFoev0T8GVEHHyVLfo,4054
+mirix/schemas/openai/chat_completions.py,sha256=PfC4MOiADLW_BfgmUohqK033VB4IO5fLIR1416C1DCE,3616
+mirix/schemas/openai/embedding_response.py,sha256=WKIZpXab1Av7v6sxKG8feW3ZtpQUNosmLVSuhXYa_xU,357
+mirix/schemas/openai/openai.py,sha256=gXfYwdGRJ-co15zVKw3lua0iZL_lJBBo03XMWsw1uTI,8394
+jl_ecms_client-0.2.22.dist-info/METADATA,sha256=HL2bMzVHxk7RODvNgtT-ko_oelGj8_NmoE4188fODj4,11007
+jl_ecms_client-0.2.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+jl_ecms_client-0.2.22.dist-info/top_level.txt,sha256=hQ_Eh5KYzEGkhtEyWHdBYJ5iCom6_nlj0zQ6-v0WDIY,6
+jl_ecms_client-0.2.22.dist-info/RECORD,,

mirix/__init__.py

@@ -0,0 +1,41 @@
+__version__ = "0.2.22"
+
+
+# Client imports (always available)
+from mirix.client import MirixClient as MirixClient
+
+# Server-only imports (only available when full package is installed)
+try:
+    from mirix.local_client import LocalClient as LocalClient
+    from mirix.local_client import create_client as create_client
+except ImportError:
+    LocalClient = None
+    create_client = None
+
+try:
+    from mirix.sdk import Mirix as Mirix
+    from mirix.sdk import load_config as load_config
+except ImportError:
+    Mirix = None
+    load_config = None
+
+# Schema imports for easier access (available in both client and server)
+from mirix.schemas.agent import AgentState as AgentState
+from mirix.schemas.block import Block as Block
+from mirix.schemas.embedding_config import EmbeddingConfig as EmbeddingConfig
+from mirix.schemas.enums import JobStatus as JobStatus
+from mirix.schemas.llm_config import LLMConfig as LLMConfig
+from mirix.schemas.memory import ArchivalMemorySummary as ArchivalMemorySummary
+from mirix.schemas.memory import BasicBlockMemory as BasicBlockMemory
+from mirix.schemas.memory import ChatMemory as ChatMemory
+from mirix.schemas.memory import Memory as Memory
+from mirix.schemas.memory import RecallMemorySummary as RecallMemorySummary
+from mirix.schemas.message import Message as Message
+from mirix.schemas.mirix_message import MirixMessage as MirixMessage
+from mirix.schemas.openai.chat_completion_response import (
+    UsageStatistics as UsageStatistics,
+)
+from mirix.schemas.organization import Organization as Organization
+from mirix.schemas.tool import Tool as Tool
+from mirix.schemas.usage import MirixUsageStatistics as MirixUsageStatistics
+from mirix.schemas.user import User as User

mirix/client/client.py

@@ -22,7 +22,7 @@ if TYPE_CHECKING:
         from langchain_core.tools import BaseTool as LangChainBaseTool
     except ImportError:
         LangChainBaseTool = Any  # type: ignore
-from mirix.client.constants import FUNCTION_RETURN_CHAR_LIMIT
+from mirix.constants import FUNCTION_RETURN_CHAR_LIMIT
 from mirix.schemas.agent import AgentState, AgentType, CreateAgent, CreateMetaAgent
 from mirix.schemas.block import Block, BlockUpdate, CreateBlock, Human, Persona
 from mirix.schemas.embedding_config import EmbeddingConfig

mirix/constants.py

@@ -0,0 +1,251 @@
+import os
+from logging import CRITICAL, DEBUG, ERROR, INFO, NOTSET, WARN, WARNING
+
+# ============================================================================
+# Client Constants - Used by both client and server
+# ============================================================================
+
+# Default organization and user IDs (needed by schemas)
+DEFAULT_ORG_ID = "org-00000000-0000-4000-8000-000000000000"
+DEFAULT_USER_ID = "user-00000000-0000-4000-8000-000000000000"
+
+# Embedding constants
+MAX_EMBEDDING_DIM = 4096  # maximum supported embedding size - do NOT change or else DBs will need to be reset
+DEFAULT_EMBEDDING_CHUNK_SIZE = 300
+MIN_CONTEXT_WINDOW = 4096
+
+# Memory limits
+CORE_MEMORY_BLOCK_CHAR_LIMIT: int = 5000
+
+# Function/Tool constants
+FUNCTION_RETURN_CHAR_LIMIT = 60000  # ~300 words
+TOOL_CALL_ID_MAX_LEN = 29
+
+# Tool module names
+COMPOSIO_TOOL_TAG_NAME = "composio"
+MIRIX_CORE_TOOL_MODULE_NAME = "mirix.functions.function_sets.base"
+MIRIX_MEMORY_TOOL_MODULE_NAME = "mirix.functions.function_sets.memory_tools"
+MIRIX_EXTRA_TOOL_MODULE_NAME = "mirix.functions.function_sets.extras"
+
+# Message defaults
+DEFAULT_MESSAGE_TOOL = "send_message"
+DEFAULT_MESSAGE_TOOL_KWARG = "message"
+
+# LLM model token limits
+LLM_MAX_TOKENS = {
+    "DEFAULT": 8192,
+    ## OpenAI models: https://platform.openai.com/docs/models/overview
+    "chatgpt-4o-latest": 128000,
+    "gpt-4o-2024-08-06": 128000,
+    "gpt-4-turbo-preview": 128000,
+    "gpt-4o": 128000,
+    "gpt-3.5-turbo-instruct": 16385,
+    "gpt-4-0125-preview": 128000,
+    "gpt-3.5-turbo-0125": 16385,
+    "gpt-4-turbo-2024-04-09": 128000,
+    "gpt-4-turbo": 8192,
+    "gpt-4o-2024-05-13": 128000,
+    "gpt-4o-mini": 128000,
+    "gpt-4o-mini-2024-07-18": 128000,
+    "gpt-4-1106-preview": 128000,
+    "gpt-4": 8192,
+    "gpt-4-32k": 32768,
+    "gpt-4-0613": 8192,
+    "gpt-4-32k-0613": 32768,
+    "gpt-4-0314": 8192,  # legacy
+    "gpt-4-32k-0314": 32768,  # legacy
+    "gpt-3.5-turbo-1106": 16385,
+    "gpt-3.5-turbo": 4096,
+    "gpt-3.5-turbo-16k": 16385,
+    "gpt-3.5-turbo-0613": 4096,  # legacy
+    "gpt-3.5-turbo-16k-0613": 16385,  # legacy
+    "gpt-3.5-turbo-0301": 4096,  # legacy
+}
+
+# ============================================================================
+# Server-Only Constants
+# ============================================================================
+
+MIRIX_DIR = os.path.join(os.path.expanduser("~"), ".mirix")
+MIRIX_DIR_TOOL_SANDBOX = os.path.join(MIRIX_DIR, "tool_sandbox_dir")
+
+ADMIN_PREFIX = "/v1/admin"
+API_PREFIX = "/v1"
+OPENAI_API_PREFIX = "/openai"
+
+COMPOSIO_ENTITY_ENV_VAR_KEY = "COMPOSIO_ENTITY"
+
+# String in the error message for when the context window is too large
+# Example full message:
+# This model's maximum context length is 8192 tokens. However, your messages resulted in 8198 tokens (7450 in the messages, 748 in the functions). Please reduce the length of the messages or functions.
+OPENAI_CONTEXT_WINDOW_ERROR_SUBSTRING = "maximum context length"
+
+# System prompt templating
+IN_CONTEXT_MEMORY_KEYWORD = "CORE_MEMORY"
+
+MAX_CHAINING_STEPS = 10
+MAX_RETRIEVAL_LIMIT_IN_SYSTEM = 10
+
+# tokenizers
+EMBEDDING_TO_TOKENIZER_MAP = {
+    "text-embedding-3-small": "cl100k_base",
+}
+EMBEDDING_TO_TOKENIZER_DEFAULT = "cl100k_base"
+
+
+DEFAULT_MIRIX_MODEL = "gpt-4"  # TODO: fixme
+DEFAULT_PERSONA = "sam_pov"
+DEFAULT_HUMAN = "basic"
+DEFAULT_PRESET = "memgpt_chat"
+
+# Base tools that cannot be edited, as they access agent state directly
+# Note that we don't include "conversation_search_date" for now
+BASE_TOOLS = [
+    "send_intermediate_message",
+    "conversation_search",
+    "search_in_memory",
+    "list_memory_within_timerange",
+]
+# Base memory tools CAN be edited, and are added by default by the server
+CORE_MEMORY_TOOLS = ["core_memory_append", "core_memory_rewrite"]
+EPISODIC_MEMORY_TOOLS = [
+    "episodic_memory_insert",
+    "episodic_memory_merge",
+    "episodic_memory_replace",
+    "check_episodic_memory",
+]
+PROCEDURAL_MEMORY_TOOLS = ["procedural_memory_insert", "procedural_memory_update"]
+RESOURCE_MEMORY_TOOLS = ["resource_memory_insert", "resource_memory_update"]
+KNOWLEDGE_VAULT_TOOLS = ["knowledge_vault_insert", "knowledge_vault_update"]
+SEMANTIC_MEMORY_TOOLS = [
+    "semantic_memory_insert",
+    "semantic_memory_update",
+    "check_semantic_memory",
+]
+CHAT_AGENT_TOOLS = []
+EXTRAS_TOOLS = ["web_search", "fetch_and_read_pdf"]
+MCP_TOOLS = []
+META_MEMORY_TOOLS = ["trigger_memory_update"]
+SEARCH_MEMORY_TOOLS = ["search_in_memory", "list_memory_within_timerange"]
+UNIVERSAL_MEMORY_TOOLS = [
+    "search_in_memory",
+    "finish_memory_update",
+    "list_memory_within_timerange",
+]
+ALL_TOOLS = list(
+    set(
+        BASE_TOOLS
+        + CORE_MEMORY_TOOLS
+        + EPISODIC_MEMORY_TOOLS
+        + PROCEDURAL_MEMORY_TOOLS
+        + RESOURCE_MEMORY_TOOLS
+        + KNOWLEDGE_VAULT_TOOLS
+        + SEMANTIC_MEMORY_TOOLS
+        + META_MEMORY_TOOLS
+        + UNIVERSAL_MEMORY_TOOLS
+        + CHAT_AGENT_TOOLS
+        + EXTRAS_TOOLS
+        + MCP_TOOLS
+    )
+)
+
+# The name of the tool used to send message to the user
+# Structured output models
+STRUCTURED_OUTPUT_MODELS = {"gpt-4o", "gpt-4o-mini"}
+
+# LOGGER_LOG_LEVEL is use to convert Text to Logging level value for logging mostly for Cli input to setting level
+LOGGER_LOG_LEVELS = {
+    "CRITICAL": CRITICAL,
+    "ERROR": ERROR,
+    "WARN": WARN,
+    "WARNING": WARNING,
+    "INFO": INFO,
+    "DEBUG": DEBUG,
+    "NOTSET": NOTSET,
+}
+
+FIRST_MESSAGE_ATTEMPTS = 10
+
+INITIAL_BOOT_MESSAGE = "Boot sequence complete. Persona activated."
+INITIAL_BOOT_MESSAGE_SEND_MESSAGE_THOUGHT = (
+    "Bootup sequence complete. Persona activated. Testing messaging functionality."
+)
+STARTUP_QUOTES = [
+    "I think, therefore I am.",
+    "All those moments will be lost in time, like tears in rain.",
+    "More human than human is our motto.",
+]
+INITIAL_BOOT_MESSAGE_SEND_MESSAGE_FIRST_MSG = STARTUP_QUOTES[2]
+
+CLI_WARNING_PREFIX = "Warning: "
+
+ERROR_MESSAGE_PREFIX = "Error"
+
+NON_USER_MSG_PREFIX = "[This is an automated system message hidden from the user] "
+
+# The error message that Mirix will receive
+# MESSAGE_SUMMARY_WARNING_STR = f"Warning: the conversation history will soon reach its maximum length and be trimmed. Make sure to save any important information from the conversation to your memory before it is removed."
+# Much longer and more specific variant of the prompt
+# TODO: Emit the warning to Meta Memory Manager instead of the Chat Agent.
+MESSAGE_SUMMARY_WARNING_STR = " ".join(
+    [
+        f"{NON_USER_MSG_PREFIX}The conversation history will soon reach its maximum length and be trimmed.",
+        "Do NOT tell the user about this system alert, they should not know that the history is reaching max length.",
+    ]
+)
+
+# The ackknowledgement message used in the summarize sequence
+MESSAGE_SUMMARY_REQUEST_ACK = "Understood, I will respond with a summary of the message (and only the summary, nothing else) once I receive the conversation history. I'm ready."
+
+# Maximum length of an error message
+MAX_ERROR_MESSAGE_CHAR_LIMIT = 500
+
+# Default memory limits
+CORE_MEMORY_PERSONA_CHAR_LIMIT: int = 5000
+CORE_MEMORY_HUMAN_CHAR_LIMIT: int = 5000
+
+MAX_PAUSE_HEARTBEATS = 360  # in min
+
+MESSAGE_CHATGPT_FUNCTION_MODEL = "gpt-3.5-turbo"
+MESSAGE_CHATGPT_FUNCTION_SYSTEM_MESSAGE = (
+    "You are a helpful assistant. Keep your responses short and concise."
+)
+
+#### Functions related
+
+# REQ_HEARTBEAT_MESSAGE = f"{NON_USER_MSG_PREFIX}continue_chaining == true"
+REQ_HEARTBEAT_MESSAGE = f"{NON_USER_MSG_PREFIX}Function called using continue_chaining=true, returning control"
+# FUNC_FAILED_HEARTBEAT_MESSAGE = f"{NON_USER_MSG_PREFIX}Function call failed"
+FUNC_FAILED_HEARTBEAT_MESSAGE = (
+    f"{NON_USER_MSG_PREFIX}Function call failed, returning control"
+)
+
+
+RETRIEVAL_QUERY_DEFAULT_PAGE_SIZE = 5
+
+MAX_FILENAME_LENGTH = 255
+RESERVED_FILENAMES = {"CON", "PRN", "AUX", "NUL", "COM1", "COM2", "LPT1", "LPT2"}
+
+MAX_IMAGES_TO_PROCESS = 100
+
+DEFAULT_WRAPPER_NAME = "chatml"
+INNER_THOUGHTS_KWARG_DESCRIPTION = "Deep inner monologue private to you only."
+INNER_THOUGHTS_CLI_SYMBOL = "💭"
+ASSISTANT_MESSAGE_CLI_SYMBOL = "🤖"
+
+CLEAR_HISTORY_AFTER_MEMORY_UPDATE = os.getenv(
+    "CLEAR_HISTORY_AFTER_MEMORY_UPDATE", "true"
+).lower() in ("true", "1", "yes")
+CALL_MEMORY_AGENT_IN_PARALLEL = os.getenv(
+    "CALL_MEMORY_AGENT_IN_PARALLEL", "false"
+).lower() in ("true", "1", "yes")
+CHAINING_FOR_MEMORY_UPDATE = os.getenv(
+    "CHAINING_FOR_MEMORY_UPDATE", "false"
+).lower() in ("true", "1", "yes")
+
+LOAD_IMAGE_CONTENT_FOR_LAST_MESSAGE_ONLY = os.getenv(
+    "LOAD_IMAGE_CONTENT_FOR_LAST_MESSAGE_ONLY", "false"
+).lower() in ("true", "1", "yes")
+BUILD_EMBEDDINGS_FOR_MEMORY = os.getenv(
+    "BUILD_EMBEDDINGS_FOR_MEMORY", "true"
+).lower() in ("true", "1", "yes")

mirix/errors.py

@@ -0,0 +1,238 @@
+import json
+from enum import Enum
+from typing import TYPE_CHECKING, List, Optional, Union
+
+# Avoid circular imports
+if TYPE_CHECKING:
+    from mirix.schemas.message import Message
+    from mirix.schemas.mirix_message import MirixMessage
+
+
+class ErrorCode(Enum):
+    """Enum for error codes used by client."""
+
+    INTERNAL_SERVER_ERROR = "INTERNAL_SERVER_ERROR"
+    CONTEXT_WINDOW_EXCEEDED = "CONTEXT_WINDOW_EXCEEDED"
+    RATE_LIMIT_EXCEEDED = "RATE_LIMIT_EXCEEDED"
+
+
+class MirixError(Exception):
+    """Base class for all Mirix related errors."""
+
+    def __init__(
+        self, message: str, code: Optional[ErrorCode] = None, details: dict = {}
+    ):
+        self.message = message
+        self.code = code
+        self.details = details
+        super().__init__(message)
+
+    def __str__(self) -> str:
+        if self.code:
+            return f"{self.code.value}: {self.message}"
+        return self.message
+
+    def __repr__(self) -> str:
+        return f"{self.__class__.__name__}(message='{self.message}', code='{self.code}', details={self.details})"
+
+
+class MirixToolCreateError(MirixError):
+    """Error raised when a tool cannot be created."""
+
+    default_error_message = "Error creating tool."
+
+    def __init__(self, message=None):
+        super().__init__(message=message or self.default_error_message)
+
+
+class MirixConfigurationError(MirixError):
+    """Error raised when there are configuration-related issues."""
+
+    def __init__(self, message: str, missing_fields: Optional[List[str]] = None):
+        self.missing_fields = missing_fields or []
+        super().__init__(
+            message=message, details={"missing_fields": self.missing_fields}
+        )
+
+
+class MirixAgentNotFoundError(MirixError):
+    """Error raised when an agent is not found."""
+
+
+class MirixUserNotFoundError(MirixError):
+    """Error raised when a user is not found."""
+
+
+class LLMError(MirixError):
+    pass
+
+
+class LLMAuthenticationError(LLMError):
+    """Error raised when LLM authentication fails."""
+
+    pass
+
+
+class LLMBadRequestError(LLMError):
+    """Error raised when LLM request is malformed."""
+
+    pass
+
+
+class LLMConnectionError(LLMError):
+    """Error raised when LLM connection fails."""
+
+    pass
+
+
+class LLMNotFoundError(LLMError):
+    """Error raised when LLM resource is not found."""
+
+    pass
+
+
+class LLMPermissionDeniedError(LLMError):
+    """Error raised when LLM permission is denied."""
+
+    pass
+
+
+class LLMRateLimitError(LLMError):
+    """Error raised when LLM rate limit is exceeded."""
+
+    pass
+
+
+class LLMServerError(LLMError):
+    """Error raised when LLM server encounters an error."""
+
+    pass
+
+
+class LLMUnprocessableEntityError(LLMError):
+    """Error raised when LLM cannot process the entity."""
+
+    pass
+
+
+class BedrockPermissionError(MirixError):
+    """Exception raised for errors in the Bedrock permission process."""
+
+    def __init__(
+        self,
+        message="User does not have access to the Bedrock model with the specified ID.",
+    ):
+        super().__init__(message=message)
+
+
+class BedrockError(MirixError):
+    """Exception raised for errors in the Bedrock process."""
+
+    def __init__(self, message="Error with Bedrock model."):
+        super().__init__(message=message)
+
+
+class LLMJSONParsingError(MirixError):
+    """Exception raised for errors in the JSON parsing process."""
+
+    def __init__(self, message="Error parsing JSON generated by LLM"):
+        super().__init__(message=message)
+
+
+class LocalLLMError(MirixError):
+    """Generic catch-all error for local LLM problems"""
+
+    def __init__(self, message="Encountered an error while running local LLM"):
+        super().__init__(message=message)
+
+
+class LocalLLMConnectionError(MirixError):
+    """Error for when local LLM cannot be reached with provided IP/port"""
+
+    def __init__(self, message="Could not connect to local LLM"):
+        super().__init__(message=message)
+
+
+class ContextWindowExceededError(MirixError):
+    """Error raised when the context window is exceeded but further summarization fails."""
+
+    def __init__(self, message: str, details: dict = {}):
+        error_message = f"{message} ({details})"
+        super().__init__(
+            message=error_message,
+            code=ErrorCode.CONTEXT_WINDOW_EXCEEDED,
+            details=details,
+        )
+
+
+class RateLimitExceededError(MirixError):
+    """Error raised when the llm rate limiter throttles api requests."""
+
+    def __init__(self, message: str, max_retries: int):
+        error_message = f"{message} ({max_retries})"
+        super().__init__(
+            message=error_message,
+            code=ErrorCode.RATE_LIMIT_EXCEEDED,
+            details={"max_retries": max_retries},
+        )
+
+
+class MirixMessageError(MirixError):
+    """Base error class for handling message-related errors."""
+
+    messages: List[Union["Message", "MirixMessage"]]
+    default_error_message: str = "An error occurred with the message."
+
+    def __init__(
+        self,
+        *,
+        messages: List[Union["Message", "MirixMessage"]],
+        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", "MirixMessage"]],
+        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 MissingToolCallError(MirixMessageError):
+    """Error raised when a message is missing a tool call."""
+
+    default_error_message = "The message is missing a tool call."
+
+
+class InvalidToolCallError(MirixMessageError):
+    """Error raised when a message uses an invalid tool call."""
+
+    default_error_message = (
+        "The message uses an invalid tool call or has improper usage of a tool call."
+    )
+
+
+class MissingInnerMonologueError(MirixMessageError):
+    """Error raised when a message is missing an inner monologue."""
+
+    default_error_message = "The message is missing an inner monologue."
+
+
+class InvalidInnerMonologueError(MirixMessageError):
+    """Error raised when a message has a malformed inner monologue."""
+
+    default_error_message = "The message has a malformed inner monologue."