Skip to main content

Bedrock

Amazon Bedrock 是一个完全托管的服务,通过 API 提供来自领先人工智能初创公司和亚马逊的 FMs,因此您可以从各种 FMs 中选择最适合您用例的模型。

%pip install boto3
from langchain.llms import Bedrock

llm = Bedrock(
credentials_profile_name="bedrock-admin",
model_id="amazon.titan-tg1-large"
)

在对话链中使用

from langchain.chains import ConversationChain
from langchain.memory import ConversationBufferMemory

conversation = ConversationChain(
llm=llm, verbose=True, memory=ConversationBufferMemory()
)

conversation.predict(input="Hi there!")