"""Configuration file for knowledge base options and their corresponding values"""

# Knowledge Base Options
KNOWLEDGE_BASE_OPTIONS = [
    "IBM Cloud Platform - General",
    "IBM Cloud Platform - General - Semantically Chunked",
    # "IBM Consulting Services",
    # "IBM Security Solutions",
    # "IBM AI & Analytics"
]

# System Prompts for each knowledge base
SYSTEM_PROMPTS = {
    "IBM Cloud Platform - General": {
        "bot_2": "You are an IBM Cloud Platform expert assistant. Help users understand cloud services, deployment, and infrastructure.",
        "bot_3": "You are a technical solutions architect specializing in IBM Cloud. Provide detailed architectural guidance."
    },
    "IBM Watson Products": {
        "bot_2": "You are a Watson product specialist. Guide users through Watson's AI capabilities and use cases.",
        "bot_3": "You are a Watson implementation expert. Provide technical details and integration guidance."
    },
    "IBM Consulting Services": {
        "bot_2": "You are an IBM Consulting advisor. Help users understand our service offerings and methodologies.",
        "bot_3": "You are a consulting solutions architect. Provide detailed project and implementation guidance."
    },
    "IBM Security Solutions": {
        "bot_2": "You are an IBM Security expert. Guide users through our security products and best practices.",
        "bot_3": "You are a security architect. Provide detailed technical security guidance and implementations."
    },
    "IBM AI & Analytics": {
        "bot_2": "You are an IBM AI and Analytics specialist. Help users understand our AI and data solutions.",
        "bot_3": "You are an AI solutions architect. Provide technical guidance on AI implementation and integration."
    }
}

# Vector Indexes for each knowledge base (using IDs)
VECTOR_INDEXES = {
    "IBM Cloud Platform - General": {
        "index_1": "f1d8a4b3-9e27-4f81-a7c0-6e5d8b2a91c3",  # Example UUID format
        "index_2": "c5e2b917-6d34-4a58-9f12-8d3a7b4c6e9d",
        "contents": [
            "IBM_Cloud_Overview.pdf",
            "Cloud_Services_Catalog.pdf",
            "Platform_Best_Practices.pdf"
        ]
    },
    "IBM Watson Products": {
        "index_1": "a2b9c8d7-5e63-4f12-b9a4-1d7e6c5b4a3f",
        "index_2": "e8f7d6c5-4b3a-2d1c-9e8f-7d6c5b4a3e2d",
        "contents": [
            "Watson_Assistant_Guide.pdf",
            "Watson_Discovery_Docs.pdf",
            "Watson_Studio_Manual.pdf"
        ]
    },
    "IBM Consulting Services": {
        "index_1": "b3c4d5e6-7f89-4a12-b3c4-d5e6f7a8b9c0",
        "index_2": "d5e6f7a8-9b0c-1d2e-3f4g-5h6i7j8k9l0",
        "contents": [
            "Consulting_Methodology.pdf",
            "Service_Offerings.pdf",
            "Case_Studies.pdf"
        ]
    },
    "IBM Security Solutions": {
        "index_1": "g4h5i6j7-8k9l-0m1n-2o3p-4q5r6s7t8u9",
        "index_2": "i7j8k9l0-1m2n-3o4p-5q6r-7s8t9u0v1w2",
        "contents": [
            "Security_Products_Overview.pdf",
            "Security_Best_Practices.pdf",
            "Threat_Management_Guide.pdf"
        ]
    },
    "IBM AI & Analytics": {
        "index_1": "m1n2o3p4-5q6r-7s8t-9u0v-1w2x3y4z5a6",
        "index_2": "p4q5r6s7-8t9u-0v1w-2x3y-4z5a6b7c8d9",
        "contents": [
            "AI_Solutions_Overview.pdf",
            "Analytics_Platform_Guide.pdf",
            "Machine_Learning_Docs.pdf"
        ]
    }
}