webui/base.py

from dataclasses import dataclass
from typing import Any


@dataclass
class GraphGenParams:
    """
    GraphGen parameters
    """

    if_trainee_model: bool
    input_file: str
    tokenizer: str
    output_data_type: str
    output_data_format: str
    bidirectional: bool
    expand_method: str
    max_extra_edges: int
    max_tokens: int
    max_depth: int
    edge_sampling: str
    isolated_node_strategy: str
    loss_strategy: str
    synthesizer_url: str
    synthesizer_model: str
    trainee_model: str
    api_key: str
    chunk_size: int
    rpm: int
    tpm: int
    quiz_samples: int
    trainee_url: str
    trainee_api_key: str
    token_counter: Any