当前位置：首页 > article >正文

两种调用智谱AI API的方式

article 2025/2/24 3:34:52

一、用智谱AI依赖包调用

from zhipuai import ZhipuAI
zhipuai_api_key='XXXXXXXXXXXXXXXXXX'  # 请填写您自己的APIKey

messages=[
    {"role": "system", "content": "你是一名经验丰富的人工智能工程师，请你解答用户的问题"},
    {"role": "user", "content": "请你为我解释什么是RAG技术？"}
]

client = ZhipuAI(api_key=zhipuai_api_key) 
response = client.chat.completions.create(
    model="glm-4", # 请填写您要调用的模型名称
    temperature=0.5,
    messages=messages
)
print(response.choices[0].message)

在这里插入图片描述

二、用langchain调用

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

chat = ChatZhipuAI(
    zhipuai_api_key = zhipuai_api_key,
    model="glm-4",
    temperature=0.5,
)
chat.invoke('请你介绍一下自己')

messages = [
    SystemMessage(content="你是一位诗人"),
    HumanMessage(content="请给我写一首关于小鸭子的古诗"),
]
response = chat.invoke(messages)
print(response.content)