Create create_knowledge_base.py

create_knowledge_base.py

@@ -0,0 +1,29 @@
+import pandas as pd
+
+import numpy as np
+
+from sentence_transformers import SentenceTransformer
+
+import faiss
+
+import pickle
+
+from datasets import load_dataset
+
+import os
+
+
+
+# Write the code below
+def create_ecommerce_knowledge_base() {
+  print("Loading Bitext e-commerce dataset...") 
+  # Load the dataset 
+  dataset = load_dataset("bitext/Bitext-customer-support-llm-chatbot-training-dataset") 
+  print(f"Dataset loaded: {len(dataset['train'])} examples") 
+}
+
+# Don't remove the following code
+
+if __name__ == "__main__":
+
+    create_ecommerce_knowledge_base()