fix: ci failures (add license headers, rename variable, restore file content)

Author: aviralgarg05

camel/toolkits/pptx_node_toolkit.py

@@ -1,4 +1,16 @@
-
+# ========= Copyright 2023-2026 @ 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-2026 @ CAMEL-AI.org. All Rights Reserved. =========
 import json
 import os
 import subprocess
@@ -15,8 +27,7 @@
 
 @MCPServer()
 class PptxNodeToolkit(BaseToolkit):
-    r"""A toolkit for creating PowerPoint presentations using PptxGenJS (Node.js).
-    """
+    r"""A toolkit for creating PowerPoint presentations using PptxGenJS (Node.js)."""
 
     def __init__(
         self,
@@ -56,15 +67,15 @@ def create_presentation(
         filename: str,
     ) -> str:
         r"""Create a PowerPoint presentation (PPTX) file using PptxGenJS.
-        
+
         The filename MUST end with ".pptx". If it does not, the toolkit will
         automatically append it, but the agent should strive to provide the
         correct extension.
 
         Args:
             content (str): The content to write to the PPTX file as a JSON
                 string. It must be a list of dictionaries representing slides.
-                
+
                 JSON Schema Example:
                 [
                     {
@@ -99,41 +110,41 @@ def create_presentation(
         script_path = Path(__file__).parent / "scripts" / "generate_pptx.js"
 
         try:
-             # Ensure content is a valid JSON string
-             try:
-                 if not isinstance(content, str):
-                     content_str = json.dumps(content)
-                 else:
-                     # Validate JSON
-                     json_obj = json.loads(content)
-                     content_str = json.dumps(json_obj)
-             except json.JSONDecodeError:
-                 return "Error: Content must be valid JSON string representing slides."
-
-             # Run node script
-             result = subprocess.run(
-                 ["node", str(script_path), str(file_path), content_str],
-                 capture_output=True,
-                 text=True,
-                 check=True
-             )
-             
-             # Parse JSON output from script
-             try:
-                 script_output = json.loads(result.stdout.strip())
-                 if script_output.get("success"):
-                     return f"Presentation created successfully. Path: {script_output.get('path')}, Slides: {script_output.get('slides')}"
-                 else:
-                     return f"Error creating presentation: {script_output.get('error')}"
-             except json.JSONDecodeError:
-                 return f"Error parsing script output: {result.stdout.strip()}"
+            # Ensure content is a valid JSON string
+            try:
+                if not isinstance(content, str):
+                    content_str = json.dumps(content)
+                else:
+                    # Validate JSON
+                    json_obj = json.loads(content)
+                    content_str = json.dumps(json_obj)
+            except json.JSONDecodeError:
+                return "Error: Content must be valid JSON string representing slides."
+
+            # Run node script
+            result = subprocess.run(
+                ["node", str(script_path), str(file_path), content_str],
+                capture_output=True,
+                text=True,
+                check=True,
+            )
+
+            # Parse JSON output from script
+            try:
+                script_output = json.loads(result.stdout.strip())
+                if script_output.get("success"):
+                    return f"Presentation created successfully. Path: {script_output.get('path')}, Slides: {script_output.get('slides')}"
+                else:
+                    return f"Error creating presentation: {script_output.get('error')}"
+            except json.JSONDecodeError:
+                return f"Error parsing script output: {result.stdout.strip()}"
 
         except subprocess.CalledProcessError as e:
             logger.error(f"Error creating presentation: {e.stderr}")
             return f"Error creating presentation subprocess: {e.stderr}"
         except Exception as e:
-            logger.error(f"Error creating presentation: {str(e)}")
-            return f"Error creating presentation: {str(e)}"
+            logger.error(f"Error creating presentation: {e!s}")
+            return f"Error creating presentation: {e!s}"
 
     def get_tools(self) -> list[FunctionTool]:
         r"""Returns a list of FunctionTool objects representing the

camel/toolkits/scripts/generate_pptx.js

@@ -21,12 +21,12 @@ try {
 }
 
 // Create Presentation
-const pres = new PptxGenJS();
+const ppt = new PptxGenJS();
 
 // Process slides
 if (Array.isArray(slidesData)) {
     slidesData.forEach((slideData) => {
-        const slide = pres.addSlide();
+        const slide = ppt.addSlide();
 
         // simple layout heuristics based on keys
         if (slideData.title) {
@@ -69,7 +69,7 @@ if (Array.isArray(slidesData)) {
 }
 
 // Save File
-pres.writeFile({ fileName: filename })
+ppt.writeFile({ fileName: filename })
     .then((fileName) => {
         const result = {
             success: true,

test/toolkits/test_pptx_node_toolkit.py

@@ -0,0 +1,71 @@
+# ========= Copyright 2023-2026 @ 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-2026 @ CAMEL-AI.org. All Rights Reserved. =========
+import json
+import unittest
+from unittest.mock import MagicMock, patch
+
+from camel.toolkits.pptx_node_toolkit import PptxNodeToolkit
+
+
+class TestPptxNodeToolkit(unittest.TestCase):
+    def setUp(self):
+        self.toolkit = PptxNodeToolkit()
+
+    @patch("subprocess.run")
+    def test_create_presentation_success(self, mock_subprocess_run):
+        # Mock successful execution
+        mock_result = MagicMock()
+        mock_result.stdout = json.dumps(
+            {"success": True, "path": "/path/to/test.pptx", "slides": 5}
+        )
+        mock_subprocess_run.return_value = mock_result
+
+        content = [{"title": "Test Slide"}]
+        filename = "test_presentation"
+
+        result = self.toolkit.create_presentation(content, filename)
+
+        expected_call_args = mock_subprocess_run.call_args[0][0]
+        self.assertEqual(expected_call_args[0], "node")
+        self.assertTrue(
+            expected_call_args[2].endswith("test_presentation.pptx")
+        )
+
+        self.assertIn("Presentation created successfully", result)
+        self.assertIn("/path/to/test.pptx", result)
+        self.assertIn("Slides: 5", result)
+
+    @patch("subprocess.run")
+    def test_create_presentation_failure(self, mock_subprocess_run):
+        # Mock failed execution
+        mock_result = MagicMock()
+        mock_result.stdout = json.dumps(
+            {"success": False, "error": "Some error occurred"}
+        )
+        mock_subprocess_run.return_value = mock_result
+
+        content = [{"title": "Test Slide"}]
+        filename = "test.pptx"
+
+        result = self.toolkit.create_presentation(content, filename)
+
+        self.assertIn(
+            "Error creating presentation: Some error occurred", result
+        )
+
+    def test_create_presentation_invalid_json(self):
+        content = "Invalid JSON"
+        filename = "test.pptx"
+        result = self.toolkit.create_presentation(content, filename)
+        self.assertIn("Error: Content must be valid JSON", result)