|
|
#coding:utf8import re
def get_content(inputdata,logging): """
重新组装参数 :param inputdata:原json数据 :return: 组装的prompt及其他参数 """
res={} admin=inputdata["metadata"]["admin"] data=inputdata["data"] prompt=admin["prompt"] if_user=re.findall("{{(.*)}}",prompt) if_data=re.findall("@@(.*)@@",prompt) user_data=inputdata["metadata"]["user"] if if_user!=[] and if_user[0] in user_data.keys(): tmp=user_data[if_user[0]] prompt=re.sub("{{(.*)}}",tmp,prompt) if if_data!=[] and if_data[0] in data.keys(): tmp1=data[if_data[0]] prompt=re.sub("@@(.*)@@",tmp1,prompt) res["prompt"]=prompt res["authorization"]=admin["authorization"] res["model"]=admin["model"] res["temperature"]=admin["temperature"] res["authorization"]=admin["authorization"] res["top_p"]=admin["top_p"] res["n"]=admin["n"] return res
if __name__=="__main__": inputdata={ "id":1, "module":"ChatGPT", "version":1, "name":"信息抽取", "describe":"此步骤进行相关信息抽取", "metadata":{ "position":[ 100, 200 ], "output":{ "output_type":"table", "label_col":[ "文件名称", "识别内容", "文件路径", "文件大小", "上传时间", "GPT处理结果" ] }, "input":{ "input_type":"text", "label":[ "3_文件名称", "3_识别内容", "3_文件路径", "3_文件大小", "3_上传时间" ] }, "admin":{ "prompt":"下面我给出一段数据，请抽取相关内容。需抽取的内容是{{tag}}。数据为@@3_识别内容@@", "authorization":"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", "model":"gpt-3.5-turbo", "temperature":"0.2", "top_p":"1", "N":"1", "user_input":[ { "keyname":"tag", "keydesc":"需抽取内容" } ] }, "user":{ "tag":"专利号，专利名称，申请人" } }, "data":{ "3_文件名称":"测试的专利文档.pdf", "3_识别内容":"\n证书号第2353566号\n发明专利证书\n发明名称：一种浅海大型复杂沙波区地形重构方法\n发 明 人：张华国；傅斌;何谢错;厉冬玲;史爱琴;楼璘林\n专 利 号：ZL 2015 1 0071764.4\n专利申请日：2015年02月11日 专利权人：国家海洋局第二海洋研究所 授权公告日：2017年01月18日\n本发明经过本局依照中华人民共和国专利法进行审查，决定授予专利权，颁发本证书 并在专利登记簿上予以登记-专利权自授权公告之日起生效。\n本专利的专利权期限为二十年，自申请日起算。专利权人应当依照专利法及其实施细 则规定缴纳年费。本专利的年费应当在每年02月11日前缴纳。未按照规定缴纳年费的， 专利权自应当缴纳年费期满之日起终止„\n专利证书记载专利权登记时的法律状况。专利权的转移、质押、无效、终止、恢复和 专利权人的姓名或名称、国籍、地址变更等事项记载在专利登记簿上。 \n", "3_文件路径":"http://10.0.32.50:/data2/lybtmp/install/知识包专利/测试的专利文档.pdf", "3_文件大小":"250KB", "3_上传时间":1687835515 }, "next_app_id":[ ], "wait_condition":[ ], "start_tag":"false"} a=get_content(inputdata) print(a)
|
