client.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_CLIENT_HPP
#define FORESTHUB_CLIENT_HPP

 
#include <map>
#include <memory>
#include <string>
 
#include "foresthub/config/config.hpp"
#include "foresthub/core/http_client.hpp"
#include "foresthub/core/model.hpp"
#include "foresthub/core/provider.hpp"
#include "foresthub/core/types.hpp"
 
namespace foresthub {

class Client : public core::LLMClient {
public:
    Client();

    static std::unique_ptr<Client> Create(const config::ClientConfig& cfg,
                                          const std::shared_ptr<core::HttpClient>& http_client = nullptr);

    void RegisterProvider(const std::shared_ptr<core::Provider>& provider);

    std::string Health() const;

    bool SupportsModel(const core::ModelID& model) const;

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

    const std::map<core::ProviderID, std::shared_ptr<core::Provider>>& providers() const { return providers_; }
 
private:
    std::map<core::ProviderID, std::shared_ptr<core::Provider>> providers_;

    std::shared_ptr<core::Provider> InferProvider(const core::ModelID& model) const;
};
 
}  // namespace foresthub
 
#endif  // FORESTHUB_CLIENT_HPP