mirror of
https://github.com/typesense/typesense.git
synced 2025-05-15 18:53:29 +08:00
d126f91aa1
Some checks failed
tests / test (push) Has been cancelled
* Refactor `get_history_collection` by overloading * Return error when model doesn't have history collection * Refactoring * Remove unnecessary function * Refactor add_conversation * Make conversation model id necessary for deleting conversation
57 lines
2.6 KiB
C++
57 lines
2.6 KiB
C++
#pragma once
|
|
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <unordered_set>
|
|
#include <shared_mutex>
|
|
#include <mutex>
|
|
#include <json.hpp>
|
|
#include "option.h"
|
|
#include "store.h"
|
|
#include "sole.hpp"
|
|
#include "collection_manager.h"
|
|
#include "conversation_model_manager.h"
|
|
|
|
class ConversationManager {
|
|
public:
|
|
ConversationManager(const ConversationManager&) = delete;
|
|
ConversationManager(ConversationManager&&) = delete;
|
|
ConversationManager& operator=(const ConversationManager&) = delete;
|
|
ConversationManager& operator=(ConversationManager&&) = delete;
|
|
static ConversationManager& get_instance() {
|
|
static ConversationManager instance;
|
|
return instance;
|
|
}
|
|
Option<std::string> add_conversation(const nlohmann::json& conversation, const nlohmann::json& model, const std::string& id = "");
|
|
Option<nlohmann::json> get_conversation(const std::string& conversation_id, const nlohmann::json& model);
|
|
Option<nlohmann::json> get_full_conversation(const std::string& question, const std::string& answer, const nlohmann::json& model, const std::string& conversation_id);
|
|
static Option<nlohmann::json> get_last_n_messages(const nlohmann::json& conversation, size_t n);
|
|
static Option<nlohmann::json> truncate_conversation(nlohmann::json conversation, size_t limit);
|
|
Option<nlohmann::json> delete_conversation(const std::string& conversation_id, const std::string& model_id);
|
|
Option<bool> check_conversation_exists(const std::string& conversation_id, Collection* collection);
|
|
Option<std::unordered_set<std::string>> get_conversation_ids();
|
|
static constexpr size_t MAX_TOKENS = 3000;
|
|
Option<bool> init(ReplicationState* raft_server);
|
|
void clear_expired_conversations();
|
|
void run();
|
|
void stop();
|
|
void _set_ttl_offset(size_t offset) {
|
|
TTL_OFFSET = offset;
|
|
}
|
|
|
|
Option<bool> validate_conversation_store_schema(Collection* collection);
|
|
Option<bool> validate_conversation_store_collection(const std::string& collection);
|
|
// get history collection by conversation model
|
|
Option<Collection*> get_history_collection(const nlohmann::json& model);
|
|
private:
|
|
ConversationManager() {}
|
|
std::mutex conversations_mutex;
|
|
|
|
ReplicationState* raft_server;
|
|
size_t TTL_OFFSET = 0;
|
|
|
|
std::atomic<bool> quit = false;
|
|
std::condition_variable cv;
|
|
|
|
Option<nlohmann::json> delete_conversation_unsafe(const std::string& conversation_id, const std::string& model_id);
|
|
}; |