# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details. from __future__ import annotations import os from typing import Any, cast import pytest from hanzoai import Hanzo, AsyncHanzo from tests.utils import assert_matches_type from hanzoai._utils import parse_datetime base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010") class TestUpdate: parametrize = pytest.mark.parametrize("client", [False, True], indirect=True, ids=["loose", "strict"]) @parametrize def test_method_full(self, client: Hanzo) -> None: update = client.model.update.full() assert_matches_type(object, update, path=["response"]) @parametrize def test_method_full_with_all_params(self, client: Hanzo) -> None: update = client.model.update.full( hanzo_params={ "api_base": "api_base", "api_key": "api_key", "api_version": "api_version", "aws_access_key_id": "aws_access_key_id", "aws_region_name": "aws_region_name", "aws_secret_access_key": "aws_secret_access_key", "budget_duration": "budget_duration", "configurable_clientside_auth_params": ["string"], "custom_llm_provider": "custom_llm_provider", "input_cost_per_second": 0, "input_cost_per_token": 0, "hanzo_trace_id": "hanzo_trace_id", "max_budget": 0, "max_file_size_mb": 0, "max_retries": 0, "merge_reasoning_content_in_choices": True, "model": "model", "model_info": {}, "organization": "organization", "output_cost_per_second": 0, "output_cost_per_token": 0, "region_name": "region_name", "rpm": 0, "stream_timeout": 0, "timeout": 0, "tpm": 0, "use_in_pass_through": True, "vertex_credentials": "string", "vertex_location": "vertex_location", "vertex_project": "vertex_project", "watsonx_region_name": "watsonx_region_name", }, model_info={ "id": "id", "base_model": "base_model", "created_at": parse_datetime("2019-12-27T18:11:19.117Z"), "created_by": "created_by", "db_model": True, "team_id": "team_id", "team_public_model_name": "team_public_model_name", "tier": "free", "updated_at": parse_datetime("2019-12-27T18:11:19.117Z"), "updated_by": "updated_by", }, model_name="model_name", ) assert_matches_type(object, update, path=["response"]) @parametrize def test_raw_response_full(self, client: Hanzo) -> None: response = client.model.update.with_raw_response.full() assert response.is_closed is True assert response.http_request.headers.get("X-Stainless-Lang") == "python" update = response.parse() assert_matches_type(object, update, path=["response"]) @parametrize def test_streaming_response_full(self, client: Hanzo) -> None: with client.model.update.with_streaming_response.full() as response: assert not response.is_closed assert response.http_request.headers.get("X-Stainless-Lang") == "python" update = response.parse() assert_matches_type(object, update, path=["response"]) assert cast(Any, response.is_closed) is True @parametrize def test_method_partial(self, client: Hanzo) -> None: update = client.model.update.partial( model_id="model_id", ) assert_matches_type(object, update, path=["response"]) @parametrize def test_method_partial_with_all_params(self, client: Hanzo) -> None: update = client.model.update.partial( model_id="model_id", hanzo_params={ "api_base": "api_base", "api_key": "api_key", "api_version": "api_version", "aws_access_key_id": "aws_access_key_id", "aws_region_name": "aws_region_name", "aws_secret_access_key": "aws_secret_access_key", "budget_duration": "budget_duration", "configurable_clientside_auth_params": ["string"], "custom_llm_provider": "custom_llm_provider", "input_cost_per_second": 0, "input_cost_per_token": 0, "hanzo_trace_id": "hanzo_trace_id", "max_budget": 0, "max_file_size_mb": 0, "max_retries": 0, "merge_reasoning_content_in_choices": True, "model": "model", "model_info": {}, "organization": "organization", "output_cost_per_second": 0, "output_cost_per_token": 0, "region_name": "region_name", "rpm": 0, "stream_timeout": 0, "timeout": 0, "tpm": 0, "use_in_pass_through": True, "vertex_credentials": "string", "vertex_location": "vertex_location", "vertex_project": "vertex_project", "watsonx_region_name": "watsonx_region_name", }, model_info={ "id": "id", "base_model": "base_model", "created_at": parse_datetime("2019-12-27T18:11:19.117Z"), "created_by": "created_by", "db_model": True, "team_id": "team_id", "team_public_model_name": "team_public_model_name", "tier": "free", "updated_at": parse_datetime("2019-12-27T18:11:19.117Z"), "updated_by": "updated_by", }, model_name="model_name", ) assert_matches_type(object, update, path=["response"]) @parametrize def test_raw_response_partial(self, client: Hanzo) -> None: response = client.model.update.with_raw_response.partial( model_id="model_id", ) assert response.is_closed is True assert response.http_request.headers.get("X-Stainless-Lang") == "python" update = response.parse() assert_matches_type(object, update, path=["response"]) @parametrize def test_streaming_response_partial(self, client: Hanzo) -> None: with client.model.update.with_streaming_response.partial( model_id="model_id", ) as response: assert not response.is_closed assert response.http_request.headers.get("X-Stainless-Lang") == "python" update = response.parse() assert_matches_type(object, update, path=["response"]) assert cast(Any, response.is_closed) is True @parametrize def test_path_params_partial(self, client: Hanzo) -> None: with pytest.raises( ValueError, match=r"Expected a non-empty value for `model_id` but received ''", ): client.model.update.with_raw_response.partial( model_id="", ) class TestAsyncUpdate: parametrize = pytest.mark.parametrize("async_client", [False, True], indirect=True, ids=["loose", "strict"]) @parametrize async def test_method_full(self, async_client: AsyncHanzo) -> None: update = await async_client.model.update.full() assert_matches_type(object, update, path=["response"]) @parametrize async def test_method_full_with_all_params(self, async_client: AsyncHanzo) -> None: update = await async_client.model.update.full( hanzo_params={ "api_base": "api_base", "api_key": "api_key", "api_version": "api_version", "aws_access_key_id": "aws_access_key_id", "aws_region_name": "aws_region_name", "aws_secret_access_key": "aws_secret_access_key", "budget_duration": "budget_duration", "configurable_clientside_auth_params": ["string"], "custom_llm_provider": "custom_llm_provider", "input_cost_per_second": 0, "input_cost_per_token": 0, "hanzo_trace_id": "hanzo_trace_id", "max_budget": 0, "max_file_size_mb": 0, "max_retries": 0, "merge_reasoning_content_in_choices": True, "model": "model", "model_info": {}, "organization": "organization", "output_cost_per_second": 0, "output_cost_per_token": 0, "region_name": "region_name", "rpm": 0, "stream_timeout": 0, "timeout": 0, "tpm": 0, "use_in_pass_through": True, "vertex_credentials": "string", "vertex_location": "vertex_location", "vertex_project": "vertex_project", "watsonx_region_name": "watsonx_region_name", }, model_info={ "id": "id", "base_model": "base_model", "created_at": parse_datetime("2019-12-27T18:11:19.117Z"), "created_by": "created_by", "db_model": True, "team_id": "team_id", "team_public_model_name": "team_public_model_name", "tier": "free", "updated_at": parse_datetime("2019-12-27T18:11:19.117Z"), "updated_by": "updated_by", }, model_name="model_name", ) assert_matches_type(object, update, path=["response"]) @parametrize async def test_raw_response_full(self, async_client: AsyncHanzo) -> None: response = await async_client.model.update.with_raw_response.full() assert response.is_closed is True assert response.http_request.headers.get("X-Stainless-Lang") == "python" update = await response.parse() assert_matches_type(object, update, path=["response"]) @parametrize async def test_streaming_response_full(self, async_client: AsyncHanzo) -> None: async with async_client.model.update.with_streaming_response.full() as response: assert not response.is_closed assert response.http_request.headers.get("X-Stainless-Lang") == "python" update = await response.parse() assert_matches_type(object, update, path=["response"]) assert cast(Any, response.is_closed) is True @parametrize async def test_method_partial(self, async_client: AsyncHanzo) -> None: update = await async_client.model.update.partial( model_id="model_id", ) assert_matches_type(object, update, path=["response"]) @parametrize async def test_method_partial_with_all_params(self, async_client: AsyncHanzo) -> None: update = await async_client.model.update.partial( model_id="model_id", hanzo_params={ "api_base": "api_base", "api_key": "api_key", "api_version": "api_version", "aws_access_key_id": "aws_access_key_id", "aws_region_name": "aws_region_name", "aws_secret_access_key": "aws_secret_access_key", "budget_duration": "budget_duration", "configurable_clientside_auth_params": ["string"], "custom_llm_provider": "custom_llm_provider", "input_cost_per_second": 0, "input_cost_per_token": 0, "hanzo_trace_id": "hanzo_trace_id", "max_budget": 0, "max_file_size_mb": 0, "max_retries": 0, "merge_reasoning_content_in_choices": True, "model": "model", "model_info": {}, "organization": "organization", "output_cost_per_second": 0, "output_cost_per_token": 0, "region_name": "region_name", "rpm": 0, "stream_timeout": 0, "timeout": 0, "tpm": 0, "use_in_pass_through": True, "vertex_credentials": "string", "vertex_location": "vertex_location", "vertex_project": "vertex_project", "watsonx_region_name": "watsonx_region_name", }, model_info={ "id": "id", "base_model": "base_model", "created_at": parse_datetime("2019-12-27T18:11:19.117Z"), "created_by": "created_by", "db_model": True, "team_id": "team_id", "team_public_model_name": "team_public_model_name", "tier": "free", "updated_at": parse_datetime("2019-12-27T18:11:19.117Z"), "updated_by": "updated_by", }, model_name="model_name", ) assert_matches_type(object, update, path=["response"]) @parametrize async def test_raw_response_partial(self, async_client: AsyncHanzo) -> None: response = await async_client.model.update.with_raw_response.partial( model_id="model_id", ) assert response.is_closed is True assert response.http_request.headers.get("X-Stainless-Lang") == "python" update = await response.parse() assert_matches_type(object, update, path=["response"]) @parametrize async def test_streaming_response_partial(self, async_client: AsyncHanzo) -> None: async with async_client.model.update.with_streaming_response.partial( model_id="model_id", ) as response: assert not response.is_closed assert response.http_request.headers.get("X-Stainless-Lang") == "python" update = await response.parse() assert_matches_type(object, update, path=["response"]) assert cast(Any, response.is_closed) is True @parametrize async def test_path_params_partial(self, async_client: AsyncHanzo) -> None: with pytest.raises( ValueError, match=r"Expected a non-empty value for `model_id` but received ''", ): await async_client.model.update.with_raw_response.partial( model_id="", )