imane0x / Customer-Support

A fine-tuned Mistral-7B model integrated with an SQL database to handle general customer support questions and manage orders.

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