3
使用langchain接入文心一言API
大模型开发/技术交流
- API
- 社区上线
- 开箱评测
2023.09.0811550看过
1、准备工作
首先到百度千帆平台<https://cloud.baidu.com/>申请API Key 和 Secret Key。
2、安装第三方python库
pip install openai langchain langchain-wenxin
langchain-wenxin的github地址<https://github.com/ninehills/langchain-wenxin>
3、调用llm
我们创建llm.py文件，内容如下：
from langchain_wenxin import WenxinWENXIN_APP_Key = "APP Key"WENXIN_APP_SECRET = "APP Secret"llm = Wenxin(temperature=0.9,model="ernie-bot-turbo",baidu_api_key = WENXIN_APP_Key,baidu_secret_key = WENXIN_APP_SECRET,verbose=True,)response = llm("你是谁？")print(response)
打开命令行运行python .\llm.py
您好，我是百度研发的知识增强大语言模型，中文名是文心一言，英文名是ERNIE Bot。我能够与人对话互动，回答问题，协助创作，高效便捷地帮助人们获取信息、知识和灵感。
4、调用chat
我们创建chat.py文件，内容如下：
from langchain.schema import (AIMessage,HumanMessage,)from langchain_wenxin import ChatWenxinWENXIN_APP_Key = "APP Key"WENXIN_APP_SECRET = "APP Secret"chat_model = ChatWenxin(temperature=0.9,model="ernie-bot-turbo",baidu_api_key = WENXIN_APP_Key,baidu_secret_key = WENXIN_APP_SECRET,verbose=True,)response = chat_model([HumanMessage(content="请给我讲一个笑话")])print(response)
打开命令行运行python .\chat.py
content='好的，这里有一个笑话：\n\n有一天，小明对他的朋友说：“你知道吗？我爸爸昨天告诉我一个笑话，他说他去参加了一个朋友的生日派对，那个朋友是个医生，他问小明爸爸有没有带礼物，小明爸爸说没有，然后那个医生就告诉他，说现在没有，但将来会有一个孩子会带着礼物来到这个世界上，那就是我。”\n\n小明爸爸听到这个笑话后哈哈大笑，小明问他爸爸：“爸爸，你为什么笑？”小明爸爸回答说：“因为我是个医生，我知道将来会有一个孩子带着礼物来到这个世界上，那就是我！”\n\n希望你喜欢这个笑话！' additional_kwargs={} example=False
小结
使用langchain接入文心一言API，我们就能在后续使用langchain其他强大的功能，如PromptTemplate，Chain, Agent等，来快速做出产品化的应用，喜欢的小伙伴记得点赞！
评论
