Skip to content

Commit 03b78c1

Browse files
SDK regeneration
1 parent 7d76fb7 commit 03b78c1

185 files changed

Lines changed: 3937 additions & 764 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

reference.md

Lines changed: 217 additions & 193 deletions
Large diffs are not rendered by default.

src/truefoundry_sdk/__init__.py

Lines changed: 277 additions & 33 deletions
Large diffs are not rendered by default.

src/truefoundry_sdk/base_client.py

Lines changed: 18 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -113,9 +113,9 @@ def __init__(
113113
self._application_versions: typing.Optional[ApplicationVersionsClient] = None
114114
self._jobs: typing.Optional[JobsClient] = None
115115
self._workspaces: typing.Optional[WorkspacesClient] = None
116-
self._environments: typing.Optional[EnvironmentsClient] = None
117116
self._secrets: typing.Optional[SecretsClient] = None
118117
self._secret_groups: typing.Optional[SecretGroupsClient] = None
118+
self._environments: typing.Optional[EnvironmentsClient] = None
119119
self._events: typing.Optional[EventsClient] = None
120120
self._alerts: typing.Optional[AlertsClient] = None
121121
self._logs: typing.Optional[LogsClient] = None
@@ -312,14 +312,6 @@ def workspaces(self):
312312
self._workspaces = WorkspacesClient(client_wrapper=self._client_wrapper)
313313
return self._workspaces
314314

315-
@property
316-
def environments(self):
317-
if self._environments is None:
318-
from .environments.client import EnvironmentsClient # noqa: E402
319-
320-
self._environments = EnvironmentsClient(client_wrapper=self._client_wrapper)
321-
return self._environments
322-
323315
@property
324316
def secrets(self):
325317
if self._secrets is None:
@@ -336,6 +328,14 @@ def secret_groups(self):
336328
self._secret_groups = SecretGroupsClient(client_wrapper=self._client_wrapper)
337329
return self._secret_groups
338330

331+
@property
332+
def environments(self):
333+
if self._environments is None:
334+
from .environments.client import EnvironmentsClient # noqa: E402
335+
336+
self._environments = EnvironmentsClient(client_wrapper=self._client_wrapper)
337+
return self._environments
338+
339339
@property
340340
def events(self):
341341
if self._events is None:
@@ -502,9 +502,9 @@ def __init__(
502502
self._application_versions: typing.Optional[AsyncApplicationVersionsClient] = None
503503
self._jobs: typing.Optional[AsyncJobsClient] = None
504504
self._workspaces: typing.Optional[AsyncWorkspacesClient] = None
505-
self._environments: typing.Optional[AsyncEnvironmentsClient] = None
506505
self._secrets: typing.Optional[AsyncSecretsClient] = None
507506
self._secret_groups: typing.Optional[AsyncSecretGroupsClient] = None
507+
self._environments: typing.Optional[AsyncEnvironmentsClient] = None
508508
self._events: typing.Optional[AsyncEventsClient] = None
509509
self._alerts: typing.Optional[AsyncAlertsClient] = None
510510
self._logs: typing.Optional[AsyncLogsClient] = None
@@ -717,14 +717,6 @@ def workspaces(self):
717717
self._workspaces = AsyncWorkspacesClient(client_wrapper=self._client_wrapper)
718718
return self._workspaces
719719

720-
@property
721-
def environments(self):
722-
if self._environments is None:
723-
from .environments.client import AsyncEnvironmentsClient # noqa: E402
724-
725-
self._environments = AsyncEnvironmentsClient(client_wrapper=self._client_wrapper)
726-
return self._environments
727-
728720
@property
729721
def secrets(self):
730722
if self._secrets is None:
@@ -741,6 +733,14 @@ def secret_groups(self):
741733
self._secret_groups = AsyncSecretGroupsClient(client_wrapper=self._client_wrapper)
742734
return self._secret_groups
743735

736+
@property
737+
def environments(self):
738+
if self._environments is None:
739+
from .environments.client import AsyncEnvironmentsClient # noqa: E402
740+
741+
self._environments = AsyncEnvironmentsClient(client_wrapper=self._client_wrapper)
742+
return self._environments
743+
744744
@property
745745
def events(self):
746746
if self._events is None:

src/truefoundry_sdk/internal/ai_gateway/types/ai_gateway_get_gateway_config_request_type.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,9 @@ class AiGatewayGetGatewayConfigRequestType(enum.StrEnum):
1414
GATEWAY_GUARDRAILS_CONFIG = "gateway-guardrails-config"
1515
GATEWAY_BUDGET_CONFIG = "gateway-budget-config"
1616
GATEWAY_OTEL_CONFIG = "gateway-otel-config"
17+
GATEWAY_GLOBAL_SETTINGS = "gateway-global-settings"
18+
GATEWAY_DATA_ACCESS_CONFIG = "gateway-data-access-config"
19+
GATEWAY_DATA_ROUTING_CONFIG = "gateway-data-routing-config"
1720
_UNKNOWN = "__AIGATEWAYGETGATEWAYCONFIGREQUESTTYPE_UNKNOWN__"
1821
"""
1922
This member is used for forward compatibility. If the value is not recognized by the enum, it will be stored here, and the raw value is accessible through `.value`.
@@ -33,6 +36,9 @@ def visit(
3336
gateway_guardrails_config: typing.Callable[[], T_Result],
3437
gateway_budget_config: typing.Callable[[], T_Result],
3538
gateway_otel_config: typing.Callable[[], T_Result],
39+
gateway_global_settings: typing.Callable[[], T_Result],
40+
gateway_data_access_config: typing.Callable[[], T_Result],
41+
gateway_data_routing_config: typing.Callable[[], T_Result],
3642
_unknown_member: typing.Callable[[str], T_Result],
3743
) -> T_Result:
3844
if self is AiGatewayGetGatewayConfigRequestType.GATEWAY_RATE_LIMITING_CONFIG:
@@ -47,4 +53,10 @@ def visit(
4753
return gateway_budget_config()
4854
if self is AiGatewayGetGatewayConfigRequestType.GATEWAY_OTEL_CONFIG:
4955
return gateway_otel_config()
56+
if self is AiGatewayGetGatewayConfigRequestType.GATEWAY_GLOBAL_SETTINGS:
57+
return gateway_global_settings()
58+
if self is AiGatewayGetGatewayConfigRequestType.GATEWAY_DATA_ACCESS_CONFIG:
59+
return gateway_data_access_config()
60+
if self is AiGatewayGetGatewayConfigRequestType.GATEWAY_DATA_ROUTING_CONFIG:
61+
return gateway_data_routing_config()
5062
return _unknown_member(self._value_)

src/truefoundry_sdk/internal/ml/types/apply_ml_entity_request_manifest.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,5 @@
66
from ....types.chat_prompt_manifest import ChatPromptManifest
77
from ....types.data_directory_manifest import DataDirectoryManifest
88
from ....types.model_manifest import ModelManifest
9-
from ....types.tracing_project_manifest import TracingProjectManifest
109

11-
ApplyMlEntityRequestManifest = typing.Union[
12-
ModelManifest, ChatPromptManifest, ArtifactManifest, DataDirectoryManifest, TracingProjectManifest
13-
]
10+
ApplyMlEntityRequestManifest = typing.Union[ModelManifest, ChatPromptManifest, ArtifactManifest, DataDirectoryManifest]

src/truefoundry_sdk/teams/__init__.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66
from importlib import import_module
77

88
if typing.TYPE_CHECKING:
9-
from .types import ApplyTeamRequestManifest, TeamsListRequestType
10-
_dynamic_imports: typing.Dict[str, str] = {"ApplyTeamRequestManifest": ".types", "TeamsListRequestType": ".types"}
9+
from .types import TeamsListRequestType
10+
_dynamic_imports: typing.Dict[str, str] = {"TeamsListRequestType": ".types"}
1111

1212

1313
def __getattr__(attr_name: str) -> typing.Any:
@@ -31,4 +31,4 @@ def __dir__():
3131
return sorted(lazy_attrs)
3232

3333

34-
__all__ = ["ApplyTeamRequestManifest", "TeamsListRequestType"]
34+
__all__ = ["TeamsListRequestType"]

src/truefoundry_sdk/teams/client.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@
99
from ..types.get_team_response import GetTeamResponse
1010
from ..types.list_teams_response import ListTeamsResponse
1111
from ..types.team import Team
12+
from ..types.team_manifest import TeamManifest
1213
from .raw_client import AsyncRawTeamsClient, RawTeamsClient
13-
from .types.apply_team_request_manifest import ApplyTeamRequestManifest
1414
from .types.teams_list_request_type import TeamsListRequestType
1515

1616
# this is used as the default value for optional parameters
@@ -87,7 +87,7 @@ def list(
8787
def create_or_update(
8888
self,
8989
*,
90-
manifest: ApplyTeamRequestManifest,
90+
manifest: TeamManifest,
9191
dry_run: typing.Optional[bool] = OMIT,
9292
request_options: typing.Optional[RequestOptions] = None,
9393
) -> GetTeamResponse:
@@ -96,7 +96,7 @@ def create_or_update(
9696
9797
Parameters
9898
----------
99-
manifest : ApplyTeamRequestManifest
99+
manifest : TeamManifest
100100
Team manifest
101101
102102
dry_run : typing.Optional[bool]
@@ -274,7 +274,7 @@ async def main() -> None:
274274
async def create_or_update(
275275
self,
276276
*,
277-
manifest: ApplyTeamRequestManifest,
277+
manifest: TeamManifest,
278278
dry_run: typing.Optional[bool] = OMIT,
279279
request_options: typing.Optional[RequestOptions] = None,
280280
) -> GetTeamResponse:
@@ -283,7 +283,7 @@ async def create_or_update(
283283
284284
Parameters
285285
----------
286-
manifest : ApplyTeamRequestManifest
286+
manifest : TeamManifest
287287
Team manifest
288288
289289
dry_run : typing.Optional[bool]

src/truefoundry_sdk/teams/raw_client.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
from ..types.http_error import HttpError
2020
from ..types.list_teams_response import ListTeamsResponse
2121
from ..types.team import Team
22-
from .types.apply_team_request_manifest import ApplyTeamRequestManifest
22+
from ..types.team_manifest import TeamManifest
2323
from .types.teams_list_request_type import TeamsListRequestType
2424

2525
# this is used as the default value for optional parameters
@@ -98,7 +98,7 @@ def list(
9898
def create_or_update(
9999
self,
100100
*,
101-
manifest: ApplyTeamRequestManifest,
101+
manifest: TeamManifest,
102102
dry_run: typing.Optional[bool] = OMIT,
103103
request_options: typing.Optional[RequestOptions] = None,
104104
) -> HttpResponse[GetTeamResponse]:
@@ -107,7 +107,7 @@ def create_or_update(
107107
108108
Parameters
109109
----------
110-
manifest : ApplyTeamRequestManifest
110+
manifest : TeamManifest
111111
Team manifest
112112
113113
dry_run : typing.Optional[bool]
@@ -126,7 +126,7 @@ def create_or_update(
126126
method="PUT",
127127
json={
128128
"manifest": convert_and_respect_annotation_metadata(
129-
object_=manifest, annotation=ApplyTeamRequestManifest, direction="write"
129+
object_=manifest, annotation=TeamManifest, direction="write"
130130
),
131131
"dryRun": dry_run,
132132
},
@@ -358,7 +358,7 @@ async def _get_next():
358358
async def create_or_update(
359359
self,
360360
*,
361-
manifest: ApplyTeamRequestManifest,
361+
manifest: TeamManifest,
362362
dry_run: typing.Optional[bool] = OMIT,
363363
request_options: typing.Optional[RequestOptions] = None,
364364
) -> AsyncHttpResponse[GetTeamResponse]:
@@ -367,7 +367,7 @@ async def create_or_update(
367367
368368
Parameters
369369
----------
370-
manifest : ApplyTeamRequestManifest
370+
manifest : TeamManifest
371371
Team manifest
372372
373373
dry_run : typing.Optional[bool]
@@ -386,7 +386,7 @@ async def create_or_update(
386386
method="PUT",
387387
json={
388388
"manifest": convert_and_respect_annotation_metadata(
389-
object_=manifest, annotation=ApplyTeamRequestManifest, direction="write"
389+
object_=manifest, annotation=TeamManifest, direction="write"
390390
),
391391
"dryRun": dry_run,
392392
},

src/truefoundry_sdk/teams/types/__init__.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,8 @@
66
from importlib import import_module
77

88
if typing.TYPE_CHECKING:
9-
from .apply_team_request_manifest import ApplyTeamRequestManifest
109
from .teams_list_request_type import TeamsListRequestType
11-
_dynamic_imports: typing.Dict[str, str] = {
12-
"ApplyTeamRequestManifest": ".apply_team_request_manifest",
13-
"TeamsListRequestType": ".teams_list_request_type",
14-
}
10+
_dynamic_imports: typing.Dict[str, str] = {"TeamsListRequestType": ".teams_list_request_type"}
1511

1612

1713
def __getattr__(attr_name: str) -> typing.Any:
@@ -35,4 +31,4 @@ def __dir__():
3531
return sorted(lazy_attrs)
3632

3733

38-
__all__ = ["ApplyTeamRequestManifest", "TeamsListRequestType"]
34+
__all__ = ["TeamsListRequestType"]

src/truefoundry_sdk/teams/types/apply_team_request_manifest.py

Lines changed: 0 additions & 8 deletions
This file was deleted.

0 commit comments

Comments
 (0)