[AI Embedchain] 集成 Streamlit

从零开始学AI

于 2024-08-13 07:35:05 发布

阅读量251

点赞数 5

分类专栏： AI 文章标签：人工智能

本文链接：https://blog.csdn.net/mahone3297/article/details/141147373

版权

在这个例子中，我们将学习如何使用mistralai/Mixtral-8x7B-Instruct-v0.1和Embedchain与Streamlit一起构建一个简单的RAG聊天机器人。

设置

安装Embedchain和Streamlit。

pip install embedchain streamlit

app.py

import os
from embedchain import App
import streamlit as st

with st.sidebar:
    huggingface_access_token = st.text_input("Hugging face Token", key="chatbot_api_key", type="password")
    "[获取Hugging Face访问令牌](https://huggingface.co/settings/tokens)"
    "[查看源代码](https://github.com/embedchain/examples/mistral-streamlit)"

st.title("💬 聊天机器人")
st.caption("🚀 一个由Mistral提供支持的Embedchain应用！")
if "messages" not in st.session_state:
    st.session_state.messages = [
        {
   
            "role": "assistant",
            "content": """
        嗨！我是一个聊天机器人。我可以回答问题并学习新事物！\n
        问我任何事情，如果你想让我学习新东西，请输入`/add <source>`。\n
        我几乎可以学习一切。:)
        "&#