用TiDB Serverless存储聊天记录：无缝集成AI应用

最新推荐文章于 2024-10-06 23:41:31 发布

llzwxh888

最新推荐文章于 2024-10-06 23:41:31 发布

阅读量296

点赞数 3

文章标签： tidb serverless 人工智能

本文链接：https://blog.csdn.net/ppoojjj/article/details/142289495

版权

# 用TiDB Serverless存储聊天记录：无缝集成AI应用

## 引言

随着AI技术的快速发展，选择合适的数据库来存储和处理数据变得尤为重要。TiDB Cloud的Serverless选项为开发者提供了一个灵活、高效的解决方案。本文将介绍如何利用TiDB来存储聊天记录，并实现动态聊天交互。

## 主要内容

### 1. 环境准备

我们首先需要安装所需的Python库：

```bash
%pip install --upgrade --quiet langchain langchain_openai langchain-community

2. 配置OpenAI API密钥

为了使用OpenAI功能，我们需要配置API密钥：

import getpass
import os

os.environ["OPENAI_API_KEY"] = getpass.getpass("Input your OpenAI API key:")

3. 连接TiDB

使用TiDB Cloud提供的标准连接方法配置数据库连接：

# 从TiDB Cloud控制台复制
tidb_connection_string_template = "mysql+pymysql://<USER>:<PASSWORD>@<HOST>:4000/<DB>?ssl_ca=/etc/ssl/cert.pem&ssl_verify_cert=true&ssl_verify_identity=true"
tidb_password = getpass.getpass("Input your TiDB password:")
tidb_connection_string = tidb_connection_string_template.replace(
    "<PASSWORD>", tidb_password
)

4. 生成历史聊天数据

我们将创建一组历史数据，用于后续的演示：

from datetime import datetime
from langchain_community.chat_message_histories import TiDBChatMessageHistory

history = TiDBChatMessageHistory(
    connection_string=tidb_connection_string,
    session_id="code_gen",
    earliest_time=datetime.utcnow()
)

history.add_user_message("How's our feature going?")
history.add_ai_message("It's going well. We are working on testing now. It will be released in Feb.")

代码示例

在这个示例中，我们会展示如何通过历史数据实现动态聊天交互：

from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
from langchain_openai import ChatOpenAI

prompt = ChatPromptTemplate.from_messages(
    [
        ("system", "You're an assistant who's good at coding. You're helping a startup build"),
        MessagesPlaceholder(variable_name="history"),
        ("human", "{question}"),
    ]
)
chain = prompt | ChatOpenAI()

from langchain_core.runnables.history import RunnableWithMessageHistory

chain_with_history = RunnableWithMessageHistory(
    chain,
    lambda session_id: TiDBChatMessageHistory(
        session_id=session_id, connection_string=tidb_connection_string
    ),
    input_messages_key="question",
    history_messages_key="history",
)

response = chain_with_history.invoke(
    {"question": "Today is Jan 1st. How many days until our feature is released?"},
    config={"configurable": {"session_id": "code_gen"}},
)
print(response.content)