#include "sentence_transformers.h"
#include <iostream>
#include <vector>
#include <string>

int main() {
    try {
        // 指定模型路径
        std::string model_path = "/path/to/bge-small-zh-v1.5";

        // 初始化 SentenceTransformer 模型
        SentenceTransformer model(model_path);

        // 中文文本示例
        std::vector<std::string> texts = {
            "This is a test sentence.", "Another example sentence.",
        };

        // 批量编码文本获取嵌入向量
        std::vector<std::vector<float>> embeddings = model.encode_batch(texts);

        // 打印嵌入向量信息
        for (size_t i = 0; i < embeddings.size(); ++i) {
            std::cout << "文本: " << texts[i] << std::endl;
            std::cout << "嵌入向量维度: " << embeddings[i].size() << std::endl;
            std::cout << "嵌入向量前5个值: ";
            for (size_t j = 0; j < 5 && j < embeddings[i].size(); ++j) {
                std::cout << embeddings[i][j] << " ";
            }
            std::cout << "...\n\n";
        }
    } catch (const std::exception& e) {
        // 错误处理
        std::cerr << "发生错误: " << e.what() << std::endl;
        return 1;
    }

    return 0;
}