monkeybrain-runtime 1.0.0__py3-none-any.whl

src/monkey_brain/kernel/intents/intent_registry.py

@@ -0,0 +1,705 @@
+from src.monkey_brain.kernel.intents.predicates.drug_research import drug_research_question_answer, is_drug_research_question
+from src.monkey_brain.kernel.intents.predicates.batch_record import batch_record_question_answer, is_batch_record_question
+from src.monkey_brain.kernel.intents.predicates.batch_creation import batch_creation_question_answer, is_batch_creation_question
+from src.monkey_brain.kernel.intents.predicates.batch_update import batch_update_question_answer, is_batch_update_question
+from src.monkey_brain.kernel.intents.predicates.batch_delete import batch_delete_question_answer, is_batch_delete_question
+from src.monkey_brain.kernel.intents.predicates.batch_hold import batch_hold_question_answer, is_batch_hold_question
+from src.monkey_brain.kernel.intents.predicates.batch_close import batch_close_question_answer, is_batch_close_question
+from src.monkey_brain.kernel.intents.predicates.work_order_create import work_order_create_question_answer, is_work_order_create_question
+from src.monkey_brain.kernel.intents.predicates.work_order_query import work_order_query_question_answer, is_work_order_query_question
+from src.monkey_brain.kernel.intents.predicates.work_order_update import work_order_update_question_answer, is_work_order_update_question
+from src.monkey_brain.kernel.intents.predicates.work_order_delete import work_order_delete_question_answer, is_work_order_delete_question
+from src.monkey_brain.kernel.intents.predicates.work_order_hold import work_order_hold_question_answer, is_work_order_hold_question
+from src.monkey_brain.kernel.intents.predicates.work_order_status import work_order_status_question_answer, is_work_order_status_question
+from src.monkey_brain.kernel.intents.predicates.approval_create import approval_create_question_answer, is_approval_create_question
+from src.monkey_brain.kernel.intents.predicates.approval_query import approval_query_question_answer, is_approval_query_question
+from src.monkey_brain.kernel.intents.predicates.approval_decision import approval_decision_question_answer, is_approval_decision_question
+from src.monkey_brain.kernel.intents.predicates.approval_hold import approval_hold_question_answer, is_approval_hold_question
+from src.monkey_brain.kernel.intents.predicates.change_control import change_control_question_answer, is_change_control_question, is_change_control_create_question, is_change_control_query_question, is_change_control_update_question, is_change_control_decision_question, change_control_create_question_answer, change_control_query_question_answer, change_control_update_question_answer, change_control_decision_question_answer
+from src.monkey_brain.kernel.intents.predicates.compliance_audit import compliance_audit_question_answer, is_compliance_audit_question, compliance_audit_approval_question_answer
+from src.monkey_brain.kernel.intents.predicates.decision_intelligence import decision_intelligence_question_answer, is_decision_intelligence_question
+from src.monkey_brain.kernel.intents.predicates.drug_research import drug_research_question_answer, is_drug_research_question
+from src.monkey_brain.kernel.intents.predicates.production_kpi import production_kpi_question_answer, is_production_kpi_question
+from src.monkey_brain.kernel.intents.predicates.warehouse_shipping import warehouse_shipping_question_answer, is_warehouse_shipping_question
+from src.monkey_brain.kernel.intents.predicates.worker import worker_question_answer, is_worker_question
+from src.monkey_brain.kernel.intents.predicates.sop_create import sop_create_question_answer, is_sop_create_question
+from src.monkey_brain.kernel.intents.predicates.sop_query import sop_query_question_answer, is_sop_query_question
+from src.monkey_brain.kernel.intents.predicates.sop_update import sop_update_question_answer, is_sop_update_question
+from src.monkey_brain.kernel.intents.predicates.fuzzy_match import fuzzy_match_pattern, extract_spelling_corrections
+
+# ------------------------------------------------------------------
+# Semantic Definition
+# ------------------------------------------------------------------
+
+from dataclasses import dataclass, field
+from typing import Any, Awaitable, Callable
+
+
+@dataclass(slots=True, frozen=True)
+class IntentDefinition:
+
+    intent: str
+
+    description: str = ""
+
+    required_inputs: list[str] = field(
+        default_factory=list
+    )
+
+    required_outputs: list[str] = field(
+        default_factory=list
+    )
+
+    default_augmentations: list[str] = field(
+        default_factory=list
+    )
+
+    constraints: dict[str, Any] = field(
+        default_factory=dict
+    )
+
+    min_confidence: float = 0.58
+
+
+DRUG_RESEARCH_DEFINITION = IntentDefinition(
+    intent="drug_research",
+    required_inputs=["compound"],
+    required_outputs=["drug_research_result"],
+    min_confidence=0.50,
+)
+
+BATCH_RECORD_DEFINITION = IntentDefinition(
+    intent="batch_record",
+    required_inputs=["batch_id"],
+    required_outputs=["batch_record"],
+    default_augmentations=[
+        "approvals",
+        "deviations",
+        "lineage",
+    ],
+)
+
+BATCH_CREATION_DEFINITION = IntentDefinition(
+    intent="batch_creation_request",
+    required_inputs=["product", "size", "route"],
+    required_outputs=["workflow"],
+    default_augmentations=[
+        "sops",
+        "process_parameters",
+        "quality_controls",
+    ],
+)
+
+BATCH_UPDATE_DEFINITION = IntentDefinition(
+    intent="batch_update",
+    required_inputs=["batch_id"],
+    required_outputs=["update_workflow"],
+    default_augmentations=[
+        "change_control",
+        "quality_review",
+        "audit_trail",
+    ],
+)
+
+BATCH_DELETE_DEFINITION = IntentDefinition(
+    intent="batch_delete",
+    required_inputs=["batch_id"],
+    required_outputs=["deletion_workflow"],
+    default_augmentations=[
+        "disposition",
+        "regulatory_compliance",
+        "audit_trail",
+    ],
+)
+
+BATCH_HOLD_DEFINITION = IntentDefinition(
+    intent="batch_hold",
+    required_inputs=["batch_id"],
+    required_outputs=["hold_workflow"],
+    default_augmentations=[
+        "stabilization",
+        "notification",
+        "monitoring",
+    ],
+)
+
+BATCH_CLOSE_DEFINITION = IntentDefinition(
+    intent="batch_close",
+    required_inputs=["batch_id"],
+    required_outputs=["close_workflow"],
+    default_augmentations=[
+        "quality_review",
+        "yield_calculation",
+        "release_approval",
+    ],
+)
+
+WORK_ORDER_CREATE_DEFINITION = IntentDefinition(
+    intent="work_order_create",
+    required_inputs=["work_order_type", "description"],
+    required_outputs=["work_order"],
+    default_augmentations=[
+        "sops",
+        "resources",
+        "schedule",
+    ],
+)
+
+WORK_ORDER_QUERY_DEFINITION = IntentDefinition(
+    intent="work_order_query",
+    required_inputs=["work_order_id"],
+    required_outputs=["work_order_details"],
+    default_augmentations=[
+        "status_history",
+        "time_tracking",
+    ],
+)
+
+WORK_ORDER_UPDATE_DEFINITION = IntentDefinition(
+    intent="work_order_update",
+    required_inputs=["work_order_id"],
+    required_outputs=["work_order"],
+    default_augmentations=[
+        "change_control",
+        "notification",
+        "audit_trail",
+    ],
+)
+
+WORK_ORDER_DELETE_DEFINITION = IntentDefinition(
+    intent="work_order_delete",
+    required_inputs=["work_order_id"],
+    required_outputs=["deletion_status"],
+    default_augmentations=[
+        "impact_assessment",
+        "approval",
+        "audit_trail",
+    ],
+)
+
+WORK_ORDER_STATUS_DEFINITION = IntentDefinition(
+    intent="work_order_status",
+    required_inputs=["work_order_id", "target_status"],
+    required_outputs=["work_order"],
+    default_augmentations=[
+        "state_machine",
+        "approval",
+        "notification",
+    ],
+)
+
+WORK_ORDER_HOLD_DEFINITION = IntentDefinition(
+    intent="work_order_hold",
+    required_inputs=["work_order_id"],
+    required_outputs=["work_order"],
+    default_augmentations=[
+        "justification",
+        "notification",
+        "monitoring",
+    ],
+)
+
+DECISION_INTELLIGENCE_DEFINITION = IntentDefinition(
+    intent="decision_intelligence",
+    required_inputs=["question"],
+    required_outputs=["decision_package"],
+    default_augmentations=[
+        "current_state",
+        "historical_outcomes",
+        "simulations",
+        "sops",
+        "policies",
+        "constraints",
+    ],
+    min_confidence=0.55,
+)
+
+APPROVAL_CREATE_DEFINITION = IntentDefinition(
+    intent="approval_create",
+    required_inputs=["approval_type"],
+    required_outputs=["approval"],
+    default_augmentations=[
+        "documentation",
+        "approver_routing",
+        "electronic_signature",
+    ],
+)
+
+APPROVAL_QUERY_DEFINITION = IntentDefinition(
+    intent="approval_query",
+    required_inputs=["approval_id"],
+    required_outputs=["approval_details"],
+    default_augmentations=[
+        "status_history",
+        "decision_history",
+    ],
+)
+
+APPROVAL_DECISION_DEFINITION = IntentDefinition(
+    intent="approval_decision",
+    required_inputs=["approval_id", "decision"],
+    required_outputs=["approval"],
+    default_augmentations=[
+        "review_checklist",
+        "electronic_signature",
+        "notification",
+    ],
+)
+
+APPROVAL_HOLD_DEFINITION = IntentDefinition(
+    intent="approval_hold",
+    required_inputs=["approval_id"],
+    required_outputs=["approval"],
+    default_augmentations=[
+        "justification",
+        "notification",
+        "monitoring",
+    ],
+)
+
+CHANGE_CONTROL_DEFINITION = IntentDefinition(
+    intent="change_control",
+    required_inputs=["change_type"],
+    required_outputs=["change_control"],
+    default_augmentations=[
+        "impact_assessment",
+        "risk_evaluation",
+        "approver_routing",
+    ],
+)
+
+CHANGE_CONTROL_CREATE_DEFINITION = IntentDefinition(
+    intent="change_control_create",
+    required_inputs=["change_type", "description"],
+    required_outputs=["change_control"],
+    default_augmentations=[
+        "impact_assessment",
+        "risk_evaluation",
+        "affected_entities",
+    ],
+)
+
+CHANGE_CONTROL_QUERY_DEFINITION = IntentDefinition(
+    intent="change_control_query",
+    required_inputs=["change_control_id"],
+    required_outputs=["change_control_details"],
+    default_augmentations=[
+        "status_history",
+        "approval_history",
+        "audit_trail",
+    ],
+)
+
+CHANGE_CONTROL_UPDATE_DEFINITION = IntentDefinition(
+    intent="change_control_update",
+    required_inputs=["change_control_id"],
+    required_outputs=["change_control"],
+    default_augmentations=[
+        "change_control",
+        "notification",
+        "audit_trail",
+    ],
+)
+
+CHANGE_CONTROL_DECISION_DEFINITION = IntentDefinition(
+    intent="change_control_decision",
+    required_inputs=["change_control_id", "decision"],
+    required_outputs=["change_control"],
+    default_augmentations=[
+        "review_checklist",
+        "electronic_signature",
+        "notification",
+    ],
+)
+
+PRODUCTION_KPI_DEFINITION = IntentDefinition(
+    intent="production_kpi",
+    required_inputs=[
+        "line",
+        "time_range",
+    ],
+    required_outputs=[
+        "kpi_report",
+    ],
+)
+
+WAREHOUSE_SHIPPING_DEFINITION = IntentDefinition(
+    intent="warehouse_shipping",
+    required_inputs=["shipment"],
+    required_outputs=["shipping_status"],
+    min_confidence=0.56,
+)
+
+WORKER_DEFINITION = IntentDefinition(
+    intent="worker",
+    required_inputs=["worker_id"],
+    required_outputs=["worker"],
+    default_augmentations=[
+        "training",
+        "certifications",
+    ],
+)
+
+APPROVAL_DEFINITION = IntentDefinition(
+    intent="approval",
+    required_inputs=["approval_id"],
+    required_outputs=["approval"],
+    min_confidence=0.52,
+)
+
+AUDIT_LOG_DEFINITION = IntentDefinition(
+    intent="audit_log",
+    required_inputs=["audit_id"],
+    required_outputs=["audit_log"],
+    min_confidence=0.52,
+)
+COUNT_DEFINITION = IntentDefinition(
+    intent="count",
+    required_inputs=["entity_type"],
+    required_outputs=["count_result"],
+    min_confidence=0.50,
+)
+
+PLANT_TOPOLOGY_DEFINITION = IntentDefinition(
+    intent="plant_topology",
+    required_inputs=["entity_name"],
+    required_outputs=["topology_info"],
+    min_confidence=0.50,
+)
+
+ASSET_MANAGEMENT_DEFINITION = IntentDefinition(
+    intent="asset_management",
+    required_inputs=["asset_id"],
+    required_outputs=["asset_info"],
+    min_confidence=0.50,
+)
+
+SOP_COMPLIANCE_DEFINITION = IntentDefinition(
+    intent="sop_compliance",
+    required_inputs=["sop_id"],
+    required_outputs=["compliance_status"],
+    min_confidence=0.50,
+)
+
+QUALITY_APPROVAL_DEFINITION = IntentDefinition(
+    intent="quality_approval",
+    required_inputs=["approval_id"],
+    required_outputs=["approval_status"],
+    min_confidence=0.50,
+)
+
+SIMULATION_DEFINITION = IntentDefinition(
+    intent="simulation",
+    required_inputs=["scenario"],
+    required_outputs=["simulation_result"],
+    min_confidence=0.50,
+)
+
+MAINTENANCE_DEFINITION = IntentDefinition(
+    intent="maintenance",
+    required_inputs=["equipment_id"],
+    required_outputs=["maintenance_status"],
+    min_confidence=0.50,
+)
+
+KNOWLEDGE_BASE_DEFINITION = IntentDefinition(
+    intent="knowledge_base",
+    required_inputs=["topic"],
+    required_outputs=["knowledge_result"],
+    min_confidence=0.50,
+)
+
+INTENT_DEFINITIONS: dict[str, IntentDefinition] = {
+    definition.intent: definition
+    for definition in (
+        DRUG_RESEARCH_DEFINITION,
+        BATCH_RECORD_DEFINITION,
+        BATCH_CREATION_DEFINITION,
+        BATCH_UPDATE_DEFINITION,
+        BATCH_DELETE_DEFINITION,
+        BATCH_HOLD_DEFINITION,
+        BATCH_CLOSE_DEFINITION,
+        WORK_ORDER_CREATE_DEFINITION,
+        WORK_ORDER_QUERY_DEFINITION,
+        WORK_ORDER_UPDATE_DEFINITION,
+        WORK_ORDER_DELETE_DEFINITION,
+        WORK_ORDER_STATUS_DEFINITION,
+        WORK_ORDER_HOLD_DEFINITION,
+        DECISION_INTELLIGENCE_DEFINITION,
+        APPROVAL_CREATE_DEFINITION,
+        APPROVAL_QUERY_DEFINITION,
+        APPROVAL_DECISION_DEFINITION,
+        APPROVAL_HOLD_DEFINITION,
+        CHANGE_CONTROL_DEFINITION,
+        CHANGE_CONTROL_CREATE_DEFINITION,
+        CHANGE_CONTROL_QUERY_DEFINITION,
+        CHANGE_CONTROL_UPDATE_DEFINITION,
+        CHANGE_CONTROL_DECISION_DEFINITION,
+        PRODUCTION_KPI_DEFINITION,
+        WAREHOUSE_SHIPPING_DEFINITION,
+        WORKER_DEFINITION,
+        AUDIT_LOG_DEFINITION,
+        COUNT_DEFINITION,
+        PLANT_TOPOLOGY_DEFINITION,
+        ASSET_MANAGEMENT_DEFINITION,
+        SOP_COMPLIANCE_DEFINITION,
+        QUALITY_APPROVAL_DEFINITION,
+        SIMULATION_DEFINITION,
+        MAINTENANCE_DEFINITION,
+        KNOWLEDGE_BASE_DEFINITION,
+    )
+}
+
+# ------------------------------------------------------------------
+# Legacy Execution Route
+# ------------------------------------------------------------------
+
+
+@dataclass(slots=True, frozen=True)
+class IntentRoute:
+    intent: str
+    handler: Callable[..., Awaitable[Any]]
+    predicate: Callable[[str], bool]
+    min_confidence: float = 0.58
+    question_source: str = "payload"
+    top_k: bool = False
+
+
+DRUG_RESEARCH_INTENT = IntentRoute(
+    intent="drug_research",
+    handler=drug_research_question_answer,
+    predicate=is_drug_research_question,
+    min_confidence=0.50,
+    question_source="visible",
+    top_k=True,
+)
+
+BATCH_RECORD_INTENT = IntentRoute(
+    intent="batch_record",
+    handler=batch_record_question_answer,
+    predicate=is_batch_record_question,
+)
+
+BATCH_CREATION_INTENT = IntentRoute(
+    intent="batch_creation_request",
+    handler=batch_creation_question_answer,
+    predicate=is_batch_creation_question,
+    question_source="visible",
+)
+
+BATCH_UPDATE_INTENT = IntentRoute(
+    intent="batch_update",
+    handler=batch_update_question_answer,
+    predicate=is_batch_update_question,
+    question_source="visible",
+)
+
+BATCH_DELETE_INTENT = IntentRoute(
+    intent="batch_delete",
+    handler=batch_delete_question_answer,
+    predicate=is_batch_delete_question,
+    question_source="visible",
+)
+
+BATCH_HOLD_INTENT = IntentRoute(
+    intent="batch_hold",
+    handler=batch_hold_question_answer,
+    predicate=is_batch_hold_question,
+    question_source="visible",
+)
+
+BATCH_CLOSE_INTENT = IntentRoute(
+    intent="batch_close",
+    handler=batch_close_question_answer,
+    predicate=is_batch_close_question,
+    question_source="visible",
+)
+
+WORK_ORDER_CREATE_INTENT = IntentRoute(
+    intent="work_order_create",
+    handler=work_order_create_question_answer,
+    predicate=is_work_order_create_question,
+    question_source="visible",
+)
+
+WORK_ORDER_QUERY_INTENT = IntentRoute(
+    intent="work_order_query",
+    handler=work_order_query_question_answer,
+    predicate=is_work_order_query_question,
+    question_source="visible",
+)
+
+WORK_ORDER_UPDATE_INTENT = IntentRoute(
+    intent="work_order_update",
+    handler=work_order_update_question_answer,
+    predicate=is_work_order_update_question,
+    question_source="visible",
+)
+
+WORK_ORDER_DELETE_INTENT = IntentRoute(
+    intent="work_order_delete",
+    handler=work_order_delete_question_answer,
+    predicate=is_work_order_delete_question,
+    question_source="visible",
+)
+
+WORK_ORDER_STATUS_INTENT = IntentRoute(
+    intent="work_order_status",
+    handler=work_order_status_question_answer,
+    predicate=is_work_order_status_question,
+    question_source="visible",
+)
+
+WORK_ORDER_HOLD_INTENT = IntentRoute(
+    intent="work_order_hold",
+    handler=work_order_hold_question_answer,
+    predicate=is_work_order_hold_question,
+    question_source="visible",
+)
+
+DECISION_INTELLIGENCE_INTENT = IntentRoute(
+    intent="decision_intelligence",
+    handler=decision_intelligence_question_answer,
+    predicate=is_decision_intelligence_question,
+    question_source="visible",
+    min_confidence=0.55,
+)
+
+APPROVAL_CREATE_INTENT = IntentRoute(
+    intent="approval_create",
+    handler=approval_create_question_answer,
+    predicate=is_approval_create_question,
+    question_source="visible",
+)
+
+APPROVAL_QUERY_INTENT = IntentRoute(
+    intent="approval_query",
+    handler=approval_query_question_answer,
+    predicate=is_approval_query_question,
+    question_source="visible",
+)
+
+APPROVAL_DECISION_INTENT = IntentRoute(
+    intent="approval_decision",
+    handler=approval_decision_question_answer,
+    predicate=is_approval_decision_question,
+    question_source="visible",
+)
+
+APPROVAL_HOLD_INTENT = IntentRoute(
+    intent="approval_hold",
+    handler=approval_hold_question_answer,
+    predicate=is_approval_hold_question,
+    question_source="visible",
+)
+
+CHANGE_CONTROL_INTENT = IntentRoute(
+    intent="change_control",
+    handler=change_control_question_answer,
+    predicate=is_change_control_question,
+    question_source="visible",
+)
+
+CHANGE_CONTROL_CREATE_INTENT = IntentRoute(
+    intent="change_control_create",
+    handler=change_control_question_answer,
+    predicate=is_change_control_create_question,
+    question_source="visible",
+)
+
+CHANGE_CONTROL_QUERY_INTENT = IntentRoute(
+    intent="change_control_query",
+    handler=change_control_question_answer,
+    predicate=is_change_control_query_question,
+    question_source="visible",
+)
+
+CHANGE_CONTROL_UPDATE_INTENT = IntentRoute(
+    intent="change_control_update",
+    handler=change_control_question_answer,
+    predicate=is_change_control_update_question,
+    question_source="visible",
+)
+
+CHANGE_CONTROL_DECISION_INTENT = IntentRoute(
+    intent="change_control_decision",
+    handler=change_control_question_answer,
+    predicate=is_change_control_decision_question,
+    question_source="visible",
+)
+
+PRODUCTION_KPI_INTENT = IntentRoute(
+    intent="production_kpi",
+    handler=production_kpi_question_answer,
+    predicate=is_production_kpi_question,
+)
+
+WAREHOUSE_SHIPPING_INTENT = IntentRoute(
+    intent="warehouse_shipping",
+    handler=warehouse_shipping_question_answer,
+    predicate=is_warehouse_shipping_question,
+    min_confidence=0.56,
+)
+
+WORKER_INTENT = IntentRoute(
+    intent="worker",
+    handler=worker_question_answer,
+    predicate=is_worker_question,
+    question_source="visible",
+)
+
+APPROVAL_INTENT = IntentRoute(
+    intent="approval",
+    handler=compliance_audit_approval_question_answer,
+    predicate=is_compliance_audit_question,
+    min_confidence=0.52,
+)
+
+AUDIT_LOG_INTENT = IntentRoute(
+    intent="audit_log",
+    handler=compliance_audit_approval_question_answer,
+    predicate=is_compliance_audit_question,
+    min_confidence=0.52,
+)
+
+INTENT_REGISTRY: dict[str, IntentRoute] = {
+    route.intent: route
+    for route in (
+        DRUG_RESEARCH_INTENT,
+        BATCH_RECORD_INTENT,
+        BATCH_CREATION_INTENT,
+        BATCH_UPDATE_INTENT,
+        BATCH_DELETE_INTENT,
+        BATCH_HOLD_INTENT,
+        BATCH_CLOSE_INTENT,
+        WORK_ORDER_CREATE_INTENT,
+        WORK_ORDER_QUERY_INTENT,
+        WORK_ORDER_UPDATE_INTENT,
+        WORK_ORDER_DELETE_INTENT,
+        WORK_ORDER_STATUS_INTENT,
+        WORK_ORDER_HOLD_INTENT,
+        DECISION_INTELLIGENCE_INTENT,
+        APPROVAL_CREATE_INTENT,
+        APPROVAL_QUERY_INTENT,
+        APPROVAL_DECISION_INTENT,
+        APPROVAL_HOLD_INTENT,
+        CHANGE_CONTROL_INTENT,
+        CHANGE_CONTROL_CREATE_INTENT,
+        CHANGE_CONTROL_QUERY_INTENT,
+        CHANGE_CONTROL_UPDATE_INTENT,
+        CHANGE_CONTROL_DECISION_INTENT,
+        PRODUCTION_KPI_INTENT,
+        WAREHOUSE_SHIPPING_INTENT,
+        WORKER_INTENT,
+        AUDIT_LOG_INTENT,
+    )
+}
+
+
+def get_intent_definition(
+    intent_name: str | None,
+) -> IntentDefinition:
+    return INTENT_DEFINITIONS.get(
+        intent_name or "",
+        IntentDefinition(intent="generic"),
+    )