def test_litellm_openai(self):
        """Check that cost tracking works for openai models through litellm."""

        os.environ["OPENAI_API_VERSION"] = "2023-07-01-preview"
        os.environ["OPENAI_API_TYPE"] = "openai"

        from trulens.providers.litellm import LiteLLM
        from trulens.providers.litellm.endpoint import LiteLLMEndpoint
        from trulens.providers.openai import OpenAI
        from trulens.providers.openai.endpoint import OpenAIEndpoint

        # Have to delete litellm endpoint singleton as it may have been created
        # with the wrong underlying litellm provider in a prior test.
        LiteLLMEndpoint.delete_instances()
        OpenAIEndpoint.delete_instances()

        provider = LiteLLM(f"openai/{OpenAI.DEFAULT_MODEL_ENGINE}")

        self._test_llm_provider_endpoint(provider)
Check that cost tracking works for openai models through litellm.
tests/e2e/test_endpoints.py