如何集成langchain
更新时间:
本文主要介绍如何将阿里云百炼集成到大模型应用开发框架Langchain中。
操作步骤
1.获取和安装python模块
pip install langchain-bailian
2. 测试代码
import os
from langchain.chains import ConversationChain
from langchain.memory import ConversationBufferMemory
from langchain_bailian import Bailian
import uuid
def test_bailian_llm():
""" 测试基础llm功能 """
# 需要在环境变量设置ACCESS_KEY_ID、ACCESS_KEY_SECRET、AGENT_KEY和APP_ID
access_key_id = os.environ.get("ACCESS_KEY_ID")
access_key_secret = os.environ.get("ACCESS_KEY_SECRET")
agent_key = os.environ.get("AGENT_KEY")
app_id = os.environ.get("APP_ID")
llm = Bailian(access_key_id=access_key_id,
access_key_secret=access_key_secret,
agent_key=agent_key,
app_id=app_id)
out = llm("1+1=?")
print(out)
def test_conversation_chain():
""" 测试包含memory的chain """
access_key_id = os.environ.get("ACCESS_KEY_ID")
access_key_secret = os.environ.get("ACCESS_KEY_SECRET")
agent_key = os.environ.get("AGENT_KEY")
app_id = os.environ.get("APP_ID")
llm = Bailian(access_key_id=access_key_id,
access_key_secret=access_key_secret,
agent_key=agent_key,
app_id=app_id)
memory = ConversationBufferMemory()
conversant_chain = ConversationChain(memory=memory, llm=llm,
verbose=True)
conversant_chain.run("我想明天去北京")
out = conversant_chain.run("那边有什么旅游景点吗")
print(out)
def test_conversant_memory():
""" 测试包含memory的conversant chain """
access_key_id = os.environ.get("ACCESS_KEY_ID")
access_key_secret = os.environ.get("ACCESS_KEY_SECRET")
agent_key = os.environ.get("AGENT_KEY")
app_id = os.environ.get("APP_ID")
llm = Bailian(access_key_id=access_key_id,
access_key_secret=access_key_secret,
agent_key=agent_key,
app_id=app_id)
memory = ConversationBufferMemory()
conversant_chain = ConversationChain(memory=memory, llm=llm,
verbose=True)
conversant_chain.run("我想明天去北京")
out = conversant_chain.run("那边有什么旅游景点吗")
print(out)
def test_conversant_bailian_session():
""" 测试基于bailian的session id保存上下文 """
access_key_id = os.environ.get("ACCESS_KEY_ID")
access_key_secret = os.environ.get("ACCESS_KEY_SECRET")
agent_key = os.environ.get("AGENT_KEY")
app_id = os.environ.get("APP_ID")
llm = Bailian(access_key_id=access_key_id,
access_key_secret=access_key_secret,
agent_key=agent_key,
app_id=app_id)
session_id = str(uuid.uuid4()).replace('-', '')
out = llm("我想明天去新疆", session_id=session_id)
print(out)
out = llm("那边有什么好玩的地方吗", session_id=session_id)
print(out)
文档内容是否对您有帮助?