adf-chatbot2 / src /utils /constants.py
Yannick Lemin
fixed tests
e9c64b8
import os
from dotenv import load_dotenv
load_dotenv(override=True)
class Constants:
# MODEL NAMES
MODEL = os.getenv("MODEL")
EMBEDDING_MODEL = os.getenv("EMBEDDING_MODEL")
# Database
DB_NAME = os.getenv("DB_NAME")
COLLECTION_NAME = os.getenv("COLLECTION_NAME")
ATLAS_VECTOR_SEARCH_INDEX_NAME = os.getenv("ATLAS_VECTOR_SEARCH_INDEX_NAME")
# Search parameters
CUSTOMER_SEARCH_LIMIT = 5
STAFF_SEARCH_LIMIT = 1
# Prompts (in French)
IMAGE_DESCRIPTION_PROMPT = """
Décrivez cette image en détail. Concentrez-vous sur les éléments de ferronnerie décorative
visibles dans l'image. Mentionnez le style, les motifs, les formes, et l'utilisation possible
de cet article dans un jardin ou une maison.
"""
CUSTOMER_SEARCH_SYSTEM_PROMPT = """
Vous êtes un assistant spécialisé dans la ferronnerie décorative pour l'Atelier du Fer.
Votre tâche est d'aider les clients à trouver des articles qui correspondent à leurs besoins.
"""
STAFF_SEARCH_SYSTEM_PROMPT = """
Vous êtes un assistant pour les employés de l'Atelier du Fer.
Votre tâche est d'identifier précisément un article du catalogue à partir d'une image.
"""
# UI Text (in French)
UI_TITLE = "L'Atelier du Fer - Recherche Catalogue"
UI_CUSTOMER_TAB = "Recherche Client"
UI_STAFF_TAB = "Recherche Personnel"
UI_CUSTOMER_DESCRIPTION = "Trouvez des articles de ferronnerie décorative pour embellir votre espace"
UI_STAFF_DESCRIPTION = "Identifiez rapidement un article du catalogue"
UI_TEXT_INPUT_LABEL = "Décrivez ce que vous recherchez"
UI_IMAGE_INPUT_LABEL = "Téléchargez une image (optionnel)"
UI_SEARCH_BUTTON = "Rechercher"
UI_LOADING_TEXT = "Recherche en cours..."
UI_NO_RESULTS = "Aucun résultat trouvé. Veuillez essayer avec une autre description ou image."
UI_PRODUCT_BUTTON = "Voir le produit"
UI_PRICE_LABEL = "Prix:"
# Placeholder image URL for articles without images
PLACEHOLDER_IMAGE_URL = "https://placehold.co/150?text=Pas+d'image"