大模型应用框架-LangChain（一）

pip install openai==0.28
pip install langchain
pip install qianfan

import  os
from langchain.chat_models import QianfanChatEndpoint
from langchain.chat_models.base import HumanMessage


os.environ['QIANFAN_AK'] = "SPPejIX4r2mEUdjdkVNwxTHc"
os.environ['QIANFAN_SK'] = "hOGdXomPZu8FRL51dkBZrEee4tqaS6PM"


chat = QianfanChatEndpoint(
    streaming=True,model="ERNIE-Bot-turbo"

)
messages = [
        HumanMessage(content="给我写一首唐诗")
]
res = chat(messages)
print(res)
# 打印结果：
'''
content='好的，以下是我为您创作的唐诗：\n\n青山依旧在，几度夕阳红。\n白发渔樵江渚上，惯看秋月春风。\n一壶浊酒喜相逢，古今多少事，都付笑谈中。'
'''

import  os
from langchain.chat_models import QianfanChatEndpoint
from langchain.prompts import ChatPromptTemplate

os.environ['QIANFAN_AK'] = "SPPejIX4r2mEUdjdkVNwxTHc"
os.environ['QIANFAN_SK'] = "hOGdXomPZu8FRL51dkBZrEee4tqaS6PM"



# 创建原始模板
template_str = """您是一位专业的鲜花店文案撰写员。\n
对于售价为 {price} 元的 {flower_name} ，您能提供一个吸引人的简短描述吗？
注意: 文字不要超过50个字符 """
# 根据原始模板创建LangChain提示模板
promp_emplate = ChatPromptTemplate.from_template(template_str)
prompt = promp_emplate.format_messages(flower_name=["玫瑰"], price='50')


print('prompt-->', prompt)
# prompt显示：
'''
prompt--> [HumanMessage(content="您是一位专业的鲜花店文案撰写员。\n\n对于售价为 50 元的 ['玫瑰'] ，您能提供一个吸引人的简短描述吗？\n注意: 文字不要超过50个字符\n# ")]
'''

# 实例化模型
chat = QianfanChatEndpoint(
    streaming=True,model="ERNIE-Bot-turbo"

)
# 打印结果
result = chat(prompt)
print(result)
# 结果展示：
'''
content='玫瑰鲜花 售价50元\n纯手工编织花束，顶级玫瑰品种\n散发浓郁香气，温暖人心扉\n白色或粉红色，娇艳欲滴\n让爱情与浪漫伴随你每一天！#'

'''

import  os
from langchain.embeddings import QianfanEmbeddingsEndpoint

os.environ['QIANFAN_AK'] = "SPPejIX4r2mEUdjdkVNwxTHc"
os.environ['QIANFAN_SK'] = "hOGdXomPZu8FRL51dkBZrEee4tqaS6PM"

embed = QianfanEmbeddingsEndpoint()
res1 = embed.embed_query('这是第一个测试文档')
print(res1)
# 打印结果：
'''
[0.039765920490026474, 0.02263435162603855, -0.01889650709927082, ...., 
'''
res2 = embed.embed_documents(['这是第一个测试文档', '这是第二个测试文档'])
print(res2)
# 打印结果：
'''
[[0.03977284952998161, 0.022625437006354332, -0.01892162673175335, ...., 
'''

from langchain import PromptTemplate
from langchain.llms import QianfanLLMEndpoint
import  os
os.environ['QIANFAN_AK'] = "SPPejIX4r2mEUdjdkVNwxTHc"
os.environ['QIANFAN_SK'] = "hOGdXomPZu8FRL51dkBZrEee4tqaS6PM"
# 定义模板
template = "我的邻居姓{lastname}，他生了个儿子，给他儿子起个名字"

prompt = PromptTemplate(
    input_variables=["lastname"],
    template=template,
)

prompt_text = prompt.format(lastname="王")
print(prompt_text)
# result: 我的邻居姓王，他生了个儿子，给他儿子起个名字

llm = QianfanLLMEndpoint()

result = llm(prompt_text)
print(result)
# 打印结果：
'''
给邻居的儿子起名字是一件非常棒的事情！在考虑名字时，通常会考虑一些基本的因素，比如名字的含义、读音、书写等。以下是一些建议：

如果您想要一个简单的名字，那么可以考虑王煦宇。这个名字寓意着阳光和宽广的宇宙，表示孩子应该像太阳一样温暖、明朗，又如宇宙般宽广包容。

另一个选择是王谦嘉。这个名字意为谦虚、高尚，同时也表示嘉奖和庆祝。如果邻居有特别期望他的儿子将来成为有道德、有修养的人，这个名字可能是一个不错的选择。

当然，这只是一些建议，最终的决定应该基于王先生的个人喜好和期望。请确保名字易于书写和发音，并且与您和邻居的姓氏搭配得当。祝王先生和他的儿子一切顺利！

'''

from langchain import PromptTemplate, FewShotPromptTemplate
from langchain.llms import QianfanLLMEndpoint
import  os
os.environ['QIANFAN_AK'] = "SPPejIX4r2mEUdjdkVNwxTHc"
os.environ['QIANFAN_SK'] = "hOGdXomPZu8FRL51dkBZrEee4tqaS6PM"

examples = [
    {"word": "开心", "antonym": "难过"},
    {"word": "高", "antonym": "矮"},
]

example_template = """
单词: {word}
反义词: {antonym}\\n
"""

example_prompt = PromptTemplate(
    input_variables=["word", "antonym"],
    template=example_template,
)

few_shot_prompt = FewShotPromptTemplate(
    examples=examples,
    example_prompt=example_prompt,
    prefix="给出每个单词的反义词",
    suffix="单词: {input}\\n反义词:",
    input_variables=["input"],
    example_separator="\\n",
)

prompt_text = few_shot_prompt.format(input="粗")
print(prompt_text)
print('*'*80)
# 给出每个单词的反义词
# 单词: 开心
# 反义词: 难过

# 单词: 高
# 反义词: 矮

# 单词: 粗
# 反义词:

# 调用OpenAI
llm = QianfanLLMEndpoint(temperature=0.9)
print(llm(prompt_text))

# 细

from langchain import PromptTemplate
from langchain.llms import QianfanLLMEndpoint
from langchain.chains import LLMChain
import  os
os.environ['QIANFAN_AK'] = "SPPejIX4r2mEUdjdkVNwxTHc"
os.environ['QIANFAN_SK'] = "hOGdXomPZu8FRL51dkBZrEee4tqaS6PM"
# 定义模板
template = "我的邻居姓{lastname}，他生了个儿子，给他儿子起个名字"

prompt = PromptTemplate(
    input_variables=["lastname"],
    template=template,
)
llm = QianfanLLMEndpoint()

chain = LLMChain(llm = llm,
                  prompt = prompt)
# 执行链
print(chain.run("王"))
# 打印结果
'''
给邻居家的新生儿起名字是一件非常重要的事情，需要考虑到很多因素，包括家庭传统、父母的偏好、名字的含义等等。在这个情况下，王先生和太太可能会想要一个既传统又具有现代感的名字。

基于这些考虑，以下是一些适合男孩的名字：

1. 王梓轩（Zi Xuan）：这个名字既有传统的含义（梓是树木的意思，轩是高远的意思），又具有现代感。
2. 王宇翔（Yu Xiang）：这个名字既包含了宇宙的含义（宇是宇宙的意思，翔是飞翔的意思），又有希望他儿子能像鸟儿一样自由飞翔的寓意。
3. 王宇轩（Yu Xuan）：这个名字也有同样的含义，而且也有一种稳重和宽广的感觉。
4. 王博远（Bo Yuan）：这个名字的含义是博学而远志，既体现了父母的期望，又有一种清新明快的感觉。

请注意，在选择名字时，还需要考虑名字在社区中的受欢迎程度，以确保这个名字不会引起任何问题或误解。此外，如果王先生和太太有任何特定的偏好或期望，他们也应该在这个过程中发挥重要作用。

以上就是我为王先生的儿子提出的一些名字建议，希望能帮助到你们。
'''

from langchain import PromptTemplate
from langchain.llms import QianfanLLMEndpoint
from langchain.chains import LLMChain, SimpleSequentialChain
import  os
os.environ['QIANFAN_AK'] = "SPPejIX4r2mEUdjdkVNwxTHc"
os.environ['QIANFAN_SK'] = "hOGdXomPZu8FRL51dkBZrEee4tqaS6PM"

# 创建第一条链
template = "我的邻居姓{lastname}，他生了个儿子，给他儿子起个名字"

first_prompt = PromptTemplate(
    input_variables=["lastname"],
    template=template,
)
llm = QianfanLLMEndpoint(temperature=0.9)

first_chain = LLMChain(llm = llm, prompt = first_prompt)

# 创建第二条链
second_prompt = PromptTemplate(
    input_variables=["child_name"],
    template="邻居的儿子名字叫{child_name}，给他起一个小名",
)

second_chain = LLMChain(llm=llm, prompt=second_prompt)


# 链接两条链
# verbose=True可以显示推理过程
overall_chain = SimpleSequentialChain(chains=[first_chain, second_chain], verbose=True)

print(overall_chain)
# 执行链，只需要传入第一个参数
catchphrase = overall_chain.run("王")
print(catchphrase)
#
'''
当然，给邻居的孩子起小名也是一个很好的方式，可以更加亲近和亲切。考虑到上述名字的含义和音韵，以下是一些小名的建议：

1. 梓轩宝宝：对应“王梓轩”这个名字，可以叫他“宝宝”，表示亲切和喜爱。
2. 宇帆小子：对应“王宇帆”这个名字，可以叫他“小子”，显得活泼可爱。
3. 瑞阳小宝：对应“王瑞阳”这个名字，可以叫他“小宝”，显得亲切温暖。
4. 博文宝贝：对应“王博文”这个名字，可以叫他“宝贝”，表示对他的喜爱和呵护。
5. 浩宇小星：对应“王浩宇”这个名字，可以叫他“小星”，显得充满活力和希望。
'''