camel-ai 0.2.65__py3-none-any.whl → 0.2.82__py3-none-any.whl

camel/prompts/video_description_prompt.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,7 +10,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 from typing import Any
 
 from camel.prompts.base import TextPrompt, TextPromptDict
@@ -28,7 +28,7 @@ class VideoDescriptionPromptTemplateDict(TextPromptDict):
     """
 
     ASSISTANT_PROMPT = TextPrompt(
-        """You are a master of video analysis. 
+        """You are a master of video analysis.
         Please provide a shot description of the content of the current video."""
     )
 

camel/responses/__init__.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,7 +10,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 from .agent_responses import ChatAgentResponse
 
 __all__ = [

camel/responses/agent_responses.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,7 +10,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 from typing import Any, Dict, List
 
 from pydantic import BaseModel, ConfigDict

camel/retrievers/__init__.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,7 +10,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # ruff: noqa: I001
 from .auto_retriever import AutoRetriever
 from .base import BaseRetriever

camel/retrievers/auto_retriever.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,7 +10,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 import re
 import uuid
 from typing import (
@@ -97,6 +97,7 @@ class AutoRetriever:
                     "URL (database url) and API key required for TiDB storage "
                     "are not provided. Format: "
                     "mysql+pymysql://<username>:<password>@<host>:4000/test"
+                    "You can get the database url from https://tidbcloud.com/console/clusters"
                 )
             return TiDBStorage(
                 vector_dim=self.embedding_model.get_output_dim(),

camel/retrievers/base.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,7 +10,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 from abc import ABC, abstractmethod
 from typing import Any, Callable
 

camel/retrievers/bm25_retriever.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,7 +10,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 from typing import Any, Dict, List
 
 from camel.loaders import UnstructuredIO

camel/retrievers/cohere_rerank_retriever.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,7 +10,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 import os
 from typing import Any, Dict, List, Optional
 

camel/retrievers/hybrid_retrival.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,7 +10,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 from typing import Any, Collection, Dict, List, Optional, Sequence, Union
 
 from camel.embeddings import BaseEmbedding

camel/retrievers/vector_retriever.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,7 +10,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 import os
 import warnings
 from io import IOBase

camel/runtimes/Dockerfile.multi-toolkit

@@ -0,0 +1,90 @@
+# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+
+# Multi-toolkit Docker image for CAMEL shared runtimes
+# Supports: BrowserToolkit, TerminalToolkit, CodeExecutionToolkit, and more
+#
+# Build (from repo root):
+#   docker build -f camel/runtimes/Dockerfile.multi-toolkit \
+#       -t camel-multi-toolkit:latest .
+#
+# Usage with DockerRuntime:
+#   runtime = (
+#       DockerRuntime("camel-multi-toolkit:latest")
+#       .add(BrowserToolkit().get_tools(), "camel.toolkits.BrowserToolkit")
+#       .add(TerminalToolkit().get_tools(), "camel.toolkits.TerminalToolkit")
+#       .add(CodeExecutionToolkit().get_tools(), "camel.toolkits.CodeExecutionToolkit")
+#       .build()
+#   )
+
+FROM python:3.10-slim
+
+# install system dependencies for various toolkits
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    # build tools (required for psutil, etc.)
+    gcc \
+    python3-dev \
+    # common tools
+    curl \
+    wget \
+    git \
+    bash \
+    # for Playwright browsers
+    libnss3 \
+    libnspr4 \
+    libatk1.0-0 \
+    libatk-bridge2.0-0 \
+    libcups2 \
+    libdrm2 \
+    libxkbcommon0 \
+    libxcomposite1 \
+    libxdamage1 \
+    libxfixes3 \
+    libxrandr2 \
+    libgbm1 \
+    libasound2 \
+    libpango-1.0-0 \
+    libcairo2 \
+    # additional dependencies
+    libdbus-1-3 \
+    libexpat1 \
+    libfontconfig1 \
+    libgcc-s1 \
+    libglib2.0-0 \
+    libgtk-3-0 \
+    libx11-6 \
+    libx11-xcb1 \
+    libxcb1 \
+    libxext6 \
+    && rm -rf /var/lib/apt/lists/*
+
+# copy local CAMEL source and install from source
+# this ensures we use local changes rather than PyPI version
+COPY . /app/camel
+WORKDIR /app/camel
+RUN pip install --no-cache-dir -e '.[all]'
+
+# install Playwright and chromium browser
+RUN pip install --no-cache-dir playwright \
+    && playwright install chromium \
+    && playwright install-deps chromium
+
+# set working directory for runtime operations
+WORKDIR /workspace
+
+# expose API port
+EXPOSE 8000
+
+# default command is sleep infinity (DockerRuntime will exec into it)
+CMD ["sleep", "infinity"]

camel/runtimes/__init__.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,7 +10,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 from .base import BaseRuntime
 from .configs import TaskConfig
 from .daytona_runtime import DaytonaRuntime

camel/runtimes/api.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,14 +10,16 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
+import asyncio
+import concurrent.futures
 import importlib
 import io
 import json
 import logging
 import os
 import sys
-from typing import Dict
+from typing import Any, Dict, List
 
 import uvicorn
 from fastapi import FastAPI, Request
@@ -25,8 +27,15 @@ from fastapi.responses import JSONResponse
 
 from camel.toolkits import BaseToolkit
 
+# thread pool for running sync tools that can't run inside async event loop
+# (e.g., Playwright sync API)
+_executor = concurrent.futures.ThreadPoolExecutor(max_workers=4)
+
 logger = logging.getLogger(__name__)
 
+# set environment variable to indicate we're running inside a CAMEL runtime
+os.environ["CAMEL_RUNTIME"] = "true"
+
 sys.path.append(os.getcwd())
 
 modules_functions = sys.argv[1:]
@@ -35,6 +44,22 @@ logger.info(f"Modules and functions: {modules_functions}")
 
 app = FastAPI()
 
+# global cache for toolkit instances to maintain state across calls
+_toolkit_instances: Dict[str, Any] = {}
+
+# track registered endpoints for health check
+_registered_endpoints: List[str] = []
+
+
+@app.get("/health")
+async def health_check():
+    r"""Health check endpoint that reports loaded toolkits and endpoints."""
+    return {
+        "status": "ok",
+        "toolkits": list(_toolkit_instances.keys()),
+        "endpoints": _registered_endpoints,
+    }
+
 
 @app.exception_handler(Exception)
 async def general_exception_handler(request: Request, exc: Exception):
@@ -49,6 +74,9 @@ async def general_exception_handler(request: Request, exc: Exception):
 
 for module_function in modules_functions:
     try:
+        # store original module_function as cache key before parsing
+        cache_key = module_function
+
         init_params = dict()
         if "{" in module_function:
             module_function, params = module_function.split("{")
@@ -62,36 +90,64 @@ for module_function in modules_functions:
         module = importlib.import_module(module_name)
         function = getattr(module, function_name)
         if isinstance(function, type) and issubclass(function, BaseToolkit):
-            function = function(**init_params).get_tools()
+            # use cached instance if available to maintain state across calls
+            if cache_key not in _toolkit_instances:
+                _toolkit_instances[cache_key] = function(**init_params)
+            function = _toolkit_instances[cache_key].get_tools()
 
         if not isinstance(function, list):
             function = [function]
 
         for func in function:
-
-            @app.post(f"/{func.get_function_name()}")
-            async def dynamic_function(data: Dict, func=func):
-                redirect_stdout = data.get('redirect_stdout', False)
-                if redirect_stdout:
-                    sys.stdout = io.StringIO()
-                response_data = func.func(*data['args'], **data['kwargs'])
-                if redirect_stdout:
-                    sys.stdout.seek(0)
-                    output = sys.stdout.read()
-                    sys.stdout = sys.__stdout__
+            endpoint_name = func.get_function_name()
+            _registered_endpoints.append(endpoint_name)
+
+            def make_endpoint(tool):
+                r"""Create endpoint with tool captured in closure."""
+
+                def run_tool(data: Dict):
+                    r"""Run tool in thread pool to avoid async event loop."""
+                    redirect_stdout = data.get('redirect_stdout', False)
+                    captured_output = None
+                    if redirect_stdout:
+                        captured_output = io.StringIO()
+                        old_stdout = sys.stdout
+                        sys.stdout = captured_output
+                    try:
+                        response_data = tool.func(
+                            *data['args'], **data['kwargs']
+                        )
+                    finally:
+                        if redirect_stdout:
+                            sys.stdout = old_stdout
+                    if redirect_stdout and captured_output is not None:
+                        captured_output.seek(0)
+                        output = captured_output.read()
+                        return {
+                            "output": json.dumps(
+                                response_data, ensure_ascii=False
+                            ),
+                            "stdout": output,
+                        }
                     return {
-                        "output": json.dumps(
-                            response_data, ensure_ascii=False
-                        ),
-                        "stdout": output,
+                        "output": json.dumps(response_data, ensure_ascii=False)
                     }
-                return {
-                    "output": json.dumps(response_data, ensure_ascii=False)
-                }
+
+                async def endpoint(data: Dict):
+                    # run in thread pool to support sync tools like Playwright
+                    loop = asyncio.get_running_loop()
+                    return await loop.run_in_executor(
+                        _executor, run_tool, data
+                    )
+
+                return endpoint
+
+            app.post(f"/{endpoint_name}")(make_endpoint(func))
 
     except (ImportError, AttributeError) as e:
         logger.error(f"Error importing {module_function}: {e}")
 
 
 if __name__ == "__main__":
-    uvicorn.run("__main__:app", host="0.0.0.0", port=8000, reload=True)
+    # reload=False to avoid conflicts with async toolkits (e.g., Playwright)
+    uvicorn.run(app, host="0.0.0.0", port=8000, reload=False)

camel/runtimes/base.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,7 +10,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 from abc import ABC, abstractmethod
 from typing import Any, List, Union
 

camel/runtimes/configs.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,7 +10,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 from typing import Dict, List, Optional, Union
 
 from pydantic import BaseModel
@@ -21,22 +21,22 @@ class TaskConfig(BaseModel):
 
     Arttributes:
         cmd (str or list): Command to be executed
-        stdout (bool): Attach to stdout. (default: :obj: `True`)
-        stderr (bool): Attach to stderr. (default: :obj: `True`)
-        stdin (bool): Attach to stdin. (default: :obj: `False`)
-        tty (bool): Allocate a pseudo-TTY. (default: :obj: `False`)
-        privileged (bool): Run as privileged. (default: :obj: `False`)
-        user (str): User to execute command as. (default: :obj: `""`)
+        stdout (bool): Attach to stdout. (default: :obj:`True`)
+        stderr (bool): Attach to stderr. (default: :obj:`True`)
+        stdin (bool): Attach to stdin. (default: :obj:`False`)
+        tty (bool): Allocate a pseudo-TTY. (default: :obj:`False`)
+        privileged (bool): Run as privileged. (default: :obj:`False`)
+        user (str): User to execute command as. (default: :obj:`""`)
         detach (bool): If true, detach from the exec command.
-            (default: :obj: `False`)
-        stream (bool): Stream response data. (default: :obj: `False`)
+            (default: :obj:`False`)
+        stream (bool): Stream response data. (default: :obj:`False`)
         socket (bool): Return the connection socket to allow custom
-            read/write operations. (default: :obj: `False`)
+            read/write operations. (default: :obj:`False`)
         environment (dict or list): A dictionary or a list of strings in
             the following format ``["PASSWORD=xxx"]`` or
-            ``{"PASSWORD": "xxx"}``. (default: :obj: `None`)
+            ``{"PASSWORD": "xxx"}``. (default: :obj:`None`)
         workdir (str): Path to working directory for this exec session.
-            (default: :obj: `None`)
+            (default: :obj:`None`)
         demux (bool): Return stdout and stderr separately. (default: :obj:
             `False`)
     """

camel/runtimes/daytona_runtime.py

@@ -1,4 +1,4 @@
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -10,13 +10,13 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# ========= Copyright 2023-2025 @ CAMEL-AI.org. All Rights Reserved. =========
 
 import inspect
 import json
 import os
 from functools import wraps
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Callable, Dict, List, Optional, Union
 
 from pydantic import BaseModel
 
@@ -34,13 +34,13 @@ class DaytonaRuntime(BaseRuntime):
     Args:
         api_key (Optional[str]): The Daytona API key for authentication. If not
             provided, it will try to use the DAYTONA_API_KEY environment
-            variable. (default: :obj: `None`)
+            variable. (default: :obj:`None`)
         api_url (Optional[str]): The URL of the Daytona server. If not
             provided, it will try to use the DAYTONA_API_URL environment
             variable. If none is provided, it will use "http://localhost:8000".
-            (default: :obj: `None`)
+            (default: :obj:`None`)
         language (Optional[str]): The programming language for the sandbox.
-            (default: :obj: `"python"`)
+            (default: :obj:`"python"`)
     """
 
     def __init__(
@@ -49,7 +49,7 @@ class DaytonaRuntime(BaseRuntime):
         api_url: Optional[str] = None,
         language: Optional[str] = "python",
     ):
-        from daytona_sdk import Daytona, DaytonaConfig
+        from daytona_sdk import Daytona, DaytonaConfig, Sandbox
 
         super().__init__()
         self.api_key = api_key or os.environ.get('DAYTONA_API_KEY')
@@ -57,7 +57,7 @@ class DaytonaRuntime(BaseRuntime):
         self.language = language
         self.config = DaytonaConfig(api_key=self.api_key, api_url=self.api_url)
         self.daytona = Daytona(self.config)
-        self.sandbox = None
+        self.sandbox: Optional[Sandbox] = None
         self.entrypoint: Dict[str, str] = dict()
 
     def build(self) -> "DaytonaRuntime":
@@ -66,10 +66,10 @@ class DaytonaRuntime(BaseRuntime):
         Returns:
             DaytonaRuntime: The current runtime.
         """
-        from daytona_sdk import CreateSandboxParams
+        from daytona_sdk import CreateSandboxBaseParams
 
         try:
-            params = CreateSandboxParams(language=self.language)
+            params = CreateSandboxBaseParams(language=self.language)
             self.sandbox = self.daytona.create(params)
             if self.sandbox is None:
                 raise RuntimeError("Failed to create sandbox.")
@@ -83,7 +83,7 @@ class DaytonaRuntime(BaseRuntime):
         r"""Clean up the sandbox when exiting."""
         if self.sandbox:
             try:
-                self.daytona.remove(self.sandbox)
+                self.daytona.delete(self.sandbox)
                 logger.info(f"Sandbox {self.sandbox.id} removed")
                 self.sandbox = None
             except Exception as e:
@@ -102,7 +102,7 @@ class DaytonaRuntime(BaseRuntime):
                 list of functions to add.
             entrypoint (str): The entrypoint for the function.
             arguments (Optional[Dict[str, Any]]): The arguments for the
-                function. (default: :obj: `None`)
+                function. (default: :obj:`None`)
 
         Returns:
             DaytonaRuntime: The current runtime.
@@ -112,7 +112,7 @@ class DaytonaRuntime(BaseRuntime):
         if arguments is not None:
             entrypoint += json.dumps(arguments, ensure_ascii=False)
 
-        def make_wrapper(inner_func, func_name, func_code):
+        def make_wrapper(inner_func: Callable, func_name: str, func_code: str):
             r"""Creates a wrapper for a function to execute it in the
             Daytona sandbox.
 
@@ -208,12 +208,11 @@ class DaytonaRuntime(BaseRuntime):
             RuntimeError: If the sandbox is not initialized.
         """
         if self.sandbox is None:
-            raise RuntimeError("Failed to create sandbox.")
-        info = self.sandbox.info()
+            raise RuntimeError("Sandbox not initialized.")
         return (
-            f"Sandbox {info.name}:\n"
-            f"State: {info.state}\n"
-            f"Resources: {info.resources.cpu} CPU, {info.resources.memory} RAM"
+            f"Sandbox {self.sandbox.id}:\n"
+            f"State: {self.sandbox.state}\n"
+            f"Resources: {self.sandbox.cpu} CPU, {self.sandbox.memory} RAM"
         )
 
     def __del__(self):