From 9503bda45ddbeebfc4b502df8100b6107fd743a8 Mon Sep 17 00:00:00 2001
From: Tianjing Zeng <39507457+StCarmen@users.noreply.github.com>
Date: Fri, 16 Jan 2026 16:09:53 +0800
Subject: [PATCH] Update the sandbox pre-installed package and sandbox timeout 
 (#104)

---
 .../alias/agent/agents/_data_science_agent.py  |  2 --
 .../agent/agents/ds_agent_utils/__init__.py    |  2 --
 .../alias/agent/agents/ds_agent_utils/utils.py | 18 ------------------
 .../box/config/nginx.conf.template             |  5 ++++-
 .../runtime/alias_sandbox/box/requirements.txt |  8 +++++---
 5 files changed, 9 insertions(+), 26 deletions(-)

diff --git a/alias/src/alias/agent/agents/_data_science_agent.py b/alias/src/alias/agent/agents/_data_science_agent.py
index 8ac192d..58e24a8 100644
--- a/alias/src/alias/agent/agents/_data_science_agent.py
+++ b/alias/src/alias/agent/agents/_data_science_agent.py
@@ -31,7 +31,6 @@ from .ds_agent_utils import (
     files_filter_pre_reply_hook,
     add_ds_specific_tool,
     set_run_ipython_cell,
-    install_package,
 )
 from .ds_agent_utils.ds_config import PROMPT_DS_BASE_PATH
 
@@ -70,7 +69,6 @@ class DataScienceAgent(AliasAgentBase):
             state_saving_dir=state_saving_dir,
         )
 
-        install_package(self.toolkit.sandbox)
         set_run_ipython_cell(self.toolkit.sandbox)
 
         self.uploaded_files: List[str] = []
diff --git a/alias/src/alias/agent/agents/ds_agent_utils/__init__.py b/alias/src/alias/agent/agents/ds_agent_utils/__init__.py
index 1c40afc..6d03ee2 100644
--- a/alias/src/alias/agent/agents/ds_agent_utils/__init__.py
+++ b/alias/src/alias/agent/agents/ds_agent_utils/__init__.py
@@ -5,7 +5,6 @@ from .utils import (
     model_call_with_retry,
     set_run_ipython_cell,
     get_prompt_from_file,
-    install_package,
 )
 from .ds_toolkit import add_ds_specific_tool
 from .prompt_selector import LLMPromptSelector
@@ -19,7 +18,6 @@ __all__ = [
     "model_call_with_retry",
     "get_prompt_from_file",
     "set_run_ipython_cell",
-    "install_package",
     "add_ds_specific_tool",
     "LLMPromptSelector",
     "files_filter_pre_reply_hook",
diff --git a/alias/src/alias/agent/agents/ds_agent_utils/utils.py b/alias/src/alias/agent/agents/ds_agent_utils/utils.py
index dbfbafd..d5ba788 100644
--- a/alias/src/alias/agent/agents/ds_agent_utils/utils.py
+++ b/alias/src/alias/agent/agents/ds_agent_utils/utils.py
@@ -95,21 +95,3 @@ def set_run_ipython_cell(sandbox):
     print(
         sandbox.call_tool("run_ipython_cell", {"code": summarize_chart_code}),
     )
-
-
-def install_package(sandbox):
-    pkgs = [
-        # "pandas",
-        # "matplotlib",
-        # "numpy",
-        # "seaborn",
-        # "scipy",
-        # "scikit-learn",
-        "agentscope",
-        "qdrant-client",
-    ]
-    command = f"pip install {' '.join(pkgs)}"
-    sandbox.call_tool(
-        name="run_shell_command",
-        arguments={"command": command},
-    )
diff --git a/alias/src/alias/runtime/alias_sandbox/box/config/nginx.conf.template b/alias/src/alias/runtime/alias_sandbox/box/config/nginx.conf.template
index 00d1110..0ac66a6 100644
--- a/alias/src/alias/runtime/alias_sandbox/box/config/nginx.conf.template
+++ b/alias/src/alias/runtime/alias_sandbox/box/config/nginx.conf.template
@@ -3,6 +3,9 @@ worker_processes 1;
 events { worker_connections 1024; }
 
 http {
+    proxy_connect_timeout 120s;
+    proxy_send_timeout 120s;
+    proxy_read_timeout 120s;
     include       /etc/nginx/mime.types;
     default_type  application/octet-stream;
 
@@ -40,4 +43,4 @@ http {
             proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
         }
     }
-}
\ No newline at end of file
+}
diff --git a/alias/src/alias/runtime/alias_sandbox/box/requirements.txt b/alias/src/alias/runtime/alias_sandbox/box/requirements.txt
index 68b239d..b70b2d9 100644
--- a/alias/src/alias/runtime/alias_sandbox/box/requirements.txt
+++ b/alias/src/alias/runtime/alias_sandbox/box/requirements.txt
@@ -1,9 +1,9 @@
 ipython==8.31.0
 fastapi==0.115.6
 uvicorn==0.34.0
-pydantic==2.10.5
+pydantic==2.12.5
 requests==2.32.3
-mcp==1.9.0
+mcp==1.25.0
 aiofiles
 uv
 gitpython
@@ -13,4 +13,6 @@ pandas
 scikit-learn
 scipy
 seaborn
-matplotlib
\ No newline at end of file
+matplotlib
+agentscope[full]==1.0.11
+qdrant-client==1.15.1