22
33from __future__ import annotations
44
5- import httpx
6-
7- from ..._types import NOT_GIVEN , Body , Query , Headers , NotGiven
85from ..._compat import cached_property
96from ..._resource import SyncAPIResource , AsyncAPIResource
10- from ..._response import (
11- to_raw_response_wrapper ,
12- to_streamed_response_wrapper ,
13- async_to_raw_response_wrapper ,
14- async_to_streamed_response_wrapper ,
15- )
16- from ..._base_client import make_request_options
177from .providers .providers import (
188 ProvidersResource ,
199 AsyncProvidersResource ,
2212 ProvidersResourceWithStreamingResponse ,
2313 AsyncProvidersResourceWithStreamingResponse ,
2414)
25- from ...types .model_list_response import ModelListResponse
26- from ...types .model_retrieve_response import ModelRetrieveResponse
2715
2816__all__ = ["ModelsResource" , "AsyncModelsResource" ]
2917
@@ -52,64 +40,6 @@ def with_streaming_response(self) -> ModelsResourceWithStreamingResponse:
5240 """
5341 return ModelsResourceWithStreamingResponse (self )
5442
55- def retrieve (
56- self ,
57- model : str ,
58- * ,
59- # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
60- # The extra values given here take precedence over values defined on the client or passed to this method.
61- extra_headers : Headers | None = None ,
62- extra_query : Query | None = None ,
63- extra_body : Body | None = None ,
64- timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
65- ) -> ModelRetrieveResponse :
66- """
67- Retrieves a model instance, providing basic information about the model such as
68- the owner and permissioning.
69-
70- Args:
71- extra_headers: Send extra headers
72-
73- extra_query: Add additional query parameters to the request
74-
75- extra_body: Add additional JSON properties to the request
76-
77- timeout: Override the client-level default timeout for this request, in seconds
78- """
79- if not model :
80- raise ValueError (f"Expected a non-empty value for `model` but received { model !r} " )
81- return self ._get (
82- f"/models/{ model } "
83- if self ._client ._base_url_overridden
84- else f"https://inference.do-ai.run/v1/models/{ model } " ,
85- options = make_request_options (
86- extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
87- ),
88- cast_to = ModelRetrieveResponse ,
89- )
90-
91- def list (
92- self ,
93- * ,
94- # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
95- # The extra values given here take precedence over values defined on the client or passed to this method.
96- extra_headers : Headers | None = None ,
97- extra_query : Query | None = None ,
98- extra_body : Body | None = None ,
99- timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
100- ) -> ModelListResponse :
101- """
102- Lists the currently available models, and provides basic information about each
103- one such as the owner and availability.
104- """
105- return self ._get (
106- "/models" if self ._client ._base_url_overridden else "https://inference.do-ai.run/v1/models" ,
107- options = make_request_options (
108- extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
109- ),
110- cast_to = ModelListResponse ,
111- )
112-
11343
11444class AsyncModelsResource (AsyncAPIResource ):
11545 @cached_property
@@ -135,76 +65,11 @@ def with_streaming_response(self) -> AsyncModelsResourceWithStreamingResponse:
13565 """
13666 return AsyncModelsResourceWithStreamingResponse (self )
13767
138- async def retrieve (
139- self ,
140- model : str ,
141- * ,
142- # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
143- # The extra values given here take precedence over values defined on the client or passed to this method.
144- extra_headers : Headers | None = None ,
145- extra_query : Query | None = None ,
146- extra_body : Body | None = None ,
147- timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
148- ) -> ModelRetrieveResponse :
149- """
150- Retrieves a model instance, providing basic information about the model such as
151- the owner and permissioning.
152-
153- Args:
154- extra_headers: Send extra headers
155-
156- extra_query: Add additional query parameters to the request
157-
158- extra_body: Add additional JSON properties to the request
159-
160- timeout: Override the client-level default timeout for this request, in seconds
161- """
162- if not model :
163- raise ValueError (f"Expected a non-empty value for `model` but received { model !r} " )
164- return await self ._get (
165- f"/models/{ model } "
166- if self ._client ._base_url_overridden
167- else f"https://inference.do-ai.run/v1/models/{ model } " ,
168- options = make_request_options (
169- extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
170- ),
171- cast_to = ModelRetrieveResponse ,
172- )
173-
174- async def list (
175- self ,
176- * ,
177- # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
178- # The extra values given here take precedence over values defined on the client or passed to this method.
179- extra_headers : Headers | None = None ,
180- extra_query : Query | None = None ,
181- extra_body : Body | None = None ,
182- timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
183- ) -> ModelListResponse :
184- """
185- Lists the currently available models, and provides basic information about each
186- one such as the owner and availability.
187- """
188- return await self ._get (
189- "/models" if self ._client ._base_url_overridden else "https://inference.do-ai.run/v1/models" ,
190- options = make_request_options (
191- extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
192- ),
193- cast_to = ModelListResponse ,
194- )
195-
19668
19769class ModelsResourceWithRawResponse :
19870 def __init__ (self , models : ModelsResource ) -> None :
19971 self ._models = models
20072
201- self .retrieve = to_raw_response_wrapper (
202- models .retrieve ,
203- )
204- self .list = to_raw_response_wrapper (
205- models .list ,
206- )
207-
20873 @cached_property
20974 def providers (self ) -> ProvidersResourceWithRawResponse :
21075 return ProvidersResourceWithRawResponse (self ._models .providers )
@@ -214,13 +79,6 @@ class AsyncModelsResourceWithRawResponse:
21479 def __init__ (self , models : AsyncModelsResource ) -> None :
21580 self ._models = models
21681
217- self .retrieve = async_to_raw_response_wrapper (
218- models .retrieve ,
219- )
220- self .list = async_to_raw_response_wrapper (
221- models .list ,
222- )
223-
22482 @cached_property
22583 def providers (self ) -> AsyncProvidersResourceWithRawResponse :
22684 return AsyncProvidersResourceWithRawResponse (self ._models .providers )
@@ -230,13 +88,6 @@ class ModelsResourceWithStreamingResponse:
23088 def __init__ (self , models : ModelsResource ) -> None :
23189 self ._models = models
23290
233- self .retrieve = to_streamed_response_wrapper (
234- models .retrieve ,
235- )
236- self .list = to_streamed_response_wrapper (
237- models .list ,
238- )
239-
24091 @cached_property
24192 def providers (self ) -> ProvidersResourceWithStreamingResponse :
24293 return ProvidersResourceWithStreamingResponse (self ._models .providers )
@@ -246,13 +97,6 @@ class AsyncModelsResourceWithStreamingResponse:
24697 def __init__ (self , models : AsyncModelsResource ) -> None :
24798 self ._models = models
24899
249- self .retrieve = async_to_streamed_response_wrapper (
250- models .retrieve ,
251- )
252- self .list = async_to_streamed_response_wrapper (
253- models .list ,
254- )
255-
256100 @cached_property
257101 def providers (self ) -> AsyncProvidersResourceWithStreamingResponse :
258102 return AsyncProvidersResourceWithStreamingResponse (self ._models .providers )
0 commit comments