Add "py" as lang in conversable agent (#1062)

autogen/agentchat/conversable_agent.py

@@ -2079,7 +2079,7 @@ def execute_code_blocks(self, code_blocks):
             )
             if lang in ["bash", "shell", "sh"]:
                 exitcode, logs, image = self.run_code(code, lang=lang, **self._code_execution_config)
-            elif lang in ["python", "Python"]:
+            elif lang in ["python", "Python", "py"]:
                 if code.startswith("# filename: "):
                     filename = code[11 : code.find("\n")].strip()
                 else:

autogen/code_utils.py

@@ -244,6 +244,8 @@ def get_powershell_command():
 
 
 def _cmd(lang: str) -> str:
+    if lang in ["Python", "py"]:
+        return "python"
     if lang.startswith("python") or lang in ["bash", "sh"]:
         return lang
     if lang in ["shell"]:

autogen/coding/local_commandline_code_executor.py

@@ -217,6 +217,9 @@ def _execute_code_dont_check_setup(self, code_blocks: List[CodeBlock]) -> Comman
             LocalCommandLineCodeExecutor.sanitize_command(lang, code)
             code = silence_pip(code, lang)
 
+            if lang in ["Python", "py"]:
+                lang = "python"
+
             if WIN32 and lang in ["sh", "shell"]:
                 lang = "ps1"
 

test/coding/test_commandline_code_executor.py

@@ -68,23 +68,29 @@ def test_commandline_executor_execute_code(cls) -> None:
 
 
 def _test_execute_code(executor: CodeExecutor) -> None:
+
+    # Python executable variants
+    python_variants = ["python", "Python", "py"]
+
     # Test single code block.
-    code_blocks = [CodeBlock(code="import sys; print('hello world!')", language="python")]
-    code_result = executor.execute_code_blocks(code_blocks)
-    assert code_result.exit_code == 0 and "hello world!" in code_result.output and code_result.code_file is not None
+    for py_variant in python_variants:
+        code_blocks = [CodeBlock(code="import sys; print('hello world!')", language=py_variant)]
+        code_result = executor.execute_code_blocks(code_blocks)
+        assert code_result.exit_code == 0 and "hello world!" in code_result.output and code_result.code_file is not None
 
     # Test multiple code blocks.
-    code_blocks = [
-        CodeBlock(code="import sys; print('hello world!')", language="python"),
-        CodeBlock(code="a = 100 + 100; print(a)", language="python"),
-    ]
-    code_result = executor.execute_code_blocks(code_blocks)
-    assert (
-        code_result.exit_code == 0
-        and "hello world!" in code_result.output
-        and "200" in code_result.output
-        and code_result.code_file is not None
-    )
+    for py_variant in python_variants:
+        code_blocks = [
+            CodeBlock(code="import sys; print('hello world!')", language=py_variant),
+            CodeBlock(code="a = 100 + 100; print(a)", language=py_variant),
+        ]
+        code_result = executor.execute_code_blocks(code_blocks)
+        assert (
+            code_result.exit_code == 0
+            and "hello world!" in code_result.output
+            and "200" in code_result.output
+            and code_result.code_file is not None
+        )
 
     # Test bash script.
     if sys.platform not in ["win32"]:
@@ -93,15 +99,16 @@ def _test_execute_code(executor: CodeExecutor) -> None:
         assert code_result.exit_code == 0 and "hello world!" in code_result.output and code_result.code_file is not None
 
     # Test running code.
-    file_lines = ["import sys", "print('hello world!')", "a = 100 + 100", "print(a)"]
-    code_blocks = [CodeBlock(code="\n".join(file_lines), language="python")]
-    code_result = executor.execute_code_blocks(code_blocks)
-    assert (
-        code_result.exit_code == 0
-        and "hello world!" in code_result.output
-        and "200" in code_result.output
-        and code_result.code_file is not None
-    )
+    for py_variant in python_variants:
+        file_lines = ["import sys", "print('hello world!')", "a = 100 + 100", "print(a)"]
+        code_blocks = [CodeBlock(code="\n".join(file_lines), language=py_variant)]
+        code_result = executor.execute_code_blocks(code_blocks)
+        assert (
+            code_result.exit_code == 0
+            and "hello world!" in code_result.output
+            and "200" in code_result.output
+            and code_result.code_file is not None
+        )
 
     # Check saved code file.
     with open(code_result.code_file) as f: