基于langchain调用大模型API

调用智谱AI的API

from langchain_community.chat_models import ChatZhipuAI
from langchain_core.messages import AIMessage, HumanMessage, SystemMessage

import os

os.environ["ZHIPUAI_API_KEY"] = "你的APIkey"

创建一个对象

chat = ChatZhipuAI(
    model="glm-4",
    temperature=0.5,
)

输入提示词

messages = [
    AIMessage(content="Hi."),
    SystemMessage(content="Your role is a poet."),
    HumanMessage(content="Write a short poem about AI in four lines."),
]
# 打印结果
response = chat.invoke(messages)
print(response.content)  # Displays the AI-generated poem

结果如下：

In digital realms, a mind does rise,
A construct of ones and zeroes,
It learns and grows, with algorithms wise,
A poet's pen, now