22
33from __future__ import annotations
44
5- from typing import Dict , Union , Optional
5+ from typing import Dict , Union , Iterable , Optional
66from typing_extensions import Literal , Required , Annotated , TypeAlias , TypedDict
77
88from ..._types import SequenceNotStr
3838 BrowserbaseGetSessionLiveURLsIntegrationDef ,
3939)
4040
41- __all__ = ["ToolCreateParams" , "APICall" , "APICallParamsSchema" , "APICallParamsSchemaProperties" , "Integration" ]
41+ __all__ = [
42+ "ToolCreateParams" ,
43+ "APICall" ,
44+ "APICallParamsSchema" ,
45+ "APICallParamsSchemaProperties" ,
46+ "Integration" ,
47+ "IntegrationGoogleSheetsIntegrationDefInput" ,
48+ "IntegrationGoogleSheetsIntegrationDefInputArguments" ,
49+ "IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsReadArguments" ,
50+ "IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsWriteArguments" ,
51+ "IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsAppendArguments" ,
52+ "IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsClearArguments" ,
53+ "IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsBatchReadArguments" ,
54+ "IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsBatchWriteArguments" ,
55+ "IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsBatchWriteArgumentsData" ,
56+ "IntegrationGoogleSheetsIntegrationDefInputSetup" ,
57+ ]
4258
4359
4460class ToolCreateParams (TypedDict , total = False ):
@@ -131,6 +147,114 @@ class APICall(TypedDict, total=False):
131147 timeout : Optional [int ]
132148
133149
150+ class IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsReadArguments (TypedDict , total = False ):
151+ range : Required [str ]
152+
153+ spreadsheet_id : Required [str ]
154+
155+ date_time_render_option : Literal ["SERIAL_NUMBER" , "FORMATTED_STRING" ]
156+
157+ major_dimension : Literal ["ROWS" , "COLUMNS" ]
158+
159+ value_render_option : Literal ["FORMATTED_VALUE" , "UNFORMATTED_VALUE" , "FORMULA" ]
160+
161+
162+ class IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsWriteArguments (TypedDict , total = False ):
163+ range : Required [str ]
164+
165+ spreadsheet_id : Required [str ]
166+
167+ values : Required [Iterable [Iterable [object ]]]
168+
169+ include_values_in_response : bool
170+
171+ insert_data_option : Literal ["OVERWRITE" , "INSERT_ROWS" ]
172+
173+ value_input_option : Literal ["RAW" , "USER_ENTERED" ]
174+
175+
176+ class IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsAppendArguments (TypedDict , total = False ):
177+ range : Required [str ]
178+
179+ spreadsheet_id : Required [str ]
180+
181+ values : Required [Iterable [Iterable [object ]]]
182+
183+ include_values_in_response : bool
184+
185+ insert_data_option : Literal ["OVERWRITE" , "INSERT_ROWS" ]
186+
187+ value_input_option : Literal ["RAW" , "USER_ENTERED" ]
188+
189+
190+ class IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsClearArguments (TypedDict , total = False ):
191+ range : Required [str ]
192+
193+ spreadsheet_id : Required [str ]
194+
195+
196+ class IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsBatchReadArguments (TypedDict , total = False ):
197+ ranges : Required [SequenceNotStr [str ]]
198+
199+ spreadsheet_id : Required [str ]
200+
201+ date_time_render_option : Literal ["SERIAL_NUMBER" , "FORMATTED_STRING" ]
202+
203+ major_dimension : Literal ["ROWS" , "COLUMNS" ]
204+
205+ value_render_option : Literal ["FORMATTED_VALUE" , "UNFORMATTED_VALUE" , "FORMULA" ]
206+
207+
208+ class IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsBatchWriteArgumentsData (TypedDict , total = False ):
209+ range : Required [str ]
210+
211+ values : Required [Iterable [Iterable [object ]]]
212+
213+ major_dimension : Literal ["ROWS" , "COLUMNS" ]
214+
215+
216+ class IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsBatchWriteArguments (TypedDict , total = False ):
217+ data : Required [Iterable [IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsBatchWriteArgumentsData ]]
218+
219+ spreadsheet_id : Required [str ]
220+
221+ include_values_in_response : bool
222+
223+ value_input_option : Literal ["RAW" , "USER_ENTERED" ]
224+
225+
226+ IntegrationGoogleSheetsIntegrationDefInputArguments : TypeAlias = Union [
227+ IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsReadArguments ,
228+ IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsWriteArguments ,
229+ IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsAppendArguments ,
230+ IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsClearArguments ,
231+ IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsBatchReadArguments ,
232+ IntegrationGoogleSheetsIntegrationDefInputArgumentsGoogleSheetsBatchWriteArguments ,
233+ ]
234+
235+
236+ class IntegrationGoogleSheetsIntegrationDefInputSetup (TypedDict , total = False ):
237+ use_julep_service : Required [bool ]
238+
239+ default_retry_count : int
240+
241+ service_account_json : Optional [str ]
242+
243+
244+ class IntegrationGoogleSheetsIntegrationDefInput (TypedDict , total = False ):
245+ arguments : Optional [IntegrationGoogleSheetsIntegrationDefInputArguments ]
246+ """Arguments for reading values from a spreadsheet"""
247+
248+ method : Optional [
249+ Literal ["read_values" , "write_values" , "append_values" , "clear_values" , "batch_read" , "batch_write" ]
250+ ]
251+
252+ provider : Literal ["google_sheets" ]
253+
254+ setup : Optional [IntegrationGoogleSheetsIntegrationDefInputSetup ]
255+ """Setup parameters for Google Sheets integration"""
256+
257+
134258Integration : TypeAlias = Union [
135259 DummyIntegrationDef ,
136260 BraveIntegrationDef ,
@@ -154,4 +278,5 @@ class APICall(TypedDict, total=False):
154278 ArxivIntegrationDef ,
155279 UnstructuredIntegrationDef ,
156280 AlgoliaIntegrationDef ,
281+ IntegrationGoogleSheetsIntegrationDefInput ,
157282]
0 commit comments