91 lines
2.7 KiB
C++
91 lines
2.7 KiB
C++
#ifndef HUMANUS_MEMORY_VECTOR_STORE_BASE_H
|
|
#define HUMANUS_MEMORY_VECTOR_STORE_BASE_H
|
|
|
|
#include "config.h"
|
|
#include <memory>
|
|
#include <unordered_map>
|
|
#include <string>
|
|
|
|
namespace humanus {
|
|
|
|
class VectorStore {
|
|
private:
|
|
static std::unordered_map<std::string, std::shared_ptr<VectorStore>> instances_;
|
|
|
|
protected:
|
|
std::shared_ptr<VectorStoreConfig> config_;
|
|
|
|
// Constructor
|
|
VectorStore(const std::shared_ptr<VectorStoreConfig>& config) : config_(config) {}
|
|
|
|
public:
|
|
// Get the singleton instance
|
|
static std::shared_ptr<VectorStore> get_instance(const std::string& config_name = "default", const std::shared_ptr<VectorStoreConfig>& config = nullptr);
|
|
|
|
virtual ~VectorStore() = default;
|
|
|
|
virtual void reset() = 0;
|
|
|
|
/**
|
|
* @brief Insert a vector with metadata
|
|
* @param vector vector data
|
|
* @param vector_id vector ID
|
|
* @param metadata metadata
|
|
*/
|
|
virtual void insert(const std::vector<float>& vector,
|
|
const size_t vector_id,
|
|
const MemoryItem& metadata = MemoryItem()) = 0;
|
|
|
|
/**
|
|
* @brief Search similar vectors
|
|
* @param query query vector
|
|
* @param limit limit of returned results
|
|
* @param filter optional filter (returns true for allowed vectors)
|
|
* @return list of similar vectors
|
|
*/
|
|
virtual std::vector<MemoryItem> search(const std::vector<float>& query,
|
|
size_t limit = 5,
|
|
const FilterFunc& filter = nullptr) = 0;
|
|
|
|
/**
|
|
* @brief Remove a vector by ID
|
|
* @param vector_id vector ID
|
|
*/
|
|
virtual void remove(size_t vector_id) = 0;
|
|
|
|
/**
|
|
* @brief Update a vector and its metadata
|
|
* @param vector_id vector ID
|
|
* @param vector new vector data
|
|
* @param metadata new metadata
|
|
*/
|
|
virtual void update(size_t vector_id, const std::vector<float>& vector = std::vector<float>(), const MemoryItem& metadata = MemoryItem()) = 0;
|
|
|
|
/**
|
|
* @brief Get a vector by ID
|
|
* @param vector_id vector ID
|
|
* @return vector data
|
|
*/
|
|
virtual MemoryItem get(size_t vector_id) = 0;
|
|
|
|
/**
|
|
* @brief Set metadata for a vector
|
|
* @param vector_id vector ID
|
|
* @param metadata new metadata
|
|
*/
|
|
virtual void set(size_t vector_id, const MemoryItem& metadata) = 0;
|
|
|
|
/**
|
|
* @brief List all memories
|
|
* @param limit optional limit of returned results
|
|
* @param filter optional filter (returns true for allowed memories)
|
|
* @return list of memories
|
|
*/
|
|
virtual std::vector<MemoryItem> list(size_t limit = 0, const FilterFunc& filter = nullptr) = 0;
|
|
};
|
|
|
|
} // namespace humanus
|
|
|
|
|
|
#endif // HUMANUS_MEMORY_VECTOR_STORE_BASE_H
|