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

JoshTrim · Mar 25, 2024 · b526232 · b526232
1 parent 2f109f5
commit b526232
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autogen/agentchat/conversable_agent.py b/autogen/agentchat/conversable_agent.py
@@ -2037,7 +2037,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: