"""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"
]

# Vector Indexes for each knowledge base
VECTOR_INDEXES = {
    "IBM Cloud Platform - General": {
        "project_id": "a8403b62-6917-4a0e-b1c6-78b696fb93c1",
        "index_1": "a7c1c5c8-4897-46fc-932d-77b8fdcc0155",
        "index_2": "a7c1c5c8-4897-46fc-932d-77b8fdcc0155",
        "contents": [
            "ibm-cloud-platform-automation-hub.pdf",
            "ibm-cloud-overview.pdf",
            "ibm-cloud-object-storage.pdf",
            "ibm-cloud-security-groups.pdf",
            "ibm-cloud-infrastructure.pdf",
            "ibm-cloud-container-registry.pdf",
            "ibm-cloud-manage-account-resources-access.pdf"
        ]
    },
    "IBM Cloud Platform - General - Semantically Chunked": {
        "project_id": "a8403b62-6917-4a0e-b1c6-78b696fb93c1",
        "index_1": "a7c1c5c8-4897-46fc-932d-77b8fdcc0155",
        "index_2": "a7c1c5c8-4897-46fc-932d-77b8fdcc0155",
        "contents": [
            "ibm-cloud-platform-automation-hub.pdf",
            "ibm-cloud-overview.pdf",
            "ibm-cloud-object-storage.pdf",
            "ibm-cloud-security-groups.pdf",
            "ibm-cloud-infrastructure.pdf",
            "ibm-cloud-container-registry.pdf",
            "ibm-cloud-manage-account-resources-access.pdf"
        ]
    },
    "IBM Consulting Services": {
        "project_id": "a8403b62-6917-4a0e-b1c6-78b696fb93c1",
        "index_1": "a7c1c5c8-4897-46fc-932d-77b8fdcc0155",
        "index_2": "a7c1c5c8-4897-46fc-932d-77b8fdcc0155",
        "contents": [
            "ibm-cloud-platform-automation-hub.pdf",
            "ibm-cloud-overview.pdf",
            "ibm-cloud-object-storage.pdf",
            "ibm-cloud-security-groups.pdf",
            "ibm-cloud-infrastructure.pdf",
            "ibm-cloud-container-registry.pdf",
            "ibm-cloud-manage-account-resources-access.pdf"
        ]
    },
    "IBM Security Solutions": {
        "project_id": "a8403b62-6917-4a0e-b1c6-78b696fb93c1",
        "index_1": "a7c1c5c8-4897-46fc-932d-77b8fdcc0155",
        "index_2": "a7c1c5c8-4897-46fc-932d-77b8fdcc0155",
        "contents": [
            "ibm-cloud-platform-automation-hub.pdf",
            "ibm-cloud-overview.pdf",
            "ibm-cloud-object-storage.pdf",
            "ibm-cloud-security-groups.pdf",
            "ibm-cloud-infrastructure.pdf",
            "ibm-cloud-container-registry.pdf",
            "ibm-cloud-manage-account-resources-access.pdf"
        ]
    },
    "IBM AI & Analytics": {
        "project_id": "a8403b62-6917-4a0e-b1c6-78b696fb93c1",
        "index_1": "a7c1c5c8-4897-46fc-932d-77b8fdcc0155",
        "index_2": "a7c1c5c8-4897-46fc-932d-77b8fdcc0155",
        "contents": [
            "ibm-cloud-platform-automation-hub.pdf",
            "ibm-cloud-overview.pdf",
            "ibm-cloud-object-storage.pdf",
            "ibm-cloud-security-groups.pdf",
            "ibm-cloud-infrastructure.pdf",
            "ibm-cloud-container-registry.pdf",
            "ibm-cloud-manage-account-resources-access.pdf"
        ]
    }
}

# 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 Cloud Platform - General - Semantically Chunked": {
        "bot_2": "You are an IBM Cloud Platform expert assistant with knowledge of semantically organized content. Help users understand cloud services, deployment, and infrastructure.",
        "bot_3": "You are a technical solutions architect specializing in IBM Cloud. Provide detailed architectural guidance using semantically structured information."
    },
    "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."
    }
}