celery/jieheceng.py

example1="""
1、用于生活泵房地面 做法详见图纸设计及相关图集规范
2、10厚600x600防滑地砖,DTG砂浆擦缝
3、30厚DS M15砂浆(1:3干硬性水泥砂浆)结合层,表面撒水泥粉
4、最薄处30厚C20细石混凝土找坡层,向排水沟、集水坑找1%坡,随打随抹平,立管根部用DS M15砂浆(1:3水泥砂浆)抹小八字角
5、含设备基础顶面及侧面
"""
answer1="""
1、用于生活泵房地面 做法详见图纸设计及相关图集规范
2、10厚600x600防滑地砖,DTG砂浆擦缝
4、最薄处30厚C20细石混凝土找坡层,向排水沟、集水坑找1%坡,随打随抹平,立管根部用DS M15砂浆(1:3水泥砂浆)抹小八字角
5、含设备基础顶面及侧面
"""
example2="""
1、用于消防水池、集水坑井地面及侧壁、排水沟地面及侧壁 做法详见图纸设计及相关图集规范
2、20厚DS M20砂浆(1:2.5水泥砂浆)保护层
3、15厚1:3水泥砂浆找平
"""
answer2="""
1、用于消防水池、集水坑井地面及侧壁、排水沟地面及侧壁 做法详见图纸设计及相关图集规范
2、20厚DS M20砂浆(1:2.5水泥砂浆)保护层
3、15厚1:3水泥砂浆找平
"""
def jieheceng(A, aiclient, qwclient):
    completion = aiclient.chat.completions.create(
        model="glm-4.5-flash",
        #model="ZhipuAI/GLM-4.5",
        messages=[
            {"role": "system", "content": "You are a helpful assistant."},
            {"role": "user", "content": "给定一段文字: " + A +   "\n 该段文字描述了包括一道或多道工序的工作内容。其中有可能一道或多道工作内容含有“结合层”字样。请你阅读这段文字，判断是否存在“结合层”工序。如果没有，请原封不动的返回给定的文字；如果有，请将含有“结合层”字样的一道或多道工作内容删除，返回剩余的工作内容，剩余的工作内容请原封不动的返回。 例如，给定一段文字：" + example1 +"\n请返回：" + answer1 + "\n 再例如，给定一段文字： " + example2 + "\n请返回： " + answer2},
        ],
        extra_body={"thinking": {"type": "disabled"}},
        #extra_body={"enable_thinking": False},
    )
    json_string = completion.choices[0].message.content
    print(json_string)
    return json_string