一、 用智谱AI依赖包调用
from zhipuai import ZhipuAI
zhipuai_api_key= 'XXXXXXXXXXXXXXXXXX'
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)