[AI Embedchain] 集成 Chainlit

在本例中，我们将学习如何一起使用Chainlit和Embedchain。

设置

首先，安装所需的包：

pip install embedchain chainlit

创建Chainlit应用

创建一个名为app.py的新文件，并添加以下代码：

import chainlit as cl
from embedchain import App

import os

os.environ["OPENAI_API_KEY"] = "sk-xxx"

@cl.on_chat_start
async def on_chat_start():
    app = App.from_config(config={
        'app': {
            'config': {
                'name': 'chainlit-app'
            }
        },
        'llm': {
            'config': {
                'stream': True,
            }
        }
    })
    # 在这里导入你的数据
    app.add("https://www.forbes.com/profile/elon-musk/")
    app.collect_metrics = False
    cl.user_session.set("app", app)

@cl.on_message
async def on_message(message: cl.Message):
    app = cl.user_session.get("app")
    msg = cl.Message(content="")
    for chunk in await cl.make_async(app.chat)(message.content):
        await msg.stream_token(chunk)
    
    await msg.send()

运行应用

chainlit run app.py

试一试

在浏览器中打开应用并开始与它聊天！

引用

• github
• doc
• AI 博客 - 从零开始学AI