fix: filter improvement

This commit is contained in:
Per Stark
2025-09-23 08:59:57 +02:00
parent c12d00edaa
commit e1d98b0c35

View File

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