options.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: AGPL-3.0-only
// Copyright (c) 2026 ForestHub. All rights reserved.
// For commercial licensing, visit https://github.com/ForestHubAI/fh-sdk
 
#ifndef FORESTHUB_CORE_OPTIONS_HPP
#define FORESTHUB_CORE_OPTIONS_HPP

 
#include "foresthub/util/optional.hpp"
 
namespace foresthub {
namespace core {

struct Options {
    foresthub::Optional<int> max_tokens;           
    foresthub::Optional<float> temperature;        
    foresthub::Optional<int> top_k;                
    foresthub::Optional<float> top_p;              
    foresthub::Optional<float> frequency_penalty;  
    foresthub::Optional<float> presence_penalty;   
    foresthub::Optional<int> seed;                 

    Options& WithMaxTokens(int max_tokens) {
        this->max_tokens = max_tokens;
        return *this;
    }

    Options& WithTemperature(float temperature) {
        this->temperature = temperature;
        return *this;
    }

    Options& WithTopK(int top_k) {
        this->top_k = top_k;
        return *this;
    }

    Options& WithTopP(float top_p) {
        this->top_p = top_p;
        return *this;
    }

    Options& WithFrequencyPenalty(float frequency_penalty) {
        this->frequency_penalty = frequency_penalty;
        return *this;
    }

    Options& WithPresencePenalty(float presence_penalty) {
        this->presence_penalty = presence_penalty;
        return *this;
    }

    Options& WithSeed(int seed) {
        this->seed = seed;
        return *this;
    }
};
 
}  // namespace core
}  // namespace foresthub
 
#endif  // FORESTHUB_CORE_OPTIONS_HPP