From 15a06521b3995dd3cfb1cfe358b3c7f89c1cc8ca Mon Sep 17 00:00:00 2001 From: stuartgo Date: Wed, 25 Feb 2026 15:56:17 +0100 Subject: [PATCH] add model validator that removes embedding field passed --- anansyTypes/structures/phrase.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/anansyTypes/structures/phrase.py b/anansyTypes/structures/phrase.py index c032944..4bd91a7 100644 --- a/anansyTypes/structures/phrase.py +++ b/anansyTypes/structures/phrase.py @@ -1,4 +1,5 @@ from typing import List, Any +from pydantic import model_validator from pydantic_core import core_schema import torch @@ -19,6 +20,12 @@ class NoEmbeddingPhrase(Lexeme): content=data["content"], original_content=data["original_content"], ) + @model_validator(mode="before") + @classmethod + def ignore_embedding(cls, values): + # Remove embedding from incoming data + values.pop("embedding", None) + return values class Phrase(Lexeme):