async http client for requesting env vars

Author: mishushakov

template/server/contexts.py

@@ -52,7 +52,7 @@ async def create_context(client, websockets: dict, language: str, cwd: str) -> C
     session_data = response.json()
     session_id = session_data["id"]
     context_id = session_data["kernel"]["id"]
-    global_env_vars = get_envs()
+    global_env_vars = await get_envs()
 
     logger.debug(f"Created context {context_id}")
 

template/server/envs.py

@@ -1,12 +1,16 @@
 import os
 
-import requests
+import httpx
 
 LOCAL = os.getenv("E2B_LOCAL", False)
 ENVD_PORT = 49983
 
 
-def get_envs() -> dict:
+async def get_envs() -> dict:
     if LOCAL:
-        return {}
-    return requests.get(f"http://localhost:{ENVD_PORT}/envs").json()
+        return {
+            "E2B_TEST_VARIABLE": "true"
+        }
+    async with httpx.AsyncClient() as client:
+        response = await client.get(f"http://localhost:{ENVD_PORT}/envs")
+        return response.json()

template/server/messaging.py

@@ -167,7 +167,7 @@ async def set_env_vars(self, env_vars: Dict[StrictStr, str]):
                     raise ExecutionError(f"Error during execution: {item}")
 
     async def reset_env_vars(self, env_vars: Dict[StrictStr, str]):
-        global_env_vars = get_envs()
+        global_env_vars = await get_envs()
 
         # Create a dict of vars to reset and a list of vars to remove
         vars_to_reset = {}