33 lines
956 B
C
33 lines
956 B
C
|
#ifndef HUMANUS_MEMORY_MEM0_EMBEDDING_MODEL_BASE_H
|
||
|
|
#define HUMANUS_MEMORY_MEM0_EMBEDDING_MODEL_BASE_H
|
||
|
|
|
||
|
|
#include "httplib.h"
|
||
|
|
#include "logger.h"
|
||
|
|
#include <vector>
|
||
|
|
#include <unordered_map>
|
||
|
|
#include <memory>
|
||
|
|
|
||
|
|
namespace humanus::mem0 {
|
||
|
|
|
||
|
|
class EmbeddingModel {
|
||
|
|
private:
|
||
|
|
static std::unordered_map<std::string, std::shared_ptr<EmbeddingModel>> instances_;
|
||
|
|
|
||
|
|
protected:
|
||
|
|
std::shared_ptr<EmbeddingModelConfig> config_;
|
||
|
|
|
||
|
|
// Constructor
|
||
|
|
EmbeddingModel(const std::shared_ptr<EmbeddingModelConfig>& config) : config_(config) {}
|
||
|
|
|
||
|
|
public:
|
||
|
|
// Get the singleton instance
|
||
|
|
static std::shared_ptr<EmbeddingModel> get_instance(const std::string& config_name = "default", const std::shared_ptr<EmbeddingModelConfig>& config = nullptr);
|
||
|
|
|
||
|
|
virtual ~EmbeddingModel() = default;
|
||
|
|
|
||
|
|
virtual std::vector<float> embed(const std::string& text, EmbeddingType type) = 0;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace humanus::mem0
|
||
|
|
|
||
|
|
#endif // HUMANUS_MEMORY_MEM0_EMBEDDING_MODEL_BASE_H
|