From 066140d7122a73b49936be6d09fb0d21ead58fb9 Mon Sep 17 00:00:00 2001 From: stainless-bot Date: Thu, 4 Jan 2024 16:24:30 +0000 Subject: [PATCH] feat(api): OpenAPI spec update --- .stats.yml | 2 +- api.md | 2 - src/dataherald/resources/golden_sqls.py | 79 +------------- src/dataherald/types/__init__.py | 2 - .../types/golden_sql_user_upload_params.py | 22 ---- .../types/golden_sql_user_upload_response.py | 9 -- tests/api_resources/test_golden_sqls.py | 101 +----------------- 7 files changed, 4 insertions(+), 213 deletions(-) delete mode 100644 src/dataherald/types/golden_sql_user_upload_params.py delete mode 100644 src/dataherald/types/golden_sql_user_upload_response.py diff --git a/.stats.yml b/.stats.yml index 37da431..1fd2991 100644 --- a/.stats.yml +++ b/.stats.yml @@ -1 +1 @@ -configured_endpoints: 42 +configured_endpoints: 41 diff --git a/api.md b/api.md index aa5f214..7769312 100644 --- a/api.md +++ b/api.md @@ -58,7 +58,6 @@ from dataherald.types import ( AdminConsoleGoldenSqlResponse, GoldenSqlListResponse, GoldenSqlDeleteResponse, - GoldenSqlUserUploadResponse, ) ``` @@ -67,7 +66,6 @@ Methods: - client.golden_sqls.retrieve(id) -> GoldenSqlResponse - client.golden_sqls.list(\*\*params) -> GoldenSqlListResponse - client.golden_sqls.delete(id) -> object -- client.golden_sqls.user_upload(\*\*params) -> GoldenSqlUserUploadResponse # Instructions diff --git a/src/dataherald/resources/golden_sqls.py b/src/dataherald/resources/golden_sqls.py index f7d79f2..e89379a 100644 --- a/src/dataherald/resources/golden_sqls.py +++ b/src/dataherald/resources/golden_sqls.py @@ -2,16 +2,11 @@ from __future__ import annotations -from typing import TYPE_CHECKING, List +from typing import TYPE_CHECKING import httpx -from ..types import ( - GoldenSqlListResponse, - GoldenSqlUserUploadResponse, - golden_sql_list_params, - golden_sql_user_upload_params, -) +from ..types import GoldenSqlListResponse, golden_sql_list_params from .._types import ( NOT_GIVEN, Body, @@ -149,38 +144,6 @@ def delete( cast_to=UnknownResponse, ) - def user_upload( - self, - *, - body: List[golden_sql_user_upload_params.Body], - # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. - # The extra values given here take precedence over values defined on the client or passed to this method. - extra_headers: Headers | None = None, - extra_query: Query | None = None, - extra_body: Body | None = None, - timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN, - ) -> GoldenSqlUserUploadResponse: - """ - Api Add User Upload Golden Sql - - Args: - extra_headers: Send extra headers - - extra_query: Add additional query parameters to the request - - extra_body: Add additional JSON properties to the request - - timeout: Override the client-level default timeout for this request, in seconds - """ - return self._post( - "/api/golden-sqls/user-upload", - body=maybe_transform(body, golden_sql_user_upload_params.GoldenSqlUserUploadParams), - options=make_request_options( - extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout - ), - cast_to=GoldenSqlUserUploadResponse, - ) - class AsyncGoldenSqls(AsyncAPIResource): with_raw_response: AsyncGoldenSqlsWithRawResponse @@ -297,38 +260,6 @@ async def delete( cast_to=UnknownResponse, ) - async def user_upload( - self, - *, - body: List[golden_sql_user_upload_params.Body], - # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. - # The extra values given here take precedence over values defined on the client or passed to this method. - extra_headers: Headers | None = None, - extra_query: Query | None = None, - extra_body: Body | None = None, - timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN, - ) -> GoldenSqlUserUploadResponse: - """ - Api Add User Upload Golden Sql - - Args: - extra_headers: Send extra headers - - extra_query: Add additional query parameters to the request - - extra_body: Add additional JSON properties to the request - - timeout: Override the client-level default timeout for this request, in seconds - """ - return await self._post( - "/api/golden-sqls/user-upload", - body=maybe_transform(body, golden_sql_user_upload_params.GoldenSqlUserUploadParams), - options=make_request_options( - extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout - ), - cast_to=GoldenSqlUserUploadResponse, - ) - class GoldenSqlsWithRawResponse: def __init__(self, golden_sqls: GoldenSqls) -> None: @@ -341,9 +272,6 @@ def __init__(self, golden_sqls: GoldenSqls) -> None: self.delete = to_raw_response_wrapper( golden_sqls.delete, ) - self.user_upload = to_raw_response_wrapper( - golden_sqls.user_upload, - ) class AsyncGoldenSqlsWithRawResponse: @@ -357,6 +285,3 @@ def __init__(self, golden_sqls: AsyncGoldenSqls) -> None: self.delete = async_to_raw_response_wrapper( golden_sqls.delete, ) - self.user_upload = async_to_raw_response_wrapper( - golden_sqls.user_upload, - ) diff --git a/src/dataherald/types/__init__.py b/src/dataherald/types/__init__.py index 33e6887..0308b86 100644 --- a/src/dataherald/types/__init__.py +++ b/src/dataherald/types/__init__.py @@ -32,10 +32,8 @@ from .nl_generation_list_response import NlGenerationListResponse as NlGenerationListResponse from .sql_generation_create_params import SqlGenerationCreateParams as SqlGenerationCreateParams from .sql_generation_list_response import SqlGenerationListResponse as SqlGenerationListResponse -from .golden_sql_user_upload_params import GoldenSqlUserUploadParams as GoldenSqlUserUploadParams from .sql_generation_execute_params import SqlGenerationExecuteParams as SqlGenerationExecuteParams from .table_description_list_params import TableDescriptionListParams as TableDescriptionListParams -from .golden_sql_user_upload_response import GoldenSqlUserUploadResponse as GoldenSqlUserUploadResponse from .sql_generation_execute_response import SqlGenerationExecuteResponse as SqlGenerationExecuteResponse from .table_description_list_response import TableDescriptionListResponse as TableDescriptionListResponse from .table_description_update_params import TableDescriptionUpdateParams as TableDescriptionUpdateParams diff --git a/src/dataherald/types/golden_sql_user_upload_params.py b/src/dataherald/types/golden_sql_user_upload_params.py deleted file mode 100644 index 544d1f5..0000000 --- a/src/dataherald/types/golden_sql_user_upload_params.py +++ /dev/null @@ -1,22 +0,0 @@ -# File generated from our OpenAPI spec by Stainless. - -from __future__ import annotations - -from typing import List -from typing_extensions import Required, TypedDict - -__all__ = ["GoldenSqlUserUploadParams", "Body"] - - -class GoldenSqlUserUploadParams(TypedDict, total=False): - body: Required[List[Body]] - - -class Body(TypedDict, total=False): - db_connection_id: Required[str] - - prompt_text: Required[str] - - sql: Required[str] - - metadata: object diff --git a/src/dataherald/types/golden_sql_user_upload_response.py b/src/dataherald/types/golden_sql_user_upload_response.py deleted file mode 100644 index 810b2cb..0000000 --- a/src/dataherald/types/golden_sql_user_upload_response.py +++ /dev/null @@ -1,9 +0,0 @@ -# File generated from our OpenAPI spec by Stainless. - -from typing import List - -from .shared import GoldenSqlResponse - -__all__ = ["GoldenSqlUserUploadResponse"] - -GoldenSqlUserUploadResponse = List[GoldenSqlResponse] diff --git a/tests/api_resources/test_golden_sqls.py b/tests/api_resources/test_golden_sqls.py index 49c6915..35dce7a 100644 --- a/tests/api_resources/test_golden_sqls.py +++ b/tests/api_resources/test_golden_sqls.py @@ -8,10 +8,7 @@ from dataherald import Dataherald, AsyncDataherald from tests.utils import assert_matches_type -from dataherald.types import ( - GoldenSqlListResponse, - GoldenSqlUserUploadResponse, -) +from dataherald.types import GoldenSqlListResponse from dataherald._client import Dataherald, AsyncDataherald from dataherald.types.shared import GoldenSqlResponse @@ -78,54 +75,6 @@ def test_raw_response_delete(self, client: Dataherald) -> None: golden_sql = response.parse() assert_matches_type(object, golden_sql, path=["response"]) - @parametrize - def test_method_user_upload(self, client: Dataherald) -> None: - golden_sql = client.golden_sqls.user_upload( - body=[ - { - "db_connection_id": "string", - "prompt_text": "string", - "sql": "string", - }, - { - "db_connection_id": "string", - "prompt_text": "string", - "sql": "string", - }, - { - "db_connection_id": "string", - "prompt_text": "string", - "sql": "string", - }, - ], - ) - assert_matches_type(GoldenSqlUserUploadResponse, golden_sql, path=["response"]) - - @parametrize - def test_raw_response_user_upload(self, client: Dataherald) -> None: - response = client.golden_sqls.with_raw_response.user_upload( - body=[ - { - "db_connection_id": "string", - "prompt_text": "string", - "sql": "string", - }, - { - "db_connection_id": "string", - "prompt_text": "string", - "sql": "string", - }, - { - "db_connection_id": "string", - "prompt_text": "string", - "sql": "string", - }, - ], - ) - assert response.http_request.headers.get("X-Stainless-Lang") == "python" - golden_sql = response.parse() - assert_matches_type(GoldenSqlUserUploadResponse, golden_sql, path=["response"]) - class TestAsyncGoldenSqls: strict_client = AsyncDataherald(base_url=base_url, api_key=api_key, _strict_response_validation=True) @@ -185,51 +134,3 @@ async def test_raw_response_delete(self, client: AsyncDataherald) -> None: assert response.http_request.headers.get("X-Stainless-Lang") == "python" golden_sql = response.parse() assert_matches_type(object, golden_sql, path=["response"]) - - @parametrize - async def test_method_user_upload(self, client: AsyncDataherald) -> None: - golden_sql = await client.golden_sqls.user_upload( - body=[ - { - "db_connection_id": "string", - "prompt_text": "string", - "sql": "string", - }, - { - "db_connection_id": "string", - "prompt_text": "string", - "sql": "string", - }, - { - "db_connection_id": "string", - "prompt_text": "string", - "sql": "string", - }, - ], - ) - assert_matches_type(GoldenSqlUserUploadResponse, golden_sql, path=["response"]) - - @parametrize - async def test_raw_response_user_upload(self, client: AsyncDataherald) -> None: - response = await client.golden_sqls.with_raw_response.user_upload( - body=[ - { - "db_connection_id": "string", - "prompt_text": "string", - "sql": "string", - }, - { - "db_connection_id": "string", - "prompt_text": "string", - "sql": "string", - }, - { - "db_connection_id": "string", - "prompt_text": "string", - "sql": "string", - }, - ], - ) - assert response.http_request.headers.get("X-Stainless-Lang") == "python" - golden_sql = response.parse() - assert_matches_type(GoldenSqlUserUploadResponse, golden_sql, path=["response"])