gemini.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_PROVIDER_REMOTE_GEMINI_HPP
#define FORESTHUB_PROVIDER_REMOTE_GEMINI_HPP

 
#include <memory>
#include <string>
#include <vector>
 
#include "foresthub/config/config.hpp"
#include "foresthub/core/http_client.hpp"
#include "foresthub/core/provider.hpp"
 
namespace foresthub {
namespace provider {
namespace remote {

class GeminiProvider : public core::Provider {
public:
    GeminiProvider(const config::ProviderConfig& cfg, std::shared_ptr<core::HttpClient> http_client);

    core::ProviderID ProviderId() const override;

    std::string Health() const override;

    std::shared_ptr<core::ChatResponse> Chat(const core::ChatRequest& req) override;

    bool SupportsModel(const core::ModelID& model) const override;
 
private:
    std::shared_ptr<core::HttpClient> http_;
    std::string api_key_;
    std::string base_url_;
    std::vector<core::ModelID> supported_models_;
    core::HttpClient::Headers cached_headers_;
};
 
}  // namespace remote
}  // namespace provider
}  // namespace foresthub
 
#endif  // FORESTHUB_PROVIDER_REMOTE_GEMINI_HPP