| #!/usr/bin/env python3 | |
| """Listwise short-list rerank + pairwise noul. Passages are invented.""" | |
| from __future__ import annotations | |
| import os | |
| os.environ.setdefault("USE_TF", "0") | |
| import laya | |
| # pip install laya==0.3.4 | |
| # Local override: LAYA_AR_MODEL=/path/to/artifacts/laya-ara-rag | |
| MODEL = os.environ.get("LAYA_AR_MODEL", "Wouze/laya-ara-rag") | |
| STATE = {"query": "ما حكم الوضوء قبل قراءة القرآن؟"} | |
| QUESTIONS = { | |
| "passage": { | |
| "type": "choice", | |
| "instructions": "Which passage is most relevant to the query?", | |
| "criteria": { | |
| "a": "الوضوء شرط للصلاة. أما قراءة القرآن من غير مس المصحف فالجمهور على الجواز.", | |
| "b": "زكاة الفطر تجب على كل مسلم قبل صلاة العيد صاعا من طعام.", | |
| "c": "صيام يوم عاشوراء سنة مؤكدة ويكفر سنة قبله.", | |
| }, | |
| }, | |
| "relevant_a": { | |
| "type": "noul", | |
| "instructions": "Is passage A relevant to the query about wudu before Quran recitation?", | |
| }, | |
| } | |
| def main() -> None: | |
| agent = laya.load(MODEL, token=os.environ.get("HF_TOKEN")) | |
| print(agent.predict(STATE, QUESTIONS)) | |
| if __name__ == "__main__": | |
| main() | |