jl-ecms-client 0.2.5__py3-none-any.whl → 0.2.19__py3-none-any.whl

mirix/log.py

@@ -0,0 +1,163 @@
+import logging
+import os
+import sys
+from logging.handlers import RotatingFileHandler
+from pathlib import Path
+from typing import Optional
+
+from mirix.settings import settings
+
+
+def get_log_level() -> int:
+    """Get the configured log level."""
+    if settings.debug:
+        return logging.DEBUG
+    
+    # Map string level to logging constant
+    level_map = {
+        "DEBUG": logging.DEBUG,
+        "INFO": logging.INFO,
+        "WARNING": logging.WARNING,
+        "ERROR": logging.ERROR,
+        "CRITICAL": logging.CRITICAL,
+    }
+    
+    return level_map.get(settings.log_level.upper(), logging.INFO)
+
+
+selected_log_level = get_log_level()
+
+
+def validate_log_file_path(log_file_path: Path) -> Path:
+    """
+    Validate that the log file path is writable.
+    
+    Checks:
+    - Path is not a directory
+    - Parent directory exists or can be created
+    - We have write permissions to the directory
+    
+    Args:
+        log_file_path: Path to the log file
+        
+    Returns:
+        Path: Validated absolute path
+        
+    Raises:
+        ValueError: If the path is invalid or not writable
+    """
+    # Convert to absolute path
+    log_file_path = log_file_path.expanduser().resolve()
+    
+    # Check if path exists and is a directory (not allowed)
+    if log_file_path.exists() and log_file_path.is_dir():
+        raise ValueError(
+            f"Invalid log file path: '{log_file_path}' is a directory. "
+            f"MIRIX_LOG_FILE must be a file path, not a directory."
+        )
+    
+    # Get parent directory
+    parent_dir = log_file_path.parent
+    
+    # Try to create parent directory if it doesn't exist
+    try:
+        parent_dir.mkdir(parents=True, exist_ok=True)
+    except (OSError, PermissionError) as e:
+        raise ValueError(
+            f"Invalid log file path: Cannot create directory '{parent_dir}'. "
+            f"Error: {e}"
+        ) from e
+    
+    # Check if parent directory is writable
+    if not os.access(parent_dir, os.W_OK):
+        raise ValueError(
+            f"Invalid log file path: Directory '{parent_dir}' is not writable. "
+            f"Check permissions for MIRIX_LOG_FILE."
+        )
+    
+    # If file exists, check if it's writable
+    if log_file_path.exists() and not os.access(log_file_path, os.W_OK):
+        raise ValueError(
+            f"Invalid log file path: File '{log_file_path}' exists but is not writable. "
+            f"Check file permissions for MIRIX_LOG_FILE."
+        )
+    
+    return log_file_path
+
+
+def get_logger(name: Optional[str] = None) -> "logging.Logger":
+    """
+    Get the Mirix logger with configured handlers.
+    
+    Log Level Configuration:
+        - Single log level (MIRIX_LOG_LEVEL) applies to ALL handlers
+        - Controlled by: MIRIX_LOG_LEVEL or MIRIX_DEBUG environment variables
+        - Same level used for both console and file output
+    
+    Handler Configuration (Default Behavior):
+        - Console: ALWAYS enabled UNLESS explicitly disabled (MIRIX_LOG_TO_CONSOLE=false)
+        - File: Automatically enabled if MIRIX_LOG_FILE is set with a valid path
+        - Handlers determine WHERE logs go, NOT what level they use
+    
+    Returns:
+        logging.Logger: Configured logger instance
+        
+    Raises:
+        ValueError: If MIRIX_LOG_FILE is set but the path is invalid or not writable
+    """
+    logger = logging.getLogger("Mirix")
+    
+    # Set the log level ONCE for the entire logger
+    # This single level applies to all handlers (console and file)
+    logger.setLevel(selected_log_level)
+    
+    # Add handlers if not already configured
+    # Handlers control WHERE logs go (console/file), not WHAT level they use
+    if not logger.handlers:
+        # Create a single formatter for consistency across all handlers
+        formatter = logging.Formatter(
+            '%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+            datefmt='%Y-%m-%d %H:%M:%S'
+        )
+        
+        handlers_added = []
+        
+        # Console handler - ALWAYS enabled unless explicitly disabled
+        # Console logging is the default behavior
+        if settings.log_to_console:
+            console_handler = logging.StreamHandler(sys.stdout)
+            console_handler.setFormatter(formatter)
+            logger.addHandler(console_handler)
+            handlers_added.append("console")
+        
+        # File handler - ONLY enabled if MIRIX_LOG_FILE is configured
+        # Automatically enabled when MIRIX_LOG_FILE is set
+        if settings.log_file is not None:
+            # Validate and get absolute path
+            # This will raise ValueError if path is invalid
+            log_file = validate_log_file_path(Path(settings.log_file))
+            
+            # Create rotating file handler
+            file_handler = RotatingFileHandler(
+                log_file,
+                maxBytes=settings.log_max_bytes,
+                backupCount=settings.log_backup_count,
+            )
+            file_handler.setFormatter(formatter)
+            logger.addHandler(file_handler)
+            handlers_added.append(f"file ({log_file})")
+        
+        # Log where logs are being written (if any handlers were added)
+        if handlers_added:
+            destinations = " and ".join(handlers_added)
+            log_level_name = logging.getLevelName(selected_log_level)
+            logger.info("Logging to: %s (level: %s)", destinations, log_level_name)
+        else:
+            # No handlers configured - add NullHandler to prevent warnings
+            # This only happens if console is explicitly disabled AND file is not configured
+            logger.addHandler(logging.NullHandler())
+        
+        # Prevent propagation to root logger to avoid duplicate messages
+        logger.propagate = False
+    
+    return logger

mirix/schemas/agent.py

@@ -14,7 +14,8 @@ from mirix.schemas.mirix_base import OrmMetadataBase
 from mirix.schemas.openai.chat_completion_response import UsageStatistics
 from mirix.schemas.tool import Tool
 from mirix.schemas.tool_rule import ToolRule
-from mirix.utils import create_random_username
+
+# Removed create_random_username import - server generates names if not provided
 
 
 class AgentType(str, Enum):
@@ -108,9 +109,9 @@ class AgentState(OrmMetadataBase, validate_assignment=True):
 
 class CreateAgent(BaseModel, validate_assignment=True):  #
     # all optional as server can generate defaults
-    name: str = Field(
-        default_factory=lambda: create_random_username(),
-        description="The name of the agent.",
+    name: Optional[str] = Field(
+        None,
+        description="The name of the agent. If not provided, server will generate one.",
     )
 
     # memory creation

mirix/schemas/cloud_file_mapping.py

@@ -2,8 +2,8 @@ from datetime import datetime
 
 from pydantic import Field
 
+from mirix.client.utils import get_utc_time
 from mirix.schemas.mirix_base import MirixBase
-from mirix.utils import get_utc_time
 
 
 class CloudFileMappingBase(MirixBase):

mirix/schemas/embedding_config.py

@@ -2,9 +2,6 @@ from typing import Literal, Optional
 
 from pydantic import BaseModel, Field
 
-from mirix.log import get_logger
-
-logger = get_logger(__name__)
 
 class EmbeddingConfig(BaseModel):
     """

mirix/schemas/enums.py

@@ -1,6 +1,18 @@
 from enum import Enum
 
 
+class ToolType(str, Enum):
+    """Types of tools in Mirix"""
+    CUSTOM = "custom"
+    MIRIX_CORE = "mirix_core"
+    MIRIX_CODER_CORE = "mirix_coder_core"
+    MIRIX_MEMORY_CORE = "mirix_memory_core"
+    MIRIX_EXTRA = "mirix_extra"
+    MIRIX_MCP = "mirix_mcp"
+    MIRIX_MULTI_AGENT_CORE = "mirix_multi_agent_core"
+    USER_DEFINED = "user_defined"
+
+
 class ProviderType(str, Enum):
     anthropic = "anthropic"
 

mirix/schemas/episodic_memory.py

@@ -3,10 +3,10 @@ from typing import Any, Dict, List, Optional
 
 from pydantic import Field, field_validator
 
+from mirix.client.utils import get_utc_time
 from mirix.constants import MAX_EMBEDDING_DIM
 from mirix.schemas.embedding_config import EmbeddingConfig
 from mirix.schemas.mirix_base import MirixBase
-from mirix.utils import get_utc_time
 
 
 class EpisodicEventBase(MirixBase):

mirix/schemas/knowledge_vault.py

@@ -3,10 +3,10 @@ from typing import Any, Dict, List, Optional
 
 from pydantic import Field, field_validator
 
+from mirix.client.utils import get_utc_time
 from mirix.constants import MAX_EMBEDDING_DIM
 from mirix.schemas.embedding_config import EmbeddingConfig
 from mirix.schemas.mirix_base import MirixBase
-from mirix.utils import get_utc_time
 
 
 class KnowledgeVaultItemBase(MirixBase):

mirix/schemas/mirix_response.py

@@ -5,10 +5,10 @@ from typing import List, Union
 
 from pydantic import BaseModel, Field
 
+from mirix.client.utils import json_dumps
 from mirix.schemas.enums import MessageStreamStatus
 from mirix.schemas.mirix_message import MirixMessage, MirixMessageUnion
 from mirix.schemas.usage import MirixUsageStatistics
-from mirix.utils import json_dumps
 
 # TODO: consider moving into own file
 

mirix/schemas/organization.py

@@ -1,11 +1,11 @@
+import uuid
 from datetime import datetime
 from typing import Optional
-import uuid
 
 from pydantic import Field
 
+from mirix.client.utils import get_utc_time
 from mirix.schemas.mirix_base import MirixBase
-from mirix.utils import create_random_username, get_utc_time
 
 
 class OrganizationBase(MirixBase):
@@ -22,9 +22,9 @@ class Organization(OrganizationBase):
         default_factory=_generate_org_id,
         description="The unique identifier of the organization.",
     )
-    name: str = Field(
-        create_random_username(),
-        description="The name of the organization.",
+    name: Optional[str] = Field(
+        None,
+        description="The name of the organization. Server will generate if not provided.",
         json_schema_extra={"default": "SincereYogurt"},
     )
     created_at: Optional[datetime] = Field(

mirix/schemas/procedural_memory.py

@@ -3,10 +3,10 @@ from typing import Any, Dict, List, Optional
 
 from pydantic import Field, field_validator
 
+from mirix.client.utils import get_utc_time
 from mirix.constants import MAX_EMBEDDING_DIM
 from mirix.schemas.embedding_config import EmbeddingConfig
 from mirix.schemas.mirix_base import MirixBase
-from mirix.utils import get_utc_time
 
 
 class ProceduralMemoryItemBase(MirixBase):

mirix/schemas/resource_memory.py

@@ -6,7 +6,7 @@ from pydantic import Field, field_validator
 from mirix.constants import MAX_EMBEDDING_DIM
 from mirix.schemas.embedding_config import EmbeddingConfig
 from mirix.schemas.mirix_base import MirixBase
-from mirix.utils import get_utc_time
+from mirix.client.utils import get_utc_time
 
 
 class ResourceMemoryItemBase(MirixBase):

mirix/schemas/sandbox_config.py

@@ -8,12 +8,9 @@ from pydantic import BaseModel, Field, model_validator
 from mirix.schemas.agent import AgentState
 from mirix.schemas.mirix_base import MirixBase, OrmMetadataBase
 from mirix.settings import tool_settings
-from mirix.log import get_logger
 
 
 # Sandbox Config
-
-logger = get_logger(__name__)
 class SandboxType(str, Enum):
     E2B = "e2b"
     LOCAL = "local"
@@ -73,6 +70,7 @@ class E2BSandboxConfig(BaseModel):
         Assign a default template value if the template field is not provided.
         """
         if data.get("template") is None:
+            # Use configured template if available
             data["template"] = tool_settings.e2b_sandbox_template_id
         return data
 

mirix/schemas/semantic_memory.py

@@ -6,7 +6,7 @@ from pydantic import Field, field_validator
 from mirix.constants import MAX_EMBEDDING_DIM
 from mirix.schemas.embedding_config import EmbeddingConfig
 from mirix.schemas.mirix_base import MirixBase
-from mirix.utils import get_utc_time
+from mirix.client.utils import get_utc_time
 
 
 class SemanticMemoryItemBase(MirixBase):