47 lines
1.3 KiB
Rust
47 lines
1.3 KiB
Rust
|
mod config;
|
||
|
|
mod controller;
|
||
|
|
mod utils;
|
||
|
|
|
||
|
|
pub use config::Config;
|
||
|
|
pub use controller::Controller;
|
||
|
|
|
||
|
|
// openai_compat needs these, so it can convert from its own config types to these
|
||
|
|
pub(super) use config::ImageGenerationConfig;
|
||
|
|
pub(super) use config::SpeechToTextConfig;
|
||
|
|
pub(super) use config::TextGenerationConfig;
|
||
|
|
pub(super) use config::TextToSpeechConfig;
|
||
|
|
|
||
|
|
use super::super::AgentInstantiationError;
|
||
|
|
use super::super::AgentInstantiationResult;
|
||
|
|
use super::controller::ControllerType;
|
||
|
|
use super::ConfigTrait;
|
||
|
|
|
||
|
|
pub fn create_controller_from_yaml_value_config(
|
||
|
|
agent_id: &str,
|
||
|
|
config: serde_yaml::Value,
|
||
|
|
) -> AgentInstantiationResult<ControllerType> {
|
||
|
|
let config = match &config {
|
||
|
|
serde_yaml::Value::Mapping(_) => {
|
||
|
|
let config: Config =
|
||
|
|
serde_yaml::from_value(config).map_err(AgentInstantiationError::Yaml)?;
|
||
|
|
|
||
|
|
config
|
||
|
|
.validate()
|
||
|
|
.map_err(AgentInstantiationError::ConfigFailsValidation)?;
|
||
|
|
|
||
|
|
config
|
||
|
|
}
|
||
|
|
_ => {
|
||
|
|
return Err(AgentInstantiationError::ConfigForAgentIsNotAMapping(
|
||
|
|
agent_id.to_owned(),
|
||
|
|
));
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
Ok(ControllerType::OpenAI(Box::new(Controller::new(config))))
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn default_config() -> Config {
|
||
|
|
Config::default()
|
||
|
|
}
|