Function run_step

tools/main.py:226–409

(cfg, state, info, plots, memory, provider, diverse_query, strategy_agents, exp_path, trading_records, mode)

    return trading_records

def run_step(cfg, state, info, plots, memory, provider, diverse_query, strategy_agents, exp_path, trading_records, mode):

    params = dict()
    save_dir = "train" if mode == "train" else "valid"

    # plot kline
    kline_path = plots.plot_kline(state=state, info=info, save_dir=save_dir)
    params.update({
        "kline_path": kline_path
    })

    # tools
    tools_params = prepared_tools_params(state=state,
                                         info=info,
                                         params=params,
                                         memory=memory,
                                         provider=provider,
                                         diverse_query=diverse_query,
                                         strategy_agents=strategy_agents,
                                         cfg=cfg,
                                         mode=mode)
    params.update(tools_params)

    # latest market intelligence
    latest_market_intelligence_summary_template = read_resource_file(cfg.train_latest_market_intelligence_summary_template_path
                                                   if mode == "train" else cfg.valid_latest_market_intelligence_summary_template_path)
    latest_market_intelligence_summary = PROMPT.build(cfg.latest_market_intelligence_summary)
    latest_market_intelligence_summary_res = latest_market_intelligence_summary.run(state = state,
                                                                      info = info,
                                                                      params = params,
                                                                      template = latest_market_intelligence_summary_template,
                                                                      memory = memory,
                                                                      provider = provider,
                                                                      diverse_query = diverse_query,
                                                                      exp_path = exp_path,
                                                                      save_dir = save_dir)

    # query past market intelligence
    prepared_latest_market_intelligence_params = prepare_latest_market_intelligence_params(state=state,
                                                                                        info=info,
                                                                                        params=params,
                                                                                        memory=memory,
                                                                                        provider=provider,
                                                                                        diverse_query=diverse_query)
    params.update(prepared_latest_market_intelligence_params)

    # add latest market intelligence to memory
    latest_market_intelligence_summary.add_to_memory(state=state,
                                                     info=info,
                                                     res=latest_market_intelligence_summary_res,
                                                     memory=memory,
                                                     provider=provider)

    # past market intelligence
    past_market_intelligence_summary_template = read_resource_file(cfg.train_past_market_intelligence_summary_template_path
                                                   if mode == "train" else cfg.valid_past_market_intelligence_summary_template_path)
    past_market_intelligence_summary = PROMPT.build(cfg.past_market_intelligence_summary)
    past_market_intelligence_summary_res = past_market_intelligence_summary.run(state=state,