# coding: utf-8

"""
    Mixpeek API

    This is the Mixpeek API, providing access to various endpoints for data processing and retrieval.

    The version of the OpenAPI document: 0.81.1
    Contact: info@mixpeek.com
    Generated by OpenAPI Generator (https://openapi-generator.tech)

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


import unittest

from mixpeek.models.stage_params_llm_enrich import StageParamsLlmEnrich

class TestStageParamsLlmEnrich(unittest.TestCase):
    """StageParamsLlmEnrich unit test stubs"""

    def setUp(self):
        pass

    def tearDown(self):
        pass

    def make_instance(self, include_optional) -> StageParamsLlmEnrich:
        """Test StageParamsLlmEnrich
            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 `StageParamsLlmEnrich`
        """
        model = StageParamsLlmEnrich()
        if include_optional:
            return StageParamsLlmEnrich(
                provider = 'openai',
                model_name = 'null',
                inference_name = 'null',
                prompt = 'Summarize the following content in 2-3 sentences: {{DOC.content}}',
                output_field = 'metadata.summary',
                batch_size = 1,
                var_schema = {
                    'key' : null
                    },
                max_tokens = 100,
                temperature = 0,
                api_key = '',
                use_vcache = True,
                when = {
                    'key' : null
                    }
            )
        else:
            return StageParamsLlmEnrich(
        )
        """

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

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