更新

app/src/doctor_workstation/services/repository.py

@@ -19,7 +19,6 @@ from doctor_workstation.core.errors import (
     ApiBusinessError,
     ApiHttpError,
     ApiProtocolError,
-    ApiTimeoutError,
     ApiTransportError,
     AuthenticationExpiredError,
 )
@@ -1413,11 +1412,11 @@ class RemoteDoctorRepository:
         task: str = "custom",
         cancelled: Callable[[], bool] | None = None,
     ) -> Iterator[dict[str, Any]]:
-        """Stream a diagnosis answer, with one legacy fallback before first content."""
+        """Stream a diagnosis answer; use the legacy route only when streaming is absent."""
 
         clean_prompt, clean_task = _diagnosis_ai_request(diagnosis_id, prompt, task)
         body = {"id": diagnosis_id, "prompt": clean_prompt, "task": clean_task}
-        received_delta = False
+        received_event = False
         received_done = False
         try:
             for raw_event in self.client.post_event_stream(
@@ -1428,13 +1427,12 @@ class RemoteDoctorRepository:
             ):
                 if cancelled is not None and cancelled():
                     return
+                received_event = True
                 event = _normalise_diagnosis_ai_event(raw_event)
                 if event is None:
                     continue
                 kind = event["event"]
-                if kind == "delta":
-                    received_delta = True
-                elif kind == "done":
+                if kind == "done":
                     received_done = True
                 yield event
                 if kind == "done":
@@ -1443,12 +1441,16 @@ class RemoteDoctorRepository:
                 return
             if not received_done:
                 raise ApiProtocolError("AI assistant stream ended before done")
-        except (ApiHttpError, ApiProtocolError, ApiTimeoutError, ApiTransportError):
-            if received_delta or (cancelled is not None and cancelled()):
+        except ApiHttpError as error:
+            if (
+                error.status_code not in {404, 405}
+                or received_event
+                or (cancelled is not None and cancelled())
+            ):
                 raise
 
-        # Older deployments do not expose the stream route. Submit exactly one
-        # request through the confirmed non-streaming endpoint in that case.
+        # Only a missing stream route on an older deployment justifies another
+        # full request. Timeouts, transport failures and malformed streams do not.
         result = self.analyze_diagnosis_ai(
             diagnosis_id,
             clean_prompt,