feat: Added field time_grain_enabled in prompt schema

valmi-io · Jun 18, 2024 · 5ad2110 · 5ad2110
1 parent e3b9f73
commit 5ad2110
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/core/schemas/schemas.py b/core/schemas/schemas.py
@@ -80,14 +80,16 @@ class Config(CamelSchemaConfig):
 class PromptSchema(ModelSchema):
     class Config(CamelSchemaConfig):
         model = Prompt
-        model_fields = ["id", "name", "description", "type", "filters", "operators", "package_id", "gated", "query", "time_grain_enabled"]
+        model_fields = ["id", "name", "description", "type", "filters", "operators",
+                        "package_id", "gated", "query", "time_grain_enabled", "time_grain_enabled"]
 
 
 class PromptByIdSchema(ModelSchema):
     class Config(CamelSchemaConfig):
         model = Prompt
-        model_fields = ["id", "name", "description", "type", "filters", "operators", "package_id", "gated", "query", "time_grain_enabled"]
-    time_grain: List[TimeGrain] = None 
+        model_fields = ["id", "name", "description", "type", "filters", "operators",
+                        "package_id", "gated", "query", "time_grain_enabled", "time_grain_enabled"]
+    time_grain: List[TimeGrain] = None
     schemas: List[Dict]