From 50e020b75db2483b9b0fa0e51bac42df0cd73641 Mon Sep 17 00:00:00 2001 From: Xu Yang Date: Fri, 14 Feb 2025 17:03:57 +0800 Subject: [PATCH] fix: auto continue small bug (#598) --- rdagent/oai/backend/deprec.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rdagent/oai/backend/deprec.py b/rdagent/oai/backend/deprec.py index 61f4d5d7..501c7add 100644 --- a/rdagent/oai/backend/deprec.py +++ b/rdagent/oai/backend/deprec.py @@ -560,7 +560,7 @@ class DeprecBackend(APIBackend): all_response = "" new_messages = deepcopy(messages) - for _ in range(10): + for _ in range(3): if "json_mode" in kwargs: del kwargs["json_mode"] response, finish_reason = self._create_chat_completion_inner_function( @@ -586,7 +586,7 @@ class DeprecBackend(APIBackend): self.cache.chat_set(input_content_json, all_response) return all_response new_messages.append({"role": "assistant", "content": response}) - return all_response + raise RuntimeError("Failed to continue the conversation after 3 retries.") def _try_create_chat_completion_or_embedding( # type: ignore[no-untyped-def] self,