使用Gradio搭建聊天UI实现质谱AI智能问答

一、调用智谱 AI API

1、获取api_key

智谱AI开放平台网址：
https://open.bigmodel.cn/overview
在这里插入图片描述

在这里插入图片描述

2、安装库pip install zhipuai
3、执行一下代码，调用质谱api进行问答

from zhipuai import ZhipuAI

client = ZhipuAI(api_key="xxxxx")  # 填写您自己的APIKey
while True:
    prompt = input("user:")
    response = client.chat.completions.create(
        model="glm-4",  # 填写需要调用的模型名称
        messages=[
            {"role": "user", "content": prompt}
        ],
    )
    answer = response.choices[0].message.content
    print("ZhipuAI:", answer)

二、使用Gradio搭建聊天UI

import gradio as gr
import random
import time

from langchain_community.chat_models import ChatZhipuAI
from zhipuai import ZhipuAI

import configure

llm = configure.chat
client = ZhipuAI(api_key="xxx")  # 填写您自己的APIKey

with gr.Blocks() as demo:
    chatbot = gr.Chatbot()
    msg = gr.Textbox()
    clear = gr.Button("清除")


    def respond(message, chat_history):
        response = client.chat.completions.create(
            model="glm-4",  # 填写需要调用的模型名称
            messages=[
                {"role": "user", "content": message}
            ],
        )
        chat_history.append((message, response.choices[0].message.content))
        return "", chat_history


    msg.submit(respond, [msg, chatbot], [msg, chatbot])
    clear.click(lambda: None, None, chatbot, queue=False)

demo.launch()