Skip to content

Commit 3074263

Browse files
MarkDaoustcopybara-github
authored andcommitted
feat: add environment files support and scotty file download helper
PiperOrigin-RevId: 969962312
1 parent 6680718 commit 3074263

10 files changed

Lines changed: 1162 additions & 8 deletions

File tree

google/genai/_gaos/environments.py

Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,10 @@
1717
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
1818

1919
from .basesdk import AsyncBaseSDK, BaseSDK
20+
from .sdkconfiguration import SDKConfiguration
2021
from . import errors, models, types, utils
2122
from ._hooks import AfterParseErrorContext, HookContext, ResponseContext
23+
from .files import AsyncFiles, Files
2224
from .types import OptionalNullable, UNSET, environments, interactions
2325
from .types.environments import (
2426
createenvironmentrequest as environments_createenvironmentrequest,
@@ -39,6 +41,18 @@ def with_raw_response(self):
3941
def with_streaming_response(self):
4042
return EnvironmentsWithStreamingResponse(self)
4143

44+
files: Files
45+
46+
def __init__(
47+
self, sdk_config: SDKConfiguration, parent_ref: Optional[object] = None
48+
) -> None:
49+
BaseSDK.__init__(self, sdk_config, parent_ref=parent_ref)
50+
self.sdk_configuration = sdk_config
51+
self._init_sdks()
52+
53+
def _init_sdks(self):
54+
self.files = Files(self.sdk_configuration, parent_ref=self.parent_ref)
55+
4256
def list_environments(
4357
self,
4458
*,
@@ -688,6 +702,10 @@ def __init__(self, sdk: Environments) -> None:
688702
sdk.get_environment, "extra_headers"
689703
)
690704

705+
@property
706+
def files(self):
707+
return self._sdk.files.with_raw_response
708+
691709

692710
class EnvironmentsWithStreamingResponse:
693711
def __init__(self, sdk: Environments) -> None:
@@ -705,6 +723,10 @@ def __init__(self, sdk: Environments) -> None:
705723
sdk.get_environment, "extra_headers"
706724
)
707725

726+
@property
727+
def files(self):
728+
return self._sdk.files.with_streaming_response
729+
708730

709731
class AsyncEnvironments(AsyncBaseSDK):
710732
@property
@@ -715,6 +737,18 @@ def with_raw_response(self):
715737
def with_streaming_response(self):
716738
return AsyncEnvironmentsWithStreamingResponse(self)
717739

740+
files: AsyncFiles
741+
742+
def __init__(
743+
self, sdk_config: SDKConfiguration, parent_ref: Optional[object] = None
744+
) -> None:
745+
AsyncBaseSDK.__init__(self, sdk_config, parent_ref=parent_ref)
746+
self.sdk_configuration = sdk_config
747+
self._init_sdks()
748+
749+
def _init_sdks(self):
750+
self.files = AsyncFiles(self.sdk_configuration, parent_ref=self.parent_ref)
751+
718752
async def list_environments(
719753
self,
720754
*,
@@ -1376,6 +1410,10 @@ def __init__(self, sdk: AsyncEnvironments) -> None:
13761410
sdk.get_environment, "extra_headers"
13771411
)
13781412

1413+
@property
1414+
def files(self):
1415+
return self._sdk.files.with_raw_response
1416+
13791417

13801418
class AsyncEnvironmentsWithStreamingResponse:
13811419
def __init__(self, sdk: AsyncEnvironments) -> None:
@@ -1392,3 +1430,7 @@ def __init__(self, sdk: AsyncEnvironments) -> None:
13921430
self.get_environment = response_helpers.async_to_streamed_response_wrapper(
13931431
sdk.get_environment, "extra_headers"
13941432
)
1433+
1434+
@property
1435+
def files(self):
1436+
return self._sdk.files.with_streaming_response

0 commit comments

Comments
 (0)