BaseDataMaintenance/BaseDataMaintenance/AIUtils/test_qwq.py

from BaseDataMaintenance.AIUtils.html2text import html2text_with_tablehtml
from BaseDataMaintenance.AIUtils.prompts import get_prompt_extract_role
from BaseDataMaintenance.AIUtils.request_qwq_32b import request_llm


a = '''
<div>
 <div> 
  <div> 
   <p>一、 <span>*</span>采购人名称：<span><a target="_blank" class="markBlue" href="/bdqyhx/785132502616420352.html" style="color: #3083EB !important;text-decoration: underline;">合肥经济技术开发区锦绣社区服务中心</a></span><br></p> 
   <p>二、 <span>*</span>履约供应商名称：<span><a target="_blank" class="markBlue" href="/bdqyhx/627394723685289984.html" style="color: #3083EB !important;text-decoration: underline;">安徽今辉科技有限公司</a></span></p> 
   <p>三、 <span>*</span>采购项目编号：<span>2071451000000740505</span></p> 
   <p>四、 <span>*</span>合同编号:</p> 
   <p>五、 <span>*</span>验收单位：<span><a target="_blank" class="markBlue" href="/bdqyhx/785132502616420352.html" style="color: #3083EB !important;text-decoration: underline;">合肥经济技术开发区锦绣社区服务中心</a></span></p> 
   <p>六、 <span>*</span>验收日期：<span>2025年3月26日</span></p> 
   <p>七、 <span>*</span>验收结果： </p> 
   <table width="100%"> 
    <tbody> 
     <tr width="100%"> 
      <td width="14.29%">序号</td> 
      <td width="14.29%">服务内容</td> 
      <td width="14.29%">验收数量</td> 
      <td width="14.29%">验收金额(元)</td> 
      <td width="14.29%">验收标准\规格型号\技术标准</td> 
      <td width="14.29%">验收结果</td> 
      <td width="14.29%" colspan="1">备注</td> 
     </tr> 
     <tr width="100%"> 
      <td width="14.29%">1</td> 
      <td width="14.29%">无毒彩泥黏土太空橡皮泥手工儿童玩具男女孩24色超轻粘土糖果袋装</td> 
      <td width="14.29%">8</td> 
      <td width="14.29%">144.0</td> 
      <td width="14.29%">无品牌\花泥</td> 
      <td width="14.29%">验收通过</td> 
      <td width="14.29%" colspan="1"></td> 
     </tr> 
     <tr width="100%"> 
      <td width="14.29%">2</td> 
      <td width="14.29%">红双喜 赛顶一星乒乓球40+ 乒乓球</td> 
      <td width="14.29%">3</td> 
      <td width="14.29%">120.0</td> 
      <td width="14.29%">红双喜/DHS\赛顶一星乒乓球40+</td> 
      <td width="14.29%">验收通过</td> 
      <td width="14.29%" colspan="1"></td> 
     </tr> 
     <tr width="100%"> 
      <td width="14.29%">3</td> 
      <td width="14.29%">新年春节红灯笼户外连串蜂窝小灯笼塑纸灯笼串节日庆典元旦装饰</td> 
      <td width="14.29%">30</td> 
      <td width="14.29%">360.0</td> 
      <td width="14.29%">无品牌\灯笼</td> 
      <td width="14.29%">验收通过</td> 
      <td width="14.29%" colspan="1"></td> 
     </tr> 
     <tr width="100%"> 
      <td width="14.29%">4</td> 
      <td width="14.29%">奥妙 3kg/瓶 洗衣液/洗衣粉</td> 
      <td width="14.29%">6</td> 
      <td width="14.29%">270.0</td> 
      <td width="14.29%">奥妙/OMO\3kg/瓶</td> 
      <td width="14.29%">验收通过</td> 
      <td width="14.29%" colspan="1"></td> 
     </tr> 
     <tr width="100%"> 
      <td width="14.29%">5</td> 
      <td width="14.29%">维达 V2182 抽纸</td> 
      <td width="14.29%">2</td> 
      <td width="14.29%">480.0</td> 
      <td width="14.29%">维达/Vinda\V2182</td> 
      <td width="14.29%">验收通过</td> 
      <td width="14.29%" colspan="1"></td> 
     </tr> 
     <tr width="100%"> 
      <td width="14.29%">6</td> 
      <td width="14.29%">横幅</td> 
      <td width="14.29%">5</td> 
      <td width="14.29%">40.0</td> 
      <td width="14.29%">无品牌\横幅</td> 
      <td width="14.29%">验收通过</td> 
      <td width="14.29%" colspan="1"></td> 
     </tr> 
     <tr width="100%"> 
      <td width="14.29%">7</td> 
      <td width="14.29%">【运费】</td> 
      <td width="14.29%">1</td> 
      <td width="14.29%">0.0</td> 
      <td width="14.29%"></td> 
      <td width="14.29%">验收通过</td> 
      <td width="14.29%" colspan="1"></td> 
     </tr> 
    </tbody> 
   </table> 
   <br> 
   <br>验收报告：  
   <br>验收人员名单： 
   <span>王玲</span> 
   <p></p> 
   <p><br></p> 
   <p><br></p> 
   <p><br></p> 
   <p><br></p> 
  </div> 
 </div>
</div>
'''

_text = html2text_with_tablehtml(a)
_text = get_prompt_extract_role(_text)
print(_text)
print(request_llm(_text,model="deepseek-r1:14b"))