Update fusion_engine.py
Browse files- 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
|
| 82 |
with _request_lock:
|
| 83 |
-
|
|
|
|
|
|
|
| 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."""
|