Fixed simple types for response_model.

Author: mr-fatalyst

fastopenapi/core/constants.py

@@ -13,6 +13,7 @@
     float: "number",
     bool: "boolean",
     str: "string",
+    dict: "object",
 }
 
 

fastopenapi/openapi/generator.py

@@ -643,13 +643,15 @@ def _add_response_model(
             inner_type = typing.get_args(response_model)[0]
             if self._is_pydantic_model(inner_type):
                 inner_schema = self.schema_builder.get_model_schema(inner_type)
-                array_schema = {"type": "array", "items": inner_schema}
-                responses[status_code]["content"] = {
-                    "application/json": {"schema": array_schema}
-                }
+                schema = {"type": "array", "items": inner_schema}
+            else:
+                schema = self.schema_builder.build_parameter_schema(response_model)
         elif self._is_pydantic_model(response_model):
             schema = self.schema_builder.get_model_schema(response_model)
-            responses[status_code]["content"] = {"application/json": {"schema": schema}}
+        else:
+            schema = self.schema_builder.build_parameter_schema(response_model)
+
+        responses[status_code]["content"] = {"application/json": {"schema": schema}}
 
     def _add_security_error_responses(
         self, responses: dict, route, has_security: bool = False

tests/openapi/test_openapi_generator.py

@@ -1281,26 +1281,26 @@ def test_response_builder_add_response_model_list_pydantic(self):
         assert "$ref" in schema["items"]
 
     def test_response_builder_add_response_model_list_non_pydantic(self):
-        """Test adding list of non-Pydantic types"""
+        """Test adding list of non-Pydantic types generates array schema"""
         builder = ResponseBuilder(self.generator.schema_builder)
 
         responses = {"200": {"description": "OK"}}
 
         builder._add_response_model(responses, "200", list[str])
 
-        # Should not add content for non-Pydantic types
-        assert "content" not in responses["200"]
+        schema = responses["200"]["content"]["application/json"]["schema"]
+        assert schema == {"type": "array", "items": {"type": "string"}}
 
     def test_response_builder_add_response_model_non_pydantic(self):
-        """Test adding non-Pydantic response model"""
+        """Test adding non-Pydantic response model generates schema"""
         builder = ResponseBuilder(self.generator.schema_builder)
 
         responses = {"200": {"description": "OK"}}
 
         builder._add_response_model(responses, "200", str)
 
-        # Should not add content for non-Pydantic types
-        assert "content" not in responses["200"]
+        schema = responses["200"]["content"]["application/json"]["schema"]
+        assert schema == {"type": "string"}
 
     def test_response_builder_add_custom_error_responses(self):
         """Test adding custom error responses"""