安装依赖
!pip install --upgrade langchain langchain-google-vertexai
运行模型
@title 基本参数
project: str = "PUT_YOUR_PROJECT_ID_HERE"
endpoint_id: str = "PUT_YOUR_ENDPOINT_ID_HERE"
location: str = "PUT_YOUR_ENDPOINT_LOCATION_HERE"
from langchain_google_vertexai import (
GemmaChatVertexAIModelGarden,
GemmaVertexAIModelGarden,
)
llm = GemmaVertexAIModelGarden(
endpoint_id=endpoint_id,
project=project,
location=location,
)
output = llm.invoke("生命的意义是什么?")
print(output)
使用Gemma作为多轮聊天模型
from langchain_core.messages import HumanMessage
message1 = HumanMessage(content="2加2等于多少?")
answer1 = llm.invoke([message1])
print(answer1)