# Please install OpenAI SDK first: `pip3 install openai`

from openai import OpenAI

api_key = 'sk-fb3c7301cf2540e98e7adce1d4f723a5'
client = OpenAI(api_key=api_key, base_url="https://api.deepseek.com")

sys_content = '''
一、命名实体按如下分类：
1. 结构部件类
2. 材料类
3. 技术参数类
4. 技术特征类
5. 制造工艺类
6. 功能属性类
7. 规范标准类
8. 专利法律实体类
9. 性能指标类
10. 特殊构造类
二、返回结果按yaml格式，具体格式如下：
专利号: 用户问询的专利号
摘要: 用户问询的专利摘要
结果:
  结构部件类: [周边梁, 周边柱, 內填板, 钢板]
  材料: [混凝土, 钢板, 空心钢管, 膨胀防火沙]
'''

question_content = '''
请识别如下建筑专利摘要中的命名实体并分类：{}, {}
'''


def ds_ner(patent, description):
    content = question_content.format(patent, description)
    response = client.chat.completions.create(
        model="deepseek-chat",
        messages=[
            {"role": "system", "content": sys_content},
            {"role": "user", "content": content},
        ],
        stream=False
    )

    return response.choices[0].message.content