End to End GenAI Retail Industry Project

GenAI零售行业项目

该项目实现了一个可以与 MySQL 数据库交互的端到端语言模型 (LLM) 系统。用户可以用自然语言提出问题，系统通过将这些问题转换为 SQL 查询并在 MySQL 数据库上执行来生成答案。

项目概况

该系统旨在处理与 MySQL 数据库中存储的零售库存、销售和折扣数据相关的查询。它可以回答以下问题：

• 我们还有多少件白色阿迪达斯 T 恤库存？
• 如果我们在应用折扣后出售所有超小尺寸 T 恤，我们的商店会产生多少销售额？

APP示例图片

技术栈

• 谷歌双子座
• 拥抱面部嵌入
• UI 的 Streamlit
• Langchain框架
• Chromadb 作为矢量存储
• 少样本学习

项目结构

 genai_retail_industry_project
 ┣ database
 ┃ ┗ create_db.sql
 ┣ exp
 ┃ ┗ testing.ipynb
 ┣ src
 ┃ ┣ __pycache__
 ┃ ┃ ┣ __init__.cpython-310.pyc
 ┃ ┃ ┣ langchain_sql.cpython-310.pyc
 ┃ ┃ ┗ utils.cpython-310.pyc
 ┃ ┣ __init__.py
 ┃ ┣ langchain_sql.py
 ┃ ┗ utils.py
 ┣ README.md
 ┣ app.py
 ┗ requirements.txt

安装

1. 将此存储库克隆到您的本地计算机：

    git clone https://github.com/yourusername/genai_retail_industry_project.git
   

2. 导航到项目目录：

    cd genai_retail_industry_project
   

3. 安装所需的依赖项：

    pip install -r requirements.txt
   

4. 在根目录中创建一个.env文件并添加您的 Google API 密钥：

    GOOGLE_API_KEY="your_api_key_here"
   

用法

1. 运行 Streamlit 应用程序：

    streamlit run app.py
   

2. 该网络应用程序将在您的浏览器中打开，您可以在其中询问有关零售数据的问题。

示例问题

• 库存中总共还剩下多少件 T 恤？
• 我们还剩下多少件 XS 码、白色的 Nike T 恤？
• 所有 S 码 T 恤的库存总价是多少？
• 如果我们今天把阿迪达斯小号衬衫打折后全部卖掉，会产生多少销售额？

执照

该项目已获得 MIT 许可证的许可。

致谢

该项目是作为在零售行业中应用大型语言模型 (LLM) 的学习练习而开发的，特别是与 SQL 数据库交互。特别感谢开源社区提供的工具和库，使之成为可能。