def test_cortex(self):
        """Check that cost (token) tracking works for Cortex LLM Functions"""
        from trulens.providers.cortex import Cortex

        snowflake_connection_parameters = {
            "account": os.environ["SNOWFLAKE_ACCOUNT"],
            "user": os.environ["SNOWFLAKE_USER"],
            "password": os.environ["SNOWFLAKE_USER_PASSWORD"],
            "database": os.environ["SNOWFLAKE_DATABASE"],
            "schema": os.environ["SNOWFLAKE_SCHEMA"],
            "warehouse": os.environ["SNOWFLAKE_WAREHOUSE"],
        }

        snowpark_session = Session.builder.configs(
            snowflake_connection_parameters
        ).create()
        provider = Cortex(
            snowpark_session=snowpark_session,
            model_engine="llama3.2-3b",
        )

        self._test_llm_provider_endpoint(provider)
Check that cost (token) tracking works for Cortex LLM Functions
tests/e2e/test_endpoints.py