fix(kubegraph): use original metadata on empty analyzer mode

ulagbulag · Jun 1, 2024 · 0b9e327 · 0b9e327
1 parent 9204ea2
commit 0b9e327
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/crates/kubegraph/analyzer/llm/src/lib.rs b/crates/kubegraph/analyzer/llm/src/lib.rs
@@ -70,9 +70,9 @@ where
         graph: Graph<LazyFrame, GraphMetadataRaw>,
     ) -> Result<Graph<LazyFrame, GraphMetadataStandard>> {
         let VirtualProblemAnalyzer {
-            original_metadata,
+            original_metadata: map_from,
             r#type: VirtualProblemAnalyzerLLM {},
-        } = problem.analyzer.clone().into_llm()?;
+        } = problem.analyzer.clone().try_into_llm()?;
         let map_to = problem.spec.metadata;
 
         // TODO: to be implemented
@@ -82,7 +82,7 @@ where
             scope,
         } = graph;
         Ok(Graph {
-            data: data.cast(&original_metadata, &map_to),
+            data: data.cast(&map_from, &map_to),
             metadata: map_to,
             scope,
         })
@@ -113,11 +113,11 @@ where
 }
 
 trait VirtualProblemAnalyzerExt {
-    fn into_llm(self) -> Result<VirtualProblemAnalyzer<VirtualProblemAnalyzerLLM>>;
+    fn try_into_llm(self) -> Result<VirtualProblemAnalyzer<VirtualProblemAnalyzerLLM>>;
 }
 
 impl VirtualProblemAnalyzerExt for VirtualProblemAnalyzer {
-    fn into_llm(self) -> Result<VirtualProblemAnalyzer<VirtualProblemAnalyzerLLM>> {
+    fn try_into_llm(self) -> Result<VirtualProblemAnalyzer<VirtualProblemAnalyzerLLM>> {
         let Self {
             original_metadata,
             r#type,

diff --git a/crates/kubegraph/vm/local/src/analyzer.rs b/crates/kubegraph/vm/local/src/analyzer.rs
@@ -6,7 +6,7 @@ use kubegraph_api::{
     analyzer::{VirtualProblemAnalyzer, VirtualProblemAnalyzerType},
     component::NetworkComponent,
     frame::LazyFrame,
-    graph::{Graph, GraphMetadataExt, GraphMetadataRaw, GraphMetadataStandard},
+    graph::{Graph, GraphMetadataRaw, GraphMetadataStandard},
     problem::VirtualProblem,
 };
 use schemars::JsonSchema;
@@ -119,13 +119,14 @@ impl ::kubegraph_api::analyzer::NetworkAnalyzer for NetworkAnalyzer {
             Self::Disabled => {
                 let Graph {
                     data,
-                    metadata: map_from,
+                    metadata: _,
                     scope,
                 } = graph;
+                let map_from = &problem.analyzer.original_metadata;
                 let map_to = problem.spec.metadata;
 
                 Ok(Graph {
-                    data: data.cast(&map_from.to_pinned(), &map_to),
+                    data: data.cast(map_from, &map_to),
                     metadata: map_to,
                     scope,
                 })