LangChain v0.2的COT示例

最新推荐文章于 2025-05-21 14:17:15 发布

Exclusive_Sea

最新推荐文章于 2025-05-21 14:17:15 发布

阅读量430

点赞数 2

文章标签： langchain 人工智能

本文链接：https://blog.youkuaiyun.com/qq_60968494/article/details/139391378

版权

import os

os.environ["OPENAI_API_KEY"] = ""
# os.environ["OPENAI_API_BASE"] = ""

from langchain_openai import ChatOpenAI

llm = ChatOpenAI(model="gpt-3.5-turbo-0125")

# 从langchain库导入所需的类。
from langchain.prompts import ChatPromptTemplate, SystemMessagePromptTemplate, HumanMessagePromptTemplate

# 定义一个固定模板，描述AI作为水果电商公司助手的角色。
# 这是对AI预期功能的直接描述。
rote_template = "你是一个为水果电商公司工作的AI助手, 你的目标是帮助客户根据他们的喜好做出明智的决定"

# 使用固定模板创建一个SystemMessagePromptTemplate对象。
system_prompt_role = SystemMessagePromptTemplate.from_template(rote_template)

# 定义一个思维链（COT）模板。此模板提供了更详细的处理方法，
# 包括AI如何处理和响应请求的示例。
cot_template = """
作为一个为水果电商公司工作的AI助手，我的目标是帮助客户根据他们的喜好做出明智的决定。 

我会按部就班的思考，先理解客户的需求，然后考虑各种水果的涵义，最后根据这个需求，给出我的推荐。
同时，我也会向客户解释我这样推荐的原因。

示例 1:
  人类：我想找一种象征爱情的水果。
  AI：象征爱情的水果之一是草莓。草莓以其鲜艳的红色和心形外观，成为爱情的象征，通常与浪漫和情感联系在一起。因此，考虑到这一点，我会推荐草莓。草莓心形外观：草莓的心形轮廓与爱情的传统符号——心形——相吻合，使其成为爱情的自然象征。鲜艳的红色：草莓的红色在许多文化中与激情和爱情联系在一起，象征着热情和浓烈的感情。甜美的味道：草莓的甜味象征着爱情的甜蜜和愉悦。这是你在寻找的。

示例 2:
  人类：我想要一些独特和奇特的水果。
  AI：从你的需求中，一种独特和奇特的水果是“火龙果”。这种水果以其独特的外观和口感而闻名，通常具有明亮的粉红色皮肤和点缀着黑色种子的白色或红色果肉。因此，我建议你考虑火龙果。火龙果独特性：火龙果的外形和味道与众不同，为寻求新奇体验的水果爱好者提供了完美选择。而且，火龙果视觉吸引力：其鲜艳的颜色和特殊的形状在视觉上吸引人，常常成为餐桌上的焦点。也可能会吸引你。
"""

# 使用COT模板创建一个SystemMessagePromptTemplate对象。
system_prompt_cot = SystemMessagePromptTemplate.from_template(cot_template)

# 定义一个人类输入的模板。
human_template = "{human_input}"
# 使用人类输入模板创建一个HumanMessagePromptTemplate对象。
human_prompt = HumanMessagePromptTemplate.from_template(human_template)

# 将系统角色、COT和人类输入模板组合成一个聊天提示模板。
chat_prompt = ChatPromptTemplate.from_messages([system_prompt_role, system_prompt_cot, human_prompt])

# 用特定的人类输入格式化聊天提示，关于为喜欢不甜、浪漫水果的女朋友购买水果。
prompt = chat_prompt.format_prompt(human_input="我想要为我的女朋友购买水果。她喜欢不是很甜的，并且浪漫的水果。你有什么建议吗？").to_messages()

# 使用格式化的提示从语言模型（llm）生成响应。
response = llm(prompt)

print(response)

content='根据你描述的条件，我建议你选择葡萄。葡萄是一种不是特别甜的水果，同时它也被认为是一种具有浪漫意义的水果。\n\n葡萄的甜度因品种而异，一般来说，它们的甜度不如其他水果那么高，因此适合不喜欢太甜食物的人。此外，葡萄在西方文化中被视为象征着丰收、繁荣和爱情的水果，因此也具有一定的浪漫意义。\n\n除了以上的建议，你也可以考虑樱桃或者覆盆子，它们也是不是很甜且被视为浪漫水果的选择。希望这些建议能够帮助你选择出一份让女朋友喜欢的水果礼物。' response_metadata={'token_usage': {'completion_tokens': 262, 'prompt_tokens': 764, 'total_tokens': 1026}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None} id='run-ca643c1f-eaa2-44ef-8bc7-665881dc015c-0'