types.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_RAG_TYPES_HPP
#define FORESTHUB_RAG_TYPES_HPP

 
#include <string>
#include <vector>
 
namespace foresthub {
namespace rag {

struct QueryRequest {
    std::string collection_id;  
    std::string query;          
    int top_k = 5;              
};

struct QueryResult {
    std::string chunk_id;     
    std::string document_id;  
    std::string content;      
    double score = 0.0;       
};

struct QueryResponse {
    std::vector<QueryResult> results;  
};

inline std::string FormatContext(const std::vector<QueryResult>& results) {
    if (results.empty()) {
        return "";
    }
    std::string out = "<context>\n";
    for (size_t i = 0; i < results.size(); ++i) {
        out += "<source id=\"" + std::to_string(i + 1) + "\">" + results[i].content + "</source>\n";
    }
    out += "</context>";
    return out;
}
 
}  // namespace rag
}  // namespace foresthub
 
#endif  // FORESTHUB_RAG_TYPES_HPP