run testes on assessments predictions pipeline

test.py

@@ -1,16 +1,7 @@
-# Example usage of the Chatbot class:
-from src.services.chatbot import Chatbot
-from src.utils.document_loader import load_document
-if __name__ == "__main__":
-    chatbot = Chatbot()
+# Example usage
+from scripts.run_assessment_prediction_trainer import CompanyModelPipeline
+company_ids = ['company_123', 'company_456', 'company_789']
+input_base_path = '/root/ds_erp_ai/data/raw/dummy_assessment_data.csv'  # The base path where the raw data for each company is stored
 
-    # Example inputs
-    path = r"C:\Users\User\Desktop\Blessing_AI\MKD\test_erp_ai\erp_ai\test\erp_ai\data\raw\coding_task_completion_document.pdf"
-
-    question = "Have you completed Task X?"
-    user_input = "Yes"
-    docs = load_document(path)
-
-    # Validate the worker's answer using the provided document
-    validation_result = chatbot.validate_worker(question, docs)
-    print(validation_result)
+pipeline = CompanyModelPipeline(company_ids=company_ids, input_base_path=input_base_path)
+pipeline.run_pipeline()