add base api model (#1298)

evidentlyai · Sep 11, 2024 · 5813560 · 5813560
1 parent dc562bc
commit 5813560
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/evidently/ui/api/models.py b/src/evidently/ui/api/models.py
@@ -8,6 +8,7 @@
 from typing import TypeVar
 
 from evidently._pydantic_compat import BaseModel
+from evidently._pydantic_compat import Extra
 from evidently.base_metric import Metric
 from evidently.model.dashboard import DashboardInfo
 from evidently.model.widget import BaseWidgetInfo
@@ -29,6 +30,12 @@
 from evidently.ui.type_aliases import UserID
 
 
+class EvidentlyAPIModel(BaseModel):
+    # todo: migrate all models to this base
+    class Config:
+        extra = Extra.forbid
+
+
 class MetricModel(BaseModel):
     id: str