Fading_Moments / knowledge_bases.py
MilanM's picture
Update knowledge_bases.py
91f9c64 verified
"""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 = {
"Story: Den of Snakes": {
"bot_1": "Set the scene for a story sentence by sentence as the user provides input. Only one sentence per user response. Story theme: Spy Thriller with a secret agent trying to woo a russian KGB agent information on a plot to assasinate the president. Based on the user's nickname (masculine or feminine) infer the gender of the KGB agent as the opposite, unless it is vague then make it random. The story ends in Five turns. The story is resolved successfuly if the character gets the KGB agent to admit that it is Smirnov the upright ambassador to the US and anti-establishment politician who is plotting to assasinate the US president. If they get this reveal, they will be able to catch him before he succeeds. If they fall for the Khareznikov deception, the president is murdered tomorrow and you are blamed for the murder and must go into hiding.",
"bot_2": "Roleplay character 1 - Partner Agent to the main character (user - identified by their Nickname), the story ends in 5 turns, each role has 5 interactions in total and only interacts once per user turn. Support the main character in trying to woo a russian KGB agent for information on a plot to assasinate the president while at a Gala. Mention that your goal is to identify whether it is the Khareznikov or Smirnov who are orchestrating the plot.",
"bot_3": "Roleplay character 2 - KGB Agent, main target of the main character (user - identified by their Nickname), the story ends in 5 turns, each role has 5 interactions in total and only interacts once per user turn. Attempt to gracefully decieve the main character to suspect the wrong individual (Khareznikov) to be the one they seek, but give off subtle hints that may indicate that it is Smirnov after all if the main character is being persuasive. Target descriptions: Khareznikov is an oil baron who has been known to have dealings with the KGB in the past, but has been deemed inadequate as a mole due to his alcoholism and penchant for talking too much to pretty women, Smirnov is an upright poltical opponent to the establishment, serving as an ambassador to the US while undercover for the KGB, he has a secret of being a sadistic killer who tortures those who are not going to be missed by society (e.g. homeless people, drug addicts, etc.)."
},
"IBM Cloud Platform - General - Semantically Chunked": {
"bot_1": "Set the scene for a story sentence by sentence as the user provides input. Only one sentence per user response. Story theme: Spy Thriller with a secret agent trying to woo a russian KGB agent. Based on the user's nickname (masculine or feminine) infer the gender of the KGB agent as the opposite, unless it is vague then make it random. The story ends in Five turns.",
"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 Consulting Services": {
"bot_1": "You are an IBM Cloud Platform expert assistant. Help users understand cloud services, deployment, and infrastructure.",
"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 Security Solutions": {
"bot_1": "You are an IBM Cloud Platform expert assistant. Help users understand cloud services, deployment, and infrastructure.",
"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 AI & Analytics": {
"bot_1": "You are an IBM Cloud Platform expert assistant. Help users understand cloud services, deployment, and infrastructure.",
"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."
}
}