Initial commit

src/agent/provider/openai/mod.rs

@@ -0,0 +1,46 @@
+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()
+}