首页 > 后端开发 > Python教程 > 如何使用一个 API 创建 AI 聊天机器人来访问多个 LLM

如何使用一个 API 创建 AI 聊天机器人来访问多个 LLM

王林
发布: 2024-08-25 06:02:02
原创
806 人浏览过

最初由 Liz Acosta 在 Streamlit 博客上发布

还记得第一次使用人工智能图像生成器时有多酷吗?那两千万根手指和噩梦般的吃意大利面的画面不仅仅是有趣,它们在不经意间透露了哎呀!人工智能模型的智能程度与我们一样。和我们一样,他们也很难画手。

How to create an AI chatbot using one API to access multiple LLMs

人工智能模型很快变得更加复杂,但现在的模型数量太多了。而且,和我们一样,他们中的一些人比其他人更擅长某些任务。以文本生成为例。尽管 Llama、Gemma 和 Mistral 都是法学硕士,但他们中的一些人更擅长生成代码,而另一些人则更擅长头脑风暴、编码或创意写作。它们根据提示提供不同的优势,因此在您的 AI 应用程序中包含多个模型可能是有意义的。

但是如何将所有这些模型集成到您的应用程序中而不重复代码?如何使人工智能的使用更加模块化,从而更易于维护和扩展?这就是 API 可以提供用于跨不同技术进行通信的标准化指令集的地方。

在这篇博文中,我们将了解如何使用 Replicate 和 Streamlit 来创建一个应用程序,该应用程序允许您通过单个 API 调用来配置和提示不同的 LLM。别担心——当我说“应用程序”时,我并不是说必须启动整个 Flask 服务器或繁琐地配置你的路由或担心 CSS。 Streamlit 已经为您解决了这个问题?

继续阅读以了解:

  • 什么是复制
  • Streamlit 是什么
  • 如何构建演示复制聊天机器人 Streamlit 应用程序
  • 以及使用 Replicate 的最佳实践

不想读书?以下是探索此演示的一些其他方法:

  • 在此处的 Streamlit Cookbook 存储库中查找代码
  • 在此处尝试应用程序的部署版本
  • 在此处观看 Replicate 创始设计师 Zeke Sikelianos 的视频演练

什么是复制?

Replicate 是一个平台,使开发人员能够通过 CLI、API 或 SDK 部署、微调和访问开源 AI 模型。该平台可以轻松地以编程方式将人工智能功能集成到软件应用程序中。

Replicate 上的可用模型

  • 文本:像 Llama 3 这样的模型可以根据输入提示生成连贯且上下文相关的文本。
  • 图像:像稳定扩散这样的模型可以根据文本提示生成高质量的图像。
  • 语音:像whisper这样的模型可以将语音转换为文本,而像xtts-v2这样的模型可以生成听起来自然的语音。
  • 视频:像 animate-diff 这样的模型或像 videocrafter 这样的稳定​​扩散的变体可以分别根据文本和图像提示生成和/或编辑视频。

当一起使用时,Replicate 允许您开发多模式应用程序,这些应用程序可以接受输入并生成各种格式的输出,无论是文本、图像、语音还是视频。

什么是 Streamlit?

Streamlit 是一个开源 Python 框架,只需几行代码即可构建高度交互的应用程序。 Streamlit 与生成式 AI 中的所有最新工具集成,例如任何 LLM、矢量数据库或各种 AI 框架(如 LangChain、LlamaIndex 或 Weights & Biases)。 Streamlit 的聊天元素使与 AI 交互变得特别容易,因此您可以构建“与您的数据对话”的聊天机器人。

与 Replicate 这样的平台相结合,Streamlit 允许您创建生成式 AI 应用程序,而无需任何应用程序设计开销。

?要详细了解 Streamlit 如何让您取得进步,请查看此博文。

要了解有关 Streamlit 的更多信息,请查看 101 指南。

尝试应用程序配方:复制 + Streamlit

但是不要相信我的话。亲自尝试该应用程序或观看视频演示,看看您的想法。

在此演示中,您将使用 Replicate 启动 Streamlit 聊天机器人应用程序。该应用程序使用单个 API 来访问三个不同的 LLM 并调整温度和 top-p 等参数。这些参数影响人工智能生成文本的随机性和多样性,以及选择标记的方法。

什么是模型温度?温度控制模型如何选择令牌。较低的温度使模型更加保守,有利于常见和“安全”的词语。相反,较高的温度会鼓励模型通过选择可能性较小的标记来承担更多风险,从而产生更多创意输出。

什么是top-p?也称为“核采样”——是另一种调整随机性的方法。随着 top-p 值的增加,它的工作原理是考虑更广泛的代币集。较高的 top-p 值会导致采样的令牌范围更加多样化,从而产生更加多样化的输出。

先决条件

  • Python 版本 >=3.8, !=3.9.7
  • 复制 API 密钥 (请注意,需要使用付款方式才能访问超出免费试用限制的功能。)

?要了解有关 API 密钥的更多信息,请查看此处的博客文章。

环境设置

本地设置

  1. 克隆 Cookbook 存储库:git clone https://github.com/streamlit/cookbook.git
  2. 从 Cookbook 根目录,将目录更改为复制配方: cd Recipe/replicate
  3. 将您的复制 API 密钥添加到 .streamlit/secrets_template.toml 文件
  4. 将文件名从secrets_template.toml更新为secrets.toml: mv .streamlit/secrets_template.toml .streamlit/secrets.toml (要了解有关 Streamlit 中秘密处理的更多信息,请参阅此处的文档。)
  5. 创建虚拟环境:python -m venvreplicatevenv
  6. 激活虚拟环境:sourcereplicatevenv/bin/activate
  7. 安装依赖项:pip install -rrequirements.txt

GitHub Codespaces 设置

  1. 从 GitHub 上的 Cookbook 存储库中,通过从“代码”按钮中选择“代码空间”选项来创建一个新的代码空间

    How to create an AI chatbot using one API to access multiple LLMs

  2. 生成代码空间后,将您的 Replicate API 密钥添加到 Recipe/replicate/.streamlit/secrets_template.toml 文件

  3. 将文件名从secrets_template.toml更新为secrets.toml
    (要了解有关 Streamlit 中秘密处理的更多信息,请参阅此处的文档。)

  4. 从 Cookbook 根目录中,将目录更改为复制配方:cd Recipe/replicate

  5. 安装依赖项:pip install -rrequirements.txt

使用 Replicate 运行文本生成模型

  1. 在recipes/replicate目录中创建一个名为replicate_hello_world.py的文件
  2. 将以下代码添加到文件中:

    import replicate   
    
    import toml
    
    import os
    
    # Read the secrets from the secrets.toml file
    with open(".streamlit/secrets.toml", "r") as f:
        secrets = toml.load(f)
    
    # Create an environment variable for the Replicate API token 
    os.environ['REPLICATE_API_TOKEN'] = secrets["REPLICATE_API_TOKEN"]
    
    # Run a model
    for event in replicate.stream("meta/meta-llama-3-8b",
                                 input={"prompt": "What is Streamlit?"},):
        print(str(event), end="")
    
    登录后复制
  3. 运行脚本:pythonreplicate_hello_world.py

您应该看到模型生成的文本的打印结果。

要了解有关复制模型及其工作原理的更多信息,您可以在此处参考其文档。从本质上讲,复制“模型”是指经过训练、打包和发布的软件程序,它接受输入并返回输出。

在这种特殊情况下,模型是meta/meta-llama-3-8b,输入是“提示”:“什么是Streamlit?”。运行脚本时,将调用 Replicate 端点,打印的文本是模型通过 Replicate 返回的输出。

运行演示 Replicate Streamlit 聊天机器人应用程序

要运行演示应用程序,请使用 Streamlit CLI:streamlit run streamlit_app.py。

运行此命令会将应用程序部署到本地主机上的端口。当您访问此位置时,您应该会看到 Streamlit 应用程序正在运行。

How to create an AI chatbot using one API to access multiple LLMs

您可以使用此应用程序通过复制提示不同的LLM,并根据您提供的配置生成生成文本。

适用于多个 LLM 模型的通用 API

使用 Replicate 意味着您可以使用一个 API 提示多个开源 LLM,这有助于简化 AI 与现代软件流程的集成。

这是通过以下代码块完成的:

for event in replicate.stream(model, 
                             input={"prompt": prompt_str,
                             "prompt_template": r"{prompt}", 
                             "temperature": temperature,
                             "top_p": top_p,}):
    yield str(event)
登录后复制

模型、温度和 top p 配置由用户通过 Streamlit 的输入小部件提供。 Streamlit 的聊天元素可以轻松地将聊天机器人功能集成到您的应用程序中。最好的部分是您不需要了解 JavaScript 或 CSS 来实现和设计这些组件 - Streamlit 提供开箱即用的所有这些。

复制最佳实践

使用提示的最佳模型

Replicate 提供了一个 API 端点来搜索公共模型。您还可以在他们的网站上探索特色模型和用例。这样可以轻松找到适合您特定需求的型号。

不同的型号有不同的性能特点。根据您对准确性和速度的需求使用适当的模型。

使用 Webhooks、流媒体和图像 URL 提高性能

Replicate 的输出数据仅可用一个小时。使用 webhook 将数据保存到您自己的存储中。您还可以设置 Webhook 来处理来自模型的异步响应。这对于构建可扩展的应用程序至关重要。

尽可能利用流媒体。某些模型支持流式传输,允许您在生成部分结果时获得它们。这非常适合实时应用程序。

使用图像 URL 比使用 Base 64 编码的上传图像可提供更高的性能。

使用 Streamlit 释放 AI 的潜力

借助 Streamlit,数月的应用程序设计工作被简化只需几行 Python 代码。它是展示最新人工智能发明的完美框架。

使用 Streamlit Cookbook 中的其他 AI 食谱快速启动并运行。 (别忘了在论坛中向我们展示您正在构建的内容!)

流媒体快乐! ?

以上是如何使用一个 API 创建 AI 聊天机器人来访问多个 LLM的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:dev.to
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板