ag2 0.9.7__py3-none-any.whl → 0.9.8.post1__py3-none-any.whl

autogen/environments/python_environment.py

@@ -17,17 +17,14 @@ class PythonEnvironment(ABC):
     _current_python_environment: ContextVar["PythonEnvironment"] = ContextVar("_current_python_environment")
 
     def __init__(self):
-        """
-        Initialize the Python environment.
-        """
+        """Initialize the Python environment."""
         self._token = None
 
         # Set up the environment
         self._setup_environment()
 
     def __enter__(self):
-        """
-        Enter the environment context.
+        """Enter the environment context.
         Sets this environment as the current one.
         """
         # Set this as the current Python environment in the context
@@ -36,8 +33,7 @@ class PythonEnvironment(ABC):
         return self
 
     def __exit__(self, exc_type, exc_val, exc_tb):
-        """
-        Exit the environment context.
+        """Exit the environment context.
         Resets the current environment and performs cleanup.
         """
         # Reset the context variable if this was the active environment
@@ -60,8 +56,7 @@ class PythonEnvironment(ABC):
 
     @abstractmethod
     def get_executable(self) -> str:
-        """
-        Get the path to the Python executable in this environment.
+        """Get the path to the Python executable in this environment.
 
         Returns:
             The full path to the Python executable.
@@ -70,8 +65,7 @@ class PythonEnvironment(ABC):
 
     @abstractmethod
     async def execute_code(self, code: str, script_path: str, timeout: int = 30) -> dict[str, Any]:
-        """
-        Execute the given code in this environment.
+        """Execute the given code in this environment.
 
         Args:
             code: The Python code to execute.
@@ -85,8 +79,7 @@ class PythonEnvironment(ABC):
 
     # Utility method for subclasses to wrap (for async support)
     def _write_to_file(self, script_path: str, content: str) -> None:
-        """
-        Write content to a file (blocking operation).
+        """Write content to a file (blocking operation).
 
         This is a helper method for use with asyncify in async contexts.
 
@@ -99,8 +92,7 @@ class PythonEnvironment(ABC):
 
     # Utility method for subclasses to wrap (for async support)
     def _run_subprocess(self, cmd: list[str], timeout: int) -> subprocess.CompletedProcess:
-        """
-        Run a subprocess (blocking operation).
+        """Run a subprocess (blocking operation).
 
         This is a helper method for use with asyncify in async contexts.
 
@@ -117,8 +109,7 @@ class PythonEnvironment(ABC):
     def get_current_python_environment(
         cls, python_environment: Optional["PythonEnvironment"] = None
     ) -> Optional["PythonEnvironment"]:
-        """
-        Get the current Python environment or the specified one if provided.
+        """Get the current Python environment or the specified one if provided.
 
         Args:
             python_environment: Optional environment to return if specified.

autogen/environments/system_python_environment.py

@@ -6,7 +6,7 @@ import logging
 import os
 import subprocess
 import sys
-from typing import Any, Optional
+from typing import Any
 
 from asyncer import asyncify
 
@@ -20,10 +20,9 @@ class SystemPythonEnvironment(PythonEnvironment):
 
     def __init__(
         self,
-        executable: Optional[str] = None,
+        executable: str | None = None,
     ):
-        """
-        Initialize a system Python environment.
+        """Initialize a system Python environment.
 
         Args:
             executable: Optional path to a specific Python executable. If None, uses the current Python executable.

autogen/environments/venv_python_environment.py

@@ -7,7 +7,7 @@ import os
 import subprocess
 import sys
 import tempfile
-from typing import Any, Optional
+from typing import Any
 
 from asyncer import asyncify
 
@@ -21,12 +21,11 @@ class VenvPythonEnvironment(PythonEnvironment):
 
     def __init__(
         self,
-        python_version: Optional[str] = None,
-        python_path: Optional[str] = None,
-        venv_path: Optional[str] = None,
+        python_version: str | None = None,
+        python_path: str | None = None,
+        venv_path: str | None = None,
     ):
-        """
-        Initialize a virtual environment for Python execution.
+        """Initialize a virtual environment for Python execution.
 
         If you pass in a venv_path the path will be checked for a valid venv. If the venv doesn't exist it will be created using the python_version or python_path provided.
 
@@ -87,8 +86,7 @@ class VenvPythonEnvironment(PythonEnvironment):
                 _ = subprocess.run(
                     [base_python, "-m", "venv", "--system-site-packages", self.venv_path],
                     check=True,
-                    stdout=subprocess.PIPE,
-                    stderr=subprocess.PIPE,
+                    capture_output=True,
                     text=True,
                 )
 
@@ -175,8 +173,7 @@ class VenvPythonEnvironment(PythonEnvironment):
             pyenv_result = subprocess.run(
                 ["pyenv", "which", f"python{self.python_version}"],
                 check=True,
-                stdout=subprocess.PIPE,
-                stderr=subprocess.PIPE,
+                capture_output=True,
                 text=True,
             )
             potential_executables.append(pyenv_result.stdout.strip())
@@ -207,8 +204,7 @@ class VenvPythonEnvironment(PythonEnvironment):
                     test_result = subprocess.run(
                         [path, "-m", "venv", "--help"],
                         check=False,  # Don't raise exception
-                        stdout=subprocess.PIPE,
-                        stderr=subprocess.PIPE,
+                        capture_output=True,
                         text=True,
                         timeout=5,  # Add timeout for safety
                     )

autogen/environments/working_directory.py

@@ -18,8 +18,7 @@ class WorkingDirectory:
     _current_working_directory: ContextVar["WorkingDirectory"] = ContextVar("_current_working_directory")
 
     def __init__(self, path: str):
-        """
-        Initialize with a directory path.
+        """Initialize with a directory path.
 
         Args:
             path: The directory path to change to.