Rodrigo Rodriguez (Pragmatismo)
cad94939c3
- Added AWS SDK S3 dependencies including aws-config, aws-sdk-s3, and related crates - Removed opendal dependency and replaced with AWS SDK S3 client - Implemented new get_file_content helper function using AWS SDK - Updated MinIOHandler to use AWS SDK client instead of opendal Operator - Modified file change detection to work with AWS SDK's S3 client The change was made to standardize on AWS's official SDK for S3 operations, which provides better maintenance and feature support compared to the opendal crate. This also aligns with AWS best practices for interacting with S3 services.
101 lines
4 KiB
Rust
101 lines
4 KiB
Rust
use crate::auth::AuthService;
|
|
use crate::channels::{ChannelAdapter, VoiceAdapter, WebChannelAdapter};
|
|
use crate::config::AppConfig;
|
|
use crate::llm::LLMProvider;
|
|
use crate::session::SessionManager;
|
|
use crate::tools::{ToolApi, ToolManager};
|
|
use crate::whatsapp::WhatsAppAdapter;
|
|
use diesel::{Connection, PgConnection};
|
|
use aws_sdk_s3::Client as S3Client;
|
|
use redis::Client as RedisClient;
|
|
use std::collections::HashMap;
|
|
use std::sync::Arc;
|
|
use std::sync::Mutex;
|
|
use tokio::sync::mpsc;
|
|
use crate::shared::models::BotResponse;
|
|
|
|
pub struct AppState {
|
|
pub s3_client: Option<S3Client>,
|
|
pub bucket_name: String,
|
|
pub config: Option<AppConfig>,
|
|
pub conn: Arc<Mutex<PgConnection>>,
|
|
pub custom_conn: Arc<Mutex<PgConnection>>,
|
|
pub redis_client: Option<Arc<RedisClient>>,
|
|
pub session_manager: Arc<tokio::sync::Mutex<SessionManager>>,
|
|
pub tool_manager: Arc<ToolManager>,
|
|
pub llm_provider: Arc<dyn LLMProvider>,
|
|
pub auth_service: Arc<tokio::sync::Mutex<AuthService>>,
|
|
pub channels: Arc<Mutex<HashMap<String, Arc<dyn ChannelAdapter>>>>,
|
|
pub response_channels: Arc<tokio::sync::Mutex<HashMap<String, mpsc::Sender<BotResponse>>>>,
|
|
pub web_adapter: Arc<WebChannelAdapter>,
|
|
pub voice_adapter: Arc<VoiceAdapter>,
|
|
pub whatsapp_adapter: Arc<WhatsAppAdapter>,
|
|
pub tool_api: Arc<ToolApi>,
|
|
}
|
|
|
|
impl Clone for AppState {
|
|
fn clone(&self) -> Self {
|
|
Self {
|
|
s3_client: self.s3_client.clone(),
|
|
bucket_name: self.bucket_name.clone(),
|
|
config: self.config.clone(),
|
|
conn: Arc::clone(&self.conn),
|
|
custom_conn: Arc::clone(&self.custom_conn),
|
|
redis_client: self.redis_client.clone(),
|
|
session_manager: Arc::clone(&self.session_manager),
|
|
tool_manager: Arc::clone(&self.tool_manager),
|
|
llm_provider: Arc::clone(&self.llm_provider),
|
|
auth_service: Arc::clone(&self.auth_service),
|
|
channels: Arc::clone(&self.channels),
|
|
response_channels: Arc::clone(&self.response_channels),
|
|
web_adapter: Arc::clone(&self.web_adapter),
|
|
voice_adapter: Arc::clone(&self.voice_adapter),
|
|
whatsapp_adapter: Arc::clone(&self.whatsapp_adapter),
|
|
tool_api: Arc::clone(&self.tool_api),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Default for AppState {
|
|
fn default() -> Self {
|
|
Self {
|
|
s3_client: None,
|
|
bucket_name: "default.gbai".to_string(),
|
|
config: None,
|
|
conn: Arc::new(Mutex::new(
|
|
diesel::PgConnection::establish("postgres://localhost/test").unwrap(),
|
|
)),
|
|
custom_conn: Arc::new(Mutex::new(
|
|
diesel::PgConnection::establish("postgres://localhost/test").unwrap(),
|
|
)),
|
|
redis_client: None,
|
|
session_manager: Arc::new(tokio::sync::Mutex::new(SessionManager::new(
|
|
diesel::PgConnection::establish("postgres://localhost/test").unwrap(),
|
|
None,
|
|
))),
|
|
tool_manager: Arc::new(ToolManager::new()),
|
|
llm_provider: Arc::new(crate::llm::OpenAIClient::new(
|
|
"empty".to_string(),
|
|
Some("http://localhost:8081".to_string()),
|
|
)),
|
|
auth_service: Arc::new(tokio::sync::Mutex::new(AuthService::new(
|
|
diesel::PgConnection::establish("postgres://localhost/test").unwrap(),
|
|
None,
|
|
))),
|
|
channels: Arc::new(Mutex::new(HashMap::new())),
|
|
response_channels: Arc::new(tokio::sync::Mutex::new(HashMap::new())),
|
|
web_adapter: Arc::new(WebChannelAdapter::new()),
|
|
voice_adapter: Arc::new(VoiceAdapter::new(
|
|
"https://livekit.example.com".to_string(),
|
|
"api_key".to_string(),
|
|
"api_secret".to_string(),
|
|
)),
|
|
whatsapp_adapter: Arc::new(WhatsAppAdapter::new(
|
|
"whatsapp_token".to_string(),
|
|
"phone_number_id".to_string(),
|
|
"verify_token".to_string(),
|
|
)),
|
|
tool_api: Arc::new(ToolApi::new()),
|
|
}
|
|
}
|
|
}
|