Skip to content

Commit 5623116

Browse files
bbathabbatha
committed
fix lints
1 parent 2dee591 commit 5623116

File tree

1 file changed

+82
-39
lines changed

1 file changed

+82
-39
lines changed

‎src/do_gradientai/_client.py‎

Lines changed: 82 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -32,56 +32,50 @@
3232
)
3333

3434
if TYPE_CHECKING:
35-
from .resources import chat, agents, models, regions, databases, inference, gpu_droplets, knowledge_bases
35+
from .resources import (
36+
chat,
37+
agents,
38+
models,
39+
regions,
40+
databases,
41+
inference,
42+
gpu_droplets,
43+
knowledge_bases,
44+
)
3645
from .resources.regions import RegionsResource, AsyncRegionsResource
3746
from .resources.chat.chat import ChatResource, AsyncChatResource
3847
from .resources.gpu_droplets import (
3948
GPUDropletsResource,
4049
AsyncGPUDropletsResource,
41-
sizes,
42-
images,
43-
account,
44-
volumes,
45-
firewalls,
46-
snapshots,
47-
floating_ips,
48-
load_balancers,
4950
)
5051
from .resources.agents.agents import AgentsResource, AsyncAgentsResource
5152
from .resources.models.models import ModelsResource, AsyncModelsResource
5253
from .resources.databases.databases import DatabasesResource, AsyncDatabasesResource
5354
from .resources.inference.inference import InferenceResource, AsyncInferenceResource
5455
from .resources.gpu_droplets.snapshots import (
5556
SnapshotsResource,
56-
AsyncSnapshotsResource,
5757
)
5858
from .resources.gpu_droplets.images.images import (
5959
ImagesResource,
60-
AsyncImagesResource,
6160
)
6261
from .resources.gpu_droplets.account.account import (
6362
AccountResource,
64-
AsyncAccountResource,
6563
)
6664
from .resources.gpu_droplets.volumes.volumes import (
6765
VolumesResource,
68-
AsyncVolumesResource,
6966
)
7067
from .resources.knowledge_bases.knowledge_bases import (
7168
KnowledgeBasesResource,
7269
AsyncKnowledgeBasesResource,
7370
)
7471
from .resources.gpu_droplets.firewalls.firewalls import (
7572
FirewallsResource,
76-
AsyncFirewallsResource,
7773
)
7874
from .resources.gpu_droplets.floating_ips.floating_ips import (
7975
FloatingIPsResource,
80-
AsyncFloatingIPsResource,
8176
)
8277
from .resources.gpu_droplets.load_balancers.load_balancers import (
8378
LoadBalancersResource,
84-
AsyncLoadBalancersResource,
8579
)
8680

8781
__all__ = [
@@ -310,7 +304,9 @@ def default_headers(self) -> dict[str, str | Omit]:
310304

311305
@override
312306
def _validate_headers(self, headers: Headers, custom_headers: Headers) -> None:
313-
if (self.api_key or self.agent_key or self.inference_key) and headers.get("Authorization"):
307+
if (self.api_key or self.agent_key or self.inference_key) and headers.get(
308+
"Authorization"
309+
):
314310
return
315311
if isinstance(custom_headers.get("Authorization"), Omit):
316312
return
@@ -340,10 +336,14 @@ def copy(
340336
Create a new client instance re-using the same options given to the current client with optional overriding.
341337
"""
342338
if default_headers is not None and set_default_headers is not None:
343-
raise ValueError("The `default_headers` and `set_default_headers` arguments are mutually exclusive")
339+
raise ValueError(
340+
"The `default_headers` and `set_default_headers` arguments are mutually exclusive"
341+
)
344342

345343
if default_query is not None and set_default_query is not None:
346-
raise ValueError("The `default_query` and `set_default_query` arguments are mutually exclusive")
344+
raise ValueError(
345+
"The `default_query` and `set_default_query` arguments are mutually exclusive"
346+
)
347347

348348
headers = self._custom_headers
349349
if default_headers is not None:
@@ -390,10 +390,14 @@ def _make_status_error(
390390
return _exceptions.BadRequestError(err_msg, response=response, body=body)
391391

392392
if response.status_code == 401:
393-
return _exceptions.AuthenticationError(err_msg, response=response, body=body)
393+
return _exceptions.AuthenticationError(
394+
err_msg, response=response, body=body
395+
)
394396

395397
if response.status_code == 403:
396-
return _exceptions.PermissionDeniedError(err_msg, response=response, body=body)
398+
return _exceptions.PermissionDeniedError(
399+
err_msg, response=response, body=body
400+
)
397401

398402
if response.status_code == 404:
399403
return _exceptions.NotFoundError(err_msg, response=response, body=body)
@@ -402,13 +406,17 @@ def _make_status_error(
402406
return _exceptions.ConflictError(err_msg, response=response, body=body)
403407

404408
if response.status_code == 422:
405-
return _exceptions.UnprocessableEntityError(err_msg, response=response, body=body)
409+
return _exceptions.UnprocessableEntityError(
410+
err_msg, response=response, body=body
411+
)
406412

407413
if response.status_code == 429:
408414
return _exceptions.RateLimitError(err_msg, response=response, body=body)
409415

410416
if response.status_code >= 500:
411-
return _exceptions.InternalServerError(err_msg, response=response, body=body)
417+
return _exceptions.InternalServerError(
418+
err_msg, response=response, body=body
419+
)
412420
return APIStatusError(err_msg, response=response, body=body)
413421

414422

@@ -578,7 +586,9 @@ def default_headers(self) -> dict[str, str | Omit]:
578586

579587
@override
580588
def _validate_headers(self, headers: Headers, custom_headers: Headers) -> None:
581-
if (self.api_key or self.agent_key or self.inference_key) and headers.get("Authorization"):
589+
if (self.api_key or self.agent_key or self.inference_key) and headers.get(
590+
"Authorization"
591+
):
582592
return
583593
if isinstance(custom_headers.get("Authorization"), Omit):
584594
return
@@ -608,10 +618,14 @@ def copy(
608618
Create a new client instance re-using the same options given to the current client with optional overriding.
609619
"""
610620
if default_headers is not None and set_default_headers is not None:
611-
raise ValueError("The `default_headers` and `set_default_headers` arguments are mutually exclusive")
621+
raise ValueError(
622+
"The `default_headers` and `set_default_headers` arguments are mutually exclusive"
623+
)
612624

613625
if default_query is not None and set_default_query is not None:
614-
raise ValueError("The `default_query` and `set_default_query` arguments are mutually exclusive")
626+
raise ValueError(
627+
"The `default_query` and `set_default_query` arguments are mutually exclusive"
628+
)
615629

616630
headers = self._custom_headers
617631
if default_headers is not None:
@@ -658,10 +672,14 @@ def _make_status_error(
658672
return _exceptions.BadRequestError(err_msg, response=response, body=body)
659673

660674
if response.status_code == 401:
661-
return _exceptions.AuthenticationError(err_msg, response=response, body=body)
675+
return _exceptions.AuthenticationError(
676+
err_msg, response=response, body=body
677+
)
662678

663679
if response.status_code == 403:
664-
return _exceptions.PermissionDeniedError(err_msg, response=response, body=body)
680+
return _exceptions.PermissionDeniedError(
681+
err_msg, response=response, body=body
682+
)
665683

666684
if response.status_code == 404:
667685
return _exceptions.NotFoundError(err_msg, response=response, body=body)
@@ -670,13 +688,17 @@ def _make_status_error(
670688
return _exceptions.ConflictError(err_msg, response=response, body=body)
671689

672690
if response.status_code == 422:
673-
return _exceptions.UnprocessableEntityError(err_msg, response=response, body=body)
691+
return _exceptions.UnprocessableEntityError(
692+
err_msg, response=response, body=body
693+
)
674694

675695
if response.status_code == 429:
676696
return _exceptions.RateLimitError(err_msg, response=response, body=body)
677697

678698
if response.status_code >= 500:
679-
return _exceptions.InternalServerError(err_msg, response=response, body=body)
699+
return _exceptions.InternalServerError(
700+
err_msg, response=response, body=body
701+
)
680702
return APIStatusError(err_msg, response=response, body=body)
681703

682704

@@ -734,6 +756,7 @@ def databases(self) -> databases.DatabasesResourceWithRawResponse:
734756

735757
return DatabasesResourceWithRawResponse(self._client.databases)
736758

759+
737760
class AsyncGradientAIWithRawResponse:
738761
_client: AsyncGradientAI
739762

@@ -765,8 +788,12 @@ def inference(self) -> inference.AsyncInferenceResourceWithRawResponse:
765788
return AsyncInferenceResourceWithRawResponse(self._client.inference)
766789

767790
@cached_property
768-
def knowledge_bases(self) -> knowledge_bases.AsyncKnowledgeBasesResourceWithRawResponse:
769-
from .resources.knowledge_bases import AsyncKnowledgeBasesResourceWithRawResponse
791+
def knowledge_bases(
792+
self,
793+
) -> knowledge_bases.AsyncKnowledgeBasesResourceWithRawResponse:
794+
from .resources.knowledge_bases import (
795+
AsyncKnowledgeBasesResourceWithRawResponse,
796+
)
770797

771798
return AsyncKnowledgeBasesResourceWithRawResponse(self._client.knowledge_bases)
772799

@@ -788,6 +815,7 @@ def databases(self) -> databases.AsyncDatabasesResourceWithRawResponse:
788815

789816
return AsyncDatabasesResourceWithRawResponse(self._client.databases)
790817

818+
791819
class GradientAIWithStreamedResponse:
792820
_client: GradientAI
793821

@@ -819,8 +847,12 @@ def inference(self) -> inference.InferenceResourceWithStreamingResponse:
819847
return InferenceResourceWithStreamingResponse(self._client.inference)
820848

821849
@cached_property
822-
def knowledge_bases(self) -> knowledge_bases.KnowledgeBasesResourceWithStreamingResponse:
823-
from .resources.knowledge_bases import KnowledgeBasesResourceWithStreamingResponse
850+
def knowledge_bases(
851+
self,
852+
) -> knowledge_bases.KnowledgeBasesResourceWithStreamingResponse:
853+
from .resources.knowledge_bases import (
854+
KnowledgeBasesResourceWithStreamingResponse,
855+
)
824856

825857
return KnowledgeBasesResourceWithStreamingResponse(self._client.knowledge_bases)
826858

@@ -842,6 +874,7 @@ def databases(self) -> databases.DatabasesResourceWithStreamingResponse:
842874

843875
return DatabasesResourceWithStreamingResponse(self._client.databases)
844876

877+
845878
class AsyncGradientAIWithStreamedResponse:
846879
_client: AsyncGradientAI
847880

@@ -861,8 +894,12 @@ def chat(self) -> chat.AsyncChatResourceWithStreamingResponse:
861894
return AsyncChatResourceWithStreamingResponse(self._client.chat)
862895

863896
@cached_property
864-
def gpu_droplets(self) -> gpu_droplets.AsyncGPUDropletsResourceWithStreamingResponse:
865-
from .resources.gpu_droplets import AsyncGPUDropletsResourceWithStreamingResponse
897+
def gpu_droplets(
898+
self,
899+
) -> gpu_droplets.AsyncGPUDropletsResourceWithStreamingResponse:
900+
from .resources.gpu_droplets import (
901+
AsyncGPUDropletsResourceWithStreamingResponse,
902+
)
866903

867904
return AsyncGPUDropletsResourceWithStreamingResponse(self._client.gpu_droplets)
868905

@@ -873,10 +910,16 @@ def inference(self) -> inference.AsyncInferenceResourceWithStreamingResponse:
873910
return AsyncInferenceResourceWithStreamingResponse(self._client.inference)
874911

875912
@cached_property
876-
def knowledge_bases(self) -> knowledge_bases.AsyncKnowledgeBasesResourceWithStreamingResponse:
877-
from .resources.knowledge_bases import AsyncKnowledgeBasesResourceWithStreamingResponse
913+
def knowledge_bases(
914+
self,
915+
) -> knowledge_bases.AsyncKnowledgeBasesResourceWithStreamingResponse:
916+
from .resources.knowledge_bases import (
917+
AsyncKnowledgeBasesResourceWithStreamingResponse,
918+
)
878919

879-
return AsyncKnowledgeBasesResourceWithStreamingResponse(self._client.knowledge_bases)
920+
return AsyncKnowledgeBasesResourceWithStreamingResponse(
921+
self._client.knowledge_bases
922+
)
880923

881924
@cached_property
882925
def models(self) -> models.AsyncModelsResourceWithStreamingResponse:

0 commit comments

Comments
 (0)