wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-03 06:12:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: add null checks for runtime.context and tighten langgraph constraint (#1326)

Browse Source 
- Add null checks for runtime.context in uploads_middleware.py and
  sandbox/middleware.py to prevent NPE when langgraph runtime context is None
- Tighten langgraph version constraint from >=1.0.6 to >=1.0.6,<1.0.10
  to avoid context=None incompatibility with langgraph-api 0.7.x

Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
This commit is contained in:
吴旭云
2026-03-25 21:01:10 +08:00
committed by GitHub
parent adc51e541c
commit d7e510763d
3 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/packages/harness/deerflow/agents/middlewares/uploads_middleware.py
View File

@@ -146,7 +146,7 @@ class UploadsMiddleware(AgentMiddleware[UploadsMiddlewareState]):
return None return None
# Resolve uploads directory for existence checks # Resolve uploads directory for existence checks
thread_id = runtime.context.get("thread_id") if runtime.context else None thread_id = (runtime.context or {}).get("thread_id")
uploads_dir = self._paths.sandbox_uploads_dir(thread_id) if thread_id else None uploads_dir = self._paths.sandbox_uploads_dir(thread_id) if thread_id else None
# Get newly uploaded files from the current message's additional_kwargs.files # Get newly uploaded files from the current message's additional_kwargs.files

6
backend/packages/harness/deerflow/sandbox/middleware.py
View File

@@ -56,7 +56,9 @@ class SandboxMiddleware(AgentMiddleware[SandboxMiddlewareState]):
# Eager initialization (original behavior) # Eager initialization (original behavior)
if "sandbox" not in state or state["sandbox"] is None: if "sandbox" not in state or state["sandbox"] is None:
thread_id = runtime.context["thread_id"] thread_id = (runtime.context or {}).get("thread_id")
if thread_id is None:
return super().before_agent(state, runtime)
sandbox_id = self._acquire_sandbox(thread_id) sandbox_id = self._acquire_sandbox(thread_id)
logger.info(f"Assigned sandbox {sandbox_id} to thread {thread_id}") logger.info(f"Assigned sandbox {sandbox_id} to thread {thread_id}")
return {"sandbox": {"sandbox_id": sandbox_id}} return {"sandbox": {"sandbox_id": sandbox_id}}
@@ -71,7 +73,7 @@ class SandboxMiddleware(AgentMiddleware[SandboxMiddlewareState]):
get_sandbox_provider().release(sandbox_id) get_sandbox_provider().release(sandbox_id)
return None return None
if runtime.context and runtime.context.get("sandbox_id") is not None: if (runtime.context or {}).get("sandbox_id") is not None:
sandbox_id = runtime.context.get("sandbox_id") sandbox_id = runtime.context.get("sandbox_id")
logger.info(f"Releasing sandbox {sandbox_id} from context") logger.info(f"Releasing sandbox {sandbox_id} from context")
get_sandbox_provider().release(sandbox_id) get_sandbox_provider().release(sandbox_id)

2
backend/packages/harness/pyproject.toml
View File

@@ -13,7 +13,7 @@ dependencies = [
"langchain-deepseek>=1.0.1", "langchain-deepseek>=1.0.1",
"langchain-mcp-adapters>=0.1.0", "langchain-mcp-adapters>=0.1.0",
"langchain-openai>=1.1.7", "langchain-openai>=1.1.7",
"langgraph>=1.0.6", "langgraph>=1.0.6,<1.0.10",
"langgraph-api>=0.7.0,<0.8.0", "langgraph-api>=0.7.0,<0.8.0",
"langgraph-cli>=0.4.14", "langgraph-cli>=0.4.14",
"langgraph-runtime-inmem>=0.22.1", "langgraph-runtime-inmem>=0.22.1",