camel-ai 0.2.51__tar.gz → 0.2.52__tar.gz

{camel_ai-0.2.51 → camel_ai-0.2.52}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: camel-ai
-Version: 0.2.51
+Version: 0.2.52
 Summary: Communicative Agents for AI Society Study
 Project-URL: Homepage, https://www.camel-ai.org/
 Project-URL: Repository, https://github.com/camel-ai/camel
@@ -36,6 +36,7 @@ Requires-Dist: dappier<0.4,>=0.3.3; extra == 'all'
 Requires-Dist: datacommons-pandas<0.0.4,>=0.0.3; extra == 'all'
 Requires-Dist: datacommons<2,>=1.4.3; extra == 'all'
 Requires-Dist: datasets<4,>=3; extra == 'all'
+Requires-Dist: daytona-sdk==0.14.0; extra == 'all'
 Requires-Dist: diffusers<0.26,>=0.25.0; extra == 'all'
 Requires-Dist: discord-py<3,>=2.3.2; extra == 'all'
 Requires-Dist: docker<8,>=7.1.0; extra == 'all'
@@ -170,6 +171,7 @@ Requires-Dist: types-tqdm<5,>=4.66.0; extra == 'dev'
 Requires-Dist: uv==0.6.5; extra == 'dev'
 Provides-Extra: dev-tools
 Requires-Dist: agentops<0.4,>=0.3.21; extra == 'dev-tools'
+Requires-Dist: daytona-sdk==0.14.0; extra == 'dev-tools'
 Requires-Dist: docker<8,>=7.1.0; extra == 'dev-tools'
 Requires-Dist: e2b-code-interpreter<2,>=1.0.3; extra == 'dev-tools'
 Requires-Dist: ipykernel<7,>=6.0.0; extra == 'dev-tools'

{camel_ai-0.2.51 → camel_ai-0.2.52}/camel/__init__.py

@@ -14,7 +14,7 @@
 
 from camel.logger import disable_logging, enable_logging, set_log_level
 
-__version__ = '0.2.51'
+__version__ = '0.2.52'
 
 __all__ = [
     '__version__',

{camel_ai-0.2.51 → camel_ai-0.2.52}/camel/runtime/__init__.py

@@ -13,6 +13,7 @@
 # ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
 from .base import BaseRuntime
 from .configs import TaskConfig
+from .daytona_runtime import DaytonaRuntime
 from .docker_runtime import DockerRuntime
 from .llm_guard_runtime import LLMGuardRuntime
 from .remote_http_runtime import RemoteHttpRuntime
@@ -28,4 +29,5 @@ __all__ = [
     "LLMGuardRuntime",
     "TaskConfig",
     "UbuntuDockerRuntime",
+    "DaytonaRuntime",
 ]

camel_ai-0.2.52/camel/runtime/daytona_runtime.py

@@ -0,0 +1,248 @@
+# ========= 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. =========
+
+import inspect
+import json
+import os
+from functools import wraps
+from typing import Any, Dict, List, Optional, Union
+
+from pydantic import BaseModel
+
+from camel.logger import get_logger
+from camel.runtime import BaseRuntime
+from camel.toolkits.function_tool import FunctionTool
+
+logger = get_logger(__name__)
+
+
+class DaytonaRuntime(BaseRuntime):
+    r"""A runtime that executes functions in a Daytona sandbox environment.
+    Requires the Daytona server to be running and an API key configured.
+
+    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`)
+        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`)
+        language (Optional[str]): The programming language for the sandbox.
+            (default: :obj: `"python"`)
+    """
+
+    def __init__(
+        self,
+        api_key: Optional[str] = None,
+        api_url: Optional[str] = None,
+        language: Optional[str] = "python",
+    ):
+        from daytona_sdk import Daytona, DaytonaConfig
+
+        super().__init__()
+        self.api_key = api_key or os.environ.get('DAYTONA_API_KEY')
+        self.api_url = api_url or os.environ.get('DAYTONA_API_URL')
+        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.entrypoint: Dict[str, str] = dict()
+
+    def build(self) -> "DaytonaRuntime":
+        r"""Create and start a Daytona sandbox.
+
+        Returns:
+            DaytonaRuntime: The current runtime.
+        """
+        from daytona_sdk import CreateSandboxParams
+
+        try:
+            params = CreateSandboxParams(language=self.language)
+            self.sandbox = self.daytona.create(params)
+            if self.sandbox is None:
+                raise RuntimeError("Failed to create sandbox.")
+            logger.info(f"Sandbox created with ID: {self.sandbox.id}")
+        except Exception as e:
+            logger.error(f"Failed to create sandbox: {e!s}")
+            raise RuntimeError(f"Daytona sandbox creation failed: {e!s}")
+        return self
+
+    def _cleanup(self):
+        r"""Clean up the sandbox when exiting."""
+        if self.sandbox:
+            try:
+                self.daytona.remove(self.sandbox)
+                logger.info(f"Sandbox {self.sandbox.id} removed")
+                self.sandbox = None
+            except Exception as e:
+                logger.error(f"Failed to remove sandbox: {e!s}")
+
+    def add(
+        self,
+        funcs: Union[FunctionTool, List[FunctionTool]],
+        entrypoint: str,
+        arguments: Optional[Dict[str, Any]] = None,
+    ) -> "DaytonaRuntime":
+        r"""Add a function or list of functions to the runtime.
+
+        Args:
+            funcs (Union[FunctionTool, List[FunctionTool]]): The function or
+                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`)
+
+        Returns:
+            DaytonaRuntime: The current runtime.
+        """
+        if not isinstance(funcs, list):
+            funcs = [funcs]
+        if arguments is not None:
+            entrypoint += json.dumps(arguments, ensure_ascii=False)
+
+        def make_wrapper(inner_func, func_name, func_code):
+            r"""Creates a wrapper for a function to execute it in the
+            Daytona sandbox.
+
+            Args:
+                inner_func (Callable): The function to wrap.
+                func_name (str): The name of the function.
+                func_code (str): The source code of the function.
+
+            Returns:
+                Callable: A wrapped function that executes in the sandbox.
+            """
+
+            @wraps(inner_func)
+            def wrapper(*args, **kwargs):
+                if not self.sandbox:
+                    raise RuntimeError(
+                        "Sandbox not initialized. Call build() first."
+                    )
+
+                try:
+                    for key, value in kwargs.items():
+                        if isinstance(value, BaseModel):
+                            kwargs[key] = value.model_dump()
+                    args_str = json.dumps(args, ensure_ascii=True)
+                    kwargs_str = json.dumps(kwargs, ensure_ascii=True)
+                except (TypeError, ValueError) as e:
+                    logger.error(f"Failed to serialize arguments: {e!s}")
+                    return {"error": f"Argument serialization failed: {e!s}"}
+
+                # Upload function code to the sandbox
+                script_path = f"/home/daytona/{func_name}.py"
+                try:
+                    self.sandbox.fs.upload_file(
+                        script_path, func_code.encode()
+                    )
+                except Exception as e:
+                    logger.error(
+                        f"Failed to upload function {func_name}: {e!s}"
+                    )
+                    return {"error": f"Upload failed: {e!s}"}
+
+                exec_code = (
+                    f"import sys\n"
+                    f"sys.path.append('/home/daytona')\n"
+                    f"import json\n"
+                    f"from {func_name} import {func_name}\n"
+                    f"args = json.loads('{args_str}')\n"
+                    f"kwargs = json.loads('{kwargs_str}')\n"
+                    f"result = {func_name}(*args, **kwargs)\n"
+                    f"print(json.dumps(result) if result is not "
+                    f"None else 'null')"
+                )
+
+                # Execute the function in the sandbox
+                try:
+                    response = self.sandbox.process.code_run(exec_code)
+                    return (
+                        json.loads(response.result)
+                        if response.result
+                        else None
+                    )
+                except json.JSONDecodeError as e:
+                    logger.error(
+                        f"Failed to decode JSON response for {func_name}"
+                    )
+                    return {"error": f"JSON decoding failed: {e!s}"}
+                except Exception as e:
+                    logger.error(
+                        f"Failed to execute function {func_name}: {e!s}"
+                    )
+                    return {"error": f"Execution failed: {e!s}"}
+
+            return wrapper
+
+        for func in funcs:
+            inner_func = func.func
+            func_name = func.get_function_name()
+            func_code = inspect.getsource(inner_func).strip()
+
+            func.func = make_wrapper(inner_func, func_name, func_code)
+            self.tools_map[func_name] = func
+            self.entrypoint[func_name] = entrypoint
+
+        return self
+
+    def info(self) -> str:
+        r"""Get information about the current sandbox.
+
+        Returns:
+            str: Information about the sandbox.
+
+        Raises:
+            RuntimeError: If the sandbox is not initialized.
+        """
+        if self.sandbox is None:
+            raise RuntimeError("Failed to create sandbox.")
+        info = self.sandbox.info()
+        return (
+            f"Sandbox {info.name}:\n"
+            f"State: {info.state}\n"
+            f"Resources: {info.resources.cpu} CPU, {info.resources.memory} RAM"
+        )
+
+    def __del__(self):
+        r"""Clean up the sandbox when the object is deleted."""
+        if hasattr(self, 'sandbox'):
+            self._cleanup()
+
+    def stop(self) -> "DaytonaRuntime":
+        r"""Stop and remove the sandbox.
+
+        Returns:
+            DaytonaRuntime: The current runtime.
+        """
+        self._cleanup()
+        return self
+
+    def reset(self) -> "DaytonaRuntime":
+        r"""Reset the sandbox by stopping and rebuilding it.
+
+        Returns:
+            DaytonaRuntime: The current runtime.
+        """
+        return self.stop().build()
+
+    @property
+    def docs(self) -> str:
+        r"""Get the URL for the Daytona API documentation.
+
+        Returns:
+            str: The URL for the API documentation.
+        """
+        return "https://www.daytona.io/docs/python-sdk/daytona/"

{camel_ai-0.2.51 → camel_ai-0.2.52}/camel/toolkits/__init__.py

@@ -68,6 +68,7 @@ from .pyautogui_toolkit import PyAutoGUIToolkit
 from .openai_agent_toolkit import OpenAIAgentToolkit
 from .searxng_toolkit import SearxNGToolkit
 from .jina_reranker_toolkit import JinaRerankerToolkit
+from .klavis_toolkit import KlavisToolkit
 
 
 __all__ = [
@@ -124,4 +125,5 @@ __all__ = [
     'OpenAIAgentToolkit',
     'SearxNGToolkit',
     'JinaRerankerToolkit',
+    'KlavisToolkit',
 ]

camel_ai-0.2.52/camel/toolkits/klavis_toolkit.py

@@ -0,0 +1,228 @@
+# ========= 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. =========
+
+import os
+from typing import Any, Dict, List, Optional
+
+import requests
+
+from camel.logger import get_logger
+from camel.toolkits.base import BaseToolkit
+from camel.toolkits.function_tool import FunctionTool
+from camel.utils import MCPServer, api_keys_required, dependencies_required
+
+logger = get_logger(__name__)
+
+
+@MCPServer()
+class KlavisToolkit(BaseToolkit):
+    r"""A class representing a toolkit for interacting with Klavis API.
+
+    This class provides methods for interacting with Klavis MCP server
+    instances, retrieving server information, managing tools, and handling
+    authentication.
+
+    Attributes:
+        api_key (str): The API key for authenticating with Klavis API.
+        base_url (str): The base URL for Klavis API endpoints.
+        timeout (Optional[float]): The timeout value for API requests
+            in seconds. If None, no timeout is applied.
+            (default: :obj:`None`)
+    """
+
+    @dependencies_required("requests")
+    @api_keys_required(
+        [
+            (None, "KLAVIS_API_KEY"),
+        ]
+    )
+    def __init__(self, timeout: Optional[float] = None) -> None:
+        r"""Initialize the KlavisToolkit with API client. The API key is
+        retrieved from environment variables.
+        """
+        super().__init__(timeout=timeout)
+        self.api_key = os.environ.get("KLAVIS_API_KEY")
+        self.base_url = "https://api.klavis.ai"
+
+    def _request(
+        self,
+        method: str,
+        endpoint: str,
+        payload: Optional[Dict[str, Any]] = None,
+        additional_headers: Optional[Dict[str, str]] = None,
+    ) -> Dict[str, Any]:
+        r"""Make an HTTP request to the Klavis API.
+
+        Args:
+            method (str): HTTP method (e.g., 'GET', 'POST', 'DELETE').
+            endpoint (str): API endpoint path.
+            payload (Optional[Dict[str, Any]]): JSON payload for POST
+                requests.
+            additional_headers (Optional[Dict[str, str]]): Additional
+                headers to include in the request.
+
+        Returns:
+            Dict[str, Any]: The JSON response from the API or an error
+                dict.
+        """
+        url = f"{self.base_url}{endpoint}"
+        headers = {
+            'accept': 'application/json',
+            'Authorization': f'Bearer {self.api_key}',
+        }
+        if additional_headers:
+            headers.update(additional_headers)
+
+        logger.debug(
+            f"Making {method} request to {url} with payload: {payload}"
+        )
+
+        try:
+            response = requests.request(
+                method,
+                url,
+                headers=headers,
+                json=payload,
+                timeout=self.timeout,
+            )
+            response.raise_for_status()
+            return response.json()
+        except requests.exceptions.RequestException as e:
+            logger.error(f"Request failed for {method} {endpoint}: {e!s}")
+            return {"error": f"Request failed: {e!s}"}
+        except ValueError:
+            logger.error(f"Response for {method} {endpoint} is not valid JSON")
+            return {"error": "Response is not valid JSON"}
+
+    def create_server_instance(
+        self, server_name: str, user_id: str, platform_name: str
+    ) -> Dict[str, Any]:
+        r"""Create a Server-Sent Events (SSE) URL for a specified MCP server.
+
+        Args:
+            server_name (str): The name of the target MCP server.
+            user_id (str): The ID for the user requesting the server URL.
+            platform_name (str): The name of the platform associated
+              with the user.
+
+        Returns:
+            Dict[str, Any]: Response containing the server instance details.
+        """
+        endpoint = "/mcp-server/instance/create"
+        payload = {
+            "serverName": server_name,
+            "userId": user_id,
+            "platformName": platform_name,
+        }
+        headers = {'Content-Type': 'application/json'}
+        return self._request(
+            'POST', endpoint, payload=payload, additional_headers=headers
+        )
+
+    def get_server_instance(self, instance_id: str) -> Dict[str, Any]:
+        r"""Get details of a specific server connection instance.
+
+        Args:
+            instance_id (str): The ID of the connection instance whose status
+                is being checked.
+
+        Returns:
+            Dict[str, Any]: Details about the server instance.
+        """
+        endpoint = f"/mcp-server/instance/get/{instance_id}"
+        return self._request('GET', endpoint)
+
+    def delete_auth_data(self, instance_id: str) -> Dict[str, Any]:
+        r"""Delete authentication metadata for a specific server
+        connection instance.
+
+        Args:
+            instance_id (str): The ID of the connection instance to
+              delete auth for.
+
+        Returns:
+            Dict[str, Any]: Status response for the operation.
+        """
+        endpoint = f"/mcp-server/instance/delete-auth/{instance_id}"
+        return self._request('DELETE', endpoint)
+
+    def delete_server_instance(self, instance_id: str) -> Dict[str, Any]:
+        r"""Completely removes a server connection instance.
+
+        Args:
+            instance_id (str): The ID of the connection instance to delete.
+
+        Returns:
+            Dict[str, Any]: Status response for the operation.
+        """
+        endpoint = f"/mcp-server/instance/delete/{instance_id}"
+        return self._request('DELETE', endpoint)
+
+    def get_server_tools(self, server_name: str) -> Dict[str, Any]:
+        r"""Get list of tool names for a specific MCP server.
+
+        Args:
+            server_name (str): The name of the target MCP server.
+
+        Returns:
+            Dict[str, Any]: List of tools available for the specified server.
+        """
+        endpoint = f"/mcp-server/tools/{server_name}"
+        return self._request('GET', endpoint)
+
+    def get_all_servers(self) -> Dict[str, Any]:
+        r"""Get all MCP servers with their basic information.
+
+        Returns:
+            Dict[str, Any]: Information about all available MCP servers.
+        """
+        endpoint = "/mcp-server/servers"
+        return self._request('GET', endpoint)
+
+    def set_auth_token(
+        self, instance_id: str, auth_token: str
+    ) -> Dict[str, Any]:
+        r"""Sets an authentication token for a specific instance.
+
+        Args:
+            instance_id (str): The ID for the connection instance.
+            auth_token (str): The authentication token to save.
+
+        Returns:
+            Dict[str, Any]: Status response for the operation.
+        """
+        endpoint = "/mcp-server/instance/set-auth-token"
+        payload = {"instanceId": instance_id, "authToken": auth_token}
+        headers = {'Content-Type': 'application/json'}
+        return self._request(
+            'POST', endpoint, payload=payload, additional_headers=headers
+        )
+
+    def get_tools(self) -> List[FunctionTool]:
+        r"""Returns a list of FunctionTool objects representing the functions
+        in the toolkit.
+
+        Returns:
+            List[FunctionTool]: A list of FunctionTool objects representing
+                the functions in the toolkit.
+        """
+        return [
+            FunctionTool(self.create_server_instance),
+            FunctionTool(self.get_server_instance),
+            FunctionTool(self.delete_auth_data),
+            FunctionTool(self.delete_server_instance),
+            FunctionTool(self.get_server_tools),
+            FunctionTool(self.get_all_servers),
+            FunctionTool(self.set_auth_token),
+        ]

{camel_ai-0.2.51 → camel_ai-0.2.52}/camel/toolkits/mcp_toolkit.py

@@ -60,6 +60,8 @@ class MCPClient(BaseToolkit):
         timeout (Optional[float]): Connection timeout. (default: :obj:`'None'`)
         headers (Dict[str, str]): Headers for the HTTP request.
             (default: :obj:`'None'`)
+        strict (Optional[bool]): Whether to enforce strict mode for the
+            function call. (default: :obj:`False`)
     """
 
     def __init__(
@@ -69,6 +71,7 @@ class MCPClient(BaseToolkit):
         env: Optional[Dict[str, str]] = None,
         timeout: Optional[float] = None,
         headers: Optional[Dict[str, str]] = None,
+        strict: Optional[bool] = False,
     ):
         from mcp import Tool
 
@@ -78,6 +81,7 @@ class MCPClient(BaseToolkit):
         self.args = args or []
         self.env = env or {}
         self.headers = headers or {}
+        self.strict = strict
 
         self._mcp_tools: List[Tool] = []
         self._session: Optional['ClientSession'] = None
@@ -317,13 +321,15 @@ class MCPClient(BaseToolkit):
             "additionalProperties": False,
         }
 
+        # Because certain parameters in MCP may include keywords that are not
+        # supported by OpenAI, it is essential to set "strict" to False.
         return {
             "type": "function",
             "function": {
                 "name": mcp_tool.name,
                 "description": mcp_tool.description
                 or "No description provided.",
-                "strict": True,
+                "strict": self.strict,
                 "parameters": parameters,
             },
         }
@@ -363,6 +369,8 @@ class MCPToolkit(BaseToolkit):
             instances to manage.
         config_path (Optional[str]): Path to a JSON configuration file
             defining MCP servers.
+        strict (Optional[bool]): Whether to enforce strict mode for the
+            function call. (default: :obj:`False`)
 
     Note:
         Either `servers` or `config_path` must be provided. If both are
@@ -397,6 +405,7 @@ class MCPToolkit(BaseToolkit):
         self,
         servers: Optional[List[MCPClient]] = None,
         config_path: Optional[str] = None,
+        strict: Optional[bool] = False,
     ):
         super().__init__()
 
@@ -409,16 +418,22 @@ class MCPToolkit(BaseToolkit):
         self.servers: List[MCPClient] = servers or []
 
         if config_path:
-            self.servers.extend(self._load_servers_from_config(config_path))
+            self.servers.extend(
+                self._load_servers_from_config(config_path, strict)
+            )
 
         self._exit_stack = AsyncExitStack()
         self._connected = False
 
-    def _load_servers_from_config(self, config_path: str) -> List[MCPClient]:
+    def _load_servers_from_config(
+        self, config_path: str, strict: Optional[bool] = False
+    ) -> List[MCPClient]:
         r"""Loads MCP server configurations from a JSON file.
 
         Args:
             config_path (str): Path to the JSON configuration file.
+            strict (bool): Whether to enforce strict mode for the
+                function call. (default: :obj:`False`)
 
         Returns:
             List[MCPClient]: List of configured MCPClient instances.
@@ -462,6 +477,7 @@ class MCPToolkit(BaseToolkit):
                 args=cfg.get("args", []),
                 env={**os.environ, **cfg.get("env", {})},
                 timeout=cfg.get("timeout", None),
+                strict=strict,
             )
             all_servers.append(server)
 

{camel_ai-0.2.51 → camel_ai-0.2.52}/camel/utils/commons.py

@@ -352,6 +352,8 @@ def api_keys_required(
                 key_way = "https://platform.lingyiwanwu.com/docs"
             elif env_var_name == 'ZHIPUAI_API_KEY':
                 key_way = "https://www.zhipuai.cn/"
+            elif env_var_name == 'KLAVIS_API_KEY':
+                key_way = "https://www.klavis.ai/docs"
 
             if missing_keys:
                 raise ValueError(

{camel_ai-0.2.51 → camel_ai-0.2.52}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "camel-ai"
-version = "0.2.51"
+version = "0.2.52"
 description = "Communicative Agents for AI Society Study"
 authors = [{ name = "CAMEL-AI.org" }]
 requires-python = ">=3.10,<3.13"
@@ -164,6 +164,7 @@ dev_tools = [
     "tree-sitter>=0.23.2,<0.24",
     "typer>=0.15.2",
     "mcp>=1.3.0",
+    "daytona-sdk==0.14.0",
 ]
 model_platforms = [
     "litellm>=1.38.1,<2",
@@ -356,6 +357,7 @@ all = [
     "pyautogui>=0.9.54,<0.10",
     "pyobvector>=0.1.18",
     "scrapegraph-py>=1.12.0,<2",
+    "daytona-sdk==0.14.0",
     "ibm-watsonx-ai>=1.3.11",
 ]
 
@@ -543,6 +545,7 @@ module = [
     "crawl4ai.*",
     "pyautogui",
     "pyobvector.*",
+    "daytona_sdk",
     "scrapegraph_py.*",
     "ibm_watsonx_ai.*",
 ]