refactoring: completed storage, now using new fn to construct

This commit is contained in:
Per Stark
2024-11-21 09:23:42 +01:00
parent 41134cfa49
commit 53b94c3569
8 changed files with 121 additions and 148 deletions

View File

@@ -1,7 +1,9 @@
use crate::{
error::ProcessingError,
models::graph_entities::{
GraphMapper, KnowledgeEntity, KnowledgeEntityType, KnowledgeRelationship,
models::graph_entities::GraphMapper,
storage::types::{
knowledge_entity::{KnowledgeEntity, KnowledgeEntityType},
knowledge_relationship::KnowledgeRelationship,
},
};
use async_openai::types::{
@@ -119,7 +121,7 @@ impl LLMGraphAnalysisResult {
entity_type: KnowledgeEntityType::from(llm_entity.entity_type.clone()),
source_id: source_id.to_string(),
metadata: None,
embedding: Some(embedding),
embedding,
};
entities.push(knowledge_entity);