Represents the result of a tool execution.

A DataProto is a data structure that aims to provide a standard protocol for data exchange between functions. It contains a batch (TensorDict

r"""Arguments pertaining to what data we are going to input our model for training and evaluation.

数据加载器类，用于加载和预处理金融领域长思维链压缩赛题数据。

Represents a single search result item

r"""Dataset attributes.

A collection of defined tools.

r"""General class for chat models. Backed by huggingface or vllm engines. Supports both sync and async methods. Sync methods: chat(), stream_

Data structure for the output of reward calculations. Attributes: reward (float): The computed reward value based on the evaluation of th

Sharding manager to support data resharding when using FSDP + Ulysses

A class for evaluating the accuracy of predictions against ground truth in various formats.

r"""Arguments pertaining to which techniques we are going to fine-tuning with.

r"""A callback for saving the processor.

A ToolResult that represents a failure.

Used to count mfu during training loop Example: flops_counter = FlopsCounter(config) flops_achieved, flops_promised = flops_

Dataset class for mathematical benchmark problems.

r"""Arguments pertaining to which model/config/tokenizer we are going to fine-tune or infer. The class on the most right will be displayed first.

r"""Data collator that supports VLMs. Features should contain input_ids, attention_mask, labels, and optionally contain images, videos and audios

We assume the dataset contains a column that contains prompts and other information

Represents a single search result item

A checkpoint manager that saves and loads - model - optimizer - lr_scheduler - extra_states in a SPMD way. We save

Define a resource pool specification. Resource pool will be initialized first. Mapping

This is an in-memory SFTDataset

Tool for retrieving market sentiment data including hot sectors and index capital flow.

Response model for TTS operation

Search the web for information using various search engines.

Dataset class for AFAC data.

Dataset class for GSM8K math problems.

Dataset class for GSM8K zero-shot learning.

Tool for retrieving hot money and market data for stocks.

Note that this trainer runs on the driver process on a single CPU/GPU node.

一个为Agent设计的工具，负责根据规则的核心概念， 从一个指定的Markdown知识库文件中查找并返回该规则的详细释义。

搜索引擎封装类

Tool for retrieving technical data for stocks.

Exception raised when the token limit is exceeded

Search the web for information using various search engines.

K-nearest-neighbors retrieval.

r"""Compute and store the average and current value.

Tool for agents to interact in the battle environment.

r"""Inherits Seq2SeqTrainer to compute generative metrics such as BLEU and ROUGE.

DataProtoFuture aims to eliminate actual data fetching on driver. By doing so, the driver doesn't have to wait for data so that asynchronous

The implementation of a simple digit completion task. The prompt is a sequence of numbers with fixed difference. The task is to complete the

r"""A dummy optimizer used for the GaLore or APOLLO algorithm.

r"""A general engine to control the behaviors of Web UI.

评估器类，用于评估模型推理结果。

基金数据集类，用于处理时间序列数据并提供给模型训练

MCP Server implementation with tool registration and management.

Configuration for MCP (Model Context Protocol)

Configuration for MCP (Model Context Protocol)

A checkpoint manager that saves and loads - model - optimizer - lr_scheduler - extra_states in a SPMD way. We save

A memory buffer is a contiguous torch tensor that may combine multiple tensors sharing with the underlying memory. It must have a unique type

r"""Data collator for pairwise data.

Data structure for input required to calculate rewards. Attributes: problem (str): The original problem text or prompt provided to the mo

Reward function for evaluating mathematical answers. This class implements the __call__ method to process the input and determine the re

Tool for retrieving risk control data for stocks.

Structured response from the web search tool, inheriting ToolResult.

Structured response from the web search tool, inheriting ToolResult.

Represents a single search result returned by a search engine.

Represents a single search result returned by a search engine.

属性: seq_len (int): 输入序列长度 pred_len (int): 预测序列长度 output_attention (bool): 是否输出注意力权重 use_norm (bool): 是否使用归一化

三维数据标准化器 (num_samples, seq_len, num_features) 在时间维度(seq_len)上独立进行标准化

Tool to fetch basic information about a stock with the current trading date.