chnjinlei/hive
1
0
Fork 0
Code Issues Pull Requests Actions 41 Packages Projects Releases Wiki Activity

chore: linter update

Browse Source
This commit is contained in:
bryan
2026-03-16 12:22:29 -07:00
parent c3f13c50eb
commit 69f0ff7ac9
4 changed files with 37 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/framework/server/routes_sessions.py
+3 -1
View File
@@ -64,7 +64,9 @@ def _session_to_live_dict(session) -> dict:
"loaded_at": session.loaded_at,
"uptime_seconds": round(time.time() - session.loaded_at, 1),
"intro_message": getattr(session.runner, "intro_message", "") or "",
"queen_phase": phase_state.phase if phase_state else ("staging" if session.worker_runtime else "planning"),
"queen_phase": phase_state.phase
if phase_state
else ("staging" if session.worker_runtime else "planning"),
}
core/framework/tools/flowchart_utils.py
+2 -6
View File
@@ -355,12 +355,8 @@ def generate_fallback_flowchart(
# Enrich with Goal metadata
if goal:
draft["goal"] = goal.description or goal.name or ""
draft["success_criteria"] = [
sc.description for sc in (goal.success_criteria or [])
]
draft["constraints"] = [
c.description for c in (goal.constraints or [])
]
draft["success_criteria"] = [sc.description for sc in (goal.success_criteria or [])]
draft["constraints"] = [c.description for c in (goal.constraints or [])]
# Use entry_node/terminal_nodes from GraphSpec if available
if graph.entry_node:
core/framework/tools/queen_lifecycle_tools.py
-2
View File
@@ -46,7 +46,6 @@ from framework.runner.preload_validation import credential_errors_to_json, valid
from framework.runtime.event_bus import AgentEvent, EventType
from framework.server.app import validate_agent_path
from framework.tools.flowchart_utils import (
FLOWCHART_FILENAME,
FLOWCHART_TYPES,
classify_flowchart_node,
load_flowchart_file,
@@ -293,7 +292,6 @@ def build_worker_profile(runtime: AgentRuntime, agent_path: Path | str | None =
return "\n".join(lines)
# FLOWCHART_TYPES is imported from framework.tools.flowchart_utils
core/tests/test_flowchart_utils.py
+32 -20
View File
@@ -1,7 +1,6 @@
"""Tests for framework/tools/flowchart_utils.py."""
import json
from pathlib import Path
from types import SimpleNamespace
from framework.tools.flowchart_utils import (
@@ -15,14 +14,27 @@ from framework.tools.flowchart_utils import (
)
def _make_node(id, name="Node", description="", node_type="event_loop",
tools=None, input_keys=None, output_keys=None,
success_criteria="", sub_agents=None):
def _make_node(
id,
name="Node",
description="",
node_type="event_loop",
tools=None,
input_keys=None,
output_keys=None,
success_criteria="",
sub_agents=None,
):
"""Create a minimal node-like object matching NodeSpec interface."""
return SimpleNamespace(
id=id, name=name, description=description, node_type=node_type,
tools=tools or [], input_keys=input_keys or [],
output_keys=output_keys or [], success_criteria=success_criteria,
id=id,
name=name,
description=description,
node_type=node_type,
tools=tools or [],
input_keys=input_keys or [],
output_keys=output_keys or [],
success_criteria=success_criteria,
sub_agents=sub_agents or [],
)
@@ -30,17 +42,20 @@ def _make_node(id, name="Node", description="", node_type="event_loop",
def _make_edge(source, target, condition="on_success", description=""):
"""Create a minimal edge-like object matching EdgeSpec interface."""
return SimpleNamespace(
source=source, target=target,
source=source,
target=target,
condition=SimpleNamespace(value=condition),
description=description,
)
def _make_goal(name="Test Goal", description="A test goal",
success_criteria=None, constraints=None):
def _make_goal(
name="Test Goal", description="A test goal", success_criteria=None, constraints=None
):
"""Create a minimal goal-like object matching Goal interface."""
return SimpleNamespace(
name=name, description=description,
name=name,
description=description,
success_criteria=success_criteria or [],
constraints=constraints or [],
)
@@ -49,7 +64,8 @@ def _make_goal(name="Test Goal", description="A test goal",
def _make_graph(nodes, edges, entry_node=None, terminal_nodes=None):
"""Create a minimal graph-like object matching GraphSpec interface."""
return SimpleNamespace(
nodes=nodes, edges=edges,
nodes=nodes,
edges=edges,
entry_node=entry_node or (nodes[0].id if nodes else ""),
terminal_nodes=terminal_nodes or [],
)
@@ -76,22 +92,19 @@ class TestClassifyFlowchartNode:
assert result == "browser"
def test_subprocess_node(self):
node = {"id": "n2", "node_type": "event_loop", "tools": [],
"sub_agents": ["sub1"]}
node = {"id": "n2", "node_type": "event_loop", "tools": [], "sub_agents": ["sub1"]}
edges = [{"source": "n1", "target": "n2"}, {"source": "n2", "target": "n3"}]
result = classify_flowchart_node(node, 1, 3, edges, set())
assert result == "subprocess"
def test_default_is_process(self):
node = {"id": "n2", "node_type": "event_loop", "tools": [],
"description": "do stuff"}
node = {"id": "n2", "node_type": "event_loop", "tools": [], "description": "do stuff"}
edges = [{"source": "n1", "target": "n2"}, {"source": "n2", "target": "n3"}]
result = classify_flowchart_node(node, 1, 3, edges, set())
assert result == "process"
def test_explicit_override(self):
node = {"id": "n2", "node_type": "event_loop", "tools": [],
"flowchart_type": "database"}
node = {"id": "n2", "node_type": "event_loop", "tools": [], "flowchart_type": "database"}
edges = [{"source": "n1", "target": "n2"}]
result = classify_flowchart_node(node, 1, 3, edges, set())
assert result == "database"
@@ -148,8 +161,7 @@ class TestSynthesizeDraftFromRuntime:
# Legend should contain all types
assert draft["flowchart_legend"] == {
k: {"shape": v["shape"], "color": v["color"]}
for k, v in FLOWCHART_TYPES.items()
k: {"shape": v["shape"], "color": v["color"]} for k, v in FLOWCHART_TYPES.items()
}
def test_graph_with_sub_agents(self):