#include <Config.h>

Public Member Functions
	ModelConfig ()

void	load_from_file (const std::string &config_path)

void	load_from_json (const Json::Value &root)

bool	is_valid () const

std::string	to_string () const

Data Fields
int	n_layers

int	hidden_size

int	num_attention_heads

int	vocab_size

int	max_position_embeddings

std::optional< int >	bos_token_id

std::optional< int >	eos_token_id

Detailed Description

Definition at line 36 of file Config.h.

Constructor & Destructor Documentation

◆ ModelConfig()

ggma::ModelConfig::ModelConfig ( )

Definition at line 58 of file Config.cc.

58{}

Member Function Documentation

◆ is_valid()

bool ggma::ModelConfig::is_valid ( ) const

Definition at line 100 of file Config.cc.

{
  // Check required fields are positive
  if (n_layers <= 0)
    return false;
  if (hidden_size <= 0)
    return false;
  if (num_attention_heads <= 0)
    return false;
  if (vocab_size <= 0)
    return false;
  if (max_position_embeddings <= 0)
    return false;
 
  // Check token IDs are non-negative (only if they have values)
  if (bos_token_id.has_value() && bos_token_id.value() < 0)
    return false;
  if (eos_token_id.has_value() && eos_token_id.value() < 0)
    return false;
 
  return true;
}

References bos_token_id, eos_token_id, hidden_size, max_position_embeddings, n_layers, num_attention_heads, and vocab_size.

◆ load_from_file()

void ggma::ModelConfig::load_from_file ( const std::string & config_path )

Definition at line 61 of file Config.cc.

{
  std::ifstream config_file(config_path);
 
  if (!config_file.is_open())
    throw std::runtime_error(
      "config.json not found in package. This file is required for ggma_context.");
 
  try
  {
    Json::Value root;
    Json::Reader reader;
 
    if (!reader.parse(config_file, root, false))
      throw std::runtime_error("Failed to parse JSON: " + reader.getFormattedErrorMessages());
 
    load_from_json(root);
  }
  catch (const std::exception &e)
  {
    // Re-throw the exception to ensure session creation fails
    throw std::runtime_error("Failed to load config.json: " + std::string(e.what()));
  }
}

References load_from_json().

◆ load_from_json()

void ggma::ModelConfig::load_from_json ( const Json::Value & root )

Definition at line 87 of file Config.cc.

{
  // Load model configuration from Hugging Face config.json
  load_config_field(root, "num_hidden_layers", n_layers);
  load_config_field(root, "hidden_size", hidden_size);
  load_config_field(root, "num_attention_heads", num_attention_heads);
  load_config_field(root, "vocab_size", vocab_size);
  load_config_field(root, "max_position_embeddings", max_position_embeddings);
  load_config_field(root, "bos_token_id", bos_token_id);
  load_config_field(root, "eos_token_id", eos_token_id);
}

References bos_token_id, eos_token_id, hidden_size, ggma::load_config_field(), max_position_embeddings, n_layers, num_attention_heads, and vocab_size.

Referenced by load_from_file().

◆ to_string()

std::string ggma::ModelConfig::to_string ( ) const

Definition at line 124 of file Config.cc.

{
  std::ostringstream oss;
  oss << "ModelConfig {\n";
  oss << "  n_layers: " << n_layers << "\n";
  oss << "  hidden_size: " << hidden_size << "\n";
  oss << "  num_attention_heads: " << num_attention_heads << "\n";
  oss << "  vocab_size: " << vocab_size << "\n";
  oss << "  max_position_embeddings: " << max_position_embeddings << "\n";
  oss << "  bos_token_id: "
      << (bos_token_id.has_value() ? std::to_string(bos_token_id.value()) : "undefined") << "\n";
  oss << "  eos_token_id: "
      << (eos_token_id.has_value() ? std::to_string(eos_token_id.value()) : "undefined") << "\n";
  oss << "}";
  return oss.str();
}