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

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.client.constants import DEFAULT_EMBEDDING_CHUNK_SIZE
+from mirix.constants import DEFAULT_EMBEDDING_CHUNK_SIZE
 from mirix.helpers import ToolRulesSolver
 from mirix.schemas.block import CreateBlock
 from mirix.schemas.embedding_config import EmbeddingConfig

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.client.constants import CORE_MEMORY_BLOCK_CHAR_LIMIT
+from mirix.constants import CORE_MEMORY_BLOCK_CHAR_LIMIT
 from mirix.schemas.mirix_base import MirixBase
 
 # block of the LLM context

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

@@ -4,7 +4,7 @@ from typing import Any, Dict, List, Optional
 from pydantic import Field, field_validator
 
 from mirix.client.utils import get_utc_time
-from mirix.client.constants import MAX_EMBEDDING_DIM
+from mirix.constants import MAX_EMBEDDING_DIM
 from mirix.schemas.embedding_config import EmbeddingConfig
 from mirix.schemas.mirix_base import MirixBase
 

mirix/schemas/knowledge_vault.py

@@ -4,7 +4,7 @@ from typing import Any, Dict, List, Optional
 from pydantic import Field, field_validator
 
 from mirix.client.utils import get_utc_time
-from mirix.client.constants import MAX_EMBEDDING_DIM
+from mirix.constants import MAX_EMBEDDING_DIM
 from mirix.schemas.embedding_config import EmbeddingConfig
 from mirix.schemas.mirix_base import 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.client.constants import CORE_MEMORY_BLOCK_CHAR_LIMIT
+from mirix.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.client.constants import (
+from mirix.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.client.constants import DEFAULT_MESSAGE_TOOL, DEFAULT_MESSAGE_TOOL_KWARG
+from mirix.constants import DEFAULT_MESSAGE_TOOL, DEFAULT_MESSAGE_TOOL_KWARG
 from mirix.schemas.message import MessageCreate
 
 

mirix/schemas/procedural_memory.py

@@ -4,7 +4,7 @@ from typing import Any, Dict, List, Optional
 from pydantic import Field, field_validator
 
 from mirix.client.utils import get_utc_time
-from mirix.client.constants import MAX_EMBEDDING_DIM
+from mirix.constants import MAX_EMBEDDING_DIM
 from mirix.schemas.embedding_config import EmbeddingConfig
 from mirix.schemas.mirix_base import MirixBase
 

mirix/schemas/providers.py

@@ -3,7 +3,7 @@ from typing import List, Optional
 
 from pydantic import Field, model_validator
 
-from mirix.client.constants import LLM_MAX_TOKENS, MIN_CONTEXT_WINDOW
+from mirix.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,7 +3,7 @@ from typing import Any, Dict, List, Optional
 
 from pydantic import Field, field_validator
 
-from mirix.client.constants import MAX_EMBEDDING_DIM
+from mirix.constants import MAX_EMBEDDING_DIM
 from mirix.schemas.embedding_config import EmbeddingConfig
 from mirix.schemas.mirix_base import MirixBase
 from mirix.client.utils import get_utc_time

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

@@ -3,7 +3,7 @@ from typing import Any, Dict, List, Optional
 
 from pydantic import Field, field_validator
 
-from mirix.client.constants import MAX_EMBEDDING_DIM
+from mirix.constants import MAX_EMBEDDING_DIM
 from mirix.schemas.embedding_config import EmbeddingConfig
 from mirix.schemas.mirix_base import MirixBase
 from mirix.client.utils import get_utc_time

mirix/schemas/tool.py

@@ -1,241 +1,241 @@
-from typing import TYPE_CHECKING, Any, Dict, List, Optional
-
-from pydantic import Field, model_validator
-
-from mirix.client.constants import (
-    COMPOSIO_TOOL_TAG_NAME,
-    FUNCTION_RETURN_CHAR_LIMIT,
-    MIRIX_CORE_TOOL_MODULE_NAME,
-    MIRIX_EXTRA_TOOL_MODULE_NAME,
-    MIRIX_MEMORY_TOOL_MODULE_NAME,
-)
-from mirix.functions.functions import (
-    derive_openai_json_schema,
-    get_json_schema_from_module,
-)
-from mirix.functions.helpers import generate_langchain_tool_wrapper
-from mirix.functions.schema_generator import generate_schema_from_args_schema_v2
-from mirix.orm.enums import ToolType
-from mirix.schemas.mirix_base import MirixBase
-
-if TYPE_CHECKING:
-    try:
-        from langchain_core.tools import BaseTool as LangChainBaseTool
-    except ImportError:
-        LangChainBaseTool = Any  # type: ignore
-
-
-class BaseTool(MirixBase):
-    __id_prefix__ = "tool"
-
-
-class Tool(BaseTool):
-    """
-    Representation of a tool, which is a function that can be called by the agent.
-
-    Parameters:
-        id (str): The unique identifier of the tool.
-        name (str): The name of the function.
-        tags (List[str]): Metadata tags.
-        source_code (str): The source code of the function.
-        json_schema (Dict): The JSON schema of the function.
-
-    """
-
-    id: str = BaseTool.generate_id_field()
-    tool_type: ToolType = Field(ToolType.CUSTOM, description="The type of the tool.")
-    description: Optional[str] = Field(None, description="The description of the tool.")
-    source_type: Optional[str] = Field(None, description="The type of the source code.")
-    organization_id: Optional[str] = Field(
-        None,
-        description="The unique identifier of the organization associated with the tool.",
-    )
-    name: Optional[str] = Field(None, description="The name of the function.")
-    tags: List[str] = Field([], description="Metadata tags.")
-
-    # code
-    source_code: Optional[str] = Field(
-        None, description="The source code of the function."
-    )
-    json_schema: Optional[Dict] = Field(
-        None, description="The JSON schema of the function."
-    )
-
-    # tool configuration
-    return_char_limit: int = Field(
-        FUNCTION_RETURN_CHAR_LIMIT,
-        description="The maximum number of characters in the response.",
-    )
-
-    # metadata fields
-    created_by_id: Optional[str] = Field(
-        None, description="The id of the user that made this Tool."
-    )
-    last_updated_by_id: Optional[str] = Field(
-        None, description="The id of the user that made this Tool."
-    )
-
-    @model_validator(mode="after")
-    def populate_missing_fields(self):
-        """
-        Populate missing fields: name, description, and json_schema.
-        """
-
-        if self.tool_type == ToolType.CUSTOM:
-            # If it's a custom tool, we need to ensure source_code is present
-            if not self.source_code:
-                raise ValueError(
-                    f"Custom tool with id={self.id} is missing source_code field."
-                )
-
-            # Always derive json_schema for freshest possible json_schema
-            # TODO: Instead of checking the tag, we should having `COMPOSIO` as a specific ToolType
-            # TODO: We skip this for Composio bc composio json schemas are derived differently
-            if COMPOSIO_TOOL_TAG_NAME not in self.tags:
-                self.json_schema = derive_openai_json_schema(
-                    source_code=self.source_code
-                )
-        elif self.tool_type in {ToolType.MIRIX_CORE}:
-            # If it's mirix core tool, we generate the json_schema on the fly here
-            self.json_schema = get_json_schema_from_module(
-                module_name=MIRIX_CORE_TOOL_MODULE_NAME, function_name=self.name
-            )
-        elif self.tool_type in {ToolType.MIRIX_MEMORY_CORE}:
-            self.json_schema = get_json_schema_from_module(
-                module_name=MIRIX_MEMORY_TOOL_MODULE_NAME, function_name=self.name
-            )
-        elif self.tool_type in {ToolType.MIRIX_EXTRA}:
-            self.json_schema = get_json_schema_from_module(
-                module_name=MIRIX_EXTRA_TOOL_MODULE_NAME, function_name=self.name
-            )
-        elif self.tool_type in {ToolType.MIRIX_MCP}:
-            # MCP tools have their json_schema already provided by MCP tool registry
-            # Skip validation since these are auto-generated tools
-            if not self.json_schema:
-                raise ValueError(f"MCP tool {self.name} is missing json_schema field")
-
-        # Derive name from the JSON schema if not provided
-        if not self.name:
-            # TODO: This in theory could error, but name should always be on json_schema
-            # TODO: Make JSON schema a typed pydantic object
-            self.name = self.json_schema.get("name")
-
-        # Derive description from the JSON schema if not provided
-        if not self.description:
-            # TODO: This in theory could error, but description should always be on json_schema
-            # TODO: Make JSON schema a typed pydantic object
-            self.description = self.json_schema.get("description")
-
-        return self
-
-
-class ToolCreate(MirixBase):
-    name: Optional[str] = Field(
-        None,
-        description="The name of the function (auto-generated from source_code if not provided).",
-    )
-    description: Optional[str] = Field(None, description="The description of the tool.")
-    tags: List[str] = Field([], description="Metadata tags.")
-    source_code: str = Field(..., description="The source code of the function.")
-    source_type: str = Field("python", description="The source type of the function.")
-    json_schema: Optional[Dict] = Field(
-        None,
-        description="The JSON schema of the function (auto-generated from source_code if not provided)",
-    )
-    return_char_limit: int = Field(
-        FUNCTION_RETURN_CHAR_LIMIT,
-        description="The maximum number of characters in the response.",
-    )
-
-    @classmethod
-    def from_langchain(
-        cls,
-        langchain_tool: "LangChainBaseTool",
-        additional_imports_module_attr_map: dict[str, str] = None,
-    ) -> "ToolCreate":
-        """
-        Class method to create an instance of Tool from a Langchain tool (must be from langchain_community.tools).
-
-        Args:
-            langchain_tool (LangChainBaseTool): An instance of a LangChain BaseTool (BaseTool from LangChain)
-            additional_imports_module_attr_map (dict[str, str]): A mapping of module names to attribute name. This is used internally to import all the required classes for the langchain tool. For example, you would pass in `{"langchain_community.utilities": "WikipediaAPIWrapper"}` for `from langchain_community.tools import WikipediaQueryRun`. NOTE: You do NOT need to specify the tool import here, that is done automatically for you.
-
-        Returns:
-            Tool: A Mirix Tool initialized with attributes derived from the provided LangChain BaseTool object.
-        """
-        description = langchain_tool.description
-        source_type = "python"
-        tags = ["langchain"]
-        # NOTE: langchain tools may come from different packages
-        wrapper_func_name, wrapper_function_str = generate_langchain_tool_wrapper(
-            langchain_tool, additional_imports_module_attr_map
-        )
-        json_schema = generate_schema_from_args_schema_v2(
-            langchain_tool.args_schema, name=wrapper_func_name, description=description
-        )
-
-        return cls(
-            name=wrapper_func_name,
-            description=description,
-            source_type=source_type,
-            tags=tags,
-            source_code=wrapper_function_str,
-            json_schema=json_schema,
-        )
-
-    @classmethod
-    def load_default_langchain_tools(cls) -> List["ToolCreate"]:
-        # For now, we only support wikipedia tool
-        from langchain_community.tools import WikipediaQueryRun
-        from langchain_community.utilities import WikipediaAPIWrapper
-
-        wikipedia_tool = ToolCreate.from_langchain(
-            WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper()),
-            {"langchain_community.utilities": "WikipediaAPIWrapper"},
-        )
-
-        return [wikipedia_tool]
-
-    @classmethod
-    def load_default_composio_tools(cls) -> List["ToolCreate"]:
-        pass
-
-        # TODO: Disable composio tools for now
-        # TODO: Naming is causing issues
-        # calculator = ToolCreate.from_composio(action_name=Action.MATHEMATICAL_CALCULATOR.name)
-        # serp_news = ToolCreate.from_composio(action_name=Action.SERPAPI_NEWS_SEARCH.name)
-        # serp_google_search = ToolCreate.from_composio(action_name=Action.SERPAPI_SEARCH.name)
-        # serp_google_maps = ToolCreate.from_composio(action_name=Action.SERPAPI_GOOGLE_MAPS_SEARCH.name)
-
-        return []
-
-
-class ToolUpdate(MirixBase):
-    description: Optional[str] = Field(None, description="The description of the tool.")
-    name: Optional[str] = Field(None, description="The name of the function.")
-    tags: Optional[List[str]] = Field(None, description="Metadata tags.")
-    source_code: Optional[str] = Field(
-        None, description="The source code of the function."
-    )
-    source_type: Optional[str] = Field(None, description="The type of the source code.")
-    json_schema: Optional[Dict] = Field(
-        None,
-        description="The JSON schema of the function (auto-generated from source_code if not provided)",
-    )
-    return_char_limit: Optional[int] = Field(
-        None, description="The maximum number of characters in the response."
-    )
-
-    class Config:
-        extra = "ignore"  # Allows extra fields without validation errors
-        # TODO: Remove this, and clean usage of ToolUpdate everywhere else
-
-
-class ToolRunFromSource(MirixBase):
-    source_code: str = Field(..., description="The source code of the function.")
-    args: Dict[str, Any] = Field(..., description="The arguments to pass to the tool.")
-    env_vars: Dict[str, str] = Field(
-        None, description="The environment variables to pass to the tool."
-    )
-    name: Optional[str] = Field(None, description="The name of the tool to run.")
-    source_type: Optional[str] = Field(None, description="The type of the source code.")
+from typing import TYPE_CHECKING, Any, Dict, List, Optional
+
+from pydantic import Field, model_validator
+
+from mirix.constants import (
+    COMPOSIO_TOOL_TAG_NAME,
+    FUNCTION_RETURN_CHAR_LIMIT,
+    MIRIX_CORE_TOOL_MODULE_NAME,
+    MIRIX_EXTRA_TOOL_MODULE_NAME,
+    MIRIX_MEMORY_TOOL_MODULE_NAME,
+)
+from mirix.functions.functions import (
+    derive_openai_json_schema,
+    get_json_schema_from_module,
+)
+from mirix.functions.helpers import generate_langchain_tool_wrapper
+from mirix.functions.schema_generator import generate_schema_from_args_schema_v2
+from mirix.schemas.enums import ToolType
+from mirix.schemas.mirix_base import MirixBase
+
+if TYPE_CHECKING:
+    try:
+        from langchain_core.tools import BaseTool as LangChainBaseTool
+    except ImportError:
+        LangChainBaseTool = Any  # type: ignore
+
+
+class BaseTool(MirixBase):
+    __id_prefix__ = "tool"
+
+
+class Tool(BaseTool):
+    """
+    Representation of a tool, which is a function that can be called by the agent.
+
+    Parameters:
+        id (str): The unique identifier of the tool.
+        name (str): The name of the function.
+        tags (List[str]): Metadata tags.
+        source_code (str): The source code of the function.
+        json_schema (Dict): The JSON schema of the function.
+
+    """
+
+    id: str = BaseTool.generate_id_field()
+    tool_type: ToolType = Field(ToolType.CUSTOM, description="The type of the tool.")
+    description: Optional[str] = Field(None, description="The description of the tool.")
+    source_type: Optional[str] = Field(None, description="The type of the source code.")
+    organization_id: Optional[str] = Field(
+        None,
+        description="The unique identifier of the organization associated with the tool.",
+    )
+    name: Optional[str] = Field(None, description="The name of the function.")
+    tags: List[str] = Field([], description="Metadata tags.")
+
+    # code
+    source_code: Optional[str] = Field(
+        None, description="The source code of the function."
+    )
+    json_schema: Optional[Dict] = Field(
+        None, description="The JSON schema of the function."
+    )
+
+    # tool configuration
+    return_char_limit: int = Field(
+        FUNCTION_RETURN_CHAR_LIMIT,
+        description="The maximum number of characters in the response.",
+    )
+
+    # metadata fields
+    created_by_id: Optional[str] = Field(
+        None, description="The id of the user that made this Tool."
+    )
+    last_updated_by_id: Optional[str] = Field(
+        None, description="The id of the user that made this Tool."
+    )
+
+    @model_validator(mode="after")
+    def populate_missing_fields(self):
+        """
+        Populate missing fields: name, description, and json_schema.
+        """
+
+        if self.tool_type == ToolType.CUSTOM:
+            # If it's a custom tool, we need to ensure source_code is present
+            if not self.source_code:
+                raise ValueError(
+                    f"Custom tool with id={self.id} is missing source_code field."
+                )
+
+            # Always derive json_schema for freshest possible json_schema
+            # TODO: Instead of checking the tag, we should having `COMPOSIO` as a specific ToolType
+            # TODO: We skip this for Composio bc composio json schemas are derived differently
+            if COMPOSIO_TOOL_TAG_NAME not in self.tags:
+                self.json_schema = derive_openai_json_schema(
+                    source_code=self.source_code
+                )
+        elif self.tool_type in {ToolType.MIRIX_CORE}:
+            # If it's mirix core tool, we generate the json_schema on the fly here
+            self.json_schema = get_json_schema_from_module(
+                module_name=MIRIX_CORE_TOOL_MODULE_NAME, function_name=self.name
+            )
+        elif self.tool_type in {ToolType.MIRIX_MEMORY_CORE}:
+            self.json_schema = get_json_schema_from_module(
+                module_name=MIRIX_MEMORY_TOOL_MODULE_NAME, function_name=self.name
+            )
+        elif self.tool_type in {ToolType.MIRIX_EXTRA}:
+            self.json_schema = get_json_schema_from_module(
+                module_name=MIRIX_EXTRA_TOOL_MODULE_NAME, function_name=self.name
+            )
+        elif self.tool_type in {ToolType.MIRIX_MCP}:
+            # MCP tools have their json_schema already provided by MCP tool registry
+            # Skip validation since these are auto-generated tools
+            if not self.json_schema:
+                raise ValueError(f"MCP tool {self.name} is missing json_schema field")
+
+        # Derive name from the JSON schema if not provided
+        if not self.name:
+            # TODO: This in theory could error, but name should always be on json_schema
+            # TODO: Make JSON schema a typed pydantic object
+            self.name = self.json_schema.get("name")
+
+        # Derive description from the JSON schema if not provided
+        if not self.description:
+            # TODO: This in theory could error, but description should always be on json_schema
+            # TODO: Make JSON schema a typed pydantic object
+            self.description = self.json_schema.get("description")
+
+        return self
+
+
+class ToolCreate(MirixBase):
+    name: Optional[str] = Field(
+        None,
+        description="The name of the function (auto-generated from source_code if not provided).",
+    )
+    description: Optional[str] = Field(None, description="The description of the tool.")
+    tags: List[str] = Field([], description="Metadata tags.")
+    source_code: str = Field(..., description="The source code of the function.")
+    source_type: str = Field("python", description="The source type of the function.")
+    json_schema: Optional[Dict] = Field(
+        None,
+        description="The JSON schema of the function (auto-generated from source_code if not provided)",
+    )
+    return_char_limit: int = Field(
+        FUNCTION_RETURN_CHAR_LIMIT,
+        description="The maximum number of characters in the response.",
+    )
+
+    @classmethod
+    def from_langchain(
+        cls,
+        langchain_tool: "LangChainBaseTool",
+        additional_imports_module_attr_map: dict[str, str] = None,
+    ) -> "ToolCreate":
+        """
+        Class method to create an instance of Tool from a Langchain tool (must be from langchain_community.tools).
+
+        Args:
+            langchain_tool (LangChainBaseTool): An instance of a LangChain BaseTool (BaseTool from LangChain)
+            additional_imports_module_attr_map (dict[str, str]): A mapping of module names to attribute name. This is used internally to import all the required classes for the langchain tool. For example, you would pass in `{"langchain_community.utilities": "WikipediaAPIWrapper"}` for `from langchain_community.tools import WikipediaQueryRun`. NOTE: You do NOT need to specify the tool import here, that is done automatically for you.
+
+        Returns:
+            Tool: A Mirix Tool initialized with attributes derived from the provided LangChain BaseTool object.
+        """
+        description = langchain_tool.description
+        source_type = "python"
+        tags = ["langchain"]
+        # NOTE: langchain tools may come from different packages
+        wrapper_func_name, wrapper_function_str = generate_langchain_tool_wrapper(
+            langchain_tool, additional_imports_module_attr_map
+        )
+        json_schema = generate_schema_from_args_schema_v2(
+            langchain_tool.args_schema, name=wrapper_func_name, description=description
+        )
+
+        return cls(
+            name=wrapper_func_name,
+            description=description,
+            source_type=source_type,
+            tags=tags,
+            source_code=wrapper_function_str,
+            json_schema=json_schema,
+        )
+
+    @classmethod
+    def load_default_langchain_tools(cls) -> List["ToolCreate"]:
+        # For now, we only support wikipedia tool
+        from langchain_community.tools import WikipediaQueryRun
+        from langchain_community.utilities import WikipediaAPIWrapper
+
+        wikipedia_tool = ToolCreate.from_langchain(
+            WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper()),
+            {"langchain_community.utilities": "WikipediaAPIWrapper"},
+        )
+
+        return [wikipedia_tool]
+
+    @classmethod
+    def load_default_composio_tools(cls) -> List["ToolCreate"]:
+        pass
+
+        # TODO: Disable composio tools for now
+        # TODO: Naming is causing issues
+        # calculator = ToolCreate.from_composio(action_name=Action.MATHEMATICAL_CALCULATOR.name)
+        # serp_news = ToolCreate.from_composio(action_name=Action.SERPAPI_NEWS_SEARCH.name)
+        # serp_google_search = ToolCreate.from_composio(action_name=Action.SERPAPI_SEARCH.name)
+        # serp_google_maps = ToolCreate.from_composio(action_name=Action.SERPAPI_GOOGLE_MAPS_SEARCH.name)
+
+        return []
+
+
+class ToolUpdate(MirixBase):
+    description: Optional[str] = Field(None, description="The description of the tool.")
+    name: Optional[str] = Field(None, description="The name of the function.")
+    tags: Optional[List[str]] = Field(None, description="Metadata tags.")
+    source_code: Optional[str] = Field(
+        None, description="The source code of the function."
+    )
+    source_type: Optional[str] = Field(None, description="The type of the source code.")
+    json_schema: Optional[Dict] = Field(
+        None,
+        description="The JSON schema of the function (auto-generated from source_code if not provided)",
+    )
+    return_char_limit: Optional[int] = Field(
+        None, description="The maximum number of characters in the response."
+    )
+
+    class Config:
+        extra = "ignore"  # Allows extra fields without validation errors
+        # TODO: Remove this, and clean usage of ToolUpdate everywhere else
+
+
+class ToolRunFromSource(MirixBase):
+    source_code: str = Field(..., description="The source code of the function.")
+    args: Dict[str, Any] = Field(..., description="The arguments to pass to the tool.")
+    env_vars: Dict[str, str] = Field(
+        None, description="The environment variables to pass to the tool."
+    )
+    name: Optional[str] = Field(None, description="The name of the tool to run.")
+    source_type: Optional[str] = Field(None, description="The type of the source code.")