Xu ly SSO

test_dce_pipeline.py

@@ -7,6 +7,7 @@ sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
 
 from core.models import IngestedDocument
 from extraction.dce import DocumentClassificationEngine
+from ingestion.providers.sharepoint_provider import SharePointProvider
 
 logging.basicConfig(level=logging.INFO, format="%(levelname)s:%(name)s:%(message)s")
 
@@ -19,20 +20,22 @@ def main():
         
     with open("ingestion_output.json", "r", encoding="utf-8") as f:
         items = json.load(f)
-        
-    dce = DocumentClassificationEngine()
+    
+    # Khởi tạo provider để download file qua Graph API auth
+    provider = SharePointProvider()
+    dce = DocumentClassificationEngine(provider=provider)
     
     print(f"Loaded {len(items)} items from ingestion_output.json\n")
     
     for item in items:
         if item.get("is_folder"):
-            continue # DCE only processes files
+            continue
             
         doc = IngestedDocument(**item)
         
         print(f"\n--- Processing: {doc.name} ---")
-        result = dce.classify(doc)
-        print(f">> Policy: {result.processing_policy.value} | Reason: {result.reason}")
+        result = dce.classify(doc, target_item=item)
+        print(f">> Type: {result.doc_type.value} | Policy: {result.processing_policy.value} | Reason: {result.reason}")
 
 if __name__ == "__main__":
     main()