2121)
2222from ._utils import is_given , get_async_library
2323from ._version import __version__
24- from .resources import pzn , models , preferences , model_router
24+ from .resources import models , prompt , preferences , model_router , custom_router
2525from ._streaming import Stream as Stream , AsyncStream as AsyncStream
2626from ._exceptions import APIStatusError , NotDiamondError
2727from ._base_client import (
2828 DEFAULT_MAX_RETRIES ,
2929 SyncAPIClient ,
3030 AsyncAPIClient ,
3131)
32- from .resources .prompt import prompt
3332from .resources .report import report
3433
3534__all__ = [
@@ -49,7 +48,7 @@ class NotDiamond(SyncAPIClient):
4948 report : report .ReportResource
5049 preferences : preferences .PreferencesResource
5150 prompt : prompt .PromptResource
52- pzn : pzn . PznResource
51+ custom_router : custom_router . CustomRouterResource
5352 models : models .ModelsResource
5453 with_raw_response : NotDiamondWithRawResponse
5554 with_streaming_response : NotDiamondWithStreamedResponse
@@ -112,7 +111,7 @@ def __init__(
112111 self .report = report .ReportResource (self )
113112 self .preferences = preferences .PreferencesResource (self )
114113 self .prompt = prompt .PromptResource (self )
115- self .pzn = pzn . PznResource (self )
114+ self .custom_router = custom_router . CustomRouterResource (self )
116115 self .models = models .ModelsResource (self )
117116 self .with_raw_response = NotDiamondWithRawResponse (self )
118117 self .with_streaming_response = NotDiamondWithStreamedResponse (self )
@@ -227,7 +226,7 @@ class AsyncNotDiamond(AsyncAPIClient):
227226 report : report .AsyncReportResource
228227 preferences : preferences .AsyncPreferencesResource
229228 prompt : prompt .AsyncPromptResource
230- pzn : pzn . AsyncPznResource
229+ custom_router : custom_router . AsyncCustomRouterResource
231230 models : models .AsyncModelsResource
232231 with_raw_response : AsyncNotDiamondWithRawResponse
233232 with_streaming_response : AsyncNotDiamondWithStreamedResponse
@@ -290,7 +289,7 @@ def __init__(
290289 self .report = report .AsyncReportResource (self )
291290 self .preferences = preferences .AsyncPreferencesResource (self )
292291 self .prompt = prompt .AsyncPromptResource (self )
293- self .pzn = pzn . AsyncPznResource (self )
292+ self .custom_router = custom_router . AsyncCustomRouterResource (self )
294293 self .models = models .AsyncModelsResource (self )
295294 self .with_raw_response = AsyncNotDiamondWithRawResponse (self )
296295 self .with_streaming_response = AsyncNotDiamondWithStreamedResponse (self )
@@ -406,7 +405,7 @@ def __init__(self, client: NotDiamond) -> None:
406405 self .report = report .ReportResourceWithRawResponse (client .report )
407406 self .preferences = preferences .PreferencesResourceWithRawResponse (client .preferences )
408407 self .prompt = prompt .PromptResourceWithRawResponse (client .prompt )
409- self .pzn = pzn . PznResourceWithRawResponse (client .pzn )
408+ self .custom_router = custom_router . CustomRouterResourceWithRawResponse (client .custom_router )
410409 self .models = models .ModelsResourceWithRawResponse (client .models )
411410
412411
@@ -416,7 +415,7 @@ def __init__(self, client: AsyncNotDiamond) -> None:
416415 self .report = report .AsyncReportResourceWithRawResponse (client .report )
417416 self .preferences = preferences .AsyncPreferencesResourceWithRawResponse (client .preferences )
418417 self .prompt = prompt .AsyncPromptResourceWithRawResponse (client .prompt )
419- self .pzn = pzn . AsyncPznResourceWithRawResponse (client .pzn )
418+ self .custom_router = custom_router . AsyncCustomRouterResourceWithRawResponse (client .custom_router )
420419 self .models = models .AsyncModelsResourceWithRawResponse (client .models )
421420
422421
@@ -426,7 +425,7 @@ def __init__(self, client: NotDiamond) -> None:
426425 self .report = report .ReportResourceWithStreamingResponse (client .report )
427426 self .preferences = preferences .PreferencesResourceWithStreamingResponse (client .preferences )
428427 self .prompt = prompt .PromptResourceWithStreamingResponse (client .prompt )
429- self .pzn = pzn . PznResourceWithStreamingResponse (client .pzn )
428+ self .custom_router = custom_router . CustomRouterResourceWithStreamingResponse (client .custom_router )
430429 self .models = models .ModelsResourceWithStreamingResponse (client .models )
431430
432431
@@ -436,7 +435,7 @@ def __init__(self, client: AsyncNotDiamond) -> None:
436435 self .report = report .AsyncReportResourceWithStreamingResponse (client .report )
437436 self .preferences = preferences .AsyncPreferencesResourceWithStreamingResponse (client .preferences )
438437 self .prompt = prompt .AsyncPromptResourceWithStreamingResponse (client .prompt )
439- self .pzn = pzn . AsyncPznResourceWithStreamingResponse (client .pzn )
438+ self .custom_router = custom_router . AsyncCustomRouterResourceWithStreamingResponse (client .custom_router )
440439 self .models = models .AsyncModelsResourceWithStreamingResponse (client .models )
441440
442441
0 commit comments