# coding: utf-8

"""
    Docling Serve

    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)

    The version of the OpenAPI document: 
    Generated by OpenAPI Generator (https://openapi-generator.tech)

    Do not edit the class manually.
"""  # noqa: E501


import unittest

from yfkm3_services_parse_client.models.picture_description_custom_config import PictureDescriptionCustomConfig

class TestPictureDescriptionCustomConfig(unittest.TestCase):
    """PictureDescriptionCustomConfig unit test stubs"""

    def setUp(self):
        pass

    def tearDown(self):
        pass

    def make_instance(self, include_optional) -> PictureDescriptionCustomConfig:
        """Test PictureDescriptionCustomConfig
            include_optional is a boolean, when False only required
            params are included, when True both required and
            optional params are included """
        # uncomment below to create an instance of `PictureDescriptionCustomConfig`
        """
        model = PictureDescriptionCustomConfig()
        if include_optional:
            return PictureDescriptionCustomConfig(
                batch_size = 1.0,
                scale = 0.0,
                picture_area_threshold = 1.337,
                classification_allow = [
                    'bar_chart'
                    ],
                classification_deny = [
                    'bar_chart'
                    ],
                classification_min_confidence = 1.337,
                engine_options = yfkm3_services_parse_client.models.base_vlm_engine_options.BaseVlmEngineOptions(
                    engine_type = null, ),
                model_spec = yfkm3_services_parse_client.models.vlm_model_spec.VlmModelSpec(
                    name = '', 
                    default_repo_id = '', 
                    revision = 'main', 
                    prompt = '', 
                    response_format = null, 
                    supported_engines = [
                        'transformers'
                        ], 
                    engine_overrides = {
                        'key' : yfkm3_services_parse_client.models.engine_model_config.EngineModelConfig(
                            repo_id = '', 
                            revision = '', 
                            torch_dtype = '', 
                            extra_config = { }, )
                        }, 
                    api_overrides = {
                        'key' : yfkm3_services_parse_client.models.api_model_config.ApiModelConfig(
                            params = { }, )
                        }, 
                    trust_remote_code = True, 
                    stop_strings = [
                        ''
                        ], 
                    max_new_tokens = 56, ),
                prompt = 'Describe this image in a few sentences.',
                generation_config = { }
            )
        else:
            return PictureDescriptionCustomConfig(
                engine_options = yfkm3_services_parse_client.models.base_vlm_engine_options.BaseVlmEngineOptions(
                    engine_type = null, ),
                model_spec = yfkm3_services_parse_client.models.vlm_model_spec.VlmModelSpec(
                    name = '', 
                    default_repo_id = '', 
                    revision = 'main', 
                    prompt = '', 
                    response_format = null, 
                    supported_engines = [
                        'transformers'
                        ], 
                    engine_overrides = {
                        'key' : yfkm3_services_parse_client.models.engine_model_config.EngineModelConfig(
                            repo_id = '', 
                            revision = '', 
                            torch_dtype = '', 
                            extra_config = { }, )
                        }, 
                    api_overrides = {
                        'key' : yfkm3_services_parse_client.models.api_model_config.ApiModelConfig(
                            params = { }, )
                        }, 
                    trust_remote_code = True, 
                    stop_strings = [
                        ''
                        ], 
                    max_new_tokens = 56, ),
        )
        """

    def testPictureDescriptionCustomConfig(self):
        """Test PictureDescriptionCustomConfig"""
        # inst_req_only = self.make_instance(include_optional=False)
        # inst_req_and_optional = self.make_instance(include_optional=True)

if __name__ == '__main__':
    unittest.main()
