[API] Update Function Minimal Format (#5692)

mod function spec Co-authored-by: quaark <a.melnick@icloud.com>
mlrun · Jun 3, 2024 · 61e0d47 · 61e0d47
1 parent e1908f1
commit 61e0d47
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/mlrun/common/formatters/function.py b/mlrun/common/formatters/function.py
@@ -33,9 +33,14 @@ def format_method(_format: str) -> typing.Optional[typing.Callable]:
                     ["metadata"],
                     ["status"],
                     ["spec", "description"],
+                    ["spec", "command"],
                     ["spec", "image"],
                     ["spec", "default_handler"],
-                    ["spec", "entry_points"],
+                    ["spec", "default_class"],
+                    ["spec", "graph"],
+                    ["spec", "preemption_mode"],
+                    ["spec", "node_selector"],
+                    ["spec", "priority_class_name"],
                 ]
             ),
         }[_format]
diff --git a/tests/api/db/test_functions.py b/tests/api/db/test_functions.py
@@ -264,9 +264,14 @@ def test_list_functions_with_format(db: DBInterface, db_session: Session):
         "status": {"state": "online"},
         "spec": {
             "description": "some_description",
+            "command": "some_command",
             "image": "some_image",
             "default_handler": "some_handler",
-            "entry_points": "some_entry_points",
+            "default_class": "some_class",
+            "graph": "some_graph",
+            "preemption_mode": "some_preemption_mode",
+            "node_selector": "some_node_selector",
+            "priority_class_name": "some_priority_class_name",
             "extra_field": "extra_field",
         },
     }