test working for pydantic responses

generator/konfig-generator-api/src/main/resources/python/endpoint.handlebars

@@ -219,10 +219,20 @@ class BaseApi(api_client.Api):
     {{> endpoint_operation_impl}}
 
 
-class {{operationIdCamelCase}}(BaseApi):
+class {{#prstv2}}Raw{{/prstv2}}{{operationIdCamelCase}}(BaseApi):
     # this class is used by api classes that refer to endpoints with operationId fn names
 
     {{> endpoint_interface_impl methodName=operationId}}
+{{#prstv2}}
+
+
+class {{operationIdCamelCase}}(BaseApi):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.raw = Raw{{operationIdCamelCase}}(*args, **kwargs)
+
+    {{> endpoint_pydantic_impl}}
+{{/prstv2}}
 
 
 class ApiFor{{httpMethod}}(BaseApi):

generator/konfig-generator-api/src/main/resources/python/endpoint_args.handlebars

@@ -77,6 +77,11 @@
     {{/if}}
     stream: bool = False,
 {{/if}}
+{{#if interface}}
+{{#if validateArg}}
+    validate: bool = False,
+{{/if}}
+{{/if}}
 ){{#unless omitReturnType}} -> {{#if isMappedArgs}}api_client.MappedArgs{{else}}typing.Union[
     {{#each responses}}
         {{#if isDefault}}

generator/konfig-generator-api/src/main/resources/python/endpoint_pydantic_impl.handlebars

@@ -0,0 +1,19 @@
+{{#if isDeprecated}}@api_client.DeprecationWarningOnce(prefix="{{clientApiName}}")
+{{/if}}async def a{{operationId}}(
+{{> endpoint_args async=true isOverload=false skipDeserialization="null" contentType="null" interface=true omitReturnType=true validateArg=true}}
+    raw_response = await self.raw.a{{operationId}}(
+        {{> endpoint_args_call_mapped_args}}
+    )
+    if validate:
+        return {{returnType}}Pydantic(**raw_response.body)
+    return {{returnType}}Pydantic.model_construct(**raw_response.body)
+
+{{#if isDeprecated}}@api_client.DeprecationWarningOnce(prefix="{{clientApiName}}")
+{{/if}}def {{operationId}}(
+{{> endpoint_args isOverload=false skipDeserialization="null" contentType="null" interface=true omitReturnType=true validateArg=true}}
+    raw_response = self.raw.{{operationId}}(
+        {{> endpoint_args_call_mapped_args}}
+    )
+    if validate:
+        return {{returnType}}Pydantic(**raw_response.body)
+    return {{returnType}}Pydantic.model_construct(**raw_response.body)

generator/konfig-generator-api/src/main/resources/python/pydantic.handlebars

@@ -3,9 +3,7 @@
 {{>partial_header}}
 
 from datetime import datetime, date
-import typing
-from enum import Enum
-from typing_extensions import TypedDict, Literal
+from pydantic import BaseModel, Field
 {{#each models}}
 {{#with model}}
 

generator/konfig-generator-api/src/main/resources/python/pydantic_templates/type.handlebars

@@ -1,6 +1,6 @@
-{{> type_templates/type_object}}
-{{> type_templates/type_composed}}
-{{> type_templates/type_enum}}
-{{> type_templates/type_list}}
-{{> type_templates/type_map}}
-{{> type_templates/type_primitive}}
+{{> pydantic_templates/type_object}}
+{{> pydantic_templates/type_composed}}
+{{> pydantic_templates/type_enum}}
+{{> pydantic_templates/type_list}}
+{{> pydantic_templates/type_map}}
+{{> pydantic_templates/type_primitive}}

generator/konfig-generator-api/src/main/resources/python/pydantic_templates/type_object.handlebars

@@ -1,18 +1,5 @@
 {{#if hasVars}}
-{{#if useDictionaryBasedTypedDict}}
-Required{{classname}} = TypedDict("Required{{classname}}", {
-    {{#each requiredVars}}
-        {{#if description}}
-    # {{{description}}}
-        {{/if}}
-    "{{baseName}}": {{dataType}},
-    {{#unless @last}}
-
-    {{/unless}}
-    {{/each}}
-    })
-{{else}}
-class Required{{classname}}(TypedDict):
+class {{classname}}(BaseModel):
     {{#each requiredVars}}
         {{#if description}}
     # {{{description}}}
@@ -22,39 +9,18 @@ class Required{{classname}}(TypedDict):
 
     {{/unless}}
     {{/each}}
-    {{#unless requiredVars}}
-    pass
-    {{/unless}}
-{{/if}}
-
-{{#if useDictionaryBasedTypedDict}}
-Optional{{classname}} = TypedDict("Optional{{classname}}", {
     {{#each optionalVars}}
         {{#if description}}
     # {{{description}}}
         {{/if}}
-    "{{baseName}}": {{dataType}},
-    {{#unless @last}}
-
-    {{/unless}}
-    {{/each}}
-    }, total=False)
-{{else}}
-class Optional{{classname}}(TypedDict, total=False):
-    {{#each optionalVars}}
-        {{#if description}}
-    # {{{description}}}
-        {{/if}}
-    {{baseName}}: {{{dataType}}}
+    {{baseName}}: {{{dataType}}} = Field(None)
     {{#unless @last}}
 
     {{/unless}}
     {{/each}}
+    {{#unless requiredVars}}
     {{#unless optionalVars}}
     pass
     {{/unless}}
-{{/if}}
-
-class {{classname}}(Required{{classname}}, Optional{{classname}}):
-    pass
+    {{/unless}}
 {{/if}}