# File generated from our OpenAPI spec by Castiron. See CONTRIBUTING.md for details.

from __future__ import annotations

import os
from typing import Any, cast

import pytest

from openai import OpenAI, AsyncOpenAI
from tests.utils import assert_matches_type
from openai.types.admin.organization import (
    UsageCostsResponse,
    UsageImagesResponse,
    UsageEmbeddingsResponse,
    UsageCompletionsResponse,
    UsageModerationsResponse,
    UsageVectorStoresResponse,
    UsageAudioSpeechesResponse,
    UsageWebSearchCallsResponse,
    UsageFileSearchCallsResponse,
    UsageAudioTranscriptionsResponse,
    UsageCodeInterpreterSessionsResponse,
)

base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")


class TestUsage:
    parametrize = pytest.mark.parametrize("client", [False, True], indirect=True, ids=["loose", "strict"])

    @parametrize
    def test_method_audio_speeches(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.audio_speeches(
            start_time=0,
        )
        assert_matches_type(UsageAudioSpeechesResponse, usage, path=["response"])

    @parametrize
    def test_method_audio_speeches_with_all_params(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.audio_speeches(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            models=["string"],
            page="page",
            project_ids=["string"],
            user_ids=["string"],
        )
        assert_matches_type(UsageAudioSpeechesResponse, usage, path=["response"])

    @parametrize
    def test_raw_response_audio_speeches(self, client: OpenAI) -> None:
        response = client.admin.organization.usage.with_raw_response.audio_speeches(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageAudioSpeechesResponse, usage, path=["response"])

    @parametrize
    def test_streaming_response_audio_speeches(self, client: OpenAI) -> None:
        with client.admin.organization.usage.with_streaming_response.audio_speeches(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = response.parse()
            assert_matches_type(UsageAudioSpeechesResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    def test_method_audio_transcriptions(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.audio_transcriptions(
            start_time=0,
        )
        assert_matches_type(UsageAudioTranscriptionsResponse, usage, path=["response"])

    @parametrize
    def test_method_audio_transcriptions_with_all_params(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.audio_transcriptions(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            models=["string"],
            page="page",
            project_ids=["string"],
            user_ids=["string"],
        )
        assert_matches_type(UsageAudioTranscriptionsResponse, usage, path=["response"])

    @parametrize
    def test_raw_response_audio_transcriptions(self, client: OpenAI) -> None:
        response = client.admin.organization.usage.with_raw_response.audio_transcriptions(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageAudioTranscriptionsResponse, usage, path=["response"])

    @parametrize
    def test_streaming_response_audio_transcriptions(self, client: OpenAI) -> None:
        with client.admin.organization.usage.with_streaming_response.audio_transcriptions(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = response.parse()
            assert_matches_type(UsageAudioTranscriptionsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    def test_method_code_interpreter_sessions(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.code_interpreter_sessions(
            start_time=0,
        )
        assert_matches_type(UsageCodeInterpreterSessionsResponse, usage, path=["response"])

    @parametrize
    def test_method_code_interpreter_sessions_with_all_params(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.code_interpreter_sessions(
            start_time=0,
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            page="page",
            project_ids=["string"],
        )
        assert_matches_type(UsageCodeInterpreterSessionsResponse, usage, path=["response"])

    @parametrize
    def test_raw_response_code_interpreter_sessions(self, client: OpenAI) -> None:
        response = client.admin.organization.usage.with_raw_response.code_interpreter_sessions(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageCodeInterpreterSessionsResponse, usage, path=["response"])

    @parametrize
    def test_streaming_response_code_interpreter_sessions(self, client: OpenAI) -> None:
        with client.admin.organization.usage.with_streaming_response.code_interpreter_sessions(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = response.parse()
            assert_matches_type(UsageCodeInterpreterSessionsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    def test_method_completions(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.completions(
            start_time=0,
        )
        assert_matches_type(UsageCompletionsResponse, usage, path=["response"])

    @parametrize
    def test_method_completions_with_all_params(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.completions(
            start_time=0,
            api_key_ids=["string"],
            batch=True,
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            models=["string"],
            page="page",
            project_ids=["string"],
            user_ids=["string"],
        )
        assert_matches_type(UsageCompletionsResponse, usage, path=["response"])

    @parametrize
    def test_raw_response_completions(self, client: OpenAI) -> None:
        response = client.admin.organization.usage.with_raw_response.completions(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageCompletionsResponse, usage, path=["response"])

    @parametrize
    def test_streaming_response_completions(self, client: OpenAI) -> None:
        with client.admin.organization.usage.with_streaming_response.completions(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = response.parse()
            assert_matches_type(UsageCompletionsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    def test_method_costs(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.costs(
            start_time=0,
        )
        assert_matches_type(UsageCostsResponse, usage, path=["response"])

    @parametrize
    def test_method_costs_with_all_params(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.costs(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1d",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            page="page",
            project_ids=["string"],
        )
        assert_matches_type(UsageCostsResponse, usage, path=["response"])

    @parametrize
    def test_raw_response_costs(self, client: OpenAI) -> None:
        response = client.admin.organization.usage.with_raw_response.costs(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageCostsResponse, usage, path=["response"])

    @parametrize
    def test_streaming_response_costs(self, client: OpenAI) -> None:
        with client.admin.organization.usage.with_streaming_response.costs(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = response.parse()
            assert_matches_type(UsageCostsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    def test_method_embeddings(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.embeddings(
            start_time=0,
        )
        assert_matches_type(UsageEmbeddingsResponse, usage, path=["response"])

    @parametrize
    def test_method_embeddings_with_all_params(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.embeddings(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            models=["string"],
            page="page",
            project_ids=["string"],
            user_ids=["string"],
        )
        assert_matches_type(UsageEmbeddingsResponse, usage, path=["response"])

    @parametrize
    def test_raw_response_embeddings(self, client: OpenAI) -> None:
        response = client.admin.organization.usage.with_raw_response.embeddings(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageEmbeddingsResponse, usage, path=["response"])

    @parametrize
    def test_streaming_response_embeddings(self, client: OpenAI) -> None:
        with client.admin.organization.usage.with_streaming_response.embeddings(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = response.parse()
            assert_matches_type(UsageEmbeddingsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    def test_method_file_search_calls(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.file_search_calls(
            start_time=0,
        )
        assert_matches_type(UsageFileSearchCallsResponse, usage, path=["response"])

    @parametrize
    def test_method_file_search_calls_with_all_params(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.file_search_calls(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            page="page",
            project_ids=["string"],
            user_ids=["string"],
            vector_store_ids=["string"],
        )
        assert_matches_type(UsageFileSearchCallsResponse, usage, path=["response"])

    @parametrize
    def test_raw_response_file_search_calls(self, client: OpenAI) -> None:
        response = client.admin.organization.usage.with_raw_response.file_search_calls(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageFileSearchCallsResponse, usage, path=["response"])

    @parametrize
    def test_streaming_response_file_search_calls(self, client: OpenAI) -> None:
        with client.admin.organization.usage.with_streaming_response.file_search_calls(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = response.parse()
            assert_matches_type(UsageFileSearchCallsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    def test_method_images(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.images(
            start_time=0,
        )
        assert_matches_type(UsageImagesResponse, usage, path=["response"])

    @parametrize
    def test_method_images_with_all_params(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.images(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            models=["string"],
            page="page",
            project_ids=["string"],
            sizes=["256x256"],
            sources=["image.generation"],
            user_ids=["string"],
        )
        assert_matches_type(UsageImagesResponse, usage, path=["response"])

    @parametrize
    def test_raw_response_images(self, client: OpenAI) -> None:
        response = client.admin.organization.usage.with_raw_response.images(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageImagesResponse, usage, path=["response"])

    @parametrize
    def test_streaming_response_images(self, client: OpenAI) -> None:
        with client.admin.organization.usage.with_streaming_response.images(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = response.parse()
            assert_matches_type(UsageImagesResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    def test_method_moderations(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.moderations(
            start_time=0,
        )
        assert_matches_type(UsageModerationsResponse, usage, path=["response"])

    @parametrize
    def test_method_moderations_with_all_params(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.moderations(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            models=["string"],
            page="page",
            project_ids=["string"],
            user_ids=["string"],
        )
        assert_matches_type(UsageModerationsResponse, usage, path=["response"])

    @parametrize
    def test_raw_response_moderations(self, client: OpenAI) -> None:
        response = client.admin.organization.usage.with_raw_response.moderations(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageModerationsResponse, usage, path=["response"])

    @parametrize
    def test_streaming_response_moderations(self, client: OpenAI) -> None:
        with client.admin.organization.usage.with_streaming_response.moderations(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = response.parse()
            assert_matches_type(UsageModerationsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    def test_method_vector_stores(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.vector_stores(
            start_time=0,
        )
        assert_matches_type(UsageVectorStoresResponse, usage, path=["response"])

    @parametrize
    def test_method_vector_stores_with_all_params(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.vector_stores(
            start_time=0,
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            page="page",
            project_ids=["string"],
        )
        assert_matches_type(UsageVectorStoresResponse, usage, path=["response"])

    @parametrize
    def test_raw_response_vector_stores(self, client: OpenAI) -> None:
        response = client.admin.organization.usage.with_raw_response.vector_stores(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageVectorStoresResponse, usage, path=["response"])

    @parametrize
    def test_streaming_response_vector_stores(self, client: OpenAI) -> None:
        with client.admin.organization.usage.with_streaming_response.vector_stores(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = response.parse()
            assert_matches_type(UsageVectorStoresResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    def test_method_web_search_calls(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.web_search_calls(
            start_time=0,
        )
        assert_matches_type(UsageWebSearchCallsResponse, usage, path=["response"])

    @parametrize
    def test_method_web_search_calls_with_all_params(self, client: OpenAI) -> None:
        usage = client.admin.organization.usage.web_search_calls(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1m",
            context_levels=["low"],
            end_time=0,
            group_by=["project_id"],
            limit=0,
            models=["string"],
            page="page",
            project_ids=["string"],
            user_ids=["string"],
        )
        assert_matches_type(UsageWebSearchCallsResponse, usage, path=["response"])

    @parametrize
    def test_raw_response_web_search_calls(self, client: OpenAI) -> None:
        response = client.admin.organization.usage.with_raw_response.web_search_calls(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageWebSearchCallsResponse, usage, path=["response"])

    @parametrize
    def test_streaming_response_web_search_calls(self, client: OpenAI) -> None:
        with client.admin.organization.usage.with_streaming_response.web_search_calls(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = response.parse()
            assert_matches_type(UsageWebSearchCallsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True


class TestAsyncUsage:
    parametrize = pytest.mark.parametrize(
        "async_client", [False, True, {"http_client": "aiohttp"}], indirect=True, ids=["loose", "strict", "aiohttp"]
    )

    @parametrize
    async def test_method_audio_speeches(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.audio_speeches(
            start_time=0,
        )
        assert_matches_type(UsageAudioSpeechesResponse, usage, path=["response"])

    @parametrize
    async def test_method_audio_speeches_with_all_params(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.audio_speeches(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            models=["string"],
            page="page",
            project_ids=["string"],
            user_ids=["string"],
        )
        assert_matches_type(UsageAudioSpeechesResponse, usage, path=["response"])

    @parametrize
    async def test_raw_response_audio_speeches(self, async_client: AsyncOpenAI) -> None:
        response = await async_client.admin.organization.usage.with_raw_response.audio_speeches(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageAudioSpeechesResponse, usage, path=["response"])

    @parametrize
    async def test_streaming_response_audio_speeches(self, async_client: AsyncOpenAI) -> None:
        async with async_client.admin.organization.usage.with_streaming_response.audio_speeches(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = await response.parse()
            assert_matches_type(UsageAudioSpeechesResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    async def test_method_audio_transcriptions(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.audio_transcriptions(
            start_time=0,
        )
        assert_matches_type(UsageAudioTranscriptionsResponse, usage, path=["response"])

    @parametrize
    async def test_method_audio_transcriptions_with_all_params(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.audio_transcriptions(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            models=["string"],
            page="page",
            project_ids=["string"],
            user_ids=["string"],
        )
        assert_matches_type(UsageAudioTranscriptionsResponse, usage, path=["response"])

    @parametrize
    async def test_raw_response_audio_transcriptions(self, async_client: AsyncOpenAI) -> None:
        response = await async_client.admin.organization.usage.with_raw_response.audio_transcriptions(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageAudioTranscriptionsResponse, usage, path=["response"])

    @parametrize
    async def test_streaming_response_audio_transcriptions(self, async_client: AsyncOpenAI) -> None:
        async with async_client.admin.organization.usage.with_streaming_response.audio_transcriptions(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = await response.parse()
            assert_matches_type(UsageAudioTranscriptionsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    async def test_method_code_interpreter_sessions(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.code_interpreter_sessions(
            start_time=0,
        )
        assert_matches_type(UsageCodeInterpreterSessionsResponse, usage, path=["response"])

    @parametrize
    async def test_method_code_interpreter_sessions_with_all_params(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.code_interpreter_sessions(
            start_time=0,
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            page="page",
            project_ids=["string"],
        )
        assert_matches_type(UsageCodeInterpreterSessionsResponse, usage, path=["response"])

    @parametrize
    async def test_raw_response_code_interpreter_sessions(self, async_client: AsyncOpenAI) -> None:
        response = await async_client.admin.organization.usage.with_raw_response.code_interpreter_sessions(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageCodeInterpreterSessionsResponse, usage, path=["response"])

    @parametrize
    async def test_streaming_response_code_interpreter_sessions(self, async_client: AsyncOpenAI) -> None:
        async with async_client.admin.organization.usage.with_streaming_response.code_interpreter_sessions(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = await response.parse()
            assert_matches_type(UsageCodeInterpreterSessionsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    async def test_method_completions(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.completions(
            start_time=0,
        )
        assert_matches_type(UsageCompletionsResponse, usage, path=["response"])

    @parametrize
    async def test_method_completions_with_all_params(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.completions(
            start_time=0,
            api_key_ids=["string"],
            batch=True,
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            models=["string"],
            page="page",
            project_ids=["string"],
            user_ids=["string"],
        )
        assert_matches_type(UsageCompletionsResponse, usage, path=["response"])

    @parametrize
    async def test_raw_response_completions(self, async_client: AsyncOpenAI) -> None:
        response = await async_client.admin.organization.usage.with_raw_response.completions(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageCompletionsResponse, usage, path=["response"])

    @parametrize
    async def test_streaming_response_completions(self, async_client: AsyncOpenAI) -> None:
        async with async_client.admin.organization.usage.with_streaming_response.completions(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = await response.parse()
            assert_matches_type(UsageCompletionsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    async def test_method_costs(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.costs(
            start_time=0,
        )
        assert_matches_type(UsageCostsResponse, usage, path=["response"])

    @parametrize
    async def test_method_costs_with_all_params(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.costs(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1d",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            page="page",
            project_ids=["string"],
        )
        assert_matches_type(UsageCostsResponse, usage, path=["response"])

    @parametrize
    async def test_raw_response_costs(self, async_client: AsyncOpenAI) -> None:
        response = await async_client.admin.organization.usage.with_raw_response.costs(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageCostsResponse, usage, path=["response"])

    @parametrize
    async def test_streaming_response_costs(self, async_client: AsyncOpenAI) -> None:
        async with async_client.admin.organization.usage.with_streaming_response.costs(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = await response.parse()
            assert_matches_type(UsageCostsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    async def test_method_embeddings(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.embeddings(
            start_time=0,
        )
        assert_matches_type(UsageEmbeddingsResponse, usage, path=["response"])

    @parametrize
    async def test_method_embeddings_with_all_params(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.embeddings(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            models=["string"],
            page="page",
            project_ids=["string"],
            user_ids=["string"],
        )
        assert_matches_type(UsageEmbeddingsResponse, usage, path=["response"])

    @parametrize
    async def test_raw_response_embeddings(self, async_client: AsyncOpenAI) -> None:
        response = await async_client.admin.organization.usage.with_raw_response.embeddings(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageEmbeddingsResponse, usage, path=["response"])

    @parametrize
    async def test_streaming_response_embeddings(self, async_client: AsyncOpenAI) -> None:
        async with async_client.admin.organization.usage.with_streaming_response.embeddings(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = await response.parse()
            assert_matches_type(UsageEmbeddingsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    async def test_method_file_search_calls(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.file_search_calls(
            start_time=0,
        )
        assert_matches_type(UsageFileSearchCallsResponse, usage, path=["response"])

    @parametrize
    async def test_method_file_search_calls_with_all_params(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.file_search_calls(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            page="page",
            project_ids=["string"],
            user_ids=["string"],
            vector_store_ids=["string"],
        )
        assert_matches_type(UsageFileSearchCallsResponse, usage, path=["response"])

    @parametrize
    async def test_raw_response_file_search_calls(self, async_client: AsyncOpenAI) -> None:
        response = await async_client.admin.organization.usage.with_raw_response.file_search_calls(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageFileSearchCallsResponse, usage, path=["response"])

    @parametrize
    async def test_streaming_response_file_search_calls(self, async_client: AsyncOpenAI) -> None:
        async with async_client.admin.organization.usage.with_streaming_response.file_search_calls(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = await response.parse()
            assert_matches_type(UsageFileSearchCallsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    async def test_method_images(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.images(
            start_time=0,
        )
        assert_matches_type(UsageImagesResponse, usage, path=["response"])

    @parametrize
    async def test_method_images_with_all_params(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.images(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            models=["string"],
            page="page",
            project_ids=["string"],
            sizes=["256x256"],
            sources=["image.generation"],
            user_ids=["string"],
        )
        assert_matches_type(UsageImagesResponse, usage, path=["response"])

    @parametrize
    async def test_raw_response_images(self, async_client: AsyncOpenAI) -> None:
        response = await async_client.admin.organization.usage.with_raw_response.images(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageImagesResponse, usage, path=["response"])

    @parametrize
    async def test_streaming_response_images(self, async_client: AsyncOpenAI) -> None:
        async with async_client.admin.organization.usage.with_streaming_response.images(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = await response.parse()
            assert_matches_type(UsageImagesResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    async def test_method_moderations(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.moderations(
            start_time=0,
        )
        assert_matches_type(UsageModerationsResponse, usage, path=["response"])

    @parametrize
    async def test_method_moderations_with_all_params(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.moderations(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            models=["string"],
            page="page",
            project_ids=["string"],
            user_ids=["string"],
        )
        assert_matches_type(UsageModerationsResponse, usage, path=["response"])

    @parametrize
    async def test_raw_response_moderations(self, async_client: AsyncOpenAI) -> None:
        response = await async_client.admin.organization.usage.with_raw_response.moderations(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageModerationsResponse, usage, path=["response"])

    @parametrize
    async def test_streaming_response_moderations(self, async_client: AsyncOpenAI) -> None:
        async with async_client.admin.organization.usage.with_streaming_response.moderations(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = await response.parse()
            assert_matches_type(UsageModerationsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    async def test_method_vector_stores(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.vector_stores(
            start_time=0,
        )
        assert_matches_type(UsageVectorStoresResponse, usage, path=["response"])

    @parametrize
    async def test_method_vector_stores_with_all_params(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.vector_stores(
            start_time=0,
            bucket_width="1m",
            end_time=0,
            group_by=["project_id"],
            limit=0,
            page="page",
            project_ids=["string"],
        )
        assert_matches_type(UsageVectorStoresResponse, usage, path=["response"])

    @parametrize
    async def test_raw_response_vector_stores(self, async_client: AsyncOpenAI) -> None:
        response = await async_client.admin.organization.usage.with_raw_response.vector_stores(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageVectorStoresResponse, usage, path=["response"])

    @parametrize
    async def test_streaming_response_vector_stores(self, async_client: AsyncOpenAI) -> None:
        async with async_client.admin.organization.usage.with_streaming_response.vector_stores(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = await response.parse()
            assert_matches_type(UsageVectorStoresResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True

    @parametrize
    async def test_method_web_search_calls(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.web_search_calls(
            start_time=0,
        )
        assert_matches_type(UsageWebSearchCallsResponse, usage, path=["response"])

    @parametrize
    async def test_method_web_search_calls_with_all_params(self, async_client: AsyncOpenAI) -> None:
        usage = await async_client.admin.organization.usage.web_search_calls(
            start_time=0,
            api_key_ids=["string"],
            bucket_width="1m",
            context_levels=["low"],
            end_time=0,
            group_by=["project_id"],
            limit=0,
            models=["string"],
            page="page",
            project_ids=["string"],
            user_ids=["string"],
        )
        assert_matches_type(UsageWebSearchCallsResponse, usage, path=["response"])

    @parametrize
    async def test_raw_response_web_search_calls(self, async_client: AsyncOpenAI) -> None:
        response = await async_client.admin.organization.usage.with_raw_response.web_search_calls(
            start_time=0,
        )

        assert response.is_closed is True
        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
        usage = response.parse()
        assert_matches_type(UsageWebSearchCallsResponse, usage, path=["response"])

    @parametrize
    async def test_streaming_response_web_search_calls(self, async_client: AsyncOpenAI) -> None:
        async with async_client.admin.organization.usage.with_streaming_response.web_search_calls(
            start_time=0,
        ) as response:
            assert not response.is_closed
            assert response.http_request.headers.get("X-Stainless-Lang") == "python"

            usage = await response.parse()
            assert_matches_type(UsageWebSearchCallsResponse, usage, path=["response"])

        assert cast(Any, response.is_closed) is True
