AshenH commited on
Commit
c02152a
·
verified ·
1 Parent(s): 852fd6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -9,6 +9,7 @@ from tools.explain_tool import ExplainTool
9
  from tools.report_tool import ReportTool
10
  from utils.tracing import Tracer
11
  from utils.config import AppConfig
 
12
 
13
  # Optional tiny CPU LLM for planning (can be disabled by not setting ORCHESTRATOR_MODEL)
14
  llm = None
@@ -30,6 +31,12 @@ predict_tool = PredictTool(cfg, tracer)
30
  explain_tool = ExplainTool(cfg, tracer)
31
  report_tool = ReportTool(cfg, tracer)
32
 
 
 
 
 
 
 
33
  SYSTEM_PROMPT = (
34
  "You are an analytical assistant for tabular data. "
35
  "Decide which tools to call in order: "
 
9
  from tools.report_tool import ReportTool
10
  from utils.tracing import Tracer
11
  from utils.config import AppConfig
12
+ from tools.ts_forecast_tool import TimeseriesForecastTool
13
 
14
  # Optional tiny CPU LLM for planning (can be disabled by not setting ORCHESTRATOR_MODEL)
15
  llm = None
 
31
  explain_tool = ExplainTool(cfg, tracer)
32
  report_tool = ReportTool(cfg, tracer)
33
 
34
+ ts_tool = TimeseriesForecastTool(cfg, tracer,
35
+ context_length=512, forecast_length=96,
36
+ target_cols=["portfolio_value"], # set after preprocessing
37
+ control_cols=["rate_deposit", "rate_asset"] # optional exogenous
38
+ )
39
+
40
  SYSTEM_PROMPT = (
41
  "You are an analytical assistant for tabular data. "
42
  "Decide which tools to call in order: "