babaTEEpe commited on
Commit
7b587e7
·
verified ·
1 Parent(s): f414faf

Update fusion_engine.py

Browse files
Files changed (1) hide show
  1. fusion_engine.py +6 -2
fusion_engine.py CHANGED
@@ -77,10 +77,14 @@ def _new_request(session_id: str) -> int:
77
  _request_counter[session_id] = _request_counter.get(session_id, 0) + 1
78
  return _request_counter[session_id]
79
 
 
 
80
  def _is_stale(session_id: str, request_id: int) -> bool:
81
- """Check if this request has been superseded by a newer one."""
82
  with _request_lock:
83
- return _request_counter.get(session_id, 0) > request_id
 
 
84
 
85
  def _cancelled_response(session_id: str):
86
  """Return a minimal response for cancelled/superseded requests."""
 
77
  _request_counter[session_id] = _request_counter.get(session_id, 0) + 1
78
  return _request_counter[session_id]
79
 
80
+ PARALLEL_LEEWAY = 4 # Allow up to 4 concurrent workers to finish
81
+
82
  def _is_stale(session_id: str, request_id: int) -> bool:
83
+ """Check if this request is significantly older than the latest one."""
84
  with _request_lock:
85
+ latest = _request_counter.get(session_id, 0)
86
+ # Only cancel if this request is more than PARALLEL_LEEWAY steps behind the latest one
87
+ return latest > (request_id + PARALLEL_LEEWAY)
88
 
89
  def _cancelled_response(session_id: str):
90
  """Return a minimal response for cancelled/superseded requests."""