#[cfg(feature = "dynamo")]
use crate::db_connectors::{dynamodb as dynamodb_connector, is_dynamodb};
#[cfg(feature = "mongo")]
use crate::db_connectors::{is_mongodb, mongodb as mongodb_connector};
use crate::error_messages::ERROR_DB_SETUP;
use crate::{Client, ConversationInfo, Database, EngineError, Memory};
pub fn add_memories(
data: &mut ConversationInfo,
memories: &[Memory],
interaction_order: i32,
) -> Result<(), EngineError> {
#[cfg(feature = "mongo")]
if is_mongodb() {
return mongodb_connector::memories::add_memories(data, &memories, interaction_order);
}
#[cfg(feature = "dynamo")]
if is_dynamodb() {
return dynamodb_connector::memories::add_memories(data, &memories, interaction_order);
}
Err(EngineError::Manager(ERROR_DB_SETUP.to_owned()))
}
pub fn get_memories(client: &Client, db: &mut Database) -> Result<serde_json::Value, EngineError> {
#[cfg(feature = "mongo")]
if is_mongodb() {
let db = mongodb_connector::get_db(db)?;
return mongodb_connector::memories::get_memories(client, db);
}
#[cfg(feature = "dynamo")]
if is_dynamodb() {
let db = dynamodb_connector::get_db(db)?;
return dynamodb_connector::memories::get_memories(client, db);
}
Err(EngineError::Manager(ERROR_DB_SETUP.to_owned()))
}