Liqun/custom image (#427)

docker/ces_container/entrypoint.sh

@@ -9,5 +9,6 @@ groupmod -g $GROUP_ID taskweaver
 
 chown -R taskweaver:taskweaver /app
 
-su taskweaver -c "python -m taskweaver.ces.kernel.launcher"
+su taskweaver -c "python -m venv --system-site-packages venv"
+su taskweaver -c "bash -c 'source venv/bin/activate; python -m taskweaver.ces.kernel.launcher'"
 

taskweaver/ces/kernel/launcher.py

@@ -83,11 +83,7 @@ def start_app():
 if __name__ == "__main__":
     if sys.path[0] == "":
         del sys.path[0]
-    import site
 
-    user_site_packages = site.getusersitepackages()
-    if user_site_packages not in sys.path:
-        sys.path.append(site.getusersitepackages())
     logger.info("Starting process...")
     logger.info("sys.path: %s", sys.path)
     logger.info("os.getcwd(): %s", os.getcwd())

tests/unit_tests/test_environment.py

@@ -161,7 +161,11 @@ def test_environment_start_outside_container():
     cwd = os.path.dirname(os.path.abspath(__file__))
     sessions = os.path.join(cwd, "sessions")
     try:
-        env = Environment("local", env_mode=EnvMode.Container)
+        env = Environment(
+            "local",
+            env_mode=EnvMode.Container,
+            custom_image="taskweavercontainers/taskweaver-executor:0.3",
+        )
         env.start_session(
             session_id="session_id",
             session_dir=os.path.join(sessions, "session_id"),
@@ -177,13 +181,11 @@ def test_environment_start_outside_container():
         connection_file = glob.glob(conn_file_glob)[0]
         ports_file = os.path.join(ces_dir, "ports.json")
         assert os.path.isfile(ports_file)
-
-        connect_and_execute_code(connection_file, ports_file)
-
-        saved_file = os.path.join(session_dir, "cwd", "filename.txt")
-        assert os.path.isfile(saved_file)
+        code = "!pip install yfinance\nimport sys\nprint(sys.path)\nimport yfinance as yf\nprint(yf.__version__)"
+        connect_and_execute_code(connection_file, ports_file, code=code)
 
         env.stop_session("session_id")
+
     finally:
         # delete sessions
         shutil.rmtree(sessions)