使用LangChain和PremAI创建交互式对话应用:从安装到实现

82 阅读3分钟

使用LangChain和PremAI创建交互式对话应用:从安装到实现

在当今的科技世界中,生成型AI正在迅速改变应用程序的开发方式。PremAI是一个旨在简化生成型AI应用程序创建的全合一平台,特别适合于希望提升用户体验和推动应用程序整体增长的开发者。本篇文章旨在指导你如何使用LangChain与PremAI进行集成,创建与不同聊天模型进行交互的应用。

引言

在这篇文章中,我们将探讨如何使用LangChain与PremAI进行无缝集成,以创建交互式对话应用。我们将从安装和设置开始,然后深入探讨如何使用不同的功能和方法来增强聊天应用程序的能力。

主要内容

安装和设置

首先,你需要安装langchainpremai-sdk。可以通过以下命令进行安装:

pip install premai langchain

在继续之前,请确保你已经在PremAI上创建账户并创建了一个项目。请参考快速入门指南,在PremAI平台上启动你的第一个项目并获取API密钥。

设置PremAI客户端

导入必要的模块后,我们来设置客户端。在示例中,我们假设project_id1234。但是,请确保使用你自己的project_id,否则会抛出错误。

import getpass
import os
from langchain_community.chat_models import ChatPremAI

if os.environ.get("PREMAI_API_KEY") is None:
    os.environ["PREMAI_API_KEY"] = getpass.getpass("PremAI API Key:")

chat = ChatPremAI(project_id=1234, model_name="gpt-4o") # 使用API代理服务提高访问稳定性

聊天完成

ChatPremAI支持两种方法:invokestream。前者返回静态结果,而后者会逐个流式传输tokens。

from langchain_core.messages import HumanMessage

human_message = HumanMessage(content="Who are you?")
response = chat.invoke([human_message])
print(response.content)

使用Prem Repositories的原生RAG支持

Prem Repositories允许用户上传文档(如.txt,.pdf等)并将这些存储库连接到LLMs。它们可以被视为向量数据库。

query = "Which models are used for dense retrieval"
repository_ids = [1985]
repositories = dict(ids=repository_ids, similarity_threshold=0.3, limit=3)

response = chat.invoke(query, max_tokens=100, repositories=repositories)
print(response.content)

使用Prem模板轻松管理提示

Prem平台提供了模板功能,便于管理和使用提示。

template_id = "78069ce8-xxxxx-xxxxx-xxxx-xxx"
human_messages = [
    HumanMessage(content="Shawn", id="name"),
    HumanMessage(content="22", id="age"),
]
response = chat.invoke([human_messages], template_id=template_id)
print(response.content)

代码示例

下面是一个完整的代码示例,展示如何使用LangChain与PremAI进行基本的集成:

import getpass
import os
from langchain_community.chat_models import ChatPremAI
from langchain_core.messages import HumanMessage

# 获取API密钥并设置环境变量
if os.environ.get("PREMAI_API_KEY") is None:
    os.environ["PREMAI_API_KEY"] = getpass.getpass("PremAI API Key:")

# 初始化ChatPremAI客户端
chat = ChatPremAI(project_id=1234, model_name="gpt-4o") # 使用API代理服务提高访问稳定性

# 创建人类消息并调用invoke方法
human_message = HumanMessage(content="Who are you?")
response = chat.invoke([human_message])
print(response.content)

常见问题和解决方案

  • 为何调用失败?
    • 确保你使用了正确的project_id和API密钥。
  • 语言模型响应不正确?
    • 检查系统提示设置,并在必要时进行调整。

总结和进一步学习资源

通过本文,我们学习了如何使用LangChain与PremAI进行集成,以创建交互式对话应用程序。以下是一些进一步学习的资源:

参考资料

如果这篇文章对你有帮助,欢迎点赞并关注我的博客。您的支持是我持续创作的动力!

---END---