Skip to content

Commit fc58a7d

Browse files
chore(internal): updated imports (#138)
1 parent 4523c9d commit fc58a7d

1 file changed

Lines changed: 60 additions & 68 deletions

File tree

src/writerai/_client.py

Lines changed: 60 additions & 68 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,35 +24,27 @@
2424
get_async_library,
2525
)
2626
from ._version import __version__
27+
from .resources import chat, files, graphs, models, completions, applications
2728
from ._streaming import Stream as Stream, AsyncStream as AsyncStream
2829
from ._exceptions import WriterError, APIStatusError
2930
from ._base_client import (
3031
DEFAULT_MAX_RETRIES,
3132
SyncAPIClient,
3233
AsyncAPIClient,
3334
)
35+
from .resources.tools import tools
3436

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

4739

4840
class Writer(SyncAPIClient):
49-
applications: resources.ApplicationsResource
50-
chat: resources.ChatResource
51-
completions: resources.CompletionsResource
52-
models: resources.ModelsResource
53-
graphs: resources.GraphsResource
54-
files: resources.FilesResource
55-
tools: resources.ToolsResource
41+
applications: applications.ApplicationsResource
42+
chat: chat.ChatResource
43+
completions: completions.CompletionsResource
44+
models: models.ModelsResource
45+
graphs: graphs.GraphsResource
46+
files: files.FilesResource
47+
tools: tools.ToolsResource
5648
with_raw_response: WriterWithRawResponse
5749
with_streaming_response: WriterWithStreamedResponse
5850

@@ -112,13 +104,13 @@ def __init__(
112104

113105
self._default_stream_cls = Stream
114106

115-
self.applications = resources.ApplicationsResource(self)
116-
self.chat = resources.ChatResource(self)
117-
self.completions = resources.CompletionsResource(self)
118-
self.models = resources.ModelsResource(self)
119-
self.graphs = resources.GraphsResource(self)
120-
self.files = resources.FilesResource(self)
121-
self.tools = resources.ToolsResource(self)
107+
self.applications = applications.ApplicationsResource(self)
108+
self.chat = chat.ChatResource(self)
109+
self.completions = completions.CompletionsResource(self)
110+
self.models = models.ModelsResource(self)
111+
self.graphs = graphs.GraphsResource(self)
112+
self.files = files.FilesResource(self)
113+
self.tools = tools.ToolsResource(self)
122114
self.with_raw_response = WriterWithRawResponse(self)
123115
self.with_streaming_response = WriterWithStreamedResponse(self)
124116

@@ -228,13 +220,13 @@ def _make_status_error(
228220

229221

230222
class AsyncWriter(AsyncAPIClient):
231-
applications: resources.AsyncApplicationsResource
232-
chat: resources.AsyncChatResource
233-
completions: resources.AsyncCompletionsResource
234-
models: resources.AsyncModelsResource
235-
graphs: resources.AsyncGraphsResource
236-
files: resources.AsyncFilesResource
237-
tools: resources.AsyncToolsResource
223+
applications: applications.AsyncApplicationsResource
224+
chat: chat.AsyncChatResource
225+
completions: completions.AsyncCompletionsResource
226+
models: models.AsyncModelsResource
227+
graphs: graphs.AsyncGraphsResource
228+
files: files.AsyncFilesResource
229+
tools: tools.AsyncToolsResource
238230
with_raw_response: AsyncWriterWithRawResponse
239231
with_streaming_response: AsyncWriterWithStreamedResponse
240232

@@ -294,13 +286,13 @@ def __init__(
294286

295287
self._default_stream_cls = AsyncStream
296288

297-
self.applications = resources.AsyncApplicationsResource(self)
298-
self.chat = resources.AsyncChatResource(self)
299-
self.completions = resources.AsyncCompletionsResource(self)
300-
self.models = resources.AsyncModelsResource(self)
301-
self.graphs = resources.AsyncGraphsResource(self)
302-
self.files = resources.AsyncFilesResource(self)
303-
self.tools = resources.AsyncToolsResource(self)
289+
self.applications = applications.AsyncApplicationsResource(self)
290+
self.chat = chat.AsyncChatResource(self)
291+
self.completions = completions.AsyncCompletionsResource(self)
292+
self.models = models.AsyncModelsResource(self)
293+
self.graphs = graphs.AsyncGraphsResource(self)
294+
self.files = files.AsyncFilesResource(self)
295+
self.tools = tools.AsyncToolsResource(self)
304296
self.with_raw_response = AsyncWriterWithRawResponse(self)
305297
self.with_streaming_response = AsyncWriterWithStreamedResponse(self)
306298

@@ -411,46 +403,46 @@ def _make_status_error(
411403

412404
class WriterWithRawResponse:
413405
def __init__(self, client: Writer) -> None:
414-
self.applications = resources.ApplicationsResourceWithRawResponse(client.applications)
415-
self.chat = resources.ChatResourceWithRawResponse(client.chat)
416-
self.completions = resources.CompletionsResourceWithRawResponse(client.completions)
417-
self.models = resources.ModelsResourceWithRawResponse(client.models)
418-
self.graphs = resources.GraphsResourceWithRawResponse(client.graphs)
419-
self.files = resources.FilesResourceWithRawResponse(client.files)
420-
self.tools = resources.ToolsResourceWithRawResponse(client.tools)
406+
self.applications = applications.ApplicationsResourceWithRawResponse(client.applications)
407+
self.chat = chat.ChatResourceWithRawResponse(client.chat)
408+
self.completions = completions.CompletionsResourceWithRawResponse(client.completions)
409+
self.models = models.ModelsResourceWithRawResponse(client.models)
410+
self.graphs = graphs.GraphsResourceWithRawResponse(client.graphs)
411+
self.files = files.FilesResourceWithRawResponse(client.files)
412+
self.tools = tools.ToolsResourceWithRawResponse(client.tools)
421413

422414

423415
class AsyncWriterWithRawResponse:
424416
def __init__(self, client: AsyncWriter) -> None:
425-
self.applications = resources.AsyncApplicationsResourceWithRawResponse(client.applications)
426-
self.chat = resources.AsyncChatResourceWithRawResponse(client.chat)
427-
self.completions = resources.AsyncCompletionsResourceWithRawResponse(client.completions)
428-
self.models = resources.AsyncModelsResourceWithRawResponse(client.models)
429-
self.graphs = resources.AsyncGraphsResourceWithRawResponse(client.graphs)
430-
self.files = resources.AsyncFilesResourceWithRawResponse(client.files)
431-
self.tools = resources.AsyncToolsResourceWithRawResponse(client.tools)
417+
self.applications = applications.AsyncApplicationsResourceWithRawResponse(client.applications)
418+
self.chat = chat.AsyncChatResourceWithRawResponse(client.chat)
419+
self.completions = completions.AsyncCompletionsResourceWithRawResponse(client.completions)
420+
self.models = models.AsyncModelsResourceWithRawResponse(client.models)
421+
self.graphs = graphs.AsyncGraphsResourceWithRawResponse(client.graphs)
422+
self.files = files.AsyncFilesResourceWithRawResponse(client.files)
423+
self.tools = tools.AsyncToolsResourceWithRawResponse(client.tools)
432424

433425

434426
class WriterWithStreamedResponse:
435427
def __init__(self, client: Writer) -> None:
436-
self.applications = resources.ApplicationsResourceWithStreamingResponse(client.applications)
437-
self.chat = resources.ChatResourceWithStreamingResponse(client.chat)
438-
self.completions = resources.CompletionsResourceWithStreamingResponse(client.completions)
439-
self.models = resources.ModelsResourceWithStreamingResponse(client.models)
440-
self.graphs = resources.GraphsResourceWithStreamingResponse(client.graphs)
441-
self.files = resources.FilesResourceWithStreamingResponse(client.files)
442-
self.tools = resources.ToolsResourceWithStreamingResponse(client.tools)
428+
self.applications = applications.ApplicationsResourceWithStreamingResponse(client.applications)
429+
self.chat = chat.ChatResourceWithStreamingResponse(client.chat)
430+
self.completions = completions.CompletionsResourceWithStreamingResponse(client.completions)
431+
self.models = models.ModelsResourceWithStreamingResponse(client.models)
432+
self.graphs = graphs.GraphsResourceWithStreamingResponse(client.graphs)
433+
self.files = files.FilesResourceWithStreamingResponse(client.files)
434+
self.tools = tools.ToolsResourceWithStreamingResponse(client.tools)
443435

444436

445437
class AsyncWriterWithStreamedResponse:
446438
def __init__(self, client: AsyncWriter) -> None:
447-
self.applications = resources.AsyncApplicationsResourceWithStreamingResponse(client.applications)
448-
self.chat = resources.AsyncChatResourceWithStreamingResponse(client.chat)
449-
self.completions = resources.AsyncCompletionsResourceWithStreamingResponse(client.completions)
450-
self.models = resources.AsyncModelsResourceWithStreamingResponse(client.models)
451-
self.graphs = resources.AsyncGraphsResourceWithStreamingResponse(client.graphs)
452-
self.files = resources.AsyncFilesResourceWithStreamingResponse(client.files)
453-
self.tools = resources.AsyncToolsResourceWithStreamingResponse(client.tools)
439+
self.applications = applications.AsyncApplicationsResourceWithStreamingResponse(client.applications)
440+
self.chat = chat.AsyncChatResourceWithStreamingResponse(client.chat)
441+
self.completions = completions.AsyncCompletionsResourceWithStreamingResponse(client.completions)
442+
self.models = models.AsyncModelsResourceWithStreamingResponse(client.models)
443+
self.graphs = graphs.AsyncGraphsResourceWithStreamingResponse(client.graphs)
444+
self.files = files.AsyncFilesResourceWithStreamingResponse(client.files)
445+
self.tools = tools.AsyncToolsResourceWithStreamingResponse(client.tools)
454446

455447

456448
Client = Writer

0 commit comments

Comments
 (0)