From 319e3dea8581d717b8b3459b954bc139b8c4f458 Mon Sep 17 00:00:00 2001 From: Daniel Wiesmann Date: Wed, 12 Feb 2025 14:27:56 +0000 Subject: [PATCH] Remove unnecessary mixin --- zeno/agents/layerfinder/agent.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/zeno/agents/layerfinder/agent.py b/zeno/agents/layerfinder/agent.py index 0be7a1a..96959d0 100644 --- a/zeno/agents/layerfinder/agent.py +++ b/zeno/agents/layerfinder/agent.py @@ -6,13 +6,7 @@ from typing_extensions import Self -# Add this mixin to all models -class JSONSerializable(BaseModel): - def to_dict(self): - return json.loads(json.dumps(self, default=lambda o: o.dict())) - - -class Dataset(JSONSerializable): +class Dataset(BaseModel): title: str = Field(description="The title of the dataset") dataset: str = Field(description="The slug of the dataset") explanation: str = Field( @@ -31,7 +25,7 @@ def set_uri_and_tilelayer(self) -> Self: return self -class LayerFinderResponse(JSONSerializable): +class LayerFinderResponse(BaseModel): datasets: List[Dataset]