Skip to content

Commit 4d218c9

Browse files
chore(internal): codegen related update (#76)
1 parent 156bc78 commit 4d218c9

File tree

1 file changed

+28
-36
lines changed

1 file changed

+28
-36
lines changed

src/lumaai/_client.py

Lines changed: 28 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
import httpx
1010

11-
from . import resources, _exceptions
11+
from . import _exceptions
1212
from ._qs import Querystring
1313
from ._types import (
1414
NOT_GIVEN,
@@ -24,31 +24,23 @@
2424
get_async_library,
2525
)
2626
from ._version import __version__
27+
from .resources import ping, credits
2728
from ._streaming import Stream as Stream, AsyncStream as AsyncStream
2829
from ._exceptions import LumaAIError, APIStatusError
2930
from ._base_client import (
3031
DEFAULT_MAX_RETRIES,
3132
SyncAPIClient,
3233
AsyncAPIClient,
3334
)
35+
from .resources.generations import generations
3436

35-
__all__ = [
36-
"Timeout",
37-
"Transport",
38-
"ProxiesTypes",
39-
"RequestOptions",
40-
"resources",
41-
"LumaAI",
42-
"AsyncLumaAI",
43-
"Client",
44-
"AsyncClient",
45-
]
37+
__all__ = ["Timeout", "Transport", "ProxiesTypes", "RequestOptions", "LumaAI", "AsyncLumaAI", "Client", "AsyncClient"]
4638

4739

4840
class LumaAI(SyncAPIClient):
49-
generations: resources.GenerationsResource
50-
ping: resources.PingResource
51-
credits: resources.CreditsResource
41+
generations: generations.GenerationsResource
42+
ping: ping.PingResource
43+
credits: credits.CreditsResource
5244
with_raw_response: LumaAIWithRawResponse
5345
with_streaming_response: LumaAIWithStreamedResponse
5446

@@ -106,9 +98,9 @@ def __init__(
10698
_strict_response_validation=_strict_response_validation,
10799
)
108100

109-
self.generations = resources.GenerationsResource(self)
110-
self.ping = resources.PingResource(self)
111-
self.credits = resources.CreditsResource(self)
101+
self.generations = generations.GenerationsResource(self)
102+
self.ping = ping.PingResource(self)
103+
self.credits = credits.CreditsResource(self)
112104
self.with_raw_response = LumaAIWithRawResponse(self)
113105
self.with_streaming_response = LumaAIWithStreamedResponse(self)
114106

@@ -218,9 +210,9 @@ def _make_status_error(
218210

219211

220212
class AsyncLumaAI(AsyncAPIClient):
221-
generations: resources.AsyncGenerationsResource
222-
ping: resources.AsyncPingResource
223-
credits: resources.AsyncCreditsResource
213+
generations: generations.AsyncGenerationsResource
214+
ping: ping.AsyncPingResource
215+
credits: credits.AsyncCreditsResource
224216
with_raw_response: AsyncLumaAIWithRawResponse
225217
with_streaming_response: AsyncLumaAIWithStreamedResponse
226218

@@ -278,9 +270,9 @@ def __init__(
278270
_strict_response_validation=_strict_response_validation,
279271
)
280272

281-
self.generations = resources.AsyncGenerationsResource(self)
282-
self.ping = resources.AsyncPingResource(self)
283-
self.credits = resources.AsyncCreditsResource(self)
273+
self.generations = generations.AsyncGenerationsResource(self)
274+
self.ping = ping.AsyncPingResource(self)
275+
self.credits = credits.AsyncCreditsResource(self)
284276
self.with_raw_response = AsyncLumaAIWithRawResponse(self)
285277
self.with_streaming_response = AsyncLumaAIWithStreamedResponse(self)
286278

@@ -391,30 +383,30 @@ def _make_status_error(
391383

392384
class LumaAIWithRawResponse:
393385
def __init__(self, client: LumaAI) -> None:
394-
self.generations = resources.GenerationsResourceWithRawResponse(client.generations)
395-
self.ping = resources.PingResourceWithRawResponse(client.ping)
396-
self.credits = resources.CreditsResourceWithRawResponse(client.credits)
386+
self.generations = generations.GenerationsResourceWithRawResponse(client.generations)
387+
self.ping = ping.PingResourceWithRawResponse(client.ping)
388+
self.credits = credits.CreditsResourceWithRawResponse(client.credits)
397389

398390

399391
class AsyncLumaAIWithRawResponse:
400392
def __init__(self, client: AsyncLumaAI) -> None:
401-
self.generations = resources.AsyncGenerationsResourceWithRawResponse(client.generations)
402-
self.ping = resources.AsyncPingResourceWithRawResponse(client.ping)
403-
self.credits = resources.AsyncCreditsResourceWithRawResponse(client.credits)
393+
self.generations = generations.AsyncGenerationsResourceWithRawResponse(client.generations)
394+
self.ping = ping.AsyncPingResourceWithRawResponse(client.ping)
395+
self.credits = credits.AsyncCreditsResourceWithRawResponse(client.credits)
404396

405397

406398
class LumaAIWithStreamedResponse:
407399
def __init__(self, client: LumaAI) -> None:
408-
self.generations = resources.GenerationsResourceWithStreamingResponse(client.generations)
409-
self.ping = resources.PingResourceWithStreamingResponse(client.ping)
410-
self.credits = resources.CreditsResourceWithStreamingResponse(client.credits)
400+
self.generations = generations.GenerationsResourceWithStreamingResponse(client.generations)
401+
self.ping = ping.PingResourceWithStreamingResponse(client.ping)
402+
self.credits = credits.CreditsResourceWithStreamingResponse(client.credits)
411403

412404

413405
class AsyncLumaAIWithStreamedResponse:
414406
def __init__(self, client: AsyncLumaAI) -> None:
415-
self.generations = resources.AsyncGenerationsResourceWithStreamingResponse(client.generations)
416-
self.ping = resources.AsyncPingResourceWithStreamingResponse(client.ping)
417-
self.credits = resources.AsyncCreditsResourceWithStreamingResponse(client.credits)
407+
self.generations = generations.AsyncGenerationsResourceWithStreamingResponse(client.generations)
408+
self.ping = ping.AsyncPingResourceWithStreamingResponse(client.ping)
409+
self.credits = credits.AsyncCreditsResourceWithStreamingResponse(client.credits)
418410

419411

420412
Client = LumaAI

0 commit comments

Comments
 (0)