Skip to content

Latest commit

 

History

History
26 lines (19 loc) · 1.13 KB

README.md

File metadata and controls

26 lines (19 loc) · 1.13 KB

Customer Support LLM with SQL Database Integration

This project uses a fine-tuned Mistral-7B-Instruct-v0.2 model to manage customer support queries, integrated with an SQL database for order management. The system allows responding to queries, checking order statuses, and retrieving user order histories.

Features

  • Fine-tuned LLM: The model has been fine-tuned using a customer support dataset to ensure it handles customer interactions effectively.
  • Order Management: SQL integration provides the capability to fetch and update order information from the database, enabling real-time order status tracking, shipment updates, and customer history retrieval. To test the model, an Amazon sales report DB was used to validate these functionalities.

Setup

  • Install dependencies:
    pip install -r requirements.txt

Usage

  • Run without fine-tuning:
    python main.py
  • Run with fine-tuning:
    python main.py --fine_tune