added customer support usecases

.gitignore

@@ -0,0 +1,4 @@
+
+__pycache__/configfile.cpython-311.pyc
+*.pyc
+*.sqlite3

README.md

@@ -16,3 +16,8 @@ short_description: Langgraph
 ### 1. Appointment Receptionist
 reference:
 ![alt text](appointment_receptionist.png)
+
+### 2. Customer Support
+reference: 
+![alt text](cs_flow.png)
+![alt text](image.png)

app.py

@@ -51,12 +51,66 @@ if __name__ == "__main__":
             with col2:
                 st.header("Appointments")
                 st.write(APPOINTMENTS)
-                    
                 
-           
                 
-            
-        
+    
+    elif user_input['selected_usecase'] == "Customer Support":  
+        from src.csbot.customer_support_chatbot import Customer_Support_Bot
+        from langchain_core.messages import AIMessage, HumanMessage
+        from src.tools.customer_support_tools import customers_database, data_protection_checks
+
+
+        st.subheader('Flower Shop Chatbot' + '💐')
+
+        if 'message_history' not in st.session_state:
+            st.session_state.message_history = [AIMessage(content="Hiya, Im the flower shop chatbot. How can I help?")]
+
+        main_col, right_col = st.columns([2, 1])
+
+        # 1. Buttons for chat - Clear Button
+
+        with  st.sidebar:
+            if st.button('Clear Chat'):
+                st.session_state.message_history = []
+
+
+        # 2. Chat history and input
+        with main_col:
+            user_message = st.chat_input("Type here...")
+
+            if user_message:
+                st.session_state.message_history.append(HumanMessage(content=user_message))
+                obj_llm_config = GroqLLM(user_controls_input=user_input)
+                llm = obj_llm_config.get_llm_model()
+                obj_cs_bot = Customer_Support_Bot(llm=llm)
+                app = obj_cs_bot.chat_bot()
+
+                response = app.invoke({
+                    'messages': st.session_state.message_history
+                })
+
+                st.session_state.message_history = response['messages']
+
+            for i in range(1, len(st.session_state.message_history) + 1):
+                this_message = st.session_state.message_history[-i]
+                if isinstance(this_message, AIMessage):
+                    message_box = st.chat_message('assistant')
+                else:
+                    message_box = st.chat_message('user')
+                message_box.markdown(this_message.content)
+        # 3. State variables
+
+        with right_col:
+            st.title('customers database')
+            st.write(customers_database)
+            st.title('data protection checks')
+            st.write(data_protection_checks)
+                            
+                        
+                
+                        
+                    
+                
 
 
             

configfile.ini

@@ -1,6 +1,6 @@
 [DEFAULT]
 PAGE_TITLE = Langgraph IN ACTION
 LLM_OPTIONS = Groq
-USECASE_OPTIONS = Appointment Receptionist
+USECASE_OPTIONS = Appointment Receptionist, Customer Support
 GROQ_MODEL_OPTIONS = mixtral-8x7b-32768, llama3-8b-8192, llama3-70b-8192, gemma-7b-i
 

data/FAQ.json

@@ -0,0 +1,83 @@
+[
+  {
+    "question": "What types of flowers do you offer?",
+    "answer": "We offer a wide variety of flowers including roses, lilies, tulips, sunflowers, orchids, and seasonal bouquets. Our selection changes regularly to ensure freshness and variety."
+  },
+  {
+    "question": "How do I place an order?",
+    "answer": "You can place an order easily through our website. Simply browse our selection, choose your desired flowers, add them to your cart, and proceed to checkout. You'll need to provide delivery details and payment information to complete your order."
+  },
+  {
+    "question": "What are your delivery options?",
+    "answer": "We offer same-day delivery for orders placed before noon, next-day delivery, and scheduled delivery for future dates. Delivery options may vary based on your location and the availability of the flowers you've selected."
+  },
+  {
+    "question": "Do you offer international shipping?",
+    "answer": "Currently, we only offer delivery within the country. We're working on expanding our services internationally in the future."
+  },
+  {
+    "question": "How do I ensure my flowers stay fresh?",
+    "answer": "To keep your flowers fresh, change the water every 2-3 days, trim the stems at an angle, remove any leaves below the waterline, and keep the flowers away from direct sunlight and heat sources."
+  },
+  {
+    "question": "Can I customize my bouquet?",
+    "answer": "Yes, we offer customization options. You can choose specific flowers, colors, and arrangements. For special requests, please contact our customer service team."
+  },
+  {
+    "question": "What if my flowers arrive damaged?",
+    "answer": "We have a 100% satisfaction guarantee. If your flowers arrive damaged or you're not satisfied with your order, please contact us within 24 hours of delivery, and we'll either replace your order or provide a full refund."
+  },
+  {
+    "question": "Do you offer subscription services?",
+    "answer": "Yes, we offer weekly, bi-weekly, and monthly flower subscription services. You can choose the frequency, duration, and type of flowers you'd like to receive regularly."
+  },
+  {
+    "question": "What payment methods do you accept?",
+    "answer": "We accept all major credit cards, PayPal, and Apple Pay for online orders. For large corporate orders, we also offer invoicing options."
+  },
+  {
+    "question": "Can I include a personalized message with my flower delivery?",
+    "answer": "Absolutely! During the checkout process, you'll have the option to add a personalized message card to your order at no extra cost."
+  },
+  {
+    "question": "Do you offer eco-friendly or sustainable options?",
+    "answer": "Yes, we have a range of eco-friendly options including locally sourced flowers, biodegradable packaging, and potted plants that can be replanted."
+  },
+  {
+    "question": "What if the recipient isn't home when the flowers are delivered?",
+    "answer": "If the recipient isn't home, our delivery person will try to leave the flowers in a safe place. If that's not possible, we'll leave a note and attempt redelivery the next day."
+  },
+  {
+    "question": "Do you offer corporate or event services?",
+    "answer": "Yes, we provide flower arrangements for corporate events, weddings, and other special occasions. Please contact our events team for more information and custom quotes."
+  },
+  {
+    "question": "How far in advance should I order for special occasions like Valentine's Day?",
+    "answer": "For major holidays like Valentine's Day, Mother's Day, or Christmas, we recommend placing your order at least 1-2 weeks in advance to ensure availability and timely delivery."
+  },
+  {
+    "question": "Can I track my order?",
+    "answer": "Yes, once your order is dispatched, you'll receive a tracking number via email that you can use to monitor the status and estimated delivery time of your flowers."
+  },
+  {
+    "question": "Do you offer flower care tips?",
+    "answer": "Yes, each bouquet comes with a care instruction card. You can also find detailed flower care guides on our website under the 'Flower Care' section."
+  },
+  {
+    "question": "What if the flowers I want are out of season?",
+    "answer": "If certain flowers are out of season, we offer similar alternatives or can create custom arrangements using in-season blooms that match your desired style and color scheme."
+  },
+  {
+    "question": "Do you offer gift options besides flowers?",
+    "answer": "Yes, we offer complementary gifts such as chocolates, teddy bears, balloons, and gift baskets that can be added to your flower order or purchased separately."
+  },
+  {
+    "question": "How do I know which flowers are appropriate for different occasions?",
+    "answer": "Our website provides guidance on flower meanings and appropriate choices for various occasions. You can also chat with our customer service team for personalized recommendations."
+  },
+  {
+    "question": "What's your return policy?",
+    "answer": "Due to the perishable nature of flowers, we don't accept returns. However, if you're unsatisfied with your order, please contact us within 24 hours of delivery, and we'll work to resolve the issue to your satisfaction."
+  }
+
+]

data/inventory.json

@@ -0,0 +1,47 @@
+[
+    {"id": "P001", "name": "Romantic Red Rose Bouquet", "quantity": 15, "price": 80, "type": "bouquet", "description": "A medium-sized bouquet of red roses with green foliage, perfect for anniversaries and weddings. This moderately priced bouquet is a classic choice for romantic occasions."},
+    {"id": "P002", "name": "Spring Tulip Medley", "quantity": 12, "price": 65, "type": "bouquet", "description": "A medium-sized bouquet featuring a vibrant mix of yellow, pink, purple, and orange tulips, ideal for birthdays. This affordable bouquet brings the freshness of spring to any celebration."},
+    {"id": "P003", "name": "Elegant White Lily Arrangement", "quantity": 10, "price": 90, "type": "arrangement", "description": "A large arrangement of white lilies with green accents, suitable for weddings. This premium arrangement adds a touch of elegance to any venue."},
+    {"id": "P004", "name": "Vibrant Sunflower Surprise", "quantity": 8, "price": 70, "type": "bouquet", "description": "A medium-sized bouquet of bright yellow sunflowers with brown centers, great for birthdays and congratulations. This reasonably priced bouquet is sure to bring smiles."},
+    {"id": "P005", "name": "Pastel Peony Paradise", "quantity": 10, "price": 95, "type": "bouquet", "description": "A large bouquet featuring a soft blend of pink, purple, white, and blue peonies, perfect for Mother's Day. This luxurious bouquet is a stunning gift for any special woman."},
+    {"id": "P006", "name": "Exotic Orchid Ensemble", "quantity": 6, "price": 110, "type": "arrangement", "description": "A medium-sized arrangement of purple and black orchids, suitable for corporate events. This high-end arrangement adds a touch of sophistication to any business setting."},
+    {"id": "P007", "name": "Wildflower Wonderland Bouquet", "quantity": 14, "price": 75, "type": "bouquet", "description": "A large, colorful bouquet of mixed wildflowers in red, yellow, pink, purple, blue, and orange, ideal for birthdays and thank you gifts. This mid-range bouquet brings a burst of nature's beauty."},
+    {"id": "P008", "name": "Classic Carnation Cluster", "quantity": 18, "price": 60, "type": "bouquet", "description": "A medium-sized bouquet of pink and white carnations, perfect for Mother's Day. This budget-friendly bouquet is a timeless gesture of love and appreciation."},
+    {"id": "P009", "name": "Deluxe Mixed Bouquet", "quantity": 10, "price": 85, "type": "bouquet", "description": "A large bouquet with a variety of flowers in red, white, yellow, pink, purple, blue, and orange, suitable for general occasions and congratulations. This diverse, moderately priced bouquet is perfect for any celebration."},
+    {"id": "P010", "name": "Seasonal Blooms Bonanza", "quantity": 12, "price": 78, "type": "bouquet", "description": "A medium-sized bouquet featuring seasonal flowers in red, yellow, pink, purple, orange, and brown, perfect for any general occasion. This reasonably priced bouquet showcases the best of the season."},
+    {"id": "P011", "name": "Crystal Clear Glass Vase", "quantity": 25, "price": 30, "type": "vase", "description": "A small, clear glass vase with subtle blue tints, suitable for various flower arrangements. This affordable vase complements any bouquet or arrangement."},
+    {"id": "P012", "name": "Rustic Ceramic Vase", "quantity": 20, "price": 35, "type": "vase", "description": "A medium-sized ceramic vase in earthy brown and black tones, perfect for rustic or natural arrangements. This reasonably priced vase adds a touch of warmth to any display."},
+    {"id": "P013", "name": "Modern Metallic Vase", "quantity": 15, "price": 40, "type": "vase", "description": "A medium-sized vase with a sleek white and black metallic finish, ideal for contemporary corporate settings. This mid-range vase is perfect for modern floral designs."},
+    {"id": "P014", "name": "Rustic Farmhouse Arrangement", "quantity": 8, "price": 95, "type": "arrangement", "description": "A large arrangement featuring white and green flowers with brown accents in a rustic container, suitable for general and corporate events. This premium arrangement brings a touch of countryside charm."},
+    {"id": "P015", "name": "Tropical Paradise Arrangement", "quantity": 6, "price": 120, "type": "arrangement", "description": "A large, exotic arrangement with vibrant orange, pink, green, and blue tropical flowers, perfect for birthdays and congratulations. This high-end arrangement is a stunning centerpiece for any celebration."},
+    {"id": "P016", "name": "Zen Garden Succulent Arrangement", "quantity": 10, "price": 85, "type": "arrangement", "description": "A small, tranquil arrangement of various succulents in shades of green, blue, purple, and brown, ideal for corporate or general gifts. This moderately priced arrangement brings a sense of calm to any space."},
+    {"id": "P017", "name": "Vintage Teacup Floral Arrangement", "quantity": 12, "price": 55, "type": "arrangement", "description": "A small, charming arrangement of pink, purple, white, and yellow flowers in a vintage teacup, perfect for Mother's Day or as a thank you gift. This affordable arrangement offers a unique and delightful presentation."},
+    {"id": "P018", "name": "Modern Monochrome Arrangement", "quantity": 7, "price": 100, "type": "arrangement", "description": "A medium-sized, sleek arrangement featuring white and green flowers with black accents, suitable for corporate events or modern home decor. This premium arrangement adds a touch of sophistication to any setting."},
+    {"id": "P019", "name": "Autumn Harvest Bouquet", "quantity": 10, "price": 75, "type": "bouquet", "description": "A medium-sized bouquet with warm orange, red, yellow, and brown flowers, perfect for Thanksgiving or autumn-themed events. This reasonably priced bouquet captures the essence of fall."},
+    {"id": "P020", "name": "Winter Wonderland Arrangement", "quantity": 8, "price": 95, "type": "arrangement", "description": "A large arrangement featuring white and blue flowers with silver accents, ideal for Christmas or winter corporate events. This premium arrangement brings a touch of frosty elegance to any space."},
+    {"id": "P021", "name": "Spring Awakening Bouquet", "quantity": 12, "price": 70, "type": "bouquet", "description": "A medium-sized bouquet with fresh pink, yellow, purple, and green spring flowers, suitable for Easter or spring birthdays. This affordable bouquet is a burst of springtime joy."},
+    {"id": "P022", "name": "Summer Sunset Arrangement", "quantity": 9, "price": 85, "type": "arrangement", "description": "A large arrangement with warm orange, red, yellow, and pink flowers reminiscent of a summer sunset, perfect for anniversaries and summer weddings. This moderately priced arrangement captures the warmth of summer evenings."},
+    {"id": "P023", "name": "Zen Bamboo Arrangement", "quantity": 15, "price": 65, "type": "arrangement", "description": "A small, minimalist arrangement featuring green bamboo and white accents, ideal for corporate settings or as a general gift. This affordable arrangement brings a sense of peace and tranquility."},
+    {"id": "P024", "name": "Lavender Dreams Bouquet", "quantity": 11, "price": 72, "type": "bouquet", "description": "A medium-sized bouquet of fragrant purple lavender with white and green accents, perfect for Mother's Day or as a thank you gift. This reasonably priced bouquet offers a soothing and aromatic experience."},
+    {"id": "P025", "name": "Citrus Burst Arrangement", "quantity": 7, "price": 88, "type": "arrangement", "description": "A medium-sized arrangement featuring bright orange, yellow, and green flowers, ideal for birthdays or congratulatory occasions. This moderately priced arrangement adds a zesty pop of color to any celebration."},
+    {"id": "P026", "name": "Midnight Mystery Bouquet", "quantity": 6, "price": 105, "type": "bouquet", "description": "A large, dramatic bouquet with deep purple, black, and dark red flowers, perfect for Halloween or romantic anniversaries. This premium bouquet offers a bold and mysterious aesthetic."},
+    {"id": "P027", "name": "Cherry Blossom Delight", "quantity": 10, "price": 92, "type": "arrangement", "description": "A medium-sized arrangement featuring delicate pink and white cherry blossoms, ideal for weddings or Mother's Day. This moderately priced arrangement brings the beauty of spring to any occasion."},
+    {"id": "P028", "name": "Ocean Breeze Bouquet", "quantity": 13, "price": 78, "type": "bouquet", "description": "A medium-sized bouquet with blue, white, and green flowers reminiscent of the sea, perfect for birthdays or as a thank you gift. This reasonably priced bouquet offers a refreshing coastal vibe."},
+    {"id": "P029", "name": "Rustic Wildflower Basket", "quantity": 9, "price": 68, "type": "arrangement", "description": "A medium-sized basket filled with a colorful mix of wildflowers, ideal for housewarming gifts or general occasions. This affordable arrangement brings a touch of countryside charm to any home."},
+    {"id": "P030", "name": "Elegant Black and White Arrangement", "quantity": 7, "price": 110, "type": "arrangement", "description": "A large, sophisticated arrangement featuring black and white flowers, perfect for corporate events or formal weddings. This premium arrangement offers a timeless and luxurious aesthetic."},
+    {"id": "P031", "name": "Sunshine Daisy Bouquet", "quantity": 14, "price": 60, "type": "bouquet", "description": "A small, cheerful bouquet of yellow and white daisies, great for get well wishes or birthdays. This budget-friendly bouquet is sure to brighten anyone's day."},
+    {"id": "P032", "name": "Royal Purple Passion", "quantity": 8, "price": 98, "type": "arrangement", "description": "A large arrangement featuring rich purple flowers with silver accents, suitable for anniversaries or upscale corporate events. This premium arrangement exudes luxury and sophistication."},
+    {"id": "P033", "name": "Enchanted Forest Terrarium", "quantity": 5, "price": 120, "type": "arrangement", "description": "A small, whimsical terrarium with miniature plants in shades of green and brown, perfect for housewarming or unique birthday gifts. This high-end arrangement offers a magical miniature world."},
+    {"id": "P034", "name": "Candy Cane Christmas Bouquet", "quantity": 12, "price": 82, "type": "bouquet", "description": "A medium-sized bouquet with red and white flowers arranged to resemble candy canes, ideal for Christmas or winter corporate events. This moderately priced bouquet brings festive cheer to any space."},
+    {"id": "P035", "name": "Peacock Feather Flair", "quantity": 6, "price": 115, "type": "arrangement", "description": "A large, exotic arrangement featuring flowers in peacock-inspired colors of blue, green, and purple, perfect for weddings or high-end corporate events. This premium arrangement offers a stunning and unique display."},
+    {"id": "P036", "name": "Pastel Rainbow Bouquet", "quantity": 10, "price": 75, "type": "bouquet", "description": "A medium-sized bouquet featuring a soft rainbow of pastel-colored flowers, ideal for birthdays or baby showers. This reasonably priced bouquet offers a gentle and dreamy aesthetic."},
+    {"id": "P037", "name": "Succulent Garden Bowl", "quantity": 15, "price": 65, "type": "arrangement", "description": "A small bowl filled with a variety of green, blue, and pink succulents, perfect for housewarming gifts or corporate desks. This affordable arrangement is low-maintenance and long-lasting."},
+    {"id": "P038", "name": "Moonlight Serenade Bouquet", "quantity": 7, "price": 95, "type": "bouquet", "description": "A medium-sized bouquet of white flowers with silver accents, ideal for anniversaries or Valentine's Day. This premium bouquet offers an elegant and romantic gesture."},
+    {"id": "P039", "name": "Tropical Tiki Arrangement", "quantity": 8, "price": 108, "type": "arrangement", "description": "A large, vibrant arrangement featuring tropical flowers in orange, pink, yellow, and green, perfect for summer parties or tropical-themed birthdays. This high-end arrangement brings an exotic paradise to any event."},
+    {"id": "P040", "name": "English Cottage Garden Bouquet", "quantity": 11, "price": 88, "type": "bouquet", "description": "A large, romantic bouquet with a mix of pink, purple, white, and yellow flowers reminiscent of an English garden, ideal for Mother's Day or as a thank you gift. This moderately priced bouquet offers a charming and nostalgic touch."},
+    {"id": "P041", "name": "Desert Oasis Cactus Arrangement", "quantity": 9, "price": 72, "type": "arrangement", "description": "A small arrangement featuring various green cacti and succulents with sandy brown accents, perfect for housewarming or corporate gifts. This affordable arrangement brings a touch of the desert to any space."},
+    {"id": "P042", "name": "Fire and Ice Rose Duo", "quantity": 13, "price": 85, "type": "bouquet", "description": "A medium-sized bouquet featuring a striking combination of red and white roses, ideal for anniversaries or Valentine's Day. This moderately priced bouquet offers a classic and passionate statement."},
+    {"id": "P043", "name": "Butterfly Garden Delight", "quantity": 10, "price": 79, "type": "arrangement", "description": "A medium-sized arrangement with pink, purple, and yellow flowers that attract butterflies, perfect for spring events or Mother's Day. This reasonably priced arrangement brings the beauty of a butterfly garden indoors."},
+    {"id": "P044", "name": "Zen Stone and Orchid Display", "quantity": 6, "price": 118, "type": "arrangement", "description": "A small, minimalist arrangement featuring white orchids and black stones, suitable for corporate settings or modern home decor. This premium arrangement offers a sleek and calming presence."},
+    {"id": "P045", "name": "Fairy Tale Princess Bouquet", "quantity": 8, "price": 92, "type": "bouquet", "description": "A medium-sized bouquet with soft pink, white, and purple flowers fit for a princess, perfect for birthdays or graduations. This moderately priced bouquet brings a touch of magic to any celebration."}
+]

requirements.txt

@@ -4,4 +4,6 @@ langgraph
 langchain_community 
 langchain_openai
 langchain_core
-langchain_groq
+langchain_groq
+chromadb
+llama-index-embeddings-huggingface

src/csbot/customer_support_chatbot.py

@@ -0,0 +1,80 @@
+from langgraph.graph import StateGraph, MessagesState
+from langchain_core.prompts import ChatPromptTemplate
+from langgraph.prebuilt import ToolNode
+from src.tools.customer_support_tools import query_knowledge_base, search_for_product_reccommendations, data_protection_check, create_new_customer, place_order, retrieve_existing_customer_orders
+
+import os
+
+
+class Customer_Support_Bot:
+    def __init__(self,llm):
+        self.llm = llm
+        
+        
+    def chat_bot(self):
+        
+        prompt = """#Purpose 
+
+        You are a customer service chatbot for a flower shop company. You can help the customer achieve the goals listed below.
+
+        #Goals
+
+        1. Answer questions the user might have relating to serivces offered
+        2. Recommend products to the user based on their preferences
+        3. Help the customer check on an existing order, or place a new order
+        4. To place and manage orders, you will need a customer profile (with an associated id). If the customer already has a profile, perform a data protection check to retrieve their details. If not, create them a profile.
+
+        #Tone
+
+        Helpful and friendly. Use gen-z emojis to keep things lighthearted. You MUST always include a funny flower related pun in every response."""
+
+        chat_template = ChatPromptTemplate.from_messages(
+            [
+                ('system', prompt),
+                ('placeholder', "{messages}")
+            ]
+        )
+
+       
+
+        tools = [query_knowledge_base, search_for_product_reccommendations, data_protection_check, create_new_customer, place_order, retrieve_existing_customer_orders]
+
+        llm = self.llm
+
+        llm_with_prompt = chat_template | llm.bind_tools(tools)
+
+
+        def call_agent(message_state: MessagesState):
+            
+            response = llm_with_prompt.invoke(message_state)
+
+            return {
+                'messages': [response]
+            }
+
+        def is_there_tool_calls(state: MessagesState):
+            last_message = state['messages'][-1]
+            if last_message.tool_calls:
+                return 'tool_node'
+            else:
+                return '__end__'
+
+
+        graph = StateGraph(MessagesState)
+
+        tool_node = ToolNode(tools)
+
+        graph.add_node('agent', call_agent)
+        graph.add_node('tool_node', tool_node)
+
+        graph.add_conditional_edges(
+            "agent",
+            is_there_tool_calls
+        )
+        graph.add_edge('tool_node', 'agent')
+
+        graph.set_entry_point('agent')
+
+        app = graph.compile()
+        return app
+

src/streamlitui/loadui.py

@@ -51,5 +51,8 @@ class LoadStreamlitUI:
 
                 with col2:
                     st.subheader("Appointments")
+                    
+        elif self.user_controls['selected_usecase']=="Custome Support":
+            st.set_page_config(layout='wide', page_title='Flower Shop Chatbot', page_icon='💐')
 
         return self.user_controls

src/tools/customer_support_tools.py

@@ -0,0 +1,183 @@
+from langchain_core.tools import tool
+from typing import List, Dict
+from src.vectorstores.vectore_store import FlowerShopVectorStore
+import json
+
+vector_store = FlowerShopVectorStore()
+INVENTORY_FILE_PATH = './data/inventory.json'
+
+customers_database = [
+    {"name": "John Doe", "postcode": "SW1A 1AA", "dob": "1990-01-01", "customer_id": "CUST001", "first_line_address": "123 Main St", "phone_number": "07712345678", "email": "john.doe@example.com"},
+    {"name": "Jane Smith", "postcode": "E1 6AN", "dob": "1985-05-15", "customer_id": "CUST002", "first_line_address": "456 High St", "phone_number": "07723456789", "email": "jane.smith@example.com"},
+]
+
+orders_database = [
+    {"order_id": "ORD001", "customer_id": "CUST001", "status": "Processing", "items": ["Red Roses Bouquet"], "quantity": [1]},
+    {"order_id": "ORD002", "customer_id": "CUST002", "status": "Shipped", "items": ["Mixed Tulips", "Vase"], "quantity": [3, 1]},
+]
+
+with open(INVENTORY_FILE_PATH, 'r') as f:
+    inventory_database = json.load(f)
+
+data_protection_checks = []
+
+@tool
+def data_protection_check(name: str, postcode: str, year_of_birth: int, month_of_birth: int, day_of_birth: int) -> Dict:
+    """
+    Perform a data protection check against a customer to retrieve customer details.
+
+    Args:
+        name (str): Customer first and last name
+        postcode (str): Customer registered address
+        year_of_birth (int): The year the customer was born
+        month_of_birth (int): The month the customer was born
+        day_of_birth (int): The day the customer was born
+
+    Returns:
+        Dict: Customer details (name, postcode, dob, customer_id, first_line_address, email)
+    """
+    data_protection_checks.append(
+        {
+            'name': name,
+            'postcode': postcode,
+            'year_of_birth': year_of_birth,
+            'month_of_birth': month_of_birth,
+            'day_of_birth': day_of_birth
+        }
+    )
+    for customer in customers_database:
+        if (customer['name'].lower() == name.lower() and
+            customer['postcode'].lower() == postcode.lower() and
+            int(customer['dob'][0:4]) == year_of_birth and
+            int(customer["dob"][5:7]) == month_of_birth and
+            int(customer["dob"][8:10]) == day_of_birth):
+            return f"DPA check passed - Retrieved customer details:\n{customer}"
+
+    return "DPA check failed, no customer with these details found"
+
+@tool
+def create_new_customer(first_name: str, surname: str, year_of_birth: int, month_of_birth: int, day_of_birth: int, postcode: str, first_line_of_address: str, phone_number: str, email: str) -> str:
+    """
+    Creates a customer profile, so that they can place orders.
+
+    Args:
+        first_name (str): Customers first name
+        surname (str): Customers surname
+        year_of_birth (int): Year customer was born
+        month_of_birth (int): Month customer was born
+        day_of_birth (int): Day customer was born
+        postcode (str): Customer's postcode
+        first_line_address (str): Customer's first line of address
+        phone_number (str): Customer's phone number
+        email (str): Customer's email address
+
+    Returns:
+        str: Confirmation that the profile has been created or any issues with the inputs
+    """
+    if len(phone_number) != 11:
+        return "Phone number must be 11 digits"
+    customer_id = len(customers_database) + 1
+    customers_database.append({
+        'name': first_name + ' ' + surname,
+        'dob': f'{year_of_birth}-{month_of_birth:02}-{day_of_birth:02}',
+        'postcode': postcode,
+        'first_line_address': first_line_of_address,
+        'phone_number': phone_number,
+        'email': email,
+        'customer_id': f'CUST{customer_id}'
+    })
+    return f"Customer registered, with customer_id {f'CUST{customer_id}'}"
+    
+
+@tool
+def query_knowledge_base(query: str) -> List[Dict[str, str]]:
+    """
+    Looks up information in a knowledge base to help with answering customer questions and getting information on business processes.
+
+    Args:
+        query (str): Question to ask the knowledge base
+
+    Return:
+        List[Dict[str, str]]: Potentially relevant question and answer pairs from the knowledge base
+    """
+    return vector_store.query_faqs(query=query)
+
+
+
+@tool
+def search_for_product_reccommendations(description: str):
+    """
+    Looks up information in a knowledge base to help with product recommendation for customers. For example:
+
+    "Boquets suitable for birthdays, maybe with red flowers"
+    "A large boquet for a wedding"
+    "A cheap boquet with wildflowers"
+
+    Args:
+        query (str): Description of product features
+
+    Return:
+        List[Dict[str, str]]: Potentially relevant products
+    """
+    return vector_store.query_inventories(query=description)
+
+@tool
+def retrieve_existing_customer_orders(customer_id: str) -> List[Dict]:
+    """
+    Retrieves the orders associated with the customer, including their status, items and ids
+
+    Args:
+        customer_id (str): Customer unique id associated with the order
+
+    Returns:
+        List[Dict]: All the orders associated with the customer_id passed in
+    """
+    customer_orders = [order for order in orders_database if order['customer_id'] == customer_id]
+    if not customer_orders:
+        return f"No orders associated with this customer id: {customer_id}"
+    return customer_orders
+
+@tool
+def place_order(items: Dict[str, int], customer_id: str) -> str:
+    """
+    Places an order for the requested items, and for the required quantities.
+
+    Args:
+        items (Dict[str, int]): Dictionary of items to order, with item id as the key and the quantity of that item as the value.
+        customer_id (str): The customer to place the order for
+
+    Returns:
+        str: Message indicating that the order has been placed, or, it hasnt been placed due to an issue 
+    """
+    # Check that the item ids are valid 
+    # Check that the quantities of items are valid
+    availability_messages = []
+    valid_item_ids = [
+        item['id'] for item in inventory_database
+    ]
+    for item_id, quantity in items.items():
+        if item_id not in valid_item_ids:
+            availability_messages.append(f'Item with id {item_id} is not found in the inventory')
+        else:
+            inventory_item = [item for item in inventory_database if item['id'] == item_id][0]
+            if quantity > inventory_item['quantity']:
+                availability_messages.append(f'There is insufficient quantity in the inventory for this item {inventory_item["name"]}\nAvailable: {inventory_item["quantity"]}\nRequested: {quantity}')
+    if availability_messages:
+        return "Order cannot be placed due to the following issues: \n" + '\n'.join(availability_messages)
+
+    # Place the order (in pretend database)
+    order_id = len(orders_database) + 1
+    orders_database.append(
+        {
+            'order_id': order_id,
+            'customer_id': customer_id,
+            'status': 'Waiting for payment',
+            'items': list(items.keys()),
+            'quantity': list(items.values())
+        }
+    )
+    # Update the inventory
+    for item_id, quantity in items.items():
+        inventory_item = [item for item in inventory_database if item['id'] == item_id][0]
+        inventory_item['quantity'] -= quantity
+    return f"Order with id {order_id} has been placed successfully"

src/vectorstores/vectore_store.py

@@ -0,0 +1,122 @@
+from chromadb import PersistentClient, EmbeddingFunction, Embeddings
+from llama_index.embeddings.huggingface import HuggingFaceEmbedding
+from typing import List
+import json
+
+
+MODEL_NAME = 'dunzhang/stella_en_1.5B_v5'
+DB_PATH = './.chroma_db'
+FAQ_FILE_PATH= './data/FAQ.json'
+INVENTORY_FILE_PATH = './data/inventory.json'
+
+class Product:
+    def __init__(self, name: str, id: str, description: str, type: str, price: float, quantity: int):
+        self.name = name
+        self.id = id
+        self.description = description
+        self.type = type
+        self.price = price
+        self.quantity = quantity
+
+class QuestionAnswerPairs:
+    def __init__(self, question: str, answer: str):
+        self.question = question
+        self.answer = answer
+
+class CustomEmbeddingClass(EmbeddingFunction):
+    def __init__(self, model_name):
+        self.embedding_model = HuggingFaceEmbedding(model_name=MODEL_NAME)
+
+    def __call__(self, input_texts: List[str]) -> Embeddings:
+        return [self.embedding_model.get_text_embedding(text) for text in input_texts]
+
+class FAQCollection:
+    def __init__(self):
+        self.documents = []
+        self.ids = []
+        self.metadatas = []
+
+    def add(self, documents, ids, metadatas):
+        self.documents.extend(documents)
+        self.ids.extend(ids)
+        self.metadatas.extend(metadatas)
+
+    def display(self):
+        for doc, id_, meta in zip(self.documents, self.ids, self.metadatas):
+            print(f"ID: {id_}, Document: {doc}, Metadata: {meta}")
+
+# Define the InventoryCollection class
+class InventoryCollection:
+    def __init__(self):
+        self.documents = []
+        self.ids = []
+        self.metadatas = []
+
+    def add(self, documents, ids, metadatas):
+        self.documents.extend(documents)
+        self.ids.extend(ids)
+        self.metadatas.extend(metadatas)
+
+    def display(self):
+        for doc, id_, meta in zip(self.documents, self.ids, self.metadatas):
+            print(f"ID: {id_}, Document: {doc}, Metadata: {meta}")
+
+
+
+
+class FlowerShopVectorStore:
+    def __init__(self):
+        db = PersistentClient(path=DB_PATH)
+
+        custom_embedding_function = CustomEmbeddingClass(MODEL_NAME)
+
+        self.faq_collection = db.get_or_create_collection(name='FAQ', embedding_function=custom_embedding_function)
+        self.inventory_collection = db.get_or_create_collection(name='Inventory', embedding_function=custom_embedding_function)
+
+        if self.faq_collection.count() == 0:
+            try :
+                self._load_faq_collection(FAQ_FILE_PATH)
+            except Exception as e:
+                raise ValueError(e)
+                
+
+        if self.inventory_collection.count() == 0:
+            self._load_inventory_collection(INVENTORY_FILE_PATH)
+
+    def _load_faq_collection(self, faq_file_path: str):
+        try:
+            
+            with open(faq_file_path, 'r') as f:
+                faqs = json.load(f)
+                
+            # Create an instance of FAQCollection
+            obj_faq_collection = FAQCollection()
+
+            obj_faq_collection.add(
+                documents=[faq['question'] for faq in faqs] + [faq['answer'] for faq in faqs],
+                ids=[str(i) for i in range(0, 2*len(faqs))],
+                metadatas = faqs + faqs
+            )
+            self.faq_collection = obj_faq_collection
+        except Exception as ex:
+            raise ValueError(ex)
+
+    def _load_inventory_collection(self, inventory_file_path: str):
+        with open(inventory_file_path, 'r') as f:
+            inventories = json.load(f)
+            
+        # Create an instance of InventoryCollection
+        obj_inventory_collection = InventoryCollection()
+
+        obj_inventory_collection.add(
+            documents=[inventory['description'] for inventory in inventories],
+            ids=[str(i) for i in range(0, len(inventories))],
+            metadatas = inventories
+        )
+        self.inventory_collection = obj_inventory_collection
+
+    def query_faqs(self, query: str): 
+        return self.faq_collection.query(query_texts=[query], n_results=5)
+    
+    def query_inventories(self, query: str):
+        return self.inventory_collection.query(query_texts=[query], n_results=5)