@@ -48,8 +48,8 @@ def __init__(self, in_background: bool = False):
4848class ContextWebSocket :
4949 _ws : Optional [WebSocketClientProtocol ] = None
5050 _receive_task : Optional [asyncio .Task ] = None
51- global_env_vars : Optional [Dict [StrictStr , str ]] = None
52- global_env_vars_set = False
51+ _global_env_vars : Optional [Dict [StrictStr , str ]] = None
52+ _global_env_vars_set = False
5353 _cleanup_task : Optional [asyncio .Task ] = None
5454
5555 def __init__ (
@@ -165,9 +165,9 @@ def _reset_env_vars_code(self, env_vars: Dict[StrictStr, str]) -> str:
165165
166166 for key in env_vars :
167167 # Check if this var exists in global env vars
168- if self .global_env_vars and key in self .global_env_vars :
168+ if self ._global_env_vars and key in self ._global_env_vars :
169169 # Reset to global value
170- value = self .global_env_vars [key ]
170+ value = self ._global_env_vars [key ]
171171 command = self ._set_env_var_snippet (key , value )
172172 else :
173173 # Remove the variable
@@ -305,11 +305,11 @@ async def execute(
305305 # Build the complete code snippet with env vars
306306 complete_code = code
307307
308- if not self .global_env_vars :
309- self .global_env_vars = await get_envs ()
308+ if not self ._global_env_vars :
309+ self ._global_env_vars = await get_envs ()
310310
311- if not self .global_env_vars_set and self .global_env_vars :
312- env_setup_code = self ._set_env_vars_code (self .global_env_vars )
311+ if not self ._global_env_vars_set and self ._global_env_vars :
312+ env_setup_code = self ._set_env_vars_code (self ._global_env_vars )
313313 if env_setup_code :
314314 indented_env_code = self ._indent_code_with_level (env_setup_code , code_indent )
315315 complete_code = f"{ indented_env_code } \n { complete_code } "
0 commit comments