add llmconfig to metastore

Author: parmesant

src/alerts/alert_types.rs

@@ -182,13 +182,13 @@ impl AlertTrait for ThresholdAlert {
         &mut self,
         new_notification_state: NotificationState,
     ) -> Result<(), AlertError> {
-        // update state in memory
-        self.notification_state = new_notification_state;
         // update on disk
         PARSEABLE
             .metastore
             .put_alert(&self.to_alert_config())
             .await?;
+        // update state in memory
+        self.notification_state = new_notification_state;
 
         Ok(())
     }

src/metastore/metastore_traits.rs

@@ -43,6 +43,11 @@ pub trait Metastore: std::fmt::Debug + Send + Sync {
     async fn put_alert(&self, obj: &dyn MetastoreObject) -> Result<(), MetastoreError>;
     async fn delete_alert(&self, obj: &dyn MetastoreObject) -> Result<(), MetastoreError>;
 
+    /// llmconfig
+    async fn get_llmconfigs(&self) -> Result<Vec<Bytes>, MetastoreError>;
+    async fn put_llmconfig(&self, obj: &dyn MetastoreObject) -> Result<(), MetastoreError>;
+    async fn delete_llmconfig(&self, obj: &dyn MetastoreObject) -> Result<(), MetastoreError>;
+
     /// targets
     async fn get_targets(&self) -> Result<Vec<Target>, MetastoreError>;
     async fn put_target(&self, obj: &dyn MetastoreObject) -> Result<(), MetastoreError>;

src/metastore/metastores/object_store_metastore.rs

@@ -114,6 +114,39 @@ impl Metastore for ObjectStoreMetastore {
             .await?)
     }
 
+    /// This function fetches all the llmconfigs from the underlying object store
+    async fn get_llmconfigs(&self) -> Result<Vec<Bytes>, MetastoreError> {
+        let base_path = RelativePathBuf::from_iter([SETTINGS_ROOT_DIRECTORY, "llmconfigs"]);
+        let conf_bytes = self
+            .storage
+            .get_objects(
+                Some(&base_path),
+                Box::new(|file_name| file_name.ends_with(".json")),
+            )
+            .await?;
+
+        Ok(conf_bytes)
+    }
+
+    /// This function puts an llmconfig in the object store at the given path
+    async fn put_llmconfig(&self, obj: &dyn MetastoreObject) -> Result<(), MetastoreError> {
+        let path = obj.get_object_path();
+
+        Ok(self
+            .storage
+            .put_object(&RelativePathBuf::from(path), to_bytes(obj))
+            .await?)
+    }
+
+    /// Delete an llmconfig
+    async fn delete_llmconfig(&self, obj: &dyn MetastoreObject) -> Result<(), MetastoreError> {
+        let path = obj.get_object_path();
+        Ok(self
+            .storage
+            .delete_object(&RelativePathBuf::from(path))
+            .await?)
+    }
+
     /// Fetch all dashboards
     async fn get_dashboards(&self) -> Result<Vec<Bytes>, MetastoreError> {
         let mut dashboards = Vec::new();