zhipuzi_pos_windows/zhipuzi_pos_windows/ai/SQLiteVecManager.h


#pragma once
#include <vector>
#include <string>
#include "../sqlite3/sqlite3.h"
#include "../sqlite3/sqlite-vec.h"

class SQLiteVecManager
{
private:
	sqlite3 * db;
	std::string dbName;
	bool useVecExtension;

public:
	SQLiteVecManager(const std::string & databaseName);
	~SQLiteVecManager();

	bool initializeDatabase(int vectorDimension);
	bool addFeatureVector(const std::vector<float> & features, const std::string & imagePath);
	std::vector<std::pair<std::string, float>> searchSimilarVectors(const std::vector<float> & queryVector, int k = 5);
	void saveDatabase();
	bool loadDatabase();
	int getFeatureCount() const;
	bool isEmpty() const;

private:
	std::string vectorToBlob(const std::vector<float> & vec);
	std::vector<float> blobToVector(const std::string & blob);
	float calculateCosineSimilarity(const std::vector<float> & vec1, const std::vector<float> & vec2);
	float distanceToSimilarity(float distance);
};