fix(environment): fix package references

Co-authored-by: Liang Wu <wuliang@google.com>
PiperOrigin-RevId: 891960308

Author: wuliang229

src/google/adk/environment/_local_environment.py

@@ -27,8 +27,8 @@
 from typing_extensions import override
 
 from ..utils.feature_decorator import experimental
-from .base_environment import BaseEnvironment
-from .base_environment import ExecutionResult
+from ._base_environment import BaseEnvironment
+from ._base_environment import ExecutionResult
 
 logger = logging.getLogger('google_adk.' + __name__)
 

src/google/adk/tools/environment/_environment_toolset.py

@@ -26,14 +26,14 @@
 from ...utils.feature_decorator import experimental
 from ..base_toolset import BaseToolset
 from ._constants import ENVIRONMENT_INSTRUCTION
-from ._tools import _EditFileTool
-from ._tools import _ExecuteTool
-from ._tools import _ReadFileTool
-from ._tools import _WriteFileTool
+from ._tools import EditFileTool
+from ._tools import ExecuteTool
+from ._tools import ReadFileTool
+from ._tools import WriteFileTool
 
 if TYPE_CHECKING:
   from ...agents.readonly_context import ReadonlyContext
-  from ...environments._base_environment import BaseEnvironment
+  from ...environment._base_environment import BaseEnvironment
   from ...models.llm_request import LlmRequest
   from ..base_tool import BaseTool
   from ..tool_context import ToolContext
@@ -82,10 +82,10 @@ async def get_tools(
       await self._environment.initialize()
       self._environment_initialized = True
     return [
-        _ExecuteTool(self._environment),
-        _ReadFileTool(self._environment),
-        _EditFileTool(self._environment),
-        _WriteFileTool(self._environment),
+        ExecuteTool(self._environment),
+        ReadFileTool(self._environment),
+        EditFileTool(self._environment),
+        WriteFileTool(self._environment),
     ]
 
   @override

src/google/adk/tools/environment/_tools.py

@@ -24,8 +24,8 @@
 from google.genai import types
 from typing_extensions import override
 
-from ...environments._base_environment import BaseEnvironment
-from ...environments._base_environment import ExecutionResult
+from ...environment._base_environment import BaseEnvironment
+from ...environment._base_environment import ExecutionResult
 from ...utils.feature_decorator import experimental
 from ..base_tool import BaseTool
 from ._constants import DEFAULT_TIMEOUT