Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

24 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

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

Topics

Resources

Stars

0 stars

Watchers

1 watching

Forks

Releases

Packages

Used by

Contributors

Languages