fuzhu_util.py

@@ -1,5 +1,6 @@
 from config import simplemodel
 def fuzhu_util(
+       city,
        label,
        options,
        work,
@@ -18,7 +19,7 @@ def fuzhu_util(
                 {"role": "system", "content": "You are a helpful assistant."},
                 {"role": "user", "content": "以下是一段工作内容的描述：" + work + '\n'},
                 {"role": "user", "content": "该工作内容的描述中包含了一道工序如下：" + label + '\n'},
-                {"role": "user", "content": "该工序与工作内容可能有稍许偏差，需要修正或补充。施工地区设定在南通市." + '; '.join(xuanxiang) + "\n请从上述选项中选择出符合工作内容描述，有明确依据确实可以修正、补充的选项并返回。例如，选项中如果有除锈等级的修正，则你需要分析工作内容描述中是否明确除锈等级。如果明确除锈等级，你应该选择对应选项。如果没有，则不得选择。再比如，选项中提及原材料每米重量5kg以内为小型构件。那么你要分析工作内容中是否给出原材料每米重量，如果给出，则你可以据此做出选择，如果没有给出，你不得选择该选项。再比如，选项中提及，（如果）设计采用螺栓的，如何如何，那么你要分析工作内容，如果工作内容明确描述设计使用螺栓的，你可以据此做出选择，如果没有，则你不得选择。再比如，选项中提到坡度大于10°的斜板，那么你要分析工作内容，如果工作内容提及斜板并且明确提及坡度，并且该坡度大于10°，则你可以据此做出选择，如果没有，则你不的选择。请给出分析并返回类似[A, B]这样的数组。如果没有任何选项符合，则返回空数组[]"},
+                {"role": "user", "content": "该工序与工作内容可能有稍许偏差，需要修正或补充。施工地区设定在"+city+"市." + '; '.join(xuanxiang) + "\n请从上述选项中选择出符合工作内容描述，有明确依据确实可以修正、补充的选项并返回。例如，选项中如果有除锈等级的修正，则你需要分析工作内容描述中是否明确除锈等级。如果明确除锈等级，你应该选择对应选项。如果没有，则不得选择。再比如，选项中提及原材料每米重量5kg以内为小型构件。那么你要分析工作内容中是否给出原材料每米重量，如果给出，则你可以据此做出选择，如果没有给出，你不得选择该选项。再比如，选项中提及，（如果）设计采用螺栓的，如何如何，那么你要分析工作内容，如果工作内容明确描述设计使用螺栓的，你可以据此做出选择，如果没有，则你不得选择。再比如，选项中提到坡度大于10°的斜板，那么你要分析工作内容，如果工作内容提及斜板并且明确提及坡度，并且该坡度大于10°，则你可以据此做出选择，如果没有，则你不的选择。请给出分析并返回类似[A, B]这样的数组。如果没有任何选项符合，则返回空数组[]"},
             ],
             extra_body={"thinking": {"type": "disabled"}},
         )

tasks.py

@@ -126,19 +126,19 @@ def analyze(json_string):
         bh_list.append([entry, bh[entry], bz['SM'][entry]])
     print(bh_list)
     return bh_list,obj['bz_selected2']
-def select_fuzhu_az(input, label, aiclient, qwclient, client):
+def select_fuzhu_az(city,input, label, aiclient, qwclient, client):
     if label_name[input[0]] in az_fuzhu_candidate:
         print(input[0])
         print(input[1][0])
         print(input[1][1])
-        return fuzhu_util(input[0], [x[2] for x in input[1][0]], label, aiclient, qwclient, client),input[1][1],input[1][0]
+        return fuzhu_util(city,input[0], [x[2] for x in input[1][0]], label, aiclient, qwclient, client),input[1][1],input[1][0]
     return [],{},[]
-def select_fuzhu(input, label, aiclient, qwclient, client):
+def select_fuzhu(city,input, label, aiclient, qwclient, client):
     if label_name[input[0]] in fuzhu_candidate:
         print(input[0])
         print(input[1][0])
         print(input[1][1])
-        return fuzhu_util(input[0], [x[2] for x in input[1][0]], label, aiclient, qwclient, client),input[1][1],input[1][0]
+        return fuzhu_util(city,input[0], [x[2] for x in input[1][0]], label, aiclient, qwclient, client),input[1][1],input[1][0]
     return [],{},[]
 def callzailonggu_0113(data):
     time.sleep(1)
@@ -972,7 +972,7 @@ def process_data(data:dict)-> dict:
     result = [(zuhe(x[0], label, data['bianma']).replace(',',''), x[1]) for x in result]  
     print("after zuhe")
     print(result)
-    if data['bianma'].startswith('01'):
+    if not data['bianma'].startswith('03'):
         result = [(x[0], x[1], dedata.read_singledexilie2(10, x[0])) for x in result]
         result = [(x[0], x[1], x[2], tihuan(x[0], x[2], label, aiclient, qwclient, simpleclient, data)) for x in result]
     else:
@@ -981,10 +981,10 @@ def process_data(data:dict)-> dict:
     fuzhu = [analyze(x[2]) for x in result]
     fuzhu = zip(selected, fuzhu)
     fuzhu = [x for x in fuzhu]
-    if data['bianma'].startswith('01'):
-        fuzhu_selected = [select_fuzhu(x, label, aiclient, qwclient, simpleclient) for x in fuzhu]
+    if not data['bianma'].startswith('03'):
+        fuzhu_selected = [select_fuzhu(data['city'],x, label, aiclient, qwclient, simpleclient) for x in fuzhu]
     else:
-        fuzhu_selected = [select_fuzhu_az(x, label, aiclient, qwclient, simpleclient) for x in fuzhu]
+        fuzhu_selected = [select_fuzhu_az(data['city'], x, label, aiclient, qwclient, simpleclient) for x in fuzhu]
 
     need = need_extra(data, aiclient, qwclient, simpleclient, result, name_label)
     replace={}