NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-23 16:54:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: harden server state management with defensive attribute access

Browse source
This commit is contained in:
RUFFY-369 2026-04-03 02:06:33 +05:30
parent 5a1ea7d3cb
commit ec64d8b375
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
atroposlib/api/server.py
View file

@ -200,6 +200,8 @@ def _process_scored_data(scored_data: ScoredData) -> Dict[str, Any]:
app.state.queue = []
if not hasattr(app.state, "buffer"):
app.state.buffer = {}
if not hasattr(app.state, "total_rollouts_processed"):
app.state.total_rollouts_processed = 0
data_dict = _scored_data_to_dict(scored_data)
env_id = data_dict.get("env_id")
@ -477,7 +479,7 @@ async def scored_data_list(scored_data_list: List[ScoredData]):
async def get_status():
try:
return {
"current_step": app.state.status_dict["step"],
"current_step": app.state.status_dict.get("step", 0),
"queue_size": len(app.state.queue),
}
except AttributeError:
@ -541,7 +543,8 @@ async def get_status_env(env: EnvIdentifier):
# Calculate total minimum allocations
total_min_allocation = 0.0
for env_config in app.state.envs:
envs = getattr(app.state, "envs", [])
for env_config in envs:
if (
env_config.get("connected", False)
and env_config.get("min_batch_allocation") is not None