第二期“贺岁灵感模型”比赛解题思路分享

  
  
  
  
  
  

   
   
   
   
   
   

    import os
   
   
   
   
   
   
   
   
   
   
   
   

    from langchain import LLMChain
   
   
   
   
   
   
   
   
   
   
   
   

    from langchain_community.llms import QianfanLLMEndpoint
   
   
   
   
   
   
   
   
   
   
   
   

    from langchain_core.language_models.chat_models import HumanMessage
   
   
   
   
   
   
   
   
   
   
   
   

    from langchain import PromptTemplate
   
   
   
   
   
   
   
   
   
   
   
   

    from langchain.prompts.chat import (
   
   
   
   
   
   
   
   
   
   
   
   

        ChatPromptTemplate,
   
   
   
   
   
   
   
   
   
   
   
   

        SystemMessagePromptTemplate,
   
   
   
   
   
   
   
   
   
   
   
   

        HumanMessagePromptTemplate,
   
   
   
   
   
   
   
   
   
   
   
   

        MessagesPlaceholder,
   
   
   
   
   
   
   
   
   
   
   
   

    )
   
   
   
   
   
   
   
   
   
   
   
   

    # 以上代码导入langchain需要的依赖
   
   
   
   
   
   
   
   
   
   
   
   

    
   
   
   
   
   
   
   
   
   
   
   
   

    os.environ["QIANFAN_AK"] = "千帆ak"
   
   
   
   
   
   
   
   
   
   
   
   

    os.environ["QIANFAN_SK"] = "千帆sk"
   
   
   
   
   
   
   
   
   
   
   
   

    llm = QianfanLLMEndpoint(
   
   
   
   
   
   
   
   
   
   
   
   

        streaming=True,
   
   
   
   
   
   
   
   
   
   
   
   

        model="ERNIE-Bot-4",
   
   
   
   
   
   
   
   
   
   
   
   

    )
   
   
   
   
   
   
   
   
   
   
   
   

    human_template="祝福的要求是：{prompt}，春节的贺词是{response},春节贺词的字数是：{count}，用户会给你一个祝福的要求，然后给你一个春节贺词，然后给你一个贺词的字数，你要对比下要求的数字和贺词字数之间数字偏差有多少,请注意允许存在正负{count2}个字以内的偏差,如果超过{count2}个字偏差，请直接重新按照祝福的要求和字数的要求写一段春节贺词，注意标点符合也算1个字符，如果未超过{count2}个字的偏差，请直接回复ok即可"
   
   
   
   
   
   
   
   
   
   
   
   

    human_message_prompt=HumanMessagePromptTemplate.from_template(human_template)
   
   
   
   
   
   
   
   
   
   
   
   

    system_template="你是一个数学高手,精通字数计算和统计"
   
   
   
   
   
   
   
   
   
   
   
   

    system_message_prompt=SystemMessagePromptTemplate.from_template(system_template)
   
   
   
   
   
   
   
   
   
   
   
   

    chat_prompt=ChatPromptTemplate.from_messages([
   
   
   
   
   
   
   
   
   
   
   
   

        system_message_prompt,
   
   
   
   
   
   
   
   
   
   
   
   

        human_message_prompt
   
   
   
   
   
   
   
   
   
   
   
   

    ])
   
   
   
   
   
   
   
   
   
   
   
   

    chain=LLMChain(llm=llm,prompt=chat_prompt) 
   
   
   
   
   
   
   
   
   
   
   
   

    # 以上代码定义了使用的大模型和提示词模板
   
   
   
   
   
   
   
   
   
   
   
   

    
   
   
   
   
   
   
   
   
   
   
   
   

    import pandas as pd  
   
   
   
   
   
   
   
   
   
   
   
   

    df = pd.read_excel('excel.xlsx') 
   
   
   
   
   
   
   
   
   
   
   
   

    for index, row in df.iterrows(): 
   
   
   
   
   
   
   
   
   
   
   
   

        isok=chain.run(
   
   
   
   
   
   
   
   
   
   
   
   

        prompt=row['prompt'],
   
   
   
   
   
   
   
   
   
   
   
   

        response=row['response'],
   
   
   
   
   
   
   
   
   
   
   
   

        count=len(row['response']),
   
   
   
   
   
   
   
   
   
   
   
   

        count2=5
   
   
   
   
   
   
   
   
   
   
   
   

        )
   
   
   
   
   
   
   
   
   
   
   
   

        df.loc[index, 'isok'] = isok  
   
   
   
   
   
   
   
   
   
   
   
   

        print(isok)
   
   
   
   
   
   
   
   
   
   
   
   

    df.to_excel('excel.xlsx')
   
   
   
   
   
   
   
   
   
   
   
   

    #以上代码对excel文件进行操作，循环遍历每一行并且根据变量内容填充提示词模#板，并且和大语言模型交互，最后的结果如果字数符合要求返回ok，不符合要求会#重新写一段新的response。