Update knowledge_bases.py

knowledge_bases.py

@@ -0,0 +1,83 @@
+"""Configuration file for knowledge base options and their corresponding values"""
+
+# Knowledge Base Options
+KNOWLEDGE_BASE_OPTIONS = [
+    "IBM Cloud Platform - General",
+    "IBM Watson Products",
+    # "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"
+        ]
+    }
+}