fix tests

This commit is contained in:
phernandez
2024-12-24 20:29:21 -06:00
parent d713097333
commit 90ca41fead
16 changed files with 137 additions and 138 deletions
+1 -9
View File
@@ -127,15 +127,7 @@ async def delete_relations(
data: DeleteRelationsRequest, knowledge_service: KnowledgeServiceDep
) -> CreateEntityResponse:
"""Delete relations between entities."""
to_delete = [
{
"from_id": relation.from_id,
"to_id": relation.to_id,
"relation_type": relation.relation_type,
}
for relation in data.relations
]
updated_entities = await knowledge_service.delete_relations(to_delete)
updated_entities = await knowledge_service.delete_relations(data.relations)
return CreateEntityResponse(
entities=[EntityResponse.model_validate(entity) for entity in updated_entities]
)