From 0541db1e345d998fce6c561fbee8afbc8dceaeea Mon Sep 17 00:00:00 2001 From: ouyangyouzhang Date: Mon, 15 Sep 2025 16:54:55 +0800 Subject: [PATCH] =?UTF-8?q?refactor:=20=E7=A7=BB=E9=99=A4llm.invoke?= =?UTF-8?q?=E4=B8=AD=E5=86=97=E4=BD=99=E7=9A=84need=5Fretry=E5=8F=82?= =?UTF-8?q?=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- rag2_0/demo/dialogue_to_workorder.py | 8 ++++---- rag2_0/dify/DifyCompareTest.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/rag2_0/demo/dialogue_to_workorder.py b/rag2_0/demo/dialogue_to_workorder.py index 43f0df9..3312f7c 100755 --- a/rag2_0/demo/dialogue_to_workorder.py +++ b/rag2_0/demo/dialogue_to_workorder.py @@ -366,7 +366,7 @@ class DialogueToWorkorder: output_format = self.user_question_and_solution_parser.get_format_instructions() llm_prompt = prompt.format(output_format=output_format, dialogue_str=dialogue_str) - response = self.llm.invoke(user_prompt=llm_prompt, need_retry=False) + response = self.llm.invoke(user_prompt=llm_prompt) clean_output = re.sub(r'.*?', '', response.content, flags=re.DOTALL) try: if clean_output.count('user_question') == 1: @@ -396,7 +396,7 @@ class DialogueToWorkorder: except Exception as e: output_format = self.user_question_and_solution_list_parser.get_format_instructions() llm_prompt = prompt.format(output_format=output_format, dialogue_str=dialogue_str) - response = self.llm.invoke(user_prompt=llm_prompt, need_retry=False) + response = self.llm.invoke(user_prompt=llm_prompt) clean_output = re.sub(r'.*?', '', response.content, flags=re.DOTALL) user_question_and_solution_temp = self.user_question_and_solution_list_parser.parse(clean_output) return user_question_and_solution_temp.user_question_list @@ -434,7 +434,7 @@ class DialogueToWorkorder: {dialogue_str} """ - response = self.llm.invoke(user_prompt=prompt, need_retry=False) + response = self.llm.invoke(user_prompt=prompt) clean_output = re.sub(r'.*?', '', response.content, flags=re.DOTALL) question_type = self.question_type_parser.parse(clean_output) @@ -471,7 +471,7 @@ class DialogueToWorkorder: """ - response = self.llm.invoke(user_prompt=prompt, need_retry=False) + response = self.llm.invoke(user_prompt=prompt) clean_output = re.sub(r'.*?', '', response.content, flags=re.DOTALL) is_complaint = self.is_complaint_parser.parse(clean_output) diff --git a/rag2_0/dify/DifyCompareTest.py b/rag2_0/dify/DifyCompareTest.py index a2e99fb..a8d3c4c 100755 --- a/rag2_0/dify/DifyCompareTest.py +++ b/rag2_0/dify/DifyCompareTest.py @@ -73,7 +73,7 @@ class DifyCompareTest: return "回答基本不相同" while retry_count < max_retries: try: - response = self.llm.invoke(user_prompt=user_prompt, need_retry=False, response_format={"type": "json_object"}) + response = self.llm.invoke(user_prompt=user_prompt, response_format={"type": "json_object"}) response.content = response.content.strip() clean_output = re.sub(r'.*?', '', response.content, flags=re.DOTALL) result = JsonOutputParser().parse(clean_output)