fix: update sandbox and template models to use Pydantic Field for TTL…

agentrun/integration/utils/tool.py

@@ -1562,8 +1562,8 @@ def _build_openapi_schema(
         if isinstance(schema, dict):
             properties[name] = {
                 **schema,
-                "description": (
-                    param.get("description") or schema.get("description", "")
+                "description": param.get("description") or schema.get(
+                    "description", ""
                 ),
             }
             if param.get("required"):

agentrun/sandbox/__sandbox_async_template.py

@@ -15,6 +15,8 @@
     Union,
 )
 
+from pydantic import Field
+
 from agentrun.sandbox.model import TemplateType
 from agentrun.utils.config import Config
 from agentrun.utils.model import BaseModel
@@ -56,7 +58,9 @@ class Sandbox(BaseModel):
     """沙箱全局唯一资源名称 / Sandbox ARN"""
     sandbox_id: Optional[str] = None
     """沙箱 ID / Sandbox ID"""
-    sandbox_idle_ttlin_seconds: Optional[int] = None
+    sandbox_idle_ttlin_seconds: Optional[int] = Field(
+        None, alias="sandboxIdleTTLInSeconds"
+    )
     """沙箱空闲 TTL（秒） / Sandbox Idle TTL (seconds)"""
     sandbox_idle_timeout_seconds: Optional[int] = None
     """沙箱空闲超时时间（秒） / Sandbox Idle Timeout (seconds)"""

agentrun/sandbox/__template_async_template.py

@@ -6,6 +6,8 @@
 
 from typing import Dict, List, Optional
 
+from pydantic import Field
+
 from agentrun.sandbox.model import (
     PageableInput,
     TemplateContainerConfiguration,
@@ -52,7 +54,9 @@ class Template(BaseModel):
     """执行角色 ARN / Execution Role ARN"""
     sandbox_idle_timeout_in_seconds: Optional[int] = None
     """沙箱空闲超时时间（秒） / Sandbox Idle Timeout (seconds)"""
-    sandbox_ttlin_seconds: Optional[int] = None
+    sandbox_ttlin_seconds: Optional[int] = Field(
+        None, alias="sandboxTTLInSeconds"
+    )
     """沙箱存活时间（秒） / Sandbox TTL (seconds)"""
     share_concurrency_limit_per_sandbox: Optional[int] = None
     """每个沙箱的最大并发会话数 / Max Concurrency Limit Per Sandbox"""

agentrun/sandbox/model.py

@@ -8,7 +8,7 @@
 from typing import Any, Dict, List, Optional, TYPE_CHECKING
 import uuid
 
-from pydantic import model_validator
+from pydantic import Field, model_validator
 
 from agentrun.utils.model import BaseModel
 
@@ -264,7 +264,9 @@ class TemplateInput(BaseModel):
     """执行角色 ARN / Execution Role ARN"""
     sandbox_idle_timeout_in_seconds: Optional[int] = 1800
     """沙箱空闲超时时间（秒） / Sandbox Idle Timeout (seconds)"""
-    sandbox_ttlin_seconds: Optional[int] = 21600
+    sandbox_ttlin_seconds: Optional[int] = Field(
+        21600, alias="sandboxTTLInSeconds"
+    )
     """沙箱存活时间（秒） / Sandbox TTL (seconds)"""
     share_concurrency_limit_per_sandbox: Optional[int] = 200
     """每个沙箱的最大并发会话数 / Max Concurrency Limit Per Sandbox"""

agentrun/sandbox/sandbox.py

@@ -25,6 +25,8 @@
     Union,
 )
 
+from pydantic import Field
+
 from agentrun.sandbox.model import TemplateType
 from agentrun.utils.config import Config
 from agentrun.utils.model import BaseModel
@@ -66,7 +68,9 @@ class Sandbox(BaseModel):
     """沙箱全局唯一资源名称 / Sandbox ARN"""
     sandbox_id: Optional[str] = None
     """沙箱 ID / Sandbox ID"""
-    sandbox_idle_ttlin_seconds: Optional[int] = None
+    sandbox_idle_ttlin_seconds: Optional[int] = Field(
+        None, alias="sandboxIdleTTLInSeconds"
+    )
     """沙箱空闲 TTL（秒） / Sandbox Idle TTL (seconds)"""
     sandbox_idle_timeout_seconds: Optional[int] = None
     """沙箱空闲超时时间（秒） / Sandbox Idle Timeout (seconds)"""

agentrun/sandbox/template.py

@@ -16,6 +16,8 @@
 
 from typing import Dict, List, Optional
 
+from pydantic import Field
+
 from agentrun.sandbox.model import (
     PageableInput,
     TemplateContainerConfiguration,
@@ -62,7 +64,9 @@ class Template(BaseModel):
     """执行角色 ARN / Execution Role ARN"""
     sandbox_idle_timeout_in_seconds: Optional[int] = None
     """沙箱空闲超时时间（秒） / Sandbox Idle Timeout (seconds)"""
-    sandbox_ttlin_seconds: Optional[int] = None
+    sandbox_ttlin_seconds: Optional[int] = Field(
+        None, alias="sandboxTTLInSeconds"
+    )
     """沙箱存活时间（秒） / Sandbox TTL (seconds)"""
     share_concurrency_limit_per_sandbox: Optional[int] = None
     """每个沙箱的最大并发会话数 / Max Concurrency Limit Per Sandbox"""

tests/unittests/integration/langchain/test_agent_invoke_methods.py

@@ -81,6 +81,37 @@ def _sse(data: Dict[str, Any]) -> str:
     return f"data: {json.dumps(data, ensure_ascii=False)}\n\n"
 
 
+def _start_server(app: FastAPI) -> tuple:
+    """启动 FastAPI 服务器并返回 (base_url, server, thread)
+
+    使用真实的 HTTP 服务器而不是 httpx.ASGITransport,
+    因为 ASGITransport 在 CI 环境中无法正确处理 SSE 流式响应,
+    会导致流式响应被提前取消 (CancelledError)。
+    """
+    port = _find_free_port()
+    config = uvicorn.Config(
+        app, host="127.0.0.1", port=port, log_level="warning"
+    )
+    server = uvicorn.Server(config)
+
+    thread = threading.Thread(target=server.run, daemon=True)
+    thread.start()
+
+    base_url = f"http://127.0.0.1:{port}"
+    for i in range(50):
+        try:
+            httpx.get(f"{base_url}/ag-ui/agent/health", timeout=0.2)
+            break
+        except Exception:
+            if i == 49:
+                raise RuntimeError(
+                    f"Server failed to start within {50 * 0.1}s: {base_url}"
+                )
+            time.sleep(0.1)
+
+    return base_url, server, thread
+
+
 def _build_mock_openai_app() -> FastAPI:
     """构建本地 OpenAI 协议兼容的简单服务"""
     app = FastAPI()
@@ -297,20 +328,34 @@ def parse_sse_events(content: str) -> List[Dict[str, Any]]:
 
 
 async def request_agui_events(
-    server_app,
+    server_url_or_app: Union[str, FastAPI],
     messages: List[Dict[str, str]],
     stream: bool = True,
 ) -> List[Dict[str, Any]]:
-    """发送 AG-UI 请求并返回事件列表"""
-    async with httpx.AsyncClient(
-        transport=httpx.ASGITransport(app=server_app),
-        base_url="http://test",
-    ) as client:
-        response = await client.post(
-            "/ag-ui/agent",
-            json={"messages": messages, "stream": stream},
-            timeout=60.0,
-        )
+    """发送 AG-UI 请求并返回事件列表
+
+    Args:
+        server_url_or_app: 服务器 URL 或 FastAPI app 对象
+        messages: 消息列表
+        stream: 是否流式响应
+    """
+    if isinstance(server_url_or_app, str):
+        async with httpx.AsyncClient(base_url=server_url_or_app) as client:
+            response = await client.post(
+                "/ag-ui/agent",
+                json={"messages": messages, "stream": stream},
+                timeout=60.0,
+            )
+    else:
+        async with httpx.AsyncClient(
+            transport=httpx.ASGITransport(app=server_url_or_app),
+            base_url="http://test",
+        ) as client:
+            response = await client.post(
+                "/ag-ui/agent",
+                json={"messages": messages, "stream": stream},
+                timeout=60.0,
+            )
 
     assert response.status_code == 200
     return parse_sse_events(response.text)
@@ -670,7 +715,7 @@ def assert_openai_tool_call_response(
 
 
 async def request_openai_events(
-    server_app,
+    server_url_or_app: Union[str, FastAPI],
     messages: List[Dict[str, str]],
     stream: bool = True,
 ) -> Union[List[Dict[str, Any]], Dict[str, Any]]:
@@ -681,15 +726,23 @@ async def request_openai_events(
         "stream": stream,
     }
 
-    async with httpx.AsyncClient(
-        transport=httpx.ASGITransport(app=server_app),
-        base_url="http://test",
-    ) as client:
-        response = await client.post(
-            "/openai/v1/chat/completions",
-            json=payload,
-            timeout=60.0,
-        )
+    if isinstance(server_url_or_app, str):
+        async with httpx.AsyncClient(base_url=server_url_or_app) as client:
+            response = await client.post(
+                "/openai/v1/chat/completions",
+                json=payload,
+                timeout=60.0,
+            )
+    else:
+        async with httpx.AsyncClient(
+            transport=httpx.ASGITransport(app=server_url_or_app),
+            base_url="http://test",
+        ) as client:
+            response = await client.post(
+                "/openai/v1/chat/completions",
+                json=payload,
+                timeout=60.0,
+            )
 
     assert response.status_code == 200
 
@@ -749,7 +802,16 @@ def agent_model(mock_openai_server: str):
 
 @pytest.fixture
 def server_app_astream_events(agent_model):
-    """创建使用 astream_events 的服务器（AG-UI/OpenAI 通用）"""
+    """创建使用 astream 的服务器（AG-UI/OpenAI 通用）
+
+    返回服务器 URL 而不是 app 对象,因为流式测试需要真实的 HTTP 连接。
+    httpx.ASGITransport 在 CI 环境中无法正确处理 SSE 流式响应。
+
+    注意: 这里使用 astream(stream_mode="updates") 而非 astream_events,
+    因为 astream_events 在 CI (Linux + uvicorn 线程) 环境中会出现
+    async generator 被提前取消或事件丢失的问题。
+    astream_events 的转换逻辑由 test_convert_python_3_10/3_12 单独覆盖。
+    """
     agent = build_agent(agent_model)
 
     async def invoke_agent(request: AgentRequest):
@@ -770,16 +832,21 @@ async def invoke_agent(request: AgentRequest):
         converter = AgentRunConverter()
 
         async def generator():
-            async for event in agent.astream_events(
-                cast(Any, input_data), version="v2"
+            async for event in agent.astream(
+                cast(Any, input_data), stream_mode="updates"
             ):
                 for item in converter.convert(event):
                     yield item
 
         return generator()
 
     server = AgentRunServer(invoke_agent=invoke_agent)
-    return server.app
+    base_url, uvicorn_server, thread = _start_server(server.app)
+
+    yield base_url
+
+    uvicorn_server.should_exit = True
+    thread.join(timeout=5)
 
 
 # =============================================================================
@@ -1753,16 +1820,21 @@ async def invoke_agent(request: AgentRequest):
             converter = AgentRunConverter()
 
             async def generator():
-                async for event in agent.astream_events(
-                    cast(Any, input_data), version="v2"
+                async for event in agent.astream(
+                    cast(Any, input_data), stream_mode="updates"
                 ):
                     for item in converter.convert(event):
                         yield item
 
             return generator()
 
         server = AgentRunServer(invoke_agent=invoke_agent)
-        return server.app
+        base_url, uvicorn_server, thread = _start_server(server.app)
+
+        yield base_url
+
+        uvicorn_server.should_exit = True
+        thread.join(timeout=5)
 
     @pytest.fixture
     def server_app_async(self, agent_model):
@@ -1787,16 +1859,21 @@ async def invoke_agent(request: AgentRequest):
             converter = AgentRunConverter()
 
             async def generator():
-                async for event in agent.astream_events(
-                    cast(Any, input_data), version="v2"
+                async for event in agent.astream(
+                    cast(Any, input_data), stream_mode="updates"
                 ):
                     for item in converter.convert(event):
                         yield item
 
             return generator()
 
         server = AgentRunServer(invoke_agent=invoke_agent)
-        return server.app
+        base_url, uvicorn_server, thread = _start_server(server.app)
+
+        yield base_url
+
+        uvicorn_server.should_exit = True
+        thread.join(timeout=5)
 
     @pytest.mark.parametrize(
         "case_key,prompt",