added sops generation for personal assessment
This commit is contained in:
@@ -1,21 +1,37 @@
|
||||
from src.services.sop_generator import SopGenerator
|
||||
from src.utils.document_loader import load_document
|
||||
from src.services.sop_generator import SopPersonalAssessment
|
||||
|
||||
|
||||
file_path = "/root/ds_erp_ai/data/raw/test_sop.pdf"
|
||||
file_path = "/root/ds_erp_ai/data/raw/document.doc"
|
||||
docs = load_document(file_path)
|
||||
sop = SopGenerator()
|
||||
from src.services.sop_generator import SopGeneratorDocument
|
||||
sop = SopPersonalAssessment()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
# Assuming 'sop' is an instance of SopGenerator and 'docs' is the loaded document content.
|
||||
|
||||
# Step 1: Get the roles from the document
|
||||
roles = sop.get_roles(docs)["roles"]
|
||||
print(f"Roles {roles}")
|
||||
|
||||
sop_status = sop.check_role_sop(roles=roles, docs=docs)
|
||||
print(sop_status)
|
||||
roles = ["cloud engineer"]
|
||||
sops = sop.generate_sops(roles=roles,docs=docs)
|
||||
|
||||
roles = [
|
||||
{
|
||||
"role": "Content Marketing Specialist",
|
||||
"sop_types": ["will", "shall"],
|
||||
"areas": ["communication", "development"]
|
||||
},
|
||||
{
|
||||
"role": "Digital Marketing Specialist",
|
||||
"sop_types": ["must"],
|
||||
"areas": ["finance", "project management"]
|
||||
},
|
||||
{
|
||||
"role": "Information Technology Officer",
|
||||
"sop_types": ["shall", "must"],
|
||||
"areas": ["development", "communication", "operations"]
|
||||
}
|
||||
]
|
||||
|
||||
|
||||
|
||||
sops = sop.generate_sops_by_role_and_area(roles)
|
||||
print(f"sops:{sops}")
|
||||
|
||||
Reference in New Issue
Block a user