LLMConfig.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstUtil/JsonValue.hpp"
#include <string>
 
AST_NAMESPACE_BEGIN
 
class JsonValue;
 
class AST_AI_API LLMConfig
{
public:
    LLMConfig() = default;
    ~LLMConfig() = default;
 
    void setModel(const std::string& model) { model_ = model; }
    const std::string& model() const { return model_; }
 
    void setTemperature(float temperature) { temperature_ = temperature; }
    float temperature() const { return temperature_; }
 
    void setExtraBody(const JsonValue& extra) { extraBody_ = extra; }
    const JsonValue& extraBody() const { return extraBody_; }
 
private:
    std::string model_ = "deepseek-v4-flash";
    float temperature_ = 0.2f;
    JsonValue extraBody_;  
};
 
AST_NAMESPACE_END