bootstrap

2026-07-12 19:46:07 +03:00
parent 2ddf2251d6
commit 940d25435e
36 changed files with 1914 additions and 0 deletions
@@ -0,0 +1,56 @@
+import json
+
+from ai import prompts
+from cashflow_model import FinancialModel
+from engine.forecast import ForecastService
+
+
+class AssistantError(Exception):
+    pass
+
+
+class AssistantService:
+    def __init__(self, model: FinancialModel):
+        self.model = model
+
+    def analyze(self, months: int = 12) -> dict:
+        forecast_service = ForecastService(self.model)
+        forecast_result = forecast_service.forecast_cashflow(months)
+        summary = forecast_service.summary(months)
+
+        prompt = prompts.format_context(
+            model_json=json.dumps(self.model.to_dict(), indent=2, ensure_ascii=False),
+            forecast_json=json.dumps(forecast_result, indent=2, ensure_ascii=False),
+            months=months,
+        )
+
+        return {
+            "prompt": prompt,
+            "summary": summary,
+            "forecast": forecast_result,
+            "ai_response": None,
+        }
+
+    def advice(self, question: str, months: int = 12) -> dict:
+        forecast_service = ForecastService(self.model)
+        forecast_result = forecast_service.forecast_cashflow(months)
+
+        prompt = prompts.ADVICE_PROMPT.format(
+            model_json=json.dumps(self.model.to_dict(), indent=2, ensure_ascii=False),
+            forecast_json=json.dumps(forecast_result, indent=2, ensure_ascii=False),
+            question=question,
+        )
+
+        return {
+            "prompt": prompt,
+            "ai_response": None,
+        }
+
+    def compare_scenarios(self, scenarios_json: str) -> dict:
+        prompt = prompts.SCENARIO_COMPARISON_PROMPT.format(
+            scenarios_json=scenarios_json,
+        )
+        return {
+            "prompt": prompt,
+            "ai_response": None,
+        }