tjdify/api/tests/integration_tests/model_runtime/__mock/openai_remote.py

from time import time
from typing import List

from openai.resources.models import Models
from openai.types.model import Model


class MockModelClass(object):
    """
        mock class for openai.models.Models
    """
    def list(
        self,
        **kwargs,
    ) -> List[Model]:
        return [
            Model(
                id='ft:gpt-3.5-turbo-0613:personal::8GYJLPDQ',
                created=int(time()),
                object='model',
                owned_by='organization:org-123',
            )
        ]