| """Example: load the model in this folder and predict. | |
| Run: python example.py | |
| The flow needs the decision-model package (see the How to use section of | |
| README.md). | |
| """ | |
| import json | |
| from pathlib import Path | |
| from decision_model.model import DecisionModel | |
| HERE = Path(__file__).resolve().parent | |
| REQUEST = { | |
| "state": "Refactor the auth module and add tests. The repo has 40 files.", | |
| "preferences": {"cost": "low cost before speed"}, | |
| "questions": [ | |
| {"id": "model", "type": "choice", "instructions": "Which model must handle this request?", | |
| "criteria": {"haiku": "Short and simple tasks", "sonnet": "Normal coding tasks", "opus": "Long or difficult tasks"}}, | |
| ], | |
| } | |
| def main(): | |
| model = DecisionModel.load(HERE) | |
| result = model.predict(REQUEST) | |
| print(json.dumps(result, indent=2)) | |
| if __name__ == "__main__": | |
| main() | |