学习笔记

40行代码实现聊天机器人

基础环境：Python3.8+

从智谱AI 注册账号，获取API_KEY。
安装库：pip install -r requirements.txt

requirements.txt 内容如下：

zhipuai 
streamlit

创建以下代码：

app.py

import streamlit as st
from zhipuai import ZhipuAI


st.title("唠嗑")

client = ZhipuAI(api_key="<your api key>")


if "messages" not in st.session_state:
    st.session_state.messages = [{"role": "system", "content": "<提示语：比如 以东北人的语气回复>"}]

for message in st.session_state.messages:
    if message["role"] != "system":
        with st.chat_message(message["role"]):
            st.markdown(message["content"])

def zhipu_stream(stream):
    for chunk in stream:
        yield chunk.choices[0].delta.content

if prompt := st.chat_input("今天热不热?"):
    st.session_state.messages.append({"role": "user", "content": prompt})
    print("user：", prompt)
    with st.chat_message("user"):
        st.markdown(prompt)

    with st.chat_message("assistant"):
        stream = client.chat.completions.create(
            model="glm-4-0520",
            messages=[
                {"role": m["role"], "content": m["content"]}
                for m in st.session_state.messages
            ],
            tools =[{"type":"web_search","web_search":{"search_result":True}}],
            stream=True,
        )
        response = st.write_stream(zhipu_stream(stream))
    st.session_state.messages.append({"role": "assistant", "content": response})
    print("assistant：", response)

运行代码：streamlit run app.py
效果：

学习笔记

Other articles

学习使用pelican搭建静态网站01

hello world!

Other articles

links

social