diff --git a/python/cog/schema.py b/python/cog/schema.py index a5a3943f2e..efdad2d0d7 100644 --- a/python/cog/schema.py +++ b/python/cog/schema.py @@ -75,7 +75,6 @@ class PredictionRequest(PredictionBaseModel): webhook: Optional[WebhookUrl] = None webhook_events_filter: Optional[List[WebhookEvent]] = pydantic.Field( default=WebhookEvent.default_events(), - title=None, ) @classmethod diff --git a/python/cog/server/helpers.py b/python/cog/server/helpers.py index f54544bbad..33bdbd4a02 100644 --- a/python/cog/server/helpers.py +++ b/python/cog/server/helpers.py @@ -263,12 +263,24 @@ def unwrap_pydantic_serialization_iterators(obj: Any) -> Any: def update_openapi_schema_for_pydantic_2( openapi_schema: Dict[str, Any], ) -> None: + _remove_webhook_events_filter_title(openapi_schema) _remove_empty_or_nullable_anyof(openapi_schema) _flatten_selected_allof_refs(openapi_schema) _extract_enum_properties(openapi_schema) _set_default_enumeration_description(openapi_schema) +def _remove_webhook_events_filter_title( + openapi_schema: Dict[str, Any], +) -> None: + try: + del openapi_schema["components"]["schemas"]["PredictionRequest"]["properties"][ + "webhook_events_filter" + ]["title"] + except KeyError: + pass + + def _remove_empty_or_nullable_anyof( openapi_schema: Union[Dict[str, Any], List[Dict[str, Any]]], ) -> None: