File size: 1,623 Bytes
a963d65 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
#!/usr/bin/env python3
"""
Direct workflow orchestrator test
"""
import sys
import asyncio
sys.path.insert(0, '.')
from src.workflow_orchestrator import workflow_orchestrator
async def test_workflow():
print("🔍 Testing workflow orchestrator directly...")
# Create a small test PDF bytes (simple mock)
test_pdf_bytes = b'%PDF-1.4\n1 0 obj\n<< /Type /Catalog /Pages 2 0 R >>\nendobj\n2 0 obj\n<< /Type /Pages /Kids [3 0 R] /Count 1 >>\nendobj\n3 0 obj\n<< /Type /Page /Parent 2 0 R /MediaBox [0 0 612 792] >>\nendobj\nxref\n0 4\n0000000000 65535 f \n0000000010 00000 n \n0000000053 00000 n \n0000000100 00000 n \ntrailer\n<< /Size 4 /Root 1 0 R >>\nstartxref\n149\n%%EOF'
try:
print(f"📄 Test document size: {len(test_pdf_bytes)} bytes")
print("🚀 Calling workflow_orchestrator.process_complete_workflow()...")
print()
result = await workflow_orchestrator.process_complete_workflow(
document_bytes=test_pdf_bytes,
user_id="test_user",
filename="test.pdf",
document_type="clinical_document",
use_mistral_ocr=True, # Enable Mistral OCR
use_advanced_llm=True,
llm_model="codellama",
generate_fhir=False # Skip FHIR for this test
)
print("✅ Workflow completed successfully!")
print(f"Result keys: {list(result.keys())}")
except Exception as e:
print(f"❌ Workflow failed: {str(e)}")
import traceback
traceback.print_exc()
if __name__ == "__main__":
asyncio.run(test_workflow()) |