diff --git a/src/dataherald/resources/database_connections/database_connections.py b/src/dataherald/resources/database_connections/database_connections.py index 3a9fb39..3e64301 100644 --- a/src/dataherald/resources/database_connections/database_connections.py +++ b/src/dataherald/resources/database_connections/database_connections.py @@ -56,7 +56,6 @@ def create( alias: str, connection_uri: str, bigquery_credential_file_content: Union[object, str] | NotGiven = NOT_GIVEN, - llm_api_key: str | NotGiven = NOT_GIVEN, metadata: object | NotGiven = NOT_GIVEN, ssh_settings: database_connection_create_params.SSHSettings | NotGiven = NOT_GIVEN, use_ssh: bool | NotGiven = NOT_GIVEN, @@ -86,7 +85,6 @@ def create( "alias": alias, "connection_uri": connection_uri, "bigquery_credential_file_content": bigquery_credential_file_content, - "llm_api_key": llm_api_key, "metadata": metadata, "ssh_settings": ssh_settings, "use_ssh": use_ssh, @@ -139,7 +137,6 @@ def update( alias: str, connection_uri: str, bigquery_credential_file_content: Union[object, str] | NotGiven = NOT_GIVEN, - llm_api_key: str | NotGiven = NOT_GIVEN, metadata: object | NotGiven = NOT_GIVEN, ssh_settings: database_connection_update_params.SSHSettings | NotGiven = NOT_GIVEN, use_ssh: bool | NotGiven = NOT_GIVEN, @@ -171,7 +168,6 @@ def update( "alias": alias, "connection_uri": connection_uri, "bigquery_credential_file_content": bigquery_credential_file_content, - "llm_api_key": llm_api_key, "metadata": metadata, "ssh_settings": ssh_settings, "use_ssh": use_ssh, @@ -194,7 +190,7 @@ def list( extra_body: Body | None = None, timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN, ) -> DatabaseConnectionListResponse: - """Api Get Db Connections""" + """Get Db Connections""" return self._get( "/api/database-connections", options=make_request_options( @@ -223,7 +219,6 @@ async def create( alias: str, connection_uri: str, bigquery_credential_file_content: Union[object, str] | NotGiven = NOT_GIVEN, - llm_api_key: str | NotGiven = NOT_GIVEN, metadata: object | NotGiven = NOT_GIVEN, ssh_settings: database_connection_create_params.SSHSettings | NotGiven = NOT_GIVEN, use_ssh: bool | NotGiven = NOT_GIVEN, @@ -253,7 +248,6 @@ async def create( "alias": alias, "connection_uri": connection_uri, "bigquery_credential_file_content": bigquery_credential_file_content, - "llm_api_key": llm_api_key, "metadata": metadata, "ssh_settings": ssh_settings, "use_ssh": use_ssh, @@ -306,7 +300,6 @@ async def update( alias: str, connection_uri: str, bigquery_credential_file_content: Union[object, str] | NotGiven = NOT_GIVEN, - llm_api_key: str | NotGiven = NOT_GIVEN, metadata: object | NotGiven = NOT_GIVEN, ssh_settings: database_connection_update_params.SSHSettings | NotGiven = NOT_GIVEN, use_ssh: bool | NotGiven = NOT_GIVEN, @@ -338,7 +331,6 @@ async def update( "alias": alias, "connection_uri": connection_uri, "bigquery_credential_file_content": bigquery_credential_file_content, - "llm_api_key": llm_api_key, "metadata": metadata, "ssh_settings": ssh_settings, "use_ssh": use_ssh, @@ -361,7 +353,7 @@ async def list( extra_body: Body | None = None, timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN, ) -> DatabaseConnectionListResponse: - """Api Get Db Connections""" + """Get Db Connections""" return await self._get( "/api/database-connections", options=make_request_options( diff --git a/src/dataherald/types/database_connection_create_params.py b/src/dataherald/types/database_connection_create_params.py index 1fa71aa..fbdf372 100644 --- a/src/dataherald/types/database_connection_create_params.py +++ b/src/dataherald/types/database_connection_create_params.py @@ -15,8 +15,6 @@ class DatabaseConnectionCreateParams(TypedDict, total=False): bigquery_credential_file_content: Union[object, str] - llm_api_key: str - metadata: object ssh_settings: SSHSettings diff --git a/src/dataherald/types/database_connection_update_params.py b/src/dataherald/types/database_connection_update_params.py index 2f51d0f..5ed096e 100644 --- a/src/dataherald/types/database_connection_update_params.py +++ b/src/dataherald/types/database_connection_update_params.py @@ -15,8 +15,6 @@ class DatabaseConnectionUpdateParams(TypedDict, total=False): bigquery_credential_file_content: Union[object, str] - llm_api_key: str - metadata: object ssh_settings: SSHSettings diff --git a/src/dataherald/types/db_connection_response.py b/src/dataherald/types/db_connection_response.py index c18f52b..da1589b 100644 --- a/src/dataherald/types/db_connection_response.py +++ b/src/dataherald/types/db_connection_response.py @@ -35,8 +35,6 @@ class DBConnectionResponse(BaseModel): created_at: Optional[datetime] = None - llm_api_key: Optional[str] = None - metadata: Optional[Metadata] = None path_to_credentials_file: Optional[str] = None diff --git a/tests/api_resources/test_database_connections.py b/tests/api_resources/test_database_connections.py index a11a6fb..cb71cb7 100644 --- a/tests/api_resources/test_database_connections.py +++ b/tests/api_resources/test_database_connections.py @@ -34,7 +34,6 @@ def test_method_create_with_all_params(self, client: Dataherald) -> None: alias="string", connection_uri="string", bigquery_credential_file_content={}, - llm_api_key="string", metadata={}, ssh_settings={ "host": "string", @@ -125,7 +124,6 @@ def test_method_update_with_all_params(self, client: Dataherald) -> None: alias="string", connection_uri="string", bigquery_credential_file_content={}, - llm_api_key="string", metadata={}, ssh_settings={ "host": "string", @@ -216,7 +214,6 @@ async def test_method_create_with_all_params(self, async_client: AsyncDataherald alias="string", connection_uri="string", bigquery_credential_file_content={}, - llm_api_key="string", metadata={}, ssh_settings={ "host": "string", @@ -307,7 +304,6 @@ async def test_method_update_with_all_params(self, async_client: AsyncDataherald alias="string", connection_uri="string", bigquery_credential_file_content={}, - llm_api_key="string", metadata={}, ssh_settings={ "host": "string",