testing new features

ai/prompts.py

@@ -1,46 +1,21 @@
-ANALYZE_PROMPT = """
-Ты — финансовый AI-ассистент. Проанализируй финансовую модель пользователя.
+from cli.i18n import t
 
-### Модель (JSON):
-{model_json}
-
-### Прогноз на {months} месяцев:
-{forecast_json}
-
-Дай анализ по пунктам:
-1. Общее финансовое состояние
-2. Тренд денежного потока (рост/падение)
-3. Достаточность ликвидности
-4. Рекомендации по улучшению
-"""
-
-ADVICE_PROMPT = """
-Ты — финансовый AI-ассистент. Дай персональные рекомендации.
-
-### Модель:
-{model_json}
-
-### Прогноз:
-{forecast_json}
-
-Вопрос пользователя: {question}
-
-Ответь как опытный финансовый консультант.
-"""
-
-SCENARIO_COMPARISON_PROMPT = """
-Ты — финансовый AI-ассистент. Сравни сценарии прогноза.
-
-### Результаты сценариев:
-{scenarios_json}
-
-Дай рекомендацию: какой сценарий наиболее вероятен и почему.
-"""
+ANALYZE_PROMPT = t("prompt.analyze")
+ADVICE_PROMPT = t("prompt.advice")
+SCENARIO_COMPARISON_PROMPT = t("prompt.scenario_comparison")
 
 
-def format_context(model_json: str, forecast_json: str, months: int = 12) -> str:
+def format_context(
+    model_json: str,
+    forecast_json: str,
+    months: int = 12,
+    base_currency: str = "RUB",
+    display_currency: str = "RUB",
+) -> str:
     return ANALYZE_PROMPT.format(
         model_json=model_json,
         forecast_json=forecast_json,
         months=months,
+        base_currency=base_currency,
+        display_currency=display_currency,
     )