import time from fallback import fallback from config import simplemodel from template import xuanxiang import json with open('zhaoping_rule', 'r') as f: content = f.read() import json obj=json.loads(content) with open('name_label', 'r') as f: content = f.read() import json name_label=json.loads(content) baohuceng = ['10-74', '10-75', '10-77', '10-78', '10-80', '10-81', '10-83', '10-84', '10-86', '10-87', '10-90'] from fallback import fallback def aifilter5(A, #options B, #data aiclient, qwclient, sfclient, dw): options=[] letters = "ABCDEFGHIJKLMN" for i in range(len(A)): options.append("给定选项" + letters[i]+"，内容为"+A[i] ) completion = aiclient.chat.completions.create( #model="THUDM/GLM-4-9B-0414", model="glm-4.5-air", #model="Qwen/Qwen3-8B", #model="ernie-speed-128k", messages=[ {"role": "system", "content": "You are a helpful assistant.请将最终答案以JSON格式输出"}, {"role": "user", "content": "问题描述： " "，".join(options) + "。请问选项中是否有龙骨选项？" + ''' 如果有，请回答 { 'answer': '有' } 如果没有，请回答 { 'answer': '没有' } ''' }, ], extra_body={"thinking": {"type": "disabled"}}, #extra_body={"enable_thinking": True}, #stream=True ) json_string = completion.choices[0].message.content print(json_string) if len(json_string.replace(" ", "")) < 10: if '没有' in json_string: return False return True completion = sfclient.chat.completions.create( model=simplemodel(), messages=[ {"role": "system", "content": "You are a helpful assistant.请将最终答案以JSON格式输出"}, {"role": "user", "content": " 给你一段文字如下， " + json_string + ",其中给出了一个“有”或者“没有”的判断，请将该中文判断输出" + ''' 如果有，请回答 { 'answer': '有' } 如果没有，请回答 { 'answer': '没有' } 你只需要输出结果，不要输出分析过程 ''' }, ], extra_body={"thinking": {"type": "disabled"}}, #extra_body={"enable_thinking": False}, ) json_string = completion.choices[0].message.content print(json_string) if '没有' in json_string: return False return True def aifilter3(A, #options B, #data aiclient, qwclient, sfclient, dw): options=[] letters = "ABCDEFGHIJKLMN" for i in range(len(A)): options.append("给定选项" + letters[i]+"，内容为"+A[i] ) completion = aiclient.chat.completions.create( #model="THUDM/GLM-4-9B-0414", model="glm-4.5-air", #model="Qwen/Qwen3-8B", #model="ernie-speed-128k", messages=[ {"role": "system", "content": "You are a helpful assistant.请将最终答案以JSON格式输出"}, {"role": "user", "content": "问题描述： 墙面装饰工程可分为外墙装饰或者内墙装饰。给定一段工作内容: " + B['label'] + " " + B['mc'] + " " + B['tz'] + "。请问该工作内容的描述指的是内墙还是外墙？" + ''' 如果是外墙，请回答 { 'answer': '外墙' } 如果是内墙，请回答 { 'answer': '内墙' } 如果无法确定，请回答 { 'answer': '不确定' } ''' }, ], extra_body={"thinking": {"type": "disabled"}}, #extra_body={"enable_thinking": True}, #stream=True ) json_string = completion.choices[0].message.content print(json_string) if len(json_string.replace(" ", "")) < 10: if '外墙' in json_string: return False return True completion = sfclient.chat.completions.create( model=simplemodel(), messages=[ {"role": "system", "content": "You are a helpful assistant.请将最终答案以JSON格式输出"}, {"role": "user", "content": " 给你一段文字如下， " + json_string + ",其中给出了一个“外墙”或者“内墙”的判断，请将该中文判断输出" + ''' 如果是外墙，请回答 { 'answer': '外墙' } 如果是内墙，请回答 { 'answer': '内墙' } 如果无法确定，请回答 { 'answer': '不确定' } 你只需要输出结果，不要输出分析过程 ''' }, ], extra_body={"thinking": {"type": "disabled"}}, #extra_body={"enable_thinking": False}, ) json_string = completion.choices[0].message.content print(json_string) if '外墙' in json_string: return False return True def aifilter1_2(A, #options B, #data aiclient, qwclient, sfclient, dw): options=[] letters = "ABCDEFGHIJKLMN" for i in range(len(A)): options.append("给定选项" + letters[i]+"，内容为"+A[i] ) completion = aiclient.chat.completions.create( model="glm-4.5-air", #model="THUDM/GLM-Z1-9B-0414", #model="ernie-speed-128k", messages=[ {"role": "system", "content": "You are a helpful assistant."}, {"role": "user", "content": " 背景知识：混凝土楼地面是面层，跟混凝土垫层是不同的施工步骤，不得混淆"}, {"role": "user", "content": " 背景知识：混凝土（砼）整体面层跟混凝土找平层是不同的施工步骤，不得混淆"}, {"role": "user", "content": " 背景知识：混凝土（砼）整体面层跟混凝土垫层是不同的施工步骤，不得混淆"}, {"role": "user", "content": " 背景知识：混凝土（砼）找平层跟混凝土垫层是不同的施工步骤，不得混淆"}, {"role": "user", "content": " 背景知识：水泥砂浆找平层跟水泥砂浆面层是不同的施工步骤，不得混淆"}, {"role": "user", "content": " 背景知识：水泥砂浆找平层跟水泥砂浆保护层是不同的施工步骤，不得混淆"}, {"role": "user", "content": " 特殊处理要求：如果工作内容描述中没有明确提到素水泥浆，则去掉所有含有“素水泥浆”字样的选项"}, {"role": "user", "content": " 特殊处理要求：如果工作内容描述中没有明确提到“加浆抹光”，则去掉所有含有“加浆抹光”字样的选项"}, {"role": "user", "content": " 特殊处理要求：如果工作内容描述中没有明确提到混凝土垫层，则去掉所有混凝土垫层的选项"}, {"role": "user", "content": " 特殊处理要求：如果选项中同时存在“冷轧带肋钢筋”选项和“抗裂基层 热镀锌钢丝网”选项，则去掉热镀锌钢丝网的选项"}, {"role": "user", "content": " 重要提示：选项指的是给定的A、B、C之类的选项，不是指的工作内容中的可能的1、2、3这样罗列的特征"}, {"role": "user", "content": " 重要提示：除特殊处理要求提及的内容外，不需考虑选项内容与工作内容是否符合，只需要根据特殊处理要求做出处理"}, {"role": "user", "content": "问题描述： 给定一段工作内容: " + B['label'] + " " + B['mc'] + " " + B['tz'] + "，".join(options) + "。请根据处理要求做出处理，并返回结果,删除选项必须对应到明确的特殊处理要求，不要擅自删除选项.例如，如果处理完后剩余A,B,C三个选项，请返回[A,B,C]"}, ], extra_body={"thinking": {"type": "disabled"}}, #extra_body={"enable_thinking": True}, #stream=True ) #done_thinking = False #json_string="" #thinking_json_string="" #for chunk in completion: # thinking_chunk = chunk.choices[0].delta.reasoning_content # answer_chunk = chunk.choices[0].delta.content # if thinking_chunk != '': # thinking_json_string = thinking_json_string + thinking_chunk # elif answer_chunk != '': # if not done_thinking: # done_thinking = True # json_string = json_string + answer_chunk json_string = completion.choices[0].message.content #print(completion.choices[0].message.reasoning_content) print(json_string) if len([x for x in json_string if x != ',' and x != '[' and x != ']' and x != ' ' and (x < 'A' or x > 'M')]) < 5: answer=[] if 'A' in json_string and len(A) > 0: answer.append(A[0]) if 'B' in json_string and len(A) > 1: answer.append(A[1]) if 'C' in json_string and len(A) > 2: answer.append(A[2]) if 'D' in json_string and len(A) > 3: answer.append(A[3]) if 'E' in json_string and len(A) > 4: answer.append(A[4]) if 'F' in json_string and len(A) > 5: answer.append(A[5]) if 'G' in json_string and len(A) > 6: answer.append(A[6]) if 'H' in json_string and len(A) > 7: answer.append(A[7]) if 'I' in json_string and len(A) > 8: answer.append(A[8]) if 'J' in json_string and len(A) > 9: answer.append(A[9]) return answer completion = sfclient.chat.completions.create( model=simplemodel(), messages=xuanxiang(json_string), extra_body={"thinking": {"type": "disabled"}}, #extra_body={"enable_thinking": False}, ) json_string = completion.choices[0].message.content print(json_string) answer=[] if 'A' in json_string and len(A) > 0: answer.append(A[0]) if 'B' in json_string and len(A) > 1: answer.append(A[1]) if 'C' in json_string and len(A) > 2: answer.append(A[2]) if 'D' in json_string and len(A) > 3: answer.append(A[3]) if 'E' in json_string and len(A) > 4: answer.append(A[4]) if 'F' in json_string and len(A) > 5: answer.append(A[5]) if 'G' in json_string and len(A) > 6: answer.append(A[6]) if 'H' in json_string and len(A) > 7: answer.append(A[7]) if 'I' in json_string and len(A) > 8: answer.append(A[8]) if 'J' in json_string and len(A) > 9: answer.append(A[9]) return answer def aifilter1_3(A, #options B, #data aiclient, qwclient, sfclient, dw): options=[] letters = "ABCDEFGHIJKLMN" for i in range(len(A)): options.append("给定选项" + letters[i]+"，内容为"+A[i] ) completion = aiclient.chat.completions.create( model="glm-4.5-air", #model="THUDM/GLM-Z1-9B-0414", #model="ernie-speed-128k", messages=[ {"role": "system", "content": "You are a helpful assistant."}, {"role": "user", "content": " 背景知识：混凝土楼地面是面层，跟混凝土垫层是不同的施工步骤，不得混淆"}, {"role": "user", "content": " 背景知识：混凝土（砼）整体面层跟混凝土找平层是不同的施工步骤，不得混淆"}, {"role": "user", "content": " 背景知识：混凝土（砼）整体面层跟混凝土垫层是不同的施工步骤，不得混淆"}, {"role": "user", "content": " 背景知识：混凝土（砼）找平层跟混凝土垫层是不同的施工步骤，不得混淆"}, {"role": "user", "content": " 背景知识：水泥砂浆找平层跟水泥砂浆面层是不同的施工步骤，不得混淆"}, {"role": "user", "content": " 背景知识：水泥砂浆找平层跟水泥砂浆保护层是不同的施工步骤，不得混淆"}, {"role": "user", "content": " 特殊处理要求：如果工作内容描述中没有明确提到石材面刷防护剂，则去掉所有含有“石材面刷防护剂”字样的选项"}, {"role": "user", "content": " 重要提示：选项指的是给定的A、B、C之类的选项，不是指的工作内容中的可能的1、2、3这样罗列的特征"}, {"role": "user", "content": " 重要提示：除特殊处理要求提及的内容外，不需考虑选项内容与工作内容是否符合，只需要根据特殊处理要求做出处理"}, {"role": "user", "content": "问题描述： 给定一段工作内容: " + B['label'] + " " + B['mc'] + " " + B['tz'] + "，".join(options) + "。请根据处理要求做出处理，并返回结果,删除选项必须对应到明确的特殊处理要求，不要擅自删除选项.例如，如果处理完后剩余A,B,C三个选项，请返回[A,B,C]"}, ], extra_body={"thinking": {"type": "disabled"}}, #extra_body={"enable_thinking": True}, #stream=True ) #done_thinking = False #json_string="" #thinking_json_string="" #for chunk in completion: # thinking_chunk = chunk.choices[0].delta.reasoning_content # answer_chunk = chunk.choices[0].delta.content # if thinking_chunk != '': # thinking_json_string = thinking_json_string + thinking_chunk # elif answer_chunk != '': # if not done_thinking: # done_thinking = True # json_string = json_string + answer_chunk json_string = completion.choices[0].message.content #print(completion.choices[0].message.reasoning_content) print(json_string) if len([x for x in json_string if x != ',' and x != '[' and x != ']' and x != ' ' and (x < 'A' or x > 'M')]) < 5: answer=[] if 'A' in json_string and len(A) > 0: answer.append(A[0]) if 'B' in json_string and len(A) > 1: answer.append(A[1]) if 'C' in json_string and len(A) > 2: answer.append(A[2]) if 'D' in json_string and len(A) > 3: answer.append(A[3]) if 'E' in json_string and len(A) > 4: answer.append(A[4]) if 'F' in json_string and len(A) > 5: answer.append(A[5]) if 'G' in json_string and len(A) > 6: answer.append(A[6]) if 'H' in json_string and len(A) > 7: answer.append(A[7]) if 'I' in json_string and len(A) > 8: answer.append(A[8]) if 'J' in json_string and len(A) > 9: answer.append(A[9]) return answer completion = sfclient.chat.completions.create( model=simplemodel(), messages=xuanxiang(json_string), extra_body={"thinking": {"type": "disabled"}}, #extra_body={"enable_thinking": False}, ) json_string = completion.choices[0].message.content print(json_string) answer=[] if 'A' in json_string and len(A) > 0: answer.append(A[0]) if 'B' in json_string and len(A) > 1: answer.append(A[1]) if 'C' in json_string and len(A) > 2: answer.append(A[2]) if 'D' in json_string and len(A) > 3: answer.append(A[3]) if 'E' in json_string and len(A) > 4: answer.append(A[4]) if 'F' in json_string and len(A) > 5: answer.append(A[5]) if 'G' in json_string and len(A) > 6: answer.append(A[6]) if 'H' in json_string and len(A) > 7: answer.append(A[7]) if 'I' in json_string and len(A) > 8: answer.append(A[8]) if 'J' in json_string and len(A) > 9: answer.append(A[9]) return answer def aifilter4(A, #options B, #data aiclient, qwclient, sfclient, dw): options=[] letters = "ABCDEFGHIJKLMN" for i in range(len(A)): options.append("给定选项" + letters[i]+"，内容为"+A[i] ) completion = aiclient.chat.completions.create( model="glm-4.5-air", #model="THUDM/GLM-Z1-9B-0414", #model="ernie-speed-128k", messages=[ {"role": "system", "content": "You are a helpful assistant."}, {"role": "user", "content": " 特殊处理要求：如果工作内容没有提及界面剂，则去掉所有含有“界面剂”字样的选项"}, {"role": "user", "content": " 重要提示：选项指的是给定的A、B、C之类的选项，不是指的工作内容中的可能的1、2、3这样罗列的特征"}, {"role": "user", "content": " 重要提示：除特殊处理要求提及的内容外，不需考虑选项内容与工作内容是否符合，只需要根据特殊处理要求做出处理"}, {"role": "user", "content": "问题描述： 给定一段工作内容: " + B['label'] + " " + B['mc'] + " " + B['tz'] + "，".join(options) + "。请根据处理要求做出处理，并返回结果, 删除选项必须对应到明确的特殊处理要求，不要擅自删除选项。例如，如果处理完后剩余A,B,C三个选项，请返回[A,B,C]"}, ], extra_body={"thinking": {"type": "disabled"}}, #extra_body={"enable_thinking": True}, #stream=True ) #done_thinking = False #json_string="" #thinking_json_string="" #for chunk in completion: # thinking_chunk = chunk.choices[0].delta.reasoning_content # answer_chunk = chunk.choices[0].delta.content # if thinking_chunk != '': # thinking_json_string = thinking_json_string + thinking_chunk # elif answer_chunk != '': # if not done_thinking: # done_thinking = True # json_string = json_string + answer_chunk json_string = completion.choices[0].message.content #print(completion.choices[0].message.reasoning_content) print(json_string) if len([x for x in json_string if x != ',' and x != '[' and x != ']' and x != ' ' and (x < 'A' or x > 'M')]) < 5: answer=[] if 'A' in json_string and len(A) > 0: answer.append(A[0]) if 'B' in json_string and len(A) > 1: answer.append(A[1]) if 'C' in json_string and len(A) > 2: answer.append(A[2]) if 'D' in json_string and len(A) > 3: answer.append(A[3]) if 'E' in json_string and len(A) > 4: answer.append(A[4]) if 'F' in json_string and len(A) > 5: answer.append(A[5]) if 'G' in json_string and len(A) > 6: answer.append(A[6]) if 'H' in json_string and len(A) > 7: answer.append(A[7]) if 'I' in json_string and len(A) > 8: answer.append(A[8]) if 'J' in json_string and len(A) > 9: answer.append(A[9]) return answer completion = sfclient.chat.completions.create( #model="glm-4.5-flash", model=simplemodel(), messages=xuanxiang(json_string), extra_body={"thinking": {"type": "disabled"}}, #extra_body={"enable_thinking": False}, ) json_string = completion.choices[0].message.content print(json_string) answer=[] if 'A' in json_string and len(A) > 0: answer.append(A[0]) if 'B' in json_string and len(A) > 1: answer.append(A[1]) if 'C' in json_string and len(A) > 2: answer.append(A[2]) if 'D' in json_string and len(A) > 3: answer.append(A[3]) if 'E' in json_string and len(A) > 4: answer.append(A[4]) if 'F' in json_string and len(A) > 5: answer.append(A[5]) if 'G' in json_string and len(A) > 6: answer.append(A[6]) if 'H' in json_string and len(A) > 7: answer.append(A[7]) if 'I' in json_string and len(A) > 8: answer.append(A[8]) if 'J' in json_string and len(A) > 9: answer.append(A[9]) return answer def aifilter1(A, #options B, #data aiclient, qwclient, sfclient, dw): options=[] letters = "ABCDEFGHIJKLMN" for i in range(len(A)): options.append("给定选项" + letters[i]+"，内容为"+A[i] ) completion = aiclient.chat.completions.create( model="glm-4.5-air", #model="THUDM/GLM-Z1-9B-0414", #model="ernie-speed-128k", messages=[ {"role": "system", "content": "You are a helpful assistant."}, {"role": "user", "content": " 特殊处理要求一：去掉所有精确含有“刷（喷）浆”字样的选项"}, {"role": "user", "content": " 特殊处理要求二：去掉所有刷素水泥浆的选项"}, {"role": "user", "content": " 重要提示：选项指的是给定的A、B、C之类的选项，不是指的工作内容中的可能的1、2、3这样罗列的特征"}, {"role": "user", "content": " 重要提示：除特殊处理要求提及的内容外，不需考虑选项内容与工作内容是否符合，只需要根据特殊处理要求做出处理"}, {"role": "user", "content": "问题描述： 给定一段工作内容: " + B['label'] + " " + B['mc'] + " " + B['tz'] + "，".join(options) + "。请根据处理要求做出处理，并返回结果, 删除选项必须对应到明确的特殊处理要求，不要擅自删除选项。例如，如果处理完后剩余A,B,C三个选项，请返回[A,B,C]"}, ], extra_body={"thinking": {"type": "disabled"}}, #extra_body={"enable_thinking": True}, #stream=True ) #done_thinking = False #json_string="" #thinking_json_string="" #for chunk in completion: # thinking_chunk = chunk.choices[0].delta.reasoning_content # answer_chunk = chunk.choices[0].delta.content # if thinking_chunk != '': # thinking_json_string = thinking_json_string + thinking_chunk # elif answer_chunk != '': # if not done_thinking: # done_thinking = True # json_string = json_string + answer_chunk json_string = completion.choices[0].message.content #print(completion.choices[0].message.reasoning_content) print(json_string) if len([x for x in json_string if x != ',' and x != '[' and x != ']' and x != ' ' and (x < 'A' or x > 'M')]) < 5: answer=[] if 'A' in json_string and len(A) > 0: answer.append(A[0]) if 'B' in json_string and len(A) > 1: answer.append(A[1]) if 'C' in json_string and len(A) > 2: answer.append(A[2]) if 'D' in json_string and len(A) > 3: answer.append(A[3]) if 'E' in json_string and len(A) > 4: answer.append(A[4]) if 'F' in json_string and len(A) > 5: answer.append(A[5]) if 'G' in json_string and len(A) > 6: answer.append(A[6]) if 'H' in json_string and len(A) > 7: answer.append(A[7]) if 'I' in json_string and len(A) > 8: answer.append(A[8]) if 'J' in json_string and len(A) > 9: answer.append(A[9]) return answer completion = sfclient.chat.completions.create( #model="glm-4.5-flash", model=simplemodel(), messages=xuanxiang(json_string), extra_body={"thinking": {"type": "disabled"}}, #extra_body={"enable_thinking": False}, ) json_string = completion.choices[0].message.content print(json_string) answer=[] if 'A' in json_string and len(A) > 0: answer.append(A[0]) if 'B' in json_string and len(A) > 1: answer.append(A[1]) if 'C' in json_string and len(A) > 2: answer.append(A[2]) if 'D' in json_string and len(A) > 3: answer.append(A[3]) if 'E' in json_string and len(A) > 4: answer.append(A[4]) if 'F' in json_string and len(A) > 5: answer.append(A[5]) if 'G' in json_string and len(A) > 6: answer.append(A[6]) if 'H' in json_string and len(A) > 7: answer.append(A[7]) if 'I' in json_string and len(A) > 8: answer.append(A[8]) if 'J' in json_string and len(A) > 9: answer.append(A[9]) return answer def aifilter2(A, #options B, #data aiclient, qwclient, dw): hit_wumian = False for entry in A: if entry in obj['wumian']: hit_wumian=True hit_loumian = False loumian_entry = '' for entry in A: if entry in obj['loumian']: hit_loumian=True loumian_entry = entry if hit_wumian and hit_loumian: return [x for x in A if x != loumian_entry] return A def postprocess0113(selected, data, aiclient, qwclient, sfclient, label_name, name_dw, candidates): prime = aifilter1(selected, data, aiclient, qwclient, sfclient, name_dw) if len(prime) == 0: selected = fallback(candidates, data, aiclient, qwclient, sfclient, None, None) return selected return prime