| from .base_agent import BaseAgent |
| from rdflib import Graph, Namespace |
| from typing import List, Dict |
| import logging |
|
|
| class ConflictAgent(BaseAgent): |
| def __init__(self, rdf_graph: Graph, namespace: Namespace): |
| super().__init__(rdf_graph, namespace) |
| self.query_template = """ |
| PREFIX ns: <http://www.example.org/DrugInteraction.owl#> |
| SELECT DISTINCT ?drug1 ?drug2 |
| WHERE {{ |
| ?drug1 ns:hasConflict ?drug2 . |
| FILTER(STRENDS(str(?drug1), "#{drug}") || STRENDS(str(?drug2), "#{drug}")) |
| }} |
| """ |
| logging.debug("ConflictAgent initialized.") |
| |
| def get_conflicts(self, drug_name: str) -> List[Dict[str, str]]: |
| query = self.query_template.format(drug=drug_name) |
| logging.info(f"Fetching conflicts for {drug_name}") |
| return self.query_ontology(query) |
|
|
