Day29 - 大模型RAG，检索增强生成

    pip install langchain-community -U

    pip install dashscope

    pip install python-dotenv

    DASHSCOPE_API_KEY="your-api-key"

 from dotenv import load_dotenv

    
 load_dotenv()
    
  
    
 # 加载阿里大模型
    
 from langchain_community.chat_models import ChatTongyi
    
  
    
 def get_tongyi_chat_model():
    
     return ChatTongyi(model="qwen-turbo",
    
                   temperature=0.1,
    
                   top_p=0.3,
    
                   max_tokens=512)
    
  
    
 if __name__ == "__main__":
    
     # 测试
    
     model = get_tongyi_chat_model()
    
     response = model.invoke("你好")
    
     print(response)

 """

    
     1. 引入模型
    
 """
    
 from utils.models import get_tongyi_chat_model
    
  
    
 model = get_tongyi_chat_model()
    
  
    
 model.invoke(input="你好")

 """

    
     2. 阅读理解
    
     给你一个段落，按照段落的描述来回答问题！
    
 """
    
 from langchain_core.messages import SystemMessage
    
 from langchain_core.messages import HumanMessage
    
  
    
 sys_msg = SystemMessage(content="""
    
 你是一个阅读理解机器人！
    
 请认真阅读用户提供的文章，并根据文章的内容来回答用户的问题！、
    
 请注意：如果问题没有在文章中出现，请直接使用你自己的知识来回答！
    
 """)
    
  
    
 user_msg = HumanMessage(content="""
    
 文章为：
    
   17. 孙中山（1866年11月12日-1925年3月12日），名文，字载之，号日新，又号逸仙，又名帝象，化名中山樵，伟大的民族英雄、伟大的爱国主义者、中国民主革命的伟大先驱 [1]，中华民国和中国国民党的缔造者，三民主义的倡导者，创立了《五权宪法》。他首举彻底反帝反封建的旗帜，“起共和而终两千年封建帝制”。 [2]
    
 孙中山1866年11月12日生于广东省广州府香山县（今中山市）翠亨村。孙中山原在香港学医，并成为西医医师。孙中山目睹中华民族有被西方列强瓜分的危险，决定抛弃“医人生涯”，进行“医国事业”。孙中山早期受郑观应的改良思想影响，后看清了清政府的腐败，决心推翻清王朝，建立民主共和国。 [3-4] [44]
    
 1894年11月24日，孙中山在檀香山创立兴中会 [4]。1905年（光绪三十一年）成立中国同盟会。1911年10月10日（宣统三年）新军中的革命党人暗中联络，决定当天晚上起义。辛亥革命后被推举为中华民国临时大总统（任期1912年1月1日——1912年4月1日）。1925年3月12日，孙中山因癌症在北京逝世。1929年6月1日，根据其生前遗愿，葬于南京紫金山中山陵。1940年，国民政府通令全国，尊称其为“中华民国国父”。
    
 孙中山著有《建国方略》、《建国大纲》、《三民主义》等。其著述在逝世后多次被结集出版，有中华书局1986年出版的十一卷本《孙中山全集》，台北1969、1973、1985年出版的《国父全集》等。
    
   22. 问题为：
    
   24. 孙中山跟毛泽东是什么关系？
    
   26. 答案为：
    
 """)
    
  
    
 messages = [sys_msg, user_msg]
    
  
    
 model.invoke(input=messages)

 # 让用户传过一篇文章和一个问题，我们根据用户的文章来回答用户的问题

    
  
    
 from langchain_core.prompts import SystemMessagePromptTemplate
    
 from langchain_core.prompts import HumanMessagePromptTemplate
    
 from langchain_core.prompts import ChatPromptTemplate
    
  
    
 sys_prompt = SystemMessagePromptTemplate.from_template(template="""
    
 你是一个阅读理解机器人！
    
 请认真阅读用户提供的文章，并根据文章的内容来回答用户的问题！
    
 请注意：
    
     1, 如果问题没有在文章中出现，请直接使用你自己的知识来回答！
    
     2, 不需要做任何的解释，直接回答问题的结果即可！
    
 """)
    
  
    
 user_prompt = HumanMessagePromptTemplate.from_template(template="""
    
 文章为：
    
 {passage}
    
 问题为：
    
 {question}
    
 答案为：
    
 """)
    
  
    
 prompt = ChatPromptTemplate.from_messages(messages=[sys_prompt, user_prompt])
    
  
    
 from langchain_core.output_parsers import StrOutputParser
    
  
    
 output_parser = StrOutputParser()
    
  
    
 chain = prompt | model | output_parser
    
  
    
 passage = """
    
 孙中山，名文，字载之，号日新，又号逸仙，又名帝象，化名中山樵，伟大的民族英雄、伟大的爱国主义者、中国民主革命的伟大先驱 [1]，中华民国和中国国民党的缔造者，三民主义的倡导者，创立了《五权宪法》。他首举彻底反帝反封建的旗帜，“起共和而终两千年封建帝制”。 [2]
    
 孙中山生于广东省广州府香山县（今中山市）翠亨村。孙中山原在香港学医，并成为西医医师。孙中山目睹中华民族有被西方列强瓜分的危险，决定抛弃“医人生涯”，进行“医国事业”。孙中山早期受郑观应的改良思想影响，后看清了清政府的腐败，决心推翻清王朝，建立民主共和国。 [3-4] [44]
    
 1894年11月24日，孙中山在檀香山创立兴中会 [4]。1905年（光绪三十一年）成立中国同盟会。1911年10月10日（宣统三年）新军中的革命党人暗中联络，决定当天晚上起义。辛亥革命后被推举为中华民国临时大总统（任期1912年1月1日——1912年4月1日）。1925年3月12日，孙中山因癌症在北京逝世。1929年6月1日，根据其生前遗愿，葬于南京紫金山中山陵。1940年，国民政府通令全国，尊称其为“中华民国国父”。
    
 孙中山著有《建国方略》、《建国大纲》、《三民主义》等。其著述在逝世后多次被结集出版，有中华书局1986年出版的十一卷本《孙中山全集》，台北1969、1973、1985年出版的《国父全集》等。
    
 """
    
  
    
 question = """
    
 Please list 孙中山's other names!
    
 """
    
  
    
 chain.invoke(input=dict(passage=passage, question=question))