chore: add knowledge field to CMD (#229)

picahq · Jan 28, 2025 · f7dbbe6 · f7dbbe6
1 parent 80642f1
commit f7dbbe6
Show file tree

Hide file tree

Showing 7 changed files with 14 additions and 3 deletions.
diff --git a/api/src/logic/connection_model_definition.rs b/api/src/logic/connection_model_definition.rs
@@ -307,6 +307,7 @@ pub struct CreateRequest {
     pub paths: Option<ModelPaths>,
     pub supported: Option<bool>,
     pub active: Option<bool>,
+    pub knowledge: Option<String>,
 }
 
 impl HookExt<ConnectionModelDefinition> for CreateRequest {}
@@ -359,6 +360,7 @@ impl RequestExt for CreateRequest {
             mapping: self.mapping.clone(),
             record_metadata: Default::default(),
             supported: self.supported.unwrap_or(false),
+            knowledge: self.knowledge.clone(),
         };
         record.record_metadata.version = self.version.clone();
         Some(record)

diff --git a/api/tests/context.rs b/api/tests/context.rs
@@ -521,6 +521,7 @@ impl TestServer {
             }),
             supported: Some(true),
             active: Some(true),
+            knowledge: None,
         };
 
         let res = self

diff --git a/api/tests/http/passthrough.rs b/api/tests/http/passthrough.rs
@@ -73,6 +73,7 @@ async fn test_passthrough_api() {
         mapping: None,
         supported: Some(true),
         active: Some(true),
+        knowledge: None,
     };
 
     let create_model_definition_response = server

diff --git a/api/tests/http/unified.rs b/api/tests/http/unified.rs
@@ -585,6 +585,7 @@ async fn create_connection_model_definition(
         mapping: Some(mapping.clone()),
         supported: Some(true),
         active: Some(true),
+        knowledge: None,
     };
 
     let create_model_definition_response = server

diff --git a/api/tests/standard/mod.rs b/api/tests/standard/mod.rs
@@ -73,6 +73,7 @@ fn test_json_connection_model_definition() {
         }),
         record_metadata: RecordMetadata::test(),
         supported: false,
+        knowledge: None,
     };
 
     assert!(

diff --git a/entities/src/domain/connection/connection_model_definition.rs b/entities/src/domain/connection/connection_model_definition.rs
@@ -47,11 +47,14 @@ pub struct ConnectionModelDefinition {
     pub is_default_crud_mapping: Option<bool>,
     pub mapping: Option<CrudMapping>,
 
-    #[serde(flatten, default)]
-    pub record_metadata: RecordMetadata,
-
     #[serde(default)]
     pub supported: bool,
+
+    #[serde(default, skip_serializing_if = "Option::is_none")]
+    pub knowledge: Option<String>,
+
+    #[serde(flatten, default)]
+    pub record_metadata: RecordMetadata,
 }
 
 #[derive(Debug, Clone, Eq, PartialEq, Hash, Deserialize, Serialize)]

diff --git a/unified/src/client.rs b/unified/src/client.rs
@@ -237,6 +237,7 @@ mod tests {
             is_default_crud_mapping: None,
             mapping: None,
             supported: true,
+            knowledge: None,
         };
 
         let client = Client::new();
@@ -312,6 +313,7 @@ mod tests {
             is_default_crud_mapping: None,
             mapping: None,
             supported: true,
+            knowledge: None,
         };
 
         let client = Client::new();