From e1d98b0c351371f5f4f0cf1a815290741f86e8dc Mon Sep 17 00:00:00 2001
From: Per Stark <per@stark.pub>
Date: Tue, 23 Sep 2025 08:59:57 +0200
Subject: [PATCH] fix: filter improvement

---
 common/src/storage/types/user.rs | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/common/src/storage/types/user.rs b/common/src/storage/types/user.rs
index 5a00c0c..f7f140c 100644
--- a/common/src/storage/types/user.rs
+++ b/common/src/storage/types/user.rs
@@ -9,7 +9,7 @@ use super::text_chunk::TextChunk;
 use super::{
     conversation::Conversation,
     ingestion_task::{IngestionTask, MAX_ATTEMPTS},
-    knowledge_entity::KnowledgeEntity,
+    knowledge_entity::{KnowledgeEntity, KnowledgeEntityType},
     knowledge_relationship::KnowledgeRelationship,
     system_settings::SystemSettings,
     text_content::TextContent,
@@ -356,7 +356,10 @@ impl User {
         // Extract the entity types from the response
         let entity_types: Vec<String> = response
             .into_iter()
-            .map(|item| format!("{:?}", item.entity_type))
+            .map(|item| {
+                let normalized = KnowledgeEntityType::from(item.entity_type.clone());
+                format!("{:?}", normalized)
+            })
             .collect();
 
         Ok(entity_types)