From a8b663505053624a011296bdb39ed5e6951ae848 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Tue, 5 Mar 2024 12:35:52 +1100
Subject: [PATCH] fix(mm): make `key` required in openapi schema for model
 config

---
 invokeai/backend/model_manager/config.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/invokeai/backend/model_manager/config.py b/invokeai/backend/model_manager/config.py
index 78149af190..749586f266 100644
--- a/invokeai/backend/model_manager/config.py
+++ b/invokeai/backend/model_manager/config.py
@@ -162,7 +162,11 @@ class ModelConfigBase(BaseModel):
         description="Default settings for this model", default=None
     )
 
-    model_config = ConfigDict(validate_assignment=True)
+    @staticmethod
+    def json_schema_extra(schema: dict[str, Any], model_class: Type[BaseModel]) -> None:
+        schema["required"].extend(["key"])
+
+    model_config = ConfigDict(validate_assignment=True, json_schema_extra=json_schema_extra)
 
 
 class CheckpointConfigBase(ModelConfigBase):