From 0a09fd02d359d5feaaea208aa03add00e42328a6 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,