camel-ai 0.2.60__py3-none-any.whl → 0.2.61__py3-none-any.whl

camel/toolkits/file_write_toolkit.py

@@ -11,8 +11,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
-
-
 import re
 from datetime import datetime
 from pathlib import Path
@@ -21,7 +19,7 @@ from typing import List, Optional, Union
 from camel.logger import get_logger
 from camel.toolkits.base import BaseToolkit
 from camel.toolkits.function_tool import FunctionTool
-from camel.utils import MCPServer
+from camel.utils import MCPServer, dependencies_required
 
 logger = get_logger(__name__)
 
@@ -148,40 +146,83 @@ class FileWriteToolkit(BaseToolkit):
         document.save(str(file_path))
         logger.debug(f"Wrote DOCX to {file_path} with default formatting")
 
-    def _write_pdf_file(self, file_path: Path, content: str, **kwargs) -> None:
+    @dependencies_required('pylatex', 'fpdf')
+    def _write_pdf_file(
+        self, file_path: Path, content: str, use_latex: bool = False
+    ) -> None:
         r"""Write text content to a PDF file with default formatting.
 
         Args:
             file_path (Path): The target file path.
             content (str): The text content to write.
+            use_latex (bool): Whether to use LaTeX for rendering. (requires
+                LaTeX toolchain). If False, uses FPDF for simpler PDF
+                generation. (default: :obj: `False`)
 
         Raises:
-            RuntimeError: If the 'fpdf' library is not installed.
+            RuntimeError: If the 'pylatex' or 'fpdf' library is not installed
+                when use_latex=True.
         """
-        from fpdf import FPDF
-
-        # Use default formatting values
-        font_family = 'Arial'
-        font_size = 12
-        font_style = ''
-        line_height = 10
-        margin = 10
-
-        pdf = FPDF()
-        pdf.set_margins(margin, margin, margin)
-
-        pdf.add_page()
-        pdf.set_font(font_family, style=font_style, size=font_size)
-
-        # Split content into paragraphs and add them
-        for para in content.split('\n'):
-            if para.strip():  # Skip empty paragraphs
-                pdf.multi_cell(0, line_height, para)
-            else:
-                pdf.ln(line_height)  # Add empty line
+        if use_latex:
+            from pylatex import (
+                Command,
+                Document,
+                Math,
+                Section,
+            )
+            from pylatex.utils import (
+                NoEscape,
+            )
 
-        pdf.output(str(file_path))
-        logger.debug(f"Wrote PDF to {file_path} with custom formatting")
+            doc = Document(documentclass="article")
+            doc.packages.append(Command('usepackage', 'amsmath'))
+
+            with doc.create(Section('Generated Content')):
+                for line in content.split('\n'):
+                    # Remove leading whitespace
+                    stripped_line = line.strip()
+                    # Check if the line is intended as a standalone math
+                    # expression
+                    if (
+                        stripped_line.startswith('$')
+                        and stripped_line.endswith('$')
+                        and len(stripped_line) > 1
+                    ):
+                        # Extract content between the '$' delimiters
+                        math_data = stripped_line[1:-1]
+                        doc.append(Math(data=math_data))
+                    else:
+                        doc.append(NoEscape(line))
+                    doc.append(NoEscape(r'\par'))
+
+            doc.generate_pdf(str(file_path), clean_tex=False)
+
+            logger.info(f"Wrote PDF (with LaTeX) to {file_path}")
+        else:
+            from fpdf import FPDF
+
+            # Use default formatting values
+            font_family = 'Arial'
+            font_size = 12
+            font_style = ''
+            line_height = 10
+            margin = 10
+
+            pdf = FPDF()
+            pdf.set_margins(margin, margin, margin)
+
+            pdf.add_page()
+            pdf.set_font(font_family, style=font_style, size=font_size)
+
+            # Split content into paragraphs and add them
+            for para in content.split('\n'):
+                if para.strip():  # Skip empty paragraphs
+                    pdf.multi_cell(0, line_height, para)
+                else:
+                    pdf.ln(line_height)  # Add empty line
+
+            pdf.output(str(file_path))
+            logger.debug(f"Wrote PDF to {file_path} with custom formatting")
 
     def _write_csv_file(
         self,
@@ -286,6 +327,7 @@ class FileWriteToolkit(BaseToolkit):
         content: Union[str, List[List[str]]],
         filename: str,
         encoding: Optional[str] = None,
+        use_latex: bool = False,
     ) -> str:
         r"""Write the given content to a file.
 
@@ -302,6 +344,9 @@ class FileWriteToolkit(BaseToolkit):
                 supplied, it is resolved to self.output_dir.
             encoding (Optional[str]): The character encoding to use. (default:
                 :obj: `None`)
+            use_latex (bool): For PDF files, whether to use LaTeX rendering
+                (True) or simple FPDF rendering (False). (default: :obj:
+                `False`)
 
         Returns:
             str: A message indicating success or error details.
@@ -326,7 +371,9 @@ class FileWriteToolkit(BaseToolkit):
             if extension in [".doc", ".docx"]:
                 self._write_docx_file(file_path, str(content))
             elif extension == ".pdf":
-                self._write_pdf_file(file_path, str(content))
+                self._write_pdf_file(
+                    file_path, str(content), use_latex=use_latex
+                )
             elif extension == ".csv":
                 self._write_csv_file(
                     file_path, content, encoding=file_encoding

camel/toolkits/mcp_toolkit.py

@@ -37,6 +37,7 @@ if TYPE_CHECKING:
 
 from camel.logger import get_logger
 from camel.toolkits import BaseToolkit, FunctionTool
+from camel.utils.commons import run_async
 
 logger = get_logger(__name__)
 
@@ -220,6 +221,10 @@ class MCPClient(BaseToolkit):
             logger.error(f"Failed to connect to MCP server: {e}")
             raise e
 
+    def connect_sync(self):
+        r"""Synchronously connect to the MCP server."""
+        return run_async(self.connect)()
+
     async def disconnect(self):
         r"""Explicitly disconnect from the MCP server."""
         # If the server is not connected, do nothing
@@ -235,6 +240,10 @@ class MCPClient(BaseToolkit):
             self._exit_stack = AsyncExitStack()
             self._session = None
 
+    def disconnect_sync(self):
+        r"""Synchronously disconnect from the MCP server."""
+        return run_async(self.disconnect)()
+
     @asynccontextmanager
     async def connection(self):
         r"""Async context manager for establishing and managing the connection
@@ -254,6 +263,10 @@ class MCPClient(BaseToolkit):
             except Exception as e:
                 logger.warning(f"Error: {e}")
 
+    def connection_sync(self):
+        r"""Synchronously connect to the MCP server."""
+        return run_async(self.connection)()
+
     async def list_mcp_tools(self) -> Union[str, "ListToolsResult"]:
         r"""Retrieves the list of available tools from the connected MCP
         server.
@@ -269,6 +282,11 @@ class MCPClient(BaseToolkit):
             logger.exception("Failed to list MCP tools")
             raise e
 
+    def list_mcp_tools_sync(self) -> Union[str, "ListToolsResult"]:
+        r"""Synchronously list the available tools from the connected MCP
+        server."""
+        return run_async(self.list_mcp_tools)()
+
     def generate_function_from_mcp_tool(self, mcp_tool: "Tool") -> Callable:
         r"""Dynamically generates a Python callable function corresponding to
         a given MCP tool.
@@ -390,6 +408,10 @@ class MCPClient(BaseToolkit):
 
         return dynamic_function
 
+    def generate_function_from_mcp_tool_sync(self, mcp_tool: "Tool") -> Any:
+        r"""Synchronously generate a function from an MCP tool."""
+        return run_async(self.generate_function_from_mcp_tool)(mcp_tool)
+
     def _build_tool_schema(self, mcp_tool: "Tool") -> Dict[str, Any]:
         input_schema = mcp_tool.inputSchema
         properties = input_schema.get("properties", {})
@@ -463,6 +485,10 @@ class MCPClient(BaseToolkit):
 
         return await self._session.call_tool(tool_name, tool_args)
 
+    def call_tool_sync(self, tool_name: str, tool_args: Dict[str, Any]) -> Any:
+        r"""Synchronously call a tool."""
+        return run_async(self.call_tool)(tool_name, tool_args)
+
     @property
     def session(self) -> Optional["ClientSession"]:
         return self._session
@@ -521,6 +547,21 @@ class MCPClient(BaseToolkit):
             logger.error(f"Failed to initialize MCPClient: {e}")
             raise RuntimeError(f"Failed to initialize MCPClient: {e}") from e
 
+    @classmethod
+    def create_sync(
+        self,
+        command_or_url: str,
+        args: Optional[List[str]] = None,
+        env: Optional[Dict[str, str]] = None,
+        timeout: Optional[float] = None,
+        headers: Optional[Dict[str, str]] = None,
+        mode: Optional[str] = None,
+    ) -> "MCPClient":
+        r"""Synchronously create and connect to the MCP server."""
+        return run_async(self.create)(
+            command_or_url, args, env, timeout, headers, mode
+        )
+
     async def __aenter__(self) -> "MCPClient":
         r"""Async context manager entry point. Automatically connects to the
         MCP server when used in an async with statement.
@@ -531,12 +572,35 @@ class MCPClient(BaseToolkit):
         await self.connect()
         return self
 
-    async def __aexit__(self, exc_type, exc_val, exc_tb) -> None:
+    def __enter__(self) -> "MCPClient":
+        r"""Synchronously enter the async context manager."""
+        return run_async(self.__aenter__)()
+
+    async def __aexit__(self) -> None:
         r"""Async context manager exit point. Automatically disconnects from
         the MCP server when exiting an async with statement.
+
+        Returns:
+            None
         """
         await self.disconnect()
 
+    def __exit__(self, exc_type, exc_val, exc_tb) -> None:
+        r"""Synchronously exit the async context manager.
+
+        Args:
+            exc_type (Optional[Type[Exception]]): The type of exception that
+                occurred during the execution of the with statement.
+            exc_val (Optional[Exception]): The exception that occurred during
+                the execution of the with statement.
+            exc_tb (Optional[TracebackType]): The traceback of the exception
+                that occurred during the execution of the with statement.
+
+        Returns:
+            None
+        """
+        return run_async(self.__aexit__)()
+
 
 class MCPToolkit(BaseToolkit):
     r"""MCPToolkit provides a unified interface for managing multiple
@@ -749,6 +813,10 @@ class MCPToolkit(BaseToolkit):
             logger.error(f"Failed to connect to one or more MCP servers: {e}")
             raise e
 
+    def connect_sync(self):
+        r"""Synchronously connect to all MCP servers."""
+        return run_async(self.connect)()
+
     async def disconnect(self):
         r"""Explicitly disconnect from all MCP servers."""
         if not self._connected:
@@ -758,6 +826,10 @@ class MCPToolkit(BaseToolkit):
             await server.disconnect()
         self._connected = False
 
+    def disconnect_sync(self):
+        r"""Synchronously disconnect from all MCP servers."""
+        return run_async(self.disconnect)()
+
     @asynccontextmanager
     async def connection(self) -> AsyncGenerator["MCPToolkit", None]:
         r"""Async context manager that simultaneously establishes connections
@@ -772,6 +844,10 @@ class MCPToolkit(BaseToolkit):
         finally:
             await self.disconnect()
 
+    def connection_sync(self):
+        r"""Synchronously connect to all MCP servers."""
+        return run_async(self.connection)()
+
     def is_connected(self) -> bool:
         r"""Checks if all the managed servers are connected.
 

camel/toolkits/wolfram_alpha_toolkit.py

@@ -117,7 +117,11 @@ class WolframAlphaToolkit(BaseToolkit):
 
         try:
             url = "https://www.wolframalpha.com/api/v1/llm-api"
-            params = {"input": query, "appid": WOLFRAMALPHA_APP_ID}
+            params = {
+                "input": query,
+                "appid": WOLFRAMALPHA_APP_ID,
+                "format": "plaintext",
+            }
 
             response = requests.get(url, params=params)
             response.raise_for_status()

camel/types/enums.py

@@ -56,6 +56,8 @@ class ModelType(UnifiedModelType, Enum):
     AWS_LLAMA_3_3_70B_INSTRUCT = "us.meta.llama3-3-70b-instruct-v1:0"
     AWS_LLAMA_3_2_90B_INSTRUCT = "us.meta.llama3-2-90b-instruct-v1:0"
     AWS_LLAMA_3_2_11B_INSTRUCT = "us.meta.llama3-2-11b-instruct-v1:0"
+    AWS_CLAUDE_SONNET_4 = "anthropic.claude-sonnet-4-20250514-v1:0"
+    AWS_CLAUDE_OPUS_4 = "anthropic.claude-opus-4-20250514-v1:0"
 
     GLM_4 = "glm-4"
     GLM_4V = "glm-4v"
@@ -145,13 +147,15 @@ class ModelType(UnifiedModelType, Enum):
     CLAUDE_2_0 = "claude-2.0"
     CLAUDE_INSTANT_1_2 = "claude-instant-1.2"
 
-    # Claude3 models
+    # Claude models
     CLAUDE_3_OPUS = "claude-3-opus-latest"
     CLAUDE_3_SONNET = "claude-3-sonnet-20240229"
     CLAUDE_3_HAIKU = "claude-3-haiku-20240307"
     CLAUDE_3_5_SONNET = "claude-3-5-sonnet-latest"
     CLAUDE_3_5_HAIKU = "claude-3-5-haiku-latest"
     CLAUDE_3_7_SONNET = "claude-3-7-sonnet-latest"
+    CLAUDE_SONNET_4 = "claude-sonnet-4-20250514"
+    CLAUDE_OPUS_4 = "claude-opus-4-20250514"
 
     # Netmind models
     NETMIND_LLAMA_4_MAVERICK_17B_128E_INSTRUCT = (
@@ -469,6 +473,8 @@ class ModelType(UnifiedModelType, Enum):
             ModelType.AWS_LLAMA_3_3_70B_INSTRUCT,
             ModelType.AWS_LLAMA_3_2_90B_INSTRUCT,
             ModelType.AWS_LLAMA_3_2_11B_INSTRUCT,
+            ModelType.AWS_CLAUDE_SONNET_4,
+            ModelType.AWS_CLAUDE_OPUS_4,
         }
 
     @property
@@ -530,6 +536,8 @@ class ModelType(UnifiedModelType, Enum):
             ModelType.CLAUDE_3_5_SONNET,
             ModelType.CLAUDE_3_5_HAIKU,
             ModelType.CLAUDE_3_7_SONNET,
+            ModelType.CLAUDE_SONNET_4,
+            ModelType.CLAUDE_OPUS_4,
         }
 
     @property
@@ -1162,11 +1170,15 @@ class ModelType(UnifiedModelType, Enum):
             ModelType.CLAUDE_3_5_SONNET,
             ModelType.CLAUDE_3_5_HAIKU,
             ModelType.CLAUDE_3_7_SONNET,
+            ModelType.CLAUDE_SONNET_4,
+            ModelType.CLAUDE_OPUS_4,
             ModelType.YI_MEDIUM_200K,
             ModelType.AWS_CLAUDE_3_5_SONNET,
             ModelType.AWS_CLAUDE_3_HAIKU,
             ModelType.AWS_CLAUDE_3_SONNET,
             ModelType.AWS_CLAUDE_3_7_SONNET,
+            ModelType.AWS_CLAUDE_SONNET_4,
+            ModelType.AWS_CLAUDE_OPUS_4,
             ModelType.O4_MINI,
             ModelType.O3,
         }:

camel/utils/__init__.py

@@ -35,6 +35,7 @@ from .commons import (
     json_to_function_code,
     print_text_animated,
     retry_on_error,
+    run_async,
     text_extract_from_web,
     to_pascal,
     track_agent,
@@ -95,4 +96,5 @@ __all__ = [
     "model_from_json_schema",
     "sanitize_filename",
     "browser_toolkit_save_auth_cookie",
+    "run_async",
 ]

camel/utils/commons.py

@@ -1103,3 +1103,30 @@ def browser_toolkit_save_auth_cookie(
     context.storage_state(path=cookie_json_path)
 
     browser.close()  # Close the browser when finished
+
+
+def run_async(func: Callable[..., Any]) -> Callable[..., Any]:
+    r"""Helper function to run async functions in synchronous context.
+
+    Args:
+        func (Callable[..., Any]): The async function to wrap.
+
+    Returns:
+        Callable[..., Any]: A synchronous wrapper for the async function.
+    """
+
+    @wraps(func)
+    def wrapper(*args: Any, **kwargs: Any) -> Any:
+        try:
+            loop = asyncio.get_event_loop()
+        except RuntimeError:
+            loop = asyncio.new_event_loop()
+            asyncio.set_event_loop(loop)
+
+        if loop.is_closed():
+            loop = asyncio.new_event_loop()
+            asyncio.set_event_loop(loop)
+
+        return loop.run_until_complete(func(*args, **kwargs))
+
+    return wrapper

{camel_ai-0.2.60.dist-info → camel_ai-0.2.61.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: camel-ai
-Version: 0.2.60
+Version: 0.2.61
 Summary: Communicative Agents for AI Society Study
 Project-URL: Homepage, https://www.camel-ai.org/
 Project-URL: Repository, https://github.com/camel-ai/camel
@@ -32,6 +32,7 @@ Requires-Dist: arxiv<3,>=2.1.3; extra == 'all'
 Requires-Dist: azure-storage-blob<13,>=12.21.0; extra == 'all'
 Requires-Dist: beautifulsoup4<5,>=4; extra == 'all'
 Requires-Dist: botocore<2,>=1.35.3; extra == 'all'
+Requires-Dist: chunkr-ai>=0.0.50; extra == 'all'
 Requires-Dist: cohere<6,>=5.11.0; extra == 'all'
 Requires-Dist: crawl4ai>=0.3.745; extra == 'all'
 Requires-Dist: dappier<0.4,>=0.3.3; extra == 'all'
@@ -47,6 +48,7 @@ Requires-Dist: docx>=0.2.4; extra == 'all'
 Requires-Dist: duckduckgo-search<7,>=6.3.5; extra == 'all'
 Requires-Dist: e2b-code-interpreter<2,>=1.0.3; extra == 'all'
 Requires-Dist: exa-py<2,>=1.10.0; extra == 'all'
+Requires-Dist: faiss-cpu<2,>=1.7.2; extra == 'all'
 Requires-Dist: fastapi>=0.115.11; extra == 'all'
 Requires-Dist: ffmpeg-python<0.3,>=0.2.0; extra == 'all'
 Requires-Dist: firecrawl-py<2,>=1.0.0; extra == 'all'
@@ -92,6 +94,7 @@ Requires-Dist: pre-commit<4,>=3; extra == 'all'
 Requires-Dist: pyautogui<0.10,>=0.9.54; extra == 'all'
 Requires-Dist: pydub<0.26,>=0.25.1; extra == 'all'
 Requires-Dist: pygithub<3,>=2.6.0; extra == 'all'
+Requires-Dist: pylatex>=1.4.2; extra == 'all'
 Requires-Dist: pymilvus<3,>=2.4.0; extra == 'all'
 Requires-Dist: pymupdf<2,>=1.22.5; extra == 'all'
 Requires-Dist: pyobvector>=0.1.18; extra == 'all'
@@ -193,6 +196,7 @@ Requires-Dist: sphinx<8,>=7; extra == 'docs'
 Requires-Dist: sphinxext-rediraffe<0.3,>=0.2.7; extra == 'docs'
 Provides-Extra: document-tools
 Requires-Dist: beautifulsoup4<5,>=4; extra == 'document-tools'
+Requires-Dist: chunkr-ai>=0.0.50; extra == 'document-tools'
 Requires-Dist: crawl4ai>=0.3.745; extra == 'document-tools'
 Requires-Dist: docx2txt<0.9,>=0.8; extra == 'document-tools'
 Requires-Dist: docx>=0.2.4; extra == 'document-tools'
@@ -203,6 +207,7 @@ Requires-Dist: openapi-spec-validator<0.8,>=0.7.1; extra == 'document-tools'
 Requires-Dist: openpyxl>=3.1.5; extra == 'document-tools'
 Requires-Dist: pandasai<3,>=2.3.0; extra == 'document-tools'
 Requires-Dist: prance<24,>=23.6.21.0; extra == 'document-tools'
+Requires-Dist: pylatex>=1.4.2; extra == 'document-tools'
 Requires-Dist: pymupdf<2,>=1.22.5; extra == 'document-tools'
 Requires-Dist: tabulate>=0.9.0; extra == 'document-tools'
 Requires-Dist: unstructured==0.16.20; extra == 'document-tools'
@@ -235,6 +240,7 @@ Requires-Dist: aci-sdk>=1.0.0b1; extra == 'owl'
 Requires-Dist: anthropic<0.50.0,>=0.47.0; extra == 'owl'
 Requires-Dist: beautifulsoup4<5,>=4; extra == 'owl'
 Requires-Dist: chunkr-ai>=0.0.41; extra == 'owl'
+Requires-Dist: chunkr-ai>=0.0.50; extra == 'owl'
 Requires-Dist: crawl4ai>=0.3.745; extra == 'owl'
 Requires-Dist: datasets<4,>=3; extra == 'owl'
 Requires-Dist: docx2txt<0.9,>=0.8; extra == 'owl'
@@ -259,6 +265,7 @@ Requires-Dist: playwright>=1.50.0; extra == 'owl'
 Requires-Dist: prance<24,>=23.6.21.0; extra == 'owl'
 Requires-Dist: pyautogui<0.10,>=0.9.54; extra == 'owl'
 Requires-Dist: pydub<0.26,>=0.25.1; extra == 'owl'
+Requires-Dist: pylatex>=1.4.2; extra == 'owl'
 Requires-Dist: pymupdf<2,>=1.22.5; extra == 'owl'
 Requires-Dist: python-dotenv<2,>=1.0.0; extra == 'owl'
 Requires-Dist: requests-oauthlib<2,>=1.3.1; extra == 'owl'
@@ -277,8 +284,10 @@ Requires-Dist: wikipedia<2,>=1; extra == 'owl'
 Requires-Dist: xls2xlsx>=0.2.0; extra == 'owl'
 Requires-Dist: yt-dlp<2025,>=2024.11.4; extra == 'owl'
 Provides-Extra: rag
+Requires-Dist: chunkr-ai>=0.0.50; extra == 'rag'
 Requires-Dist: cohere<6,>=5.11.0; extra == 'rag'
 Requires-Dist: crawl4ai>=0.3.745; extra == 'rag'
+Requires-Dist: faiss-cpu<2,>=1.7.2; extra == 'rag'
 Requires-Dist: google-genai>=1.13.0; extra == 'rag'
 Requires-Dist: nebula3-python==3.8.2; extra == 'rag'
 Requires-Dist: neo4j<6,>=5.18.0; extra == 'rag'
@@ -298,6 +307,7 @@ Requires-Dist: scholarly[tor]==1.7.11; extra == 'research-tools'
 Provides-Extra: storage
 Requires-Dist: azure-storage-blob<13,>=12.21.0; extra == 'storage'
 Requires-Dist: botocore<2,>=1.35.3; extra == 'storage'
+Requires-Dist: faiss-cpu<2,>=1.7.2; extra == 'storage'
 Requires-Dist: google-cloud-storage<3,>=2.18.0; extra == 'storage'
 Requires-Dist: mem0ai>=0.1.73; extra == 'storage'
 Requires-Dist: nebula3-python==3.8.2; extra == 'storage'