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.

Files changed (38) hide show
  1. groundx/__init__.py +14 -18
  2. groundx/client.py +3 -3
  3. groundx/core/client_wrapper.py +2 -2
  4. groundx/extract/classes/document.py +33 -16
  5. groundx/extract/classes/groundx.py +37 -18
  6. groundx/extract/services/logging_cfg.py +0 -2
  7. groundx/extract/services/upload.py +1 -6
  8. groundx/extract/services/upload_s3.py +10 -3
  9. groundx/extract/settings/settings.py +51 -9
  10. groundx/extract/settings/test_settings.py +0 -3
  11. groundx/ingest.py +100 -37
  12. groundx/types/__init__.py +10 -14
  13. groundx/types/workflow_detail.py +4 -0
  14. groundx/types/workflow_detail_chunk_strategy.py +5 -0
  15. groundx/types/workflow_prompt.py +1 -3
  16. groundx/types/workflow_prompt_role.py +1 -1
  17. groundx/types/{workflow_steps_doc_summary.py → workflow_request.py} +12 -4
  18. groundx/types/workflow_request_chunk_strategy.py +5 -0
  19. groundx/types/workflow_step.py +11 -4
  20. groundx/types/workflow_step_config.py +33 -0
  21. groundx/types/workflow_step_config_field.py +8 -0
  22. groundx/types/workflow_steps.py +12 -24
  23. groundx/{workflow → workflows}/__init__.py +2 -2
  24. groundx/{workflow → workflows}/client.py +67 -74
  25. groundx/{workflow → workflows}/raw_client.py +30 -23
  26. groundx/workflows/types/__init__.py +7 -0
  27. groundx/{workflow/types/workflow_get_request_id.py → workflows/types/workflows_get_request_id.py} +1 -1
  28. {groundx-2.6.0.dist-info → groundx-2.7.8.dist-info}/METADATA +1 -1
  29. {groundx-2.6.0.dist-info → groundx-2.7.8.dist-info}/RECORD +31 -33
  30. groundx/types/workflow_steps_chunk_instruct.py +0 -24
  31. groundx/types/workflow_steps_chunk_summary.py +0 -26
  32. groundx/types/workflow_steps_doc_keys.py +0 -22
  33. groundx/types/workflow_steps_search_query.py +0 -22
  34. groundx/types/workflow_steps_sect_instruct.py +0 -20
  35. groundx/types/workflow_steps_sect_summary.py +0 -23
  36. groundx/workflow/types/__init__.py +0 -7
  37. {groundx-2.6.0.dist-info → groundx-2.7.8.dist-info}/LICENSE +0 -0
  38. {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 AsyncRawWorkflowClient, RawWorkflowClient
12
- from .types.workflow_get_request_id import WorkflowGetRequestId
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 WorkflowClient:
19
+ class WorkflowsClient:
19
20
  def __init__(self, *, client_wrapper: SyncClientWrapper):
20
- self._raw_client = RawWorkflowClient(client_wrapper=client_wrapper)
21
+ self._raw_client = RawWorkflowsClient(client_wrapper=client_wrapper)
21
22
 
22
23
  @property
23
- def with_raw_response(self) -> RawWorkflowClient:
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
- RawWorkflowClient
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.workflow.list()
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.workflow.create()
95
+ client.workflows.create()
92
96
  """
93
- _response = self._raw_client.create(name=name, steps=steps, request_options=request_options)
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 relationship_add_to_account(
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.workflow.relationship_add_to_account(
128
+ client.workflows.add_to_account(
123
129
  workflow_id="workflowId",
124
130
  )
125
131
  """
126
- _response = self._raw_client.relationship_add_to_account(
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 relationship_remove_from_account(
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.workflow.relationship_remove_from_account()
156
+ client.workflows.remove_from_account()
155
157
  """
156
- _response = self._raw_client.relationship_remove_from_account(request_options=request_options)
158
+ _response = self._raw_client.remove_from_account(request_options=request_options)
157
159
  return _response.data
158
160
 
159
- def relationship_add_to_id(
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.workflow.relationship_add_to_id(
190
+ client.workflows.add_to_id(
189
191
  id=1,
190
192
  workflow_id="workflowId",
191
193
  )
192
194
  """
193
- _response = self._raw_client.relationship_add_to_id(
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 relationship_remove_from_id(
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.workflow.relationship_remove_from_id(
222
+ client.workflows.remove_from_id(
225
223
  id=1,
226
224
  )
227
225
  """
228
- _response = self._raw_client.relationship_remove_from_id(id, request_options=request_options)
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: WorkflowGetRequestId, *, request_options: typing.Optional[RequestOptions] = None
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 : WorkflowGetRequestId
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.workflow.get(
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
- workflow_id: str,
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
- workflow_id : str
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.workflow.update(
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, workflow_id=workflow_id, name=name, steps=steps, request_options=request_options
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.workflow.delete(
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 AsyncWorkflowClient:
342
+ class AsyncWorkflowsClient:
347
343
  def __init__(self, *, client_wrapper: AsyncClientWrapper):
348
- self._raw_client = AsyncRawWorkflowClient(client_wrapper=client_wrapper)
344
+ self._raw_client = AsyncRawWorkflowsClient(client_wrapper=client_wrapper)
349
345
 
350
346
  @property
351
- def with_raw_response(self) -> AsyncRawWorkflowClient:
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
- AsyncRawWorkflowClient
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.workflow.list()
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.workflow.create()
431
+ await client.workflows.create()
433
432
 
434
433
 
435
434
  asyncio.run(main())
436
435
  """
437
- _response = await self._raw_client.create(name=name, steps=steps, request_options=request_options)
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 relationship_add_to_account(
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.workflow.relationship_add_to_account(
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.relationship_add_to_account(
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 relationship_remove_from_account(
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.workflow.relationship_remove_from_account()
508
+ await client.workflows.remove_from_account()
512
509
 
513
510
 
514
511
  asyncio.run(main())
515
512
  """
516
- _response = await self._raw_client.relationship_remove_from_account(request_options=request_options)
513
+ _response = await self._raw_client.remove_from_account(request_options=request_options)
517
514
  return _response.data
518
515
 
519
- async def relationship_add_to_id(
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.workflow.relationship_add_to_id(
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.relationship_add_to_id(
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 relationship_remove_from_id(
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.workflow.relationship_remove_from_id(
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.relationship_remove_from_id(id, request_options=request_options)
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: WorkflowGetRequestId, *, request_options: typing.Optional[RequestOptions] = None
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 : WorkflowGetRequestId
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.workflow.get(
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
- workflow_id: str,
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
- workflow_id : str
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.workflow.update(
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, workflow_id=workflow_id, name=name, steps=steps, request_options=request_options
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.workflow.delete(
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.workflow_get_request_id import WorkflowGetRequestId
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 RawWorkflowClient:
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 relationship_add_to_account(
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 relationship_remove_from_account(
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 relationship_add_to_id(
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 relationship_remove_from_id(
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: WorkflowGetRequestId, *, request_options: typing.Optional[RequestOptions] = None
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 : WorkflowGetRequestId
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
- workflow_id: str,
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
- workflow_id : str
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 AsyncRawWorkflowClient:
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 relationship_add_to_account(
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 relationship_remove_from_account(
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 relationship_add_to_id(
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 relationship_remove_from_id(
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: WorkflowGetRequestId, *, request_options: typing.Optional[RequestOptions] = None
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 : WorkflowGetRequestId
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
- workflow_id: str,
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
- workflow_id : str
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",
@@ -0,0 +1,7 @@
1
+ # This file was auto-generated by Fern from our API Definition.
2
+
3
+ # isort: skip_file
4
+
5
+ from .workflows_get_request_id import WorkflowsGetRequestId
6
+
7
+ __all__ = ["WorkflowsGetRequestId"]