jl-ecms-client 0.2.3__py3-none-any.whl → 0.2.7__py3-none-any.whl

{jl_ecms_client-0.2.3.dist-info → jl_ecms_client-0.2.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jl-ecms-client
-Version: 0.2.3
+Version: 0.2.7
 Summary: Intuit ECMS Client - Lightweight Python client for Intuit ECMS server
 Home-page: https://github.com/Mirix-AI/MIRIX
 Author: Mirix AI

{jl_ecms_client-0.2.3.dist-info → jl_ecms_client-0.2.7.dist-info}/RECORD

@@ -1,7 +1,9 @@
-jl_ecms_client-0.2.3.dist-info/licenses/LICENSE,sha256=8fOrt8t-9Lb_RKUa_xmj1_bVzbbluGEXERWSfM7rXfY,10758
+jl_ecms_client-0.2.7.dist-info/licenses/LICENSE,sha256=8fOrt8t-9Lb_RKUa_xmj1_bVzbbluGEXERWSfM7rXfY,10758
 mirix/client/__init__.py,sha256=QOjO6HJdl5CBK9QrSvAILmlCWfoobt8CNBnRWsoZFBs,2383
-mirix/client/client.py,sha256=PCD5NAglP9rV8aWaKBvJ0X1G0vs_FHqBDvVEhvJJM8k,88722
+mirix/client/client.py,sha256=9RK7BP5Fjonn5wKfUK5WK2fZS5--1mtdbmO50gQmdNs,88727
+mirix/client/constants.py,sha256=6NVFu3yZqm1HW6vKA1HCpNc8l-ds6yGPyUGf-tqXJ78,2011
 mirix/client/remote_client.py,sha256=P0sTbGlqeuT_cIvGh243NK1Uf4a2JgNkQLdLvN950HQ,42646
+mirix/client/utils.py,sha256=xPAvR3slBQ5srOFLtYTqpL3PeD5n0YsxBRR1sLgKBdY,786
 mirix/helpers/__init__.py,sha256=2QXDAywb0wF3JNLC7GgMDJb_kQpwvlb6ZCnjuLmOfPw,78
 mirix/helpers/converters.py,sha256=WuYkra4c9-JRC2lRo7LmmFMH42qf3k6ox3OGOYlDQLk,13504
 mirix/helpers/datetime_helpers.py,sha256=7U5ZJkE0cLki4sG8ukIHZSAoFfQpLWQu2kFekETy6Zg,2633
@@ -9,33 +11,33 @@ mirix/helpers/json_helpers.py,sha256=2iq1Lyxa8Ofes38QHAOBZHE7I2jS65h-zkAXWJzpm3k
 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=kcG2VEJZQIFIapwvYXjcuvf8MU8jlIHoVswYUQra7Xc,15802
-mirix/schemas/block.py,sha256=F8G8pOO-dQuYQK7X7DNRdbRo5cTkfOp-vcDIora_VV8,5138
-mirix/schemas/cloud_file_mapping.py,sha256=RF3AvGWORULxxpykq4nL4PdZUJZqDah3k0JWmgAIxLo,970
+mirix/schemas/agent.py,sha256=SYiLURW1PYAHSXAaBOx8N8kaP_5Qt7ujWVxf2uA4BDM,15854
+mirix/schemas/block.py,sha256=NhqaSCOHKACHb77F77NhET-Sy8bZqSJJnenQaC7pBR4,5145
+mirix/schemas/cloud_file_mapping.py,sha256=i_5N41hNt0rEl-fvnGoPggLMbFdEk97wI71qQTjHKIg,977
 mirix/schemas/embedding_config.py,sha256=aHySSdkn5BeL_i6UQLuyDmjo40FCWvKW1Yfc01hQYak,4050
 mirix/schemas/enums.py,sha256=7Z4Wotyndp_eghhIvSdjFA3nMp_Vh_pcK1A_MrkBFRA,1357
 mirix/schemas/environment_variables.py,sha256=mSL5U2NYxJHO4fb2dF3YgPyINgYgelsSzMqQ6nsjzng,2573
-mirix/schemas/episodic_memory.py,sha256=SNguRpOY6tGm9yKUtlej6CGSgK1zdKeqeFUdwihzeUY,6316
+mirix/schemas/episodic_memory.py,sha256=KXr1hHXEqceF5wjMDpAN6MMkRthY-GBnFUcW_QDYUTg,6330
 mirix/schemas/file.py,sha256=J92aWus84jPoF71wRtKIvoAPSdIDt4McEM6yDWzHiqg,2064
 mirix/schemas/health.py,sha256=zT6mYovvD17iJRuu2rcaQQzbEEYrkwvAE9TB7iU824c,139
-mirix/schemas/knowledge_vault.py,sha256=S-q271_qomQwsmk5oukVRlf2DAbj2OjyUFimVmC6F-0,6302
+mirix/schemas/knowledge_vault.py,sha256=i9qOXp7D8tOgjpRlThDIBDBB4TvOtO7fpSCgaLZW3Lg,6316
 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/memory.py,sha256=ORWX_qxQBirsHN9vTlr3d_UYLiUB9tlP8DnIh_obIfA,11991
+mirix/schemas/message.py,sha256=_alOnt4BjdBw92Etg_i4vUhcWwdvD-54r5yuuydPBkc,53990
 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=q0csTjlwLKWhWKFdQpTtXqdYGGc9Nnapx5tTuvQVipg,7437
-mirix/schemas/organization.py,sha256=RRwoWq4Y1DzNsuKta_rNKI9K75MeuiS836mzrCh-akE,1109
-mirix/schemas/procedural_memory.py,sha256=tG5VOih3MsY04Hj1UtNJQPy2556AyODjz43frieeiro,5495
-mirix/schemas/providers.py,sha256=kZmC2-RTb8Hzhd7LXvwg0yyL2-d4JVFoYIZ7l04DPWs,30545
-mirix/schemas/resource_memory.py,sha256=Jm70q5hD_R_SnpG1SW46L1Qe8KBleNIXD1ni1W32epk,4893
+mirix/schemas/mirix_request.py,sha256=A2CwbKaIQcc1KchT0U3Ro82mBe6RerpDPFguDN_JFCo,1399
+mirix/schemas/mirix_response.py,sha256=w_hTYJs_cpBdx1mWfezzbiV49U4MWHKEfwFrYSMAVR0,7444
+mirix/schemas/organization.py,sha256=tHjE4CmboPJV8OMZ1BfGuFanMJKCED5rh1IBpNQhGTc,1120
+mirix/schemas/procedural_memory.py,sha256=XdX0tfX3LLRXngEP7c3Y2A6KEx7b9dY_ADKOrCdSqQg,5509
+mirix/schemas/providers.py,sha256=PNXCRGw2UqAwNgk3y2xqXa9bbH9ySVInLeU8qftTca4,30552
+mirix/schemas/resource_memory.py,sha256=0w6tRynshimSviPd7fkXG77KqCxEBeuHsnjpYXILTVQ,4907
 mirix/schemas/sandbox_config.py,sha256=X89bm-6QCE7n-ugQ7GyNfan6ZxKXU1UOwdWP0_PXnc4,4460
-mirix/schemas/semantic_memory.py,sha256=S8NWt-U6VjJhNpKkj_H4hFZuWKK-0KaPP4MOlKgI8v0,5979
+mirix/schemas/semantic_memory.py,sha256=vpDs9X1h6BRaAWnGkF0Il6oiYqosbmxahFh1OFb8WA0,5993
 mirix/schemas/source.py,sha256=9Qe-E15sDXcopd3UPW0-sC3lFHIjlV44t_lV7KxkCWY,3050
 mirix/schemas/step.py,sha256=msWn5uBdaR3dYmjTuCwmvq3Mm-kP0Ha8e3-2sfCvG1Y,1826
-mirix/schemas/tool.py,sha256=ryC1klcdVwKZUwThCZOjwzRHSLWBsag7OzEnZR2ZCBM,10480
+mirix/schemas/tool.py,sha256=ypkDYr8dikfyUU7tgnEF-CCam97TYYbTtaJ3K6PTebg,10487
 mirix/schemas/tool_rule.py,sha256=uwyrOX38Vtp-6V1CbHnvBI3DzmWpW2zy7UPtYPmewus,6877
 mirix/schemas/usage.py,sha256=QX7uIaK7-GgrZDMgeot-1ae2n-ol0yif3E7JsLF0Sxk,1166
 mirix/schemas/user.py,sha256=9ywAkPX__q3Ja4lfs-YIiCKQkFmE3-poIimt0owGz_E,2476
@@ -45,7 +47,7 @@ mirix/schemas/openai/chat_completion_response.py,sha256=1hF0Gaxvd03oBdKupz37qk_2
 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.3.dist-info/METADATA,sha256=bnyKOUUg7doG3KSypWG9aSYLxUoEs7pq1np__HdCi2w,10843
-jl_ecms_client-0.2.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-jl_ecms_client-0.2.3.dist-info/top_level.txt,sha256=hQ_Eh5KYzEGkhtEyWHdBYJ5iCom6_nlj0zQ6-v0WDIY,6
-jl_ecms_client-0.2.3.dist-info/RECORD,,
+jl_ecms_client-0.2.7.dist-info/METADATA,sha256=T7mM1EjJebIwmHGR_8Ma5ln6dKEUfaY1fRsGOUPFO2Q,10843
+jl_ecms_client-0.2.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+jl_ecms_client-0.2.7.dist-info/top_level.txt,sha256=hQ_Eh5KYzEGkhtEyWHdBYJ5iCom6_nlj0zQ6-v0WDIY,6
+jl_ecms_client-0.2.7.dist-info/RECORD,,

mirix/client/client.py

@@ -9,8 +9,6 @@ from urllib.parse import urlparse
 
 import requests
 
-import mirix.utils
-
 if TYPE_CHECKING:
     try:
         from composio import ActionType
@@ -34,12 +32,7 @@ from mirix.constants import (
 from mirix.functions.functions import parse_source_code
 from mirix.interface import QueuingInterface
 from mirix.orm.errors import NoResultFound
-from mirix.schemas.agent import (
-    AgentState,
-    AgentType,
-    CreateAgent,
-    CreateMetaAgent,
-)
+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
 
@@ -53,11 +46,7 @@ from mirix.schemas.environment_variables import (
 from mirix.schemas.file import FileMetadata
 from mirix.schemas.file import FileMetadata as PydanticFileMetadata
 from mirix.schemas.llm_config import LLMConfig
-from mirix.schemas.memory import (
-    ArchivalMemorySummary,
-    Memory,
-    RecallMemorySummary,
-)
+from mirix.schemas.memory import ArchivalMemorySummary, Memory, RecallMemorySummary
 from mirix.schemas.message import Message, MessageCreate
 from mirix.schemas.mirix_message_content import (
     CloudFileContent,
@@ -77,7 +66,8 @@ from mirix.schemas.sandbox_config import (
 )
 from mirix.schemas.tool import Tool, ToolCreate, ToolUpdate
 from mirix.schemas.tool_rule import BaseToolRule
-from mirix.schemas.user import User as PydanticUser, UserCreate
+from mirix.schemas.user import User as PydanticUser
+from mirix.schemas.user import UserCreate
 
 
 def create_client():
@@ -461,7 +451,7 @@ class LocalClient(AbstractClient):
         from mirix.server.server import SyncServer
 
         # set logging levels
-        mirix.utils.DEBUG = debug
+        # DEBUG flag not needed for LocalClient (server-side only)
         logging.getLogger().setLevel(logging.CRITICAL)
 
         # save default model config

mirix/client/constants.py

@@ -0,0 +1,60 @@
+"""
+Client-side constants - minimal set needed by schemas and client code.
+
+These are the constants required by the client package (schemas, helpers, client).
+The full server constants module (mirix.constants) imports from here and adds
+additional server-only constants.
+"""
+
+# 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
+}

mirix/client/utils.py

@@ -0,0 +1,34 @@
+"""
+Lightweight utilities for Mirix client package.
+
+Contains only essential utility functions needed by the client,
+without heavy dependencies.
+"""
+
+import json
+from datetime import datetime, timezone
+
+
+def get_utc_time() -> datetime:
+    """Get the current UTC time"""
+    return datetime.now(timezone.utc)
+
+
+def json_dumps(data, indent=2):
+    """
+    JSON serializer that handles datetime objects.
+
+    Args:
+        data: Data to serialize
+        indent: JSON indentation level
+
+    Returns:
+        str: JSON string
+    """
+
+    def safe_serializer(obj):
+        if isinstance(obj, datetime):
+            return obj.isoformat()
+        raise TypeError(f"Type {type(obj)} not serializable")
+
+    return json.dumps(data, indent=indent, default=safe_serializer, ensure_ascii=False)

mirix/schemas/agent.py

@@ -3,7 +3,7 @@ from typing import Any, Dict, List, Optional, Union
 
 from pydantic import BaseModel, Field, field_validator
 
-from mirix.constants import DEFAULT_EMBEDDING_CHUNK_SIZE
+from mirix.client.constants import DEFAULT_EMBEDDING_CHUNK_SIZE
 from mirix.helpers import ToolRulesSolver
 from mirix.schemas.block import CreateBlock
 from mirix.schemas.embedding_config import EmbeddingConfig
@@ -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/block.py

@@ -3,7 +3,7 @@ from typing import Optional
 from pydantic import BaseModel, Field, model_validator
 from typing_extensions import Self
 
-from mirix.constants import CORE_MEMORY_BLOCK_CHAR_LIMIT
+from mirix.client.constants import CORE_MEMORY_BLOCK_CHAR_LIMIT
 from mirix.schemas.mirix_base import MirixBase
 
 # block of the LLM context

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/episodic_memory.py

@@ -3,10 +3,10 @@ from typing import Any, Dict, List, Optional
 
 from pydantic import Field, field_validator
 
-from mirix.constants import MAX_EMBEDDING_DIM
+from mirix.client.utils import get_utc_time
+from mirix.client.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.constants import MAX_EMBEDDING_DIM
+from mirix.client.utils import get_utc_time
+from mirix.client.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/memory.py

@@ -7,7 +7,7 @@ from pydantic import BaseModel, Field
 if TYPE_CHECKING:
     from mirix.schemas.agent import AgentState
 
-from mirix.constants import CORE_MEMORY_BLOCK_CHAR_LIMIT
+from mirix.client.constants import CORE_MEMORY_BLOCK_CHAR_LIMIT
 from mirix.schemas.block import Block
 from mirix.schemas.message import Message
 from mirix.schemas.openai.chat_completion_request import Tool

mirix/schemas/message.py

@@ -10,7 +10,7 @@ from typing import Any, Dict, List, Literal, Optional, Union
 
 from pydantic import BaseModel, Field, field_validator
 
-from mirix.constants import (
+from mirix.client.constants import (
     DEFAULT_MESSAGE_TOOL,
     DEFAULT_MESSAGE_TOOL_KWARG,
     TOOL_CALL_ID_MAX_LEN,

mirix/schemas/mirix_request.py

@@ -2,7 +2,7 @@ from typing import List
 
 from pydantic import BaseModel, Field
 
-from mirix.constants import DEFAULT_MESSAGE_TOOL, DEFAULT_MESSAGE_TOOL_KWARG
+from mirix.client.constants import DEFAULT_MESSAGE_TOOL, DEFAULT_MESSAGE_TOOL_KWARG
 from mirix.schemas.message import MessageCreate
 
 

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.constants import MAX_EMBEDDING_DIM
+from mirix.client.utils import get_utc_time
+from mirix.client.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/providers.py

@@ -3,7 +3,7 @@ from typing import List, Optional
 
 from pydantic import Field, model_validator
 
-from mirix.constants import LLM_MAX_TOKENS, MIN_CONTEXT_WINDOW
+from mirix.client.constants import LLM_MAX_TOKENS, MIN_CONTEXT_WINDOW
 from mirix.llm_api.azure_openai import (
     get_azure_chat_completions_endpoint,
     get_azure_embeddings_endpoint,

mirix/schemas/resource_memory.py

@@ -3,10 +3,10 @@ from typing import Any, Dict, List, Optional
 
 from pydantic import Field, field_validator
 
-from mirix.constants import MAX_EMBEDDING_DIM
+from mirix.client.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/semantic_memory.py

@@ -3,10 +3,10 @@ from typing import Any, Dict, List, Optional
 
 from pydantic import Field, field_validator
 
-from mirix.constants import MAX_EMBEDDING_DIM
+from mirix.client.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):

mirix/schemas/tool.py

@@ -2,7 +2,7 @@ from typing import TYPE_CHECKING, Any, Dict, List, Optional
 
 from pydantic import Field, model_validator
 
-from mirix.constants import (
+from mirix.client.constants import (
     COMPOSIO_TOOL_TAG_NAME,
     FUNCTION_RETURN_CHAR_LIMIT,
     MIRIX_CORE_TOOL_MODULE_NAME,