groundx 2.6.0__py3-none-any.whl → 2.7.8__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of groundx might be problematic. Click here for more details.
- groundx/__init__.py +14 -18
- groundx/client.py +3 -3
- groundx/core/client_wrapper.py +2 -2
- groundx/extract/classes/document.py +33 -16
- groundx/extract/classes/groundx.py +37 -18
- groundx/extract/services/logging_cfg.py +0 -2
- groundx/extract/services/upload.py +1 -6
- groundx/extract/services/upload_s3.py +10 -3
- groundx/extract/settings/settings.py +51 -9
- groundx/extract/settings/test_settings.py +0 -3
- groundx/ingest.py +100 -37
- groundx/types/__init__.py +10 -14
- groundx/types/workflow_detail.py +4 -0
- groundx/types/workflow_detail_chunk_strategy.py +5 -0
- groundx/types/workflow_prompt.py +1 -3
- groundx/types/workflow_prompt_role.py +1 -1
- groundx/types/{workflow_steps_doc_summary.py → workflow_request.py} +12 -4
- groundx/types/workflow_request_chunk_strategy.py +5 -0
- groundx/types/workflow_step.py +11 -4
- groundx/types/workflow_step_config.py +33 -0
- groundx/types/workflow_step_config_field.py +8 -0
- groundx/types/workflow_steps.py +12 -24
- groundx/{workflow → workflows}/__init__.py +2 -2
- groundx/{workflow → workflows}/client.py +67 -74
- groundx/{workflow → workflows}/raw_client.py +30 -23
- groundx/workflows/types/__init__.py +7 -0
- groundx/{workflow/types/workflow_get_request_id.py → workflows/types/workflows_get_request_id.py} +1 -1
- {groundx-2.6.0.dist-info → groundx-2.7.8.dist-info}/METADATA +1 -1
- {groundx-2.6.0.dist-info → groundx-2.7.8.dist-info}/RECORD +31 -33
- groundx/types/workflow_steps_chunk_instruct.py +0 -24
- groundx/types/workflow_steps_chunk_summary.py +0 -26
- groundx/types/workflow_steps_doc_keys.py +0 -22
- groundx/types/workflow_steps_search_query.py +0 -22
- groundx/types/workflow_steps_sect_instruct.py +0 -20
- groundx/types/workflow_steps_sect_summary.py +0 -23
- groundx/workflow/types/__init__.py +0 -7
- {groundx-2.6.0.dist-info → groundx-2.7.8.dist-info}/LICENSE +0 -0
- {groundx-2.6.0.dist-info → groundx-2.7.8.dist-info}/WHEEL +0 -0
|
@@ -5,28 +5,29 @@ import typing
|
|
|
5
5
|
from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
|
|
6
6
|
from ..core.request_options import RequestOptions
|
|
7
7
|
from ..types.message_response import MessageResponse
|
|
8
|
+
from ..types.workflow_request_chunk_strategy import WorkflowRequestChunkStrategy
|
|
8
9
|
from ..types.workflow_response import WorkflowResponse
|
|
9
10
|
from ..types.workflow_steps import WorkflowSteps
|
|
10
11
|
from ..types.workflows_response import WorkflowsResponse
|
|
11
|
-
from .raw_client import
|
|
12
|
-
from .types.
|
|
12
|
+
from .raw_client import AsyncRawWorkflowsClient, RawWorkflowsClient
|
|
13
|
+
from .types.workflows_get_request_id import WorkflowsGetRequestId
|
|
13
14
|
|
|
14
15
|
# this is used as the default value for optional parameters
|
|
15
16
|
OMIT = typing.cast(typing.Any, ...)
|
|
16
17
|
|
|
17
18
|
|
|
18
|
-
class
|
|
19
|
+
class WorkflowsClient:
|
|
19
20
|
def __init__(self, *, client_wrapper: SyncClientWrapper):
|
|
20
|
-
self._raw_client =
|
|
21
|
+
self._raw_client = RawWorkflowsClient(client_wrapper=client_wrapper)
|
|
21
22
|
|
|
22
23
|
@property
|
|
23
|
-
def with_raw_response(self) ->
|
|
24
|
+
def with_raw_response(self) -> RawWorkflowsClient:
|
|
24
25
|
"""
|
|
25
26
|
Retrieves a raw implementation of this client that returns raw responses.
|
|
26
27
|
|
|
27
28
|
Returns
|
|
28
29
|
-------
|
|
29
|
-
|
|
30
|
+
RawWorkflowsClient
|
|
30
31
|
"""
|
|
31
32
|
return self._raw_client
|
|
32
33
|
|
|
@@ -51,7 +52,7 @@ class WorkflowClient:
|
|
|
51
52
|
client = GroundX(
|
|
52
53
|
api_key="YOUR_API_KEY",
|
|
53
54
|
)
|
|
54
|
-
client.
|
|
55
|
+
client.workflows.list()
|
|
55
56
|
"""
|
|
56
57
|
_response = self._raw_client.list(request_options=request_options)
|
|
57
58
|
return _response.data
|
|
@@ -59,6 +60,7 @@ class WorkflowClient:
|
|
|
59
60
|
def create(
|
|
60
61
|
self,
|
|
61
62
|
*,
|
|
63
|
+
chunk_strategy: typing.Optional[WorkflowRequestChunkStrategy] = OMIT,
|
|
62
64
|
name: typing.Optional[str] = OMIT,
|
|
63
65
|
steps: typing.Optional[WorkflowSteps] = OMIT,
|
|
64
66
|
request_options: typing.Optional[RequestOptions] = None,
|
|
@@ -68,6 +70,8 @@ class WorkflowClient:
|
|
|
68
70
|
|
|
69
71
|
Parameters
|
|
70
72
|
----------
|
|
73
|
+
chunk_strategy : typing.Optional[WorkflowRequestChunkStrategy]
|
|
74
|
+
|
|
71
75
|
name : typing.Optional[str]
|
|
72
76
|
The name of the workflow being created.
|
|
73
77
|
|
|
@@ -88,12 +92,14 @@ class WorkflowClient:
|
|
|
88
92
|
client = GroundX(
|
|
89
93
|
api_key="YOUR_API_KEY",
|
|
90
94
|
)
|
|
91
|
-
client.
|
|
95
|
+
client.workflows.create()
|
|
92
96
|
"""
|
|
93
|
-
_response = self._raw_client.create(
|
|
97
|
+
_response = self._raw_client.create(
|
|
98
|
+
chunk_strategy=chunk_strategy, name=name, steps=steps, request_options=request_options
|
|
99
|
+
)
|
|
94
100
|
return _response.data
|
|
95
101
|
|
|
96
|
-
def
|
|
102
|
+
def add_to_account(
|
|
97
103
|
self, *, workflow_id: str, request_options: typing.Optional[RequestOptions] = None
|
|
98
104
|
) -> MessageResponse:
|
|
99
105
|
"""
|
|
@@ -119,18 +125,14 @@ class WorkflowClient:
|
|
|
119
125
|
client = GroundX(
|
|
120
126
|
api_key="YOUR_API_KEY",
|
|
121
127
|
)
|
|
122
|
-
client.
|
|
128
|
+
client.workflows.add_to_account(
|
|
123
129
|
workflow_id="workflowId",
|
|
124
130
|
)
|
|
125
131
|
"""
|
|
126
|
-
_response = self._raw_client.
|
|
127
|
-
workflow_id=workflow_id, request_options=request_options
|
|
128
|
-
)
|
|
132
|
+
_response = self._raw_client.add_to_account(workflow_id=workflow_id, request_options=request_options)
|
|
129
133
|
return _response.data
|
|
130
134
|
|
|
131
|
-
def
|
|
132
|
-
self, *, request_options: typing.Optional[RequestOptions] = None
|
|
133
|
-
) -> MessageResponse:
|
|
135
|
+
def remove_from_account(self, *, request_options: typing.Optional[RequestOptions] = None) -> MessageResponse:
|
|
134
136
|
"""
|
|
135
137
|
Removes the assigned workflow from the customer account.
|
|
136
138
|
|
|
@@ -151,12 +153,12 @@ class WorkflowClient:
|
|
|
151
153
|
client = GroundX(
|
|
152
154
|
api_key="YOUR_API_KEY",
|
|
153
155
|
)
|
|
154
|
-
client.
|
|
156
|
+
client.workflows.remove_from_account()
|
|
155
157
|
"""
|
|
156
|
-
_response = self._raw_client.
|
|
158
|
+
_response = self._raw_client.remove_from_account(request_options=request_options)
|
|
157
159
|
return _response.data
|
|
158
160
|
|
|
159
|
-
def
|
|
161
|
+
def add_to_id(
|
|
160
162
|
self, id: int, *, workflow_id: str, request_options: typing.Optional[RequestOptions] = None
|
|
161
163
|
) -> MessageResponse:
|
|
162
164
|
"""
|
|
@@ -185,19 +187,15 @@ class WorkflowClient:
|
|
|
185
187
|
client = GroundX(
|
|
186
188
|
api_key="YOUR_API_KEY",
|
|
187
189
|
)
|
|
188
|
-
client.
|
|
190
|
+
client.workflows.add_to_id(
|
|
189
191
|
id=1,
|
|
190
192
|
workflow_id="workflowId",
|
|
191
193
|
)
|
|
192
194
|
"""
|
|
193
|
-
_response = self._raw_client.
|
|
194
|
-
id, workflow_id=workflow_id, request_options=request_options
|
|
195
|
-
)
|
|
195
|
+
_response = self._raw_client.add_to_id(id, workflow_id=workflow_id, request_options=request_options)
|
|
196
196
|
return _response.data
|
|
197
197
|
|
|
198
|
-
def
|
|
199
|
-
self, id: int, *, request_options: typing.Optional[RequestOptions] = None
|
|
200
|
-
) -> MessageResponse:
|
|
198
|
+
def remove_from_id(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> MessageResponse:
|
|
201
199
|
"""
|
|
202
200
|
Removes the assigned workflow from the customer account.
|
|
203
201
|
|
|
@@ -221,22 +219,22 @@ class WorkflowClient:
|
|
|
221
219
|
client = GroundX(
|
|
222
220
|
api_key="YOUR_API_KEY",
|
|
223
221
|
)
|
|
224
|
-
client.
|
|
222
|
+
client.workflows.remove_from_id(
|
|
225
223
|
id=1,
|
|
226
224
|
)
|
|
227
225
|
"""
|
|
228
|
-
_response = self._raw_client.
|
|
226
|
+
_response = self._raw_client.remove_from_id(id, request_options=request_options)
|
|
229
227
|
return _response.data
|
|
230
228
|
|
|
231
229
|
def get(
|
|
232
|
-
self, id:
|
|
230
|
+
self, id: WorkflowsGetRequestId, *, request_options: typing.Optional[RequestOptions] = None
|
|
233
231
|
) -> WorkflowResponse:
|
|
234
232
|
"""
|
|
235
233
|
look up a specific workflow by groupId, bucketId, or workflowId.
|
|
236
234
|
|
|
237
235
|
Parameters
|
|
238
236
|
----------
|
|
239
|
-
id :
|
|
237
|
+
id : WorkflowsGetRequestId
|
|
240
238
|
The id of the group, bucket, or workflow to look up.
|
|
241
239
|
|
|
242
240
|
request_options : typing.Optional[RequestOptions]
|
|
@@ -254,7 +252,7 @@ class WorkflowClient:
|
|
|
254
252
|
client = GroundX(
|
|
255
253
|
api_key="YOUR_API_KEY",
|
|
256
254
|
)
|
|
257
|
-
client.
|
|
255
|
+
client.workflows.get(
|
|
258
256
|
id=1,
|
|
259
257
|
)
|
|
260
258
|
"""
|
|
@@ -265,7 +263,7 @@ class WorkflowClient:
|
|
|
265
263
|
self,
|
|
266
264
|
id: str,
|
|
267
265
|
*,
|
|
268
|
-
|
|
266
|
+
chunk_strategy: typing.Optional[WorkflowRequestChunkStrategy] = OMIT,
|
|
269
267
|
name: typing.Optional[str] = OMIT,
|
|
270
268
|
steps: typing.Optional[WorkflowSteps] = OMIT,
|
|
271
269
|
request_options: typing.Optional[RequestOptions] = None,
|
|
@@ -278,8 +276,7 @@ class WorkflowClient:
|
|
|
278
276
|
id : str
|
|
279
277
|
The workflowId of the workflow being updated.
|
|
280
278
|
|
|
281
|
-
|
|
282
|
-
The id of the workflow that is being updated.
|
|
279
|
+
chunk_strategy : typing.Optional[WorkflowRequestChunkStrategy]
|
|
283
280
|
|
|
284
281
|
name : typing.Optional[str]
|
|
285
282
|
The name of the workflow being created.
|
|
@@ -301,13 +298,12 @@ class WorkflowClient:
|
|
|
301
298
|
client = GroundX(
|
|
302
299
|
api_key="YOUR_API_KEY",
|
|
303
300
|
)
|
|
304
|
-
client.
|
|
301
|
+
client.workflows.update(
|
|
305
302
|
id="id",
|
|
306
|
-
workflow_id="workflowId",
|
|
307
303
|
)
|
|
308
304
|
"""
|
|
309
305
|
_response = self._raw_client.update(
|
|
310
|
-
id,
|
|
306
|
+
id, chunk_strategy=chunk_strategy, name=name, steps=steps, request_options=request_options
|
|
311
307
|
)
|
|
312
308
|
return _response.data
|
|
313
309
|
|
|
@@ -335,7 +331,7 @@ class WorkflowClient:
|
|
|
335
331
|
client = GroundX(
|
|
336
332
|
api_key="YOUR_API_KEY",
|
|
337
333
|
)
|
|
338
|
-
client.
|
|
334
|
+
client.workflows.delete(
|
|
339
335
|
id="id",
|
|
340
336
|
)
|
|
341
337
|
"""
|
|
@@ -343,18 +339,18 @@ class WorkflowClient:
|
|
|
343
339
|
return _response.data
|
|
344
340
|
|
|
345
341
|
|
|
346
|
-
class
|
|
342
|
+
class AsyncWorkflowsClient:
|
|
347
343
|
def __init__(self, *, client_wrapper: AsyncClientWrapper):
|
|
348
|
-
self._raw_client =
|
|
344
|
+
self._raw_client = AsyncRawWorkflowsClient(client_wrapper=client_wrapper)
|
|
349
345
|
|
|
350
346
|
@property
|
|
351
|
-
def with_raw_response(self) ->
|
|
347
|
+
def with_raw_response(self) -> AsyncRawWorkflowsClient:
|
|
352
348
|
"""
|
|
353
349
|
Retrieves a raw implementation of this client that returns raw responses.
|
|
354
350
|
|
|
355
351
|
Returns
|
|
356
352
|
-------
|
|
357
|
-
|
|
353
|
+
AsyncRawWorkflowsClient
|
|
358
354
|
"""
|
|
359
355
|
return self._raw_client
|
|
360
356
|
|
|
@@ -384,7 +380,7 @@ class AsyncWorkflowClient:
|
|
|
384
380
|
|
|
385
381
|
|
|
386
382
|
async def main() -> None:
|
|
387
|
-
await client.
|
|
383
|
+
await client.workflows.list()
|
|
388
384
|
|
|
389
385
|
|
|
390
386
|
asyncio.run(main())
|
|
@@ -395,6 +391,7 @@ class AsyncWorkflowClient:
|
|
|
395
391
|
async def create(
|
|
396
392
|
self,
|
|
397
393
|
*,
|
|
394
|
+
chunk_strategy: typing.Optional[WorkflowRequestChunkStrategy] = OMIT,
|
|
398
395
|
name: typing.Optional[str] = OMIT,
|
|
399
396
|
steps: typing.Optional[WorkflowSteps] = OMIT,
|
|
400
397
|
request_options: typing.Optional[RequestOptions] = None,
|
|
@@ -404,6 +401,8 @@ class AsyncWorkflowClient:
|
|
|
404
401
|
|
|
405
402
|
Parameters
|
|
406
403
|
----------
|
|
404
|
+
chunk_strategy : typing.Optional[WorkflowRequestChunkStrategy]
|
|
405
|
+
|
|
407
406
|
name : typing.Optional[str]
|
|
408
407
|
The name of the workflow being created.
|
|
409
408
|
|
|
@@ -429,15 +428,17 @@ class AsyncWorkflowClient:
|
|
|
429
428
|
|
|
430
429
|
|
|
431
430
|
async def main() -> None:
|
|
432
|
-
await client.
|
|
431
|
+
await client.workflows.create()
|
|
433
432
|
|
|
434
433
|
|
|
435
434
|
asyncio.run(main())
|
|
436
435
|
"""
|
|
437
|
-
_response = await self._raw_client.create(
|
|
436
|
+
_response = await self._raw_client.create(
|
|
437
|
+
chunk_strategy=chunk_strategy, name=name, steps=steps, request_options=request_options
|
|
438
|
+
)
|
|
438
439
|
return _response.data
|
|
439
440
|
|
|
440
|
-
async def
|
|
441
|
+
async def add_to_account(
|
|
441
442
|
self, *, workflow_id: str, request_options: typing.Optional[RequestOptions] = None
|
|
442
443
|
) -> MessageResponse:
|
|
443
444
|
"""
|
|
@@ -468,21 +469,17 @@ class AsyncWorkflowClient:
|
|
|
468
469
|
|
|
469
470
|
|
|
470
471
|
async def main() -> None:
|
|
471
|
-
await client.
|
|
472
|
+
await client.workflows.add_to_account(
|
|
472
473
|
workflow_id="workflowId",
|
|
473
474
|
)
|
|
474
475
|
|
|
475
476
|
|
|
476
477
|
asyncio.run(main())
|
|
477
478
|
"""
|
|
478
|
-
_response = await self._raw_client.
|
|
479
|
-
workflow_id=workflow_id, request_options=request_options
|
|
480
|
-
)
|
|
479
|
+
_response = await self._raw_client.add_to_account(workflow_id=workflow_id, request_options=request_options)
|
|
481
480
|
return _response.data
|
|
482
481
|
|
|
483
|
-
async def
|
|
484
|
-
self, *, request_options: typing.Optional[RequestOptions] = None
|
|
485
|
-
) -> MessageResponse:
|
|
482
|
+
async def remove_from_account(self, *, request_options: typing.Optional[RequestOptions] = None) -> MessageResponse:
|
|
486
483
|
"""
|
|
487
484
|
Removes the assigned workflow from the customer account.
|
|
488
485
|
|
|
@@ -508,15 +505,15 @@ class AsyncWorkflowClient:
|
|
|
508
505
|
|
|
509
506
|
|
|
510
507
|
async def main() -> None:
|
|
511
|
-
await client.
|
|
508
|
+
await client.workflows.remove_from_account()
|
|
512
509
|
|
|
513
510
|
|
|
514
511
|
asyncio.run(main())
|
|
515
512
|
"""
|
|
516
|
-
_response = await self._raw_client.
|
|
513
|
+
_response = await self._raw_client.remove_from_account(request_options=request_options)
|
|
517
514
|
return _response.data
|
|
518
515
|
|
|
519
|
-
async def
|
|
516
|
+
async def add_to_id(
|
|
520
517
|
self, id: int, *, workflow_id: str, request_options: typing.Optional[RequestOptions] = None
|
|
521
518
|
) -> MessageResponse:
|
|
522
519
|
"""
|
|
@@ -550,7 +547,7 @@ class AsyncWorkflowClient:
|
|
|
550
547
|
|
|
551
548
|
|
|
552
549
|
async def main() -> None:
|
|
553
|
-
await client.
|
|
550
|
+
await client.workflows.add_to_id(
|
|
554
551
|
id=1,
|
|
555
552
|
workflow_id="workflowId",
|
|
556
553
|
)
|
|
@@ -558,12 +555,10 @@ class AsyncWorkflowClient:
|
|
|
558
555
|
|
|
559
556
|
asyncio.run(main())
|
|
560
557
|
"""
|
|
561
|
-
_response = await self._raw_client.
|
|
562
|
-
id, workflow_id=workflow_id, request_options=request_options
|
|
563
|
-
)
|
|
558
|
+
_response = await self._raw_client.add_to_id(id, workflow_id=workflow_id, request_options=request_options)
|
|
564
559
|
return _response.data
|
|
565
560
|
|
|
566
|
-
async def
|
|
561
|
+
async def remove_from_id(
|
|
567
562
|
self, id: int, *, request_options: typing.Optional[RequestOptions] = None
|
|
568
563
|
) -> MessageResponse:
|
|
569
564
|
"""
|
|
@@ -594,25 +589,25 @@ class AsyncWorkflowClient:
|
|
|
594
589
|
|
|
595
590
|
|
|
596
591
|
async def main() -> None:
|
|
597
|
-
await client.
|
|
592
|
+
await client.workflows.remove_from_id(
|
|
598
593
|
id=1,
|
|
599
594
|
)
|
|
600
595
|
|
|
601
596
|
|
|
602
597
|
asyncio.run(main())
|
|
603
598
|
"""
|
|
604
|
-
_response = await self._raw_client.
|
|
599
|
+
_response = await self._raw_client.remove_from_id(id, request_options=request_options)
|
|
605
600
|
return _response.data
|
|
606
601
|
|
|
607
602
|
async def get(
|
|
608
|
-
self, id:
|
|
603
|
+
self, id: WorkflowsGetRequestId, *, request_options: typing.Optional[RequestOptions] = None
|
|
609
604
|
) -> WorkflowResponse:
|
|
610
605
|
"""
|
|
611
606
|
look up a specific workflow by groupId, bucketId, or workflowId.
|
|
612
607
|
|
|
613
608
|
Parameters
|
|
614
609
|
----------
|
|
615
|
-
id :
|
|
610
|
+
id : WorkflowsGetRequestId
|
|
616
611
|
The id of the group, bucket, or workflow to look up.
|
|
617
612
|
|
|
618
613
|
request_options : typing.Optional[RequestOptions]
|
|
@@ -635,7 +630,7 @@ class AsyncWorkflowClient:
|
|
|
635
630
|
|
|
636
631
|
|
|
637
632
|
async def main() -> None:
|
|
638
|
-
await client.
|
|
633
|
+
await client.workflows.get(
|
|
639
634
|
id=1,
|
|
640
635
|
)
|
|
641
636
|
|
|
@@ -649,7 +644,7 @@ class AsyncWorkflowClient:
|
|
|
649
644
|
self,
|
|
650
645
|
id: str,
|
|
651
646
|
*,
|
|
652
|
-
|
|
647
|
+
chunk_strategy: typing.Optional[WorkflowRequestChunkStrategy] = OMIT,
|
|
653
648
|
name: typing.Optional[str] = OMIT,
|
|
654
649
|
steps: typing.Optional[WorkflowSteps] = OMIT,
|
|
655
650
|
request_options: typing.Optional[RequestOptions] = None,
|
|
@@ -662,8 +657,7 @@ class AsyncWorkflowClient:
|
|
|
662
657
|
id : str
|
|
663
658
|
The workflowId of the workflow being updated.
|
|
664
659
|
|
|
665
|
-
|
|
666
|
-
The id of the workflow that is being updated.
|
|
660
|
+
chunk_strategy : typing.Optional[WorkflowRequestChunkStrategy]
|
|
667
661
|
|
|
668
662
|
name : typing.Optional[str]
|
|
669
663
|
The name of the workflow being created.
|
|
@@ -690,16 +684,15 @@ class AsyncWorkflowClient:
|
|
|
690
684
|
|
|
691
685
|
|
|
692
686
|
async def main() -> None:
|
|
693
|
-
await client.
|
|
687
|
+
await client.workflows.update(
|
|
694
688
|
id="id",
|
|
695
|
-
workflow_id="workflowId",
|
|
696
689
|
)
|
|
697
690
|
|
|
698
691
|
|
|
699
692
|
asyncio.run(main())
|
|
700
693
|
"""
|
|
701
694
|
_response = await self._raw_client.update(
|
|
702
|
-
id,
|
|
695
|
+
id, chunk_strategy=chunk_strategy, name=name, steps=steps, request_options=request_options
|
|
703
696
|
)
|
|
704
697
|
return _response.data
|
|
705
698
|
|
|
@@ -732,7 +725,7 @@ class AsyncWorkflowClient:
|
|
|
732
725
|
|
|
733
726
|
|
|
734
727
|
async def main() -> None:
|
|
735
|
-
await client.
|
|
728
|
+
await client.workflows.delete(
|
|
736
729
|
id="id",
|
|
737
730
|
)
|
|
738
731
|
|
|
@@ -11,16 +11,17 @@ from ..core.pydantic_utilities import parse_obj_as
|
|
|
11
11
|
from ..core.request_options import RequestOptions
|
|
12
12
|
from ..core.serialization import convert_and_respect_annotation_metadata
|
|
13
13
|
from ..types.message_response import MessageResponse
|
|
14
|
+
from ..types.workflow_request_chunk_strategy import WorkflowRequestChunkStrategy
|
|
14
15
|
from ..types.workflow_response import WorkflowResponse
|
|
15
16
|
from ..types.workflow_steps import WorkflowSteps
|
|
16
17
|
from ..types.workflows_response import WorkflowsResponse
|
|
17
|
-
from .types.
|
|
18
|
+
from .types.workflows_get_request_id import WorkflowsGetRequestId
|
|
18
19
|
|
|
19
20
|
# this is used as the default value for optional parameters
|
|
20
21
|
OMIT = typing.cast(typing.Any, ...)
|
|
21
22
|
|
|
22
23
|
|
|
23
|
-
class
|
|
24
|
+
class RawWorkflowsClient:
|
|
24
25
|
def __init__(self, *, client_wrapper: SyncClientWrapper):
|
|
25
26
|
self._client_wrapper = client_wrapper
|
|
26
27
|
|
|
@@ -61,6 +62,7 @@ class RawWorkflowClient:
|
|
|
61
62
|
def create(
|
|
62
63
|
self,
|
|
63
64
|
*,
|
|
65
|
+
chunk_strategy: typing.Optional[WorkflowRequestChunkStrategy] = OMIT,
|
|
64
66
|
name: typing.Optional[str] = OMIT,
|
|
65
67
|
steps: typing.Optional[WorkflowSteps] = OMIT,
|
|
66
68
|
request_options: typing.Optional[RequestOptions] = None,
|
|
@@ -70,6 +72,8 @@ class RawWorkflowClient:
|
|
|
70
72
|
|
|
71
73
|
Parameters
|
|
72
74
|
----------
|
|
75
|
+
chunk_strategy : typing.Optional[WorkflowRequestChunkStrategy]
|
|
76
|
+
|
|
73
77
|
name : typing.Optional[str]
|
|
74
78
|
The name of the workflow being created.
|
|
75
79
|
|
|
@@ -87,6 +91,7 @@ class RawWorkflowClient:
|
|
|
87
91
|
"v1/workflow",
|
|
88
92
|
method="POST",
|
|
89
93
|
json={
|
|
94
|
+
"chunkStrategy": chunk_strategy,
|
|
90
95
|
"name": name,
|
|
91
96
|
"steps": convert_and_respect_annotation_metadata(
|
|
92
97
|
object_=steps, annotation=WorkflowSteps, direction="write"
|
|
@@ -113,7 +118,7 @@ class RawWorkflowClient:
|
|
|
113
118
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
|
|
114
119
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
|
|
115
120
|
|
|
116
|
-
def
|
|
121
|
+
def add_to_account(
|
|
117
122
|
self, *, workflow_id: str, request_options: typing.Optional[RequestOptions] = None
|
|
118
123
|
) -> HttpResponse[MessageResponse]:
|
|
119
124
|
"""
|
|
@@ -159,7 +164,7 @@ class RawWorkflowClient:
|
|
|
159
164
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
|
|
160
165
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
|
|
161
166
|
|
|
162
|
-
def
|
|
167
|
+
def remove_from_account(
|
|
163
168
|
self, *, request_options: typing.Optional[RequestOptions] = None
|
|
164
169
|
) -> HttpResponse[MessageResponse]:
|
|
165
170
|
"""
|
|
@@ -195,7 +200,7 @@ class RawWorkflowClient:
|
|
|
195
200
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
|
|
196
201
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
|
|
197
202
|
|
|
198
|
-
def
|
|
203
|
+
def add_to_id(
|
|
199
204
|
self, id: int, *, workflow_id: str, request_options: typing.Optional[RequestOptions] = None
|
|
200
205
|
) -> HttpResponse[MessageResponse]:
|
|
201
206
|
"""
|
|
@@ -244,7 +249,7 @@ class RawWorkflowClient:
|
|
|
244
249
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
|
|
245
250
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
|
|
246
251
|
|
|
247
|
-
def
|
|
252
|
+
def remove_from_id(
|
|
248
253
|
self, id: int, *, request_options: typing.Optional[RequestOptions] = None
|
|
249
254
|
) -> HttpResponse[MessageResponse]:
|
|
250
255
|
"""
|
|
@@ -284,14 +289,14 @@ class RawWorkflowClient:
|
|
|
284
289
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
|
|
285
290
|
|
|
286
291
|
def get(
|
|
287
|
-
self, id:
|
|
292
|
+
self, id: WorkflowsGetRequestId, *, request_options: typing.Optional[RequestOptions] = None
|
|
288
293
|
) -> HttpResponse[WorkflowResponse]:
|
|
289
294
|
"""
|
|
290
295
|
look up a specific workflow by groupId, bucketId, or workflowId.
|
|
291
296
|
|
|
292
297
|
Parameters
|
|
293
298
|
----------
|
|
294
|
-
id :
|
|
299
|
+
id : WorkflowsGetRequestId
|
|
295
300
|
The id of the group, bucket, or workflow to look up.
|
|
296
301
|
|
|
297
302
|
request_options : typing.Optional[RequestOptions]
|
|
@@ -326,7 +331,7 @@ class RawWorkflowClient:
|
|
|
326
331
|
self,
|
|
327
332
|
id: str,
|
|
328
333
|
*,
|
|
329
|
-
|
|
334
|
+
chunk_strategy: typing.Optional[WorkflowRequestChunkStrategy] = OMIT,
|
|
330
335
|
name: typing.Optional[str] = OMIT,
|
|
331
336
|
steps: typing.Optional[WorkflowSteps] = OMIT,
|
|
332
337
|
request_options: typing.Optional[RequestOptions] = None,
|
|
@@ -339,8 +344,7 @@ class RawWorkflowClient:
|
|
|
339
344
|
id : str
|
|
340
345
|
The workflowId of the workflow being updated.
|
|
341
346
|
|
|
342
|
-
|
|
343
|
-
The id of the workflow that is being updated.
|
|
347
|
+
chunk_strategy : typing.Optional[WorkflowRequestChunkStrategy]
|
|
344
348
|
|
|
345
349
|
name : typing.Optional[str]
|
|
346
350
|
The name of the workflow being created.
|
|
@@ -359,11 +363,11 @@ class RawWorkflowClient:
|
|
|
359
363
|
f"v1/workflow/{jsonable_encoder(id)}",
|
|
360
364
|
method="PUT",
|
|
361
365
|
json={
|
|
366
|
+
"chunkStrategy": chunk_strategy,
|
|
362
367
|
"name": name,
|
|
363
368
|
"steps": convert_and_respect_annotation_metadata(
|
|
364
369
|
object_=steps, annotation=WorkflowSteps, direction="write"
|
|
365
370
|
),
|
|
366
|
-
"workflowId": workflow_id,
|
|
367
371
|
},
|
|
368
372
|
headers={
|
|
369
373
|
"content-type": "application/json",
|
|
@@ -426,7 +430,7 @@ class RawWorkflowClient:
|
|
|
426
430
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
|
|
427
431
|
|
|
428
432
|
|
|
429
|
-
class
|
|
433
|
+
class AsyncRawWorkflowsClient:
|
|
430
434
|
def __init__(self, *, client_wrapper: AsyncClientWrapper):
|
|
431
435
|
self._client_wrapper = client_wrapper
|
|
432
436
|
|
|
@@ -469,6 +473,7 @@ class AsyncRawWorkflowClient:
|
|
|
469
473
|
async def create(
|
|
470
474
|
self,
|
|
471
475
|
*,
|
|
476
|
+
chunk_strategy: typing.Optional[WorkflowRequestChunkStrategy] = OMIT,
|
|
472
477
|
name: typing.Optional[str] = OMIT,
|
|
473
478
|
steps: typing.Optional[WorkflowSteps] = OMIT,
|
|
474
479
|
request_options: typing.Optional[RequestOptions] = None,
|
|
@@ -478,6 +483,8 @@ class AsyncRawWorkflowClient:
|
|
|
478
483
|
|
|
479
484
|
Parameters
|
|
480
485
|
----------
|
|
486
|
+
chunk_strategy : typing.Optional[WorkflowRequestChunkStrategy]
|
|
487
|
+
|
|
481
488
|
name : typing.Optional[str]
|
|
482
489
|
The name of the workflow being created.
|
|
483
490
|
|
|
@@ -495,6 +502,7 @@ class AsyncRawWorkflowClient:
|
|
|
495
502
|
"v1/workflow",
|
|
496
503
|
method="POST",
|
|
497
504
|
json={
|
|
505
|
+
"chunkStrategy": chunk_strategy,
|
|
498
506
|
"name": name,
|
|
499
507
|
"steps": convert_and_respect_annotation_metadata(
|
|
500
508
|
object_=steps, annotation=WorkflowSteps, direction="write"
|
|
@@ -521,7 +529,7 @@ class AsyncRawWorkflowClient:
|
|
|
521
529
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
|
|
522
530
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
|
|
523
531
|
|
|
524
|
-
async def
|
|
532
|
+
async def add_to_account(
|
|
525
533
|
self, *, workflow_id: str, request_options: typing.Optional[RequestOptions] = None
|
|
526
534
|
) -> AsyncHttpResponse[MessageResponse]:
|
|
527
535
|
"""
|
|
@@ -567,7 +575,7 @@ class AsyncRawWorkflowClient:
|
|
|
567
575
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
|
|
568
576
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
|
|
569
577
|
|
|
570
|
-
async def
|
|
578
|
+
async def remove_from_account(
|
|
571
579
|
self, *, request_options: typing.Optional[RequestOptions] = None
|
|
572
580
|
) -> AsyncHttpResponse[MessageResponse]:
|
|
573
581
|
"""
|
|
@@ -603,7 +611,7 @@ class AsyncRawWorkflowClient:
|
|
|
603
611
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
|
|
604
612
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
|
|
605
613
|
|
|
606
|
-
async def
|
|
614
|
+
async def add_to_id(
|
|
607
615
|
self, id: int, *, workflow_id: str, request_options: typing.Optional[RequestOptions] = None
|
|
608
616
|
) -> AsyncHttpResponse[MessageResponse]:
|
|
609
617
|
"""
|
|
@@ -652,7 +660,7 @@ class AsyncRawWorkflowClient:
|
|
|
652
660
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
|
|
653
661
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
|
|
654
662
|
|
|
655
|
-
async def
|
|
663
|
+
async def remove_from_id(
|
|
656
664
|
self, id: int, *, request_options: typing.Optional[RequestOptions] = None
|
|
657
665
|
) -> AsyncHttpResponse[MessageResponse]:
|
|
658
666
|
"""
|
|
@@ -692,14 +700,14 @@ class AsyncRawWorkflowClient:
|
|
|
692
700
|
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
|
|
693
701
|
|
|
694
702
|
async def get(
|
|
695
|
-
self, id:
|
|
703
|
+
self, id: WorkflowsGetRequestId, *, request_options: typing.Optional[RequestOptions] = None
|
|
696
704
|
) -> AsyncHttpResponse[WorkflowResponse]:
|
|
697
705
|
"""
|
|
698
706
|
look up a specific workflow by groupId, bucketId, or workflowId.
|
|
699
707
|
|
|
700
708
|
Parameters
|
|
701
709
|
----------
|
|
702
|
-
id :
|
|
710
|
+
id : WorkflowsGetRequestId
|
|
703
711
|
The id of the group, bucket, or workflow to look up.
|
|
704
712
|
|
|
705
713
|
request_options : typing.Optional[RequestOptions]
|
|
@@ -734,7 +742,7 @@ class AsyncRawWorkflowClient:
|
|
|
734
742
|
self,
|
|
735
743
|
id: str,
|
|
736
744
|
*,
|
|
737
|
-
|
|
745
|
+
chunk_strategy: typing.Optional[WorkflowRequestChunkStrategy] = OMIT,
|
|
738
746
|
name: typing.Optional[str] = OMIT,
|
|
739
747
|
steps: typing.Optional[WorkflowSteps] = OMIT,
|
|
740
748
|
request_options: typing.Optional[RequestOptions] = None,
|
|
@@ -747,8 +755,7 @@ class AsyncRawWorkflowClient:
|
|
|
747
755
|
id : str
|
|
748
756
|
The workflowId of the workflow being updated.
|
|
749
757
|
|
|
750
|
-
|
|
751
|
-
The id of the workflow that is being updated.
|
|
758
|
+
chunk_strategy : typing.Optional[WorkflowRequestChunkStrategy]
|
|
752
759
|
|
|
753
760
|
name : typing.Optional[str]
|
|
754
761
|
The name of the workflow being created.
|
|
@@ -767,11 +774,11 @@ class AsyncRawWorkflowClient:
|
|
|
767
774
|
f"v1/workflow/{jsonable_encoder(id)}",
|
|
768
775
|
method="PUT",
|
|
769
776
|
json={
|
|
777
|
+
"chunkStrategy": chunk_strategy,
|
|
770
778
|
"name": name,
|
|
771
779
|
"steps": convert_and_respect_annotation_metadata(
|
|
772
780
|
object_=steps, annotation=WorkflowSteps, direction="write"
|
|
773
781
|
),
|
|
774
|
-
"workflowId": workflow_id,
|
|
775
782
|
},
|
|
776
783
|
headers={
|
|
777
784
|
"content-type": "application/json",
|