Merge pull request #91 from Haidra-Org/main

fix: don't crash when extra/new fields are returned due to api changes
Haidra-Org · Dec 12, 2023 · ad3a5a3 · ad3a5a3
2 parents 6fd9497 + 8e3258e
commit ad3a5a3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/horde_sdk/ai_horde_api/apimodels/base.py b/horde_sdk/ai_horde_api/apimodels/base.py
@@ -110,7 +110,7 @@ class ImageGenerateParamMixin(BaseModel):
     v2 API Model: `ModelPayloadStable`
     """
 
-    model_config = ConfigDict(frozen=True, extra="forbid")
+    model_config = ConfigDict(frozen=True)  # , extra="forbid")
 
     sampler_name: KNOWN_SAMPLERS = KNOWN_SAMPLERS.k_lms
     """The sampler to use for this generation. Defaults to `KNOWN_SAMPLERS.k_lms`."""

diff --git a/horde_sdk/generic_api/apimodels.py b/horde_sdk/generic_api/apimodels.py
@@ -46,7 +46,7 @@ class HordeResponse(HordeAPIMessage):
 
 
 class HordeResponseBaseModel(HordeResponse, BaseModel):
-    model_config = ConfigDict(frozen=True, extra="forbid")
+    model_config = ConfigDict(frozen=True)  # , extra="forbid")
 
 
 class ResponseRequiringFollowUpMixin(abc.ABC):