kchen707/wedding-bundle-builder
0
1"""2Configuration constants for the Wedding Bundle Builder.3 4Originally Cell 4 (1c. Configuration) of the notebook. Kept identical5so all downstream modules see the same values.6"""7 8# -----------------------------------------------------------------9# Models (OpenRouter slugs)10# -----------------------------------------------------------------11# NOTE: These slugs are passed through OpenRouter. If a slug is wrong12# the API will return an error at the first call. The fallback inside13# generate_explanation() will try CHAT_MODEL if EXPLANATION_MODEL fails.14EMBEDDING_MODEL = "openai/text-embedding-3-large"15CHAT_MODEL = "google/gemini-3.1-flash-lite-preview"16EXPLANATION_MODEL = "anthropic/claude-sonnet-4-5"17VISION_MODEL = "google/gemini-3-flash-preview"18EMBEDDING_DIM = 307219 20# -----------------------------------------------------------------21# Matching parameters22# -----------------------------------------------------------------23TOP_PICK = 124RUNNERS_UP = 225SIMILARITY_THRESHOLD = 0.2526RERANK_CANDIDATES = 1527NUM_QUERY_EXPANSIONS = 328 29# -----------------------------------------------------------------30# Vendor categories31# -----------------------------------------------------------------32VENDOR_CATEGORIES = [33 "Venue", "Caterer/Food", "Photographer", "Florist",34 "Bakery/Wedding Cake", "DJ/Live Music", "Makeup Artist/Beauty", "Attire",35]36 37CATEGORY_ICONS = {38 "Venue": "๐๏ธ",39 "Caterer/Food": "๐ฝ๏ธ",40 "Photographer": "๐ธ",41 "Florist": "๐",42 "Bakery/Wedding Cake": "๐",43 "DJ/Live Music": "๐ต",44 "Makeup Artist/Beauty": "๐",45 "Attire": "๐",46}47 