fix: ModelGenerationInputStable / ModelPayloadRootStable

Author: tazlin

horde_sdk/ai_horde_api/apimodels/base.py

@@ -227,11 +227,11 @@ def get_api_model_name(cls) -> str | None:
         return "RequestSingleWarning"
 
 
-class ImageGenerateParamMixin(HordeAPIData):
+class ImageGenerateParamMixin(HordeAPIObjectBaseModel):
     """Mix-in class of some of the data included in a request to the `/v2/generate/async` endpoint.
 
     Also is the corresponding information returned on a job pop to the `/v2/generate/pop` endpoint.
-    v2 API Model: `ModelPayloadStable`
+    v2 API Model: `ModelPayloadRootStable`
     """
 
     model_config = (
@@ -361,6 +361,11 @@ def control_type_must_be_known(cls, v: str | KNOWN_CONTROLNETS | None) -> str |
 
         return v
 
+    @override
+    @classmethod
+    def get_api_model_name(cls) -> str | None:
+        return "ModelPayloadRootStable"
+
 
 class JobSubmitResponse(HordeResponseBaseModel):
     """The response to a job submission request, indicating the number of kudos gained.

horde_sdk/ai_horde_api/apimodels/generate/_async.py

@@ -98,7 +98,7 @@ def get_api_model_name(cls) -> str | None:
         return _ANONYMOUS_MODEL
 
 
-class ImageGenerationInputPayload(HordeAPIObject, ImageGenerateParamMixin):
+class ImageGenerationInputPayload(ImageGenerateParamMixin):
     """Represents the 'params' field in the `/v2/generate/async` endpoint.
 
     v2 API Model: `ModelGenerationInputStable`