openparser-sdk 0.0.1__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.
Files changed (173) hide show
  1. openparser/__init__.py +54 -0
  2. openparser/_files.py +43 -0
  3. openparser/_generated/__init__.py +8 -0
  4. openparser/_generated/api/__init__.py +1 -0
  5. openparser/_generated/api/extract/__init__.py +1 -0
  6. openparser/_generated/api/extract/extract_async.py +280 -0
  7. openparser/_generated/api/extract/extract_batch.py +284 -0
  8. openparser/_generated/api/extract/extract_sync.py +315 -0
  9. openparser/_generated/api/extract/suggest_schema.py +260 -0
  10. openparser/_generated/api/files/__init__.py +1 -0
  11. openparser/_generated/api/files/create_file.py +237 -0
  12. openparser/_generated/api/files/delete_file.py +200 -0
  13. openparser/_generated/api/files/get_file.py +200 -0
  14. openparser/_generated/api/files/get_file_content.py +198 -0
  15. openparser/_generated/api/jobs/__init__.py +1 -0
  16. openparser/_generated/api/jobs/get_job.py +261 -0
  17. openparser/_generated/api/jobs/get_job_result.py +305 -0
  18. openparser/_generated/api/jobs/get_job_source.py +233 -0
  19. openparser/_generated/api/jobs/list_jobs.py +340 -0
  20. openparser/_generated/api/models/__init__.py +1 -0
  21. openparser/_generated/api/models/list_llm_models.py +288 -0
  22. openparser/_generated/api/models/list_ocr_models.py +184 -0
  23. openparser/_generated/api/parse/__init__.py +1 -0
  24. openparser/_generated/api/parse/parse_async.py +316 -0
  25. openparser/_generated/api/parse/parse_batch.py +280 -0
  26. openparser/_generated/api/parse/parse_sync.py +369 -0
  27. openparser/_generated/api/pipelines/__init__.py +1 -0
  28. openparser/_generated/api/pipelines/create_extraction_pipeline.py +242 -0
  29. openparser/_generated/api/pipelines/delete_extraction_pipeline.py +212 -0
  30. openparser/_generated/api/pipelines/get_extraction_pipeline.py +212 -0
  31. openparser/_generated/api/pipelines/list_extraction_pipelines.py +180 -0
  32. openparser/_generated/api/pipelines/update_extraction_pipeline.py +258 -0
  33. openparser/_generated/client.py +270 -0
  34. openparser/_generated/errors.py +14 -0
  35. openparser/_generated/models/__init__.py +261 -0
  36. openparser/_generated/models/batch_child_page.py +97 -0
  37. openparser/_generated/models/batch_child_summary.py +195 -0
  38. openparser/_generated/models/batch_job_accepted.py +139 -0
  39. openparser/_generated/models/batch_job_accepted_operation.py +8 -0
  40. openparser/_generated/models/batch_summary_counts.py +97 -0
  41. openparser/_generated/models/bounding_box.py +82 -0
  42. openparser/_generated/models/chunk_provenance_span.py +97 -0
  43. openparser/_generated/models/content_kind.py +10 -0
  44. openparser/_generated/models/create_extraction_pipeline_request.py +177 -0
  45. openparser/_generated/models/create_extraction_pipeline_request_grounding.py +8 -0
  46. openparser/_generated/models/create_extraction_pipeline_request_llm_options.py +84 -0
  47. openparser/_generated/models/create_extraction_pipeline_request_llm_options_reasoning_effort_type_1.py +13 -0
  48. openparser/_generated/models/create_extraction_pipeline_request_ocr_options.py +78 -0
  49. openparser/_generated/models/create_extraction_pipeline_request_schema.py +68 -0
  50. openparser/_generated/models/create_file_body.py +78 -0
  51. openparser/_generated/models/delete_extraction_pipeline_response.py +66 -0
  52. openparser/_generated/models/delete_file_response.py +57 -0
  53. openparser/_generated/models/error_body.py +106 -0
  54. openparser/_generated/models/error_body_details.py +65 -0
  55. openparser/_generated/models/error_response.py +65 -0
  56. openparser/_generated/models/extract_async_body.py +136 -0
  57. openparser/_generated/models/extract_batch_body.py +143 -0
  58. openparser/_generated/models/extract_batch_item.py +207 -0
  59. openparser/_generated/models/extract_batch_item_grounding.py +8 -0
  60. openparser/_generated/models/extract_batch_item_llm_options.py +84 -0
  61. openparser/_generated/models/extract_batch_item_llm_options_reasoning_effort_type_1.py +13 -0
  62. openparser/_generated/models/extract_batch_item_ocr_options.py +78 -0
  63. openparser/_generated/models/extract_batch_item_schema.py +65 -0
  64. openparser/_generated/models/extract_batch_request.py +99 -0
  65. openparser/_generated/models/extract_batch_request_items_item.py +201 -0
  66. openparser/_generated/models/extract_batch_request_items_item_grounding.py +8 -0
  67. openparser/_generated/models/extract_batch_request_items_item_llm_options.py +84 -0
  68. openparser/_generated/models/extract_batch_request_items_item_llm_options_reasoning_effort_type_1.py +13 -0
  69. openparser/_generated/models/extract_batch_request_items_item_ocr_options.py +78 -0
  70. openparser/_generated/models/extract_batch_request_items_item_schema.py +65 -0
  71. openparser/_generated/models/extract_batch_request_output_format.py +8 -0
  72. openparser/_generated/models/extract_request.py +226 -0
  73. openparser/_generated/models/extract_request_grounding.py +8 -0
  74. openparser/_generated/models/extract_request_llm_options.py +84 -0
  75. openparser/_generated/models/extract_request_llm_options_reasoning_effort_type_1.py +13 -0
  76. openparser/_generated/models/extract_request_ocr_options.py +78 -0
  77. openparser/_generated/models/extract_request_output_format.py +8 -0
  78. openparser/_generated/models/extract_request_schema.py +65 -0
  79. openparser/_generated/models/extract_sync_body.py +136 -0
  80. openparser/_generated/models/extraction_attempt.py +122 -0
  81. openparser/_generated/models/extraction_attempt_kind.py +8 -0
  82. openparser/_generated/models/extraction_attempt_status.py +9 -0
  83. openparser/_generated/models/extraction_chunk.py +184 -0
  84. openparser/_generated/models/extraction_citation.py +169 -0
  85. openparser/_generated/models/extraction_citation_granularity.py +8 -0
  86. openparser/_generated/models/extraction_citation_source_type.py +16 -0
  87. openparser/_generated/models/extraction_grounding_field.py +102 -0
  88. openparser/_generated/models/extraction_grounding_mode.py +8 -0
  89. openparser/_generated/models/extraction_grounding_result.py +87 -0
  90. openparser/_generated/models/extraction_pipeline.py +201 -0
  91. openparser/_generated/models/extraction_pipeline_list_response.py +76 -0
  92. openparser/_generated/models/extraction_pipeline_llm_options.py +84 -0
  93. openparser/_generated/models/extraction_pipeline_llm_options_reasoning_effort_type_0_type_1.py +13 -0
  94. openparser/_generated/models/extraction_pipeline_ocr_options.py +73 -0
  95. openparser/_generated/models/extraction_pipeline_schema.py +67 -0
  96. openparser/_generated/models/extraction_terminal_result.py +206 -0
  97. openparser/_generated/models/extraction_terminal_result_reasoning_effort_type_0.py +13 -0
  98. openparser/_generated/models/extraction_usage_totals.py +80 -0
  99. openparser/_generated/models/job.py +437 -0
  100. openparser/_generated/models/job_accepted.py +131 -0
  101. openparser/_generated/models/job_accepted_operation.py +8 -0
  102. openparser/_generated/models/job_extraction_schema.py +65 -0
  103. openparser/_generated/models/job_failure.py +98 -0
  104. openparser/_generated/models/job_failure_details.py +65 -0
  105. openparser/_generated/models/job_list_response.py +91 -0
  106. openparser/_generated/models/job_operation.py +10 -0
  107. openparser/_generated/models/job_progress.py +68 -0
  108. openparser/_generated/models/job_related_extractions_item.py +145 -0
  109. openparser/_generated/models/job_status.py +11 -0
  110. openparser/_generated/models/job_summary.py +264 -0
  111. openparser/_generated/models/json_schema_object.py +66 -0
  112. openparser/_generated/models/list_llm_models_mode.py +8 -0
  113. openparser/_generated/models/ocr_llm_model_catalog_entry.py +257 -0
  114. openparser/_generated/models/ocr_llm_model_catalog_entry_pricing.py +76 -0
  115. openparser/_generated/models/ocr_llm_model_catalog_entry_reasoning_type_0.py +134 -0
  116. openparser/_generated/models/ocr_llm_model_catalog_entry_reasoning_type_0_default_effort_type_0.py +13 -0
  117. openparser/_generated/models/ocr_llm_model_catalog_entry_reasoning_type_0_supported_efforts_type_0_item.py +13 -0
  118. openparser/_generated/models/ocr_llm_model_catalog_entry_recommendation.py +8 -0
  119. openparser/_generated/models/ocr_llm_models_response.py +150 -0
  120. openparser/_generated/models/ocr_llm_models_response_mode.py +8 -0
  121. openparser/_generated/models/ocr_model.py +7 -0
  122. openparser/_generated/models/ocr_model_catalog_entry.py +131 -0
  123. openparser/_generated/models/ocr_model_catalog_entry_availability.py +9 -0
  124. openparser/_generated/models/ocr_model_catalog_entry_capabilities.py +97 -0
  125. openparser/_generated/models/ocr_model_catalog_entry_capabilities_options.py +73 -0
  126. openparser/_generated/models/ocr_model_catalog_entry_option_defaults.py +73 -0
  127. openparser/_generated/models/ocr_model_catalog_entry_pricing.py +68 -0
  128. openparser/_generated/models/ocr_models_response.py +76 -0
  129. openparser/_generated/models/ocr_output_format.py +8 -0
  130. openparser/_generated/models/paddle_raw_profile.py +76 -0
  131. openparser/_generated/models/paddle_raw_profile_options.py +115 -0
  132. openparser/_generated/models/page_block.py +332 -0
  133. openparser/_generated/models/page_block_kind.py +9 -0
  134. openparser/_generated/models/page_block_polygon_type_0_item.py +65 -0
  135. openparser/_generated/models/parse_async_body.py +133 -0
  136. openparser/_generated/models/parse_batch_body.py +143 -0
  137. openparser/_generated/models/parse_batch_item.py +109 -0
  138. openparser/_generated/models/parse_batch_item_ocr_options.py +78 -0
  139. openparser/_generated/models/parse_batch_request.py +100 -0
  140. openparser/_generated/models/parse_batch_request_items_item.py +108 -0
  141. openparser/_generated/models/parse_batch_request_items_item_ocr_options.py +78 -0
  142. openparser/_generated/models/parse_batch_request_output_format.py +8 -0
  143. openparser/_generated/models/parse_request.py +112 -0
  144. openparser/_generated/models/parse_request_ocr_options.py +78 -0
  145. openparser/_generated/models/parse_request_output_format.py +8 -0
  146. openparser/_generated/models/parse_sync_body.py +133 -0
  147. openparser/_generated/models/parsed_document.py +187 -0
  148. openparser/_generated/models/public_file.py +117 -0
  149. openparser/_generated/models/raw_parse_result.py +110 -0
  150. openparser/_generated/models/raw_parse_result_result.py +65 -0
  151. openparser/_generated/models/region.py +224 -0
  152. openparser/_generated/models/region_content.py +155 -0
  153. openparser/_generated/models/region_polygon_type_0_item.py +65 -0
  154. openparser/_generated/models/region_type.py +16 -0
  155. openparser/_generated/models/suggest_schema_request.py +69 -0
  156. openparser/_generated/models/suggest_schema_response.py +74 -0
  157. openparser/_generated/models/suggest_schema_response_schema.py +66 -0
  158. openparser/_generated/models/update_extraction_pipeline_request.py +227 -0
  159. openparser/_generated/models/update_extraction_pipeline_request_grounding.py +8 -0
  160. openparser/_generated/models/update_extraction_pipeline_request_llm_options_type_0.py +84 -0
  161. openparser/_generated/models/update_extraction_pipeline_request_llm_options_type_0_reasoning_effort_type_1.py +13 -0
  162. openparser/_generated/models/update_extraction_pipeline_request_ocr_options_type_0.py +78 -0
  163. openparser/_generated/models/update_extraction_pipeline_request_schema.py +67 -0
  164. openparser/_generated/py.typed +1 -0
  165. openparser/_generated/types.py +53 -0
  166. openparser/_telemetry.py +34 -0
  167. openparser/client.py +541 -0
  168. openparser/errors.py +172 -0
  169. openparser/py.typed +0 -0
  170. openparser_sdk-0.0.1.dist-info/METADATA +178 -0
  171. openparser_sdk-0.0.1.dist-info/RECORD +173 -0
  172. openparser_sdk-0.0.1.dist-info/WHEEL +4 -0
  173. openparser_sdk-0.0.1.dist-info/licenses/LICENSE +201 -0
@@ -0,0 +1,180 @@
1
+ from http import HTTPStatus
2
+ from typing import Any, cast
3
+ from urllib.parse import quote
4
+
5
+ import httpx
6
+
7
+ from ...client import AuthenticatedClient, Client
8
+ from ...types import Response, UNSET
9
+ from ... import errors
10
+
11
+ from ...models.error_response import ErrorResponse
12
+ from ...models.extraction_pipeline_list_response import ExtractionPipelineListResponse
13
+ from typing import cast
14
+
15
+
16
+
17
+ def _get_kwargs(
18
+
19
+ ) -> dict[str, Any]:
20
+
21
+
22
+
23
+
24
+
25
+
26
+ _kwargs: dict[str, Any] = {
27
+ "method": "get",
28
+ "url": "/pipelines",
29
+ }
30
+
31
+
32
+ return _kwargs
33
+
34
+
35
+
36
+ def _parse_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> ErrorResponse | ExtractionPipelineListResponse | None:
37
+ if response.status_code == 200:
38
+ response_200 = ExtractionPipelineListResponse.from_dict(response.json())
39
+
40
+
41
+
42
+ return response_200
43
+
44
+ if response.status_code == 401:
45
+ response_401 = ErrorResponse.from_dict(response.json())
46
+
47
+
48
+
49
+ return response_401
50
+
51
+ if response.status_code == 403:
52
+ response_403 = ErrorResponse.from_dict(response.json())
53
+
54
+
55
+
56
+ return response_403
57
+
58
+ if response.status_code == 429:
59
+ response_429 = ErrorResponse.from_dict(response.json())
60
+
61
+
62
+
63
+ return response_429
64
+
65
+ if client.raise_on_unexpected_status:
66
+ raise errors.UnexpectedStatus(response.status_code, response.content)
67
+ else:
68
+ return None
69
+
70
+
71
+ def _build_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> Response[ErrorResponse | ExtractionPipelineListResponse]:
72
+ return Response(
73
+ status_code=HTTPStatus(response.status_code),
74
+ content=response.content,
75
+ headers=response.headers,
76
+ parsed=_parse_response(client=client, response=response),
77
+ )
78
+
79
+
80
+ def sync_detailed(
81
+ *,
82
+ client: AuthenticatedClient | Client,
83
+
84
+ ) -> Response[ErrorResponse | ExtractionPipelineListResponse]:
85
+ """ List saved extraction pipelines
86
+
87
+ Return active (non-deleted) pipelines for the authenticated tenant, newest updates first.
88
+
89
+ Raises:
90
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
91
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
92
+
93
+ Returns:
94
+ Response[ErrorResponse | ExtractionPipelineListResponse]
95
+ """
96
+
97
+
98
+ kwargs = _get_kwargs(
99
+
100
+ )
101
+
102
+ response = client.get_httpx_client().request(
103
+ **kwargs,
104
+ )
105
+
106
+ return _build_response(client=client, response=response)
107
+
108
+ def sync(
109
+ *,
110
+ client: AuthenticatedClient | Client,
111
+
112
+ ) -> ErrorResponse | ExtractionPipelineListResponse | None:
113
+ """ List saved extraction pipelines
114
+
115
+ Return active (non-deleted) pipelines for the authenticated tenant, newest updates first.
116
+
117
+ Raises:
118
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
119
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
120
+
121
+ Returns:
122
+ ErrorResponse | ExtractionPipelineListResponse
123
+ """
124
+
125
+
126
+ return sync_detailed(
127
+ client=client,
128
+
129
+ ).parsed
130
+
131
+ async def asyncio_detailed(
132
+ *,
133
+ client: AuthenticatedClient | Client,
134
+
135
+ ) -> Response[ErrorResponse | ExtractionPipelineListResponse]:
136
+ """ List saved extraction pipelines
137
+
138
+ Return active (non-deleted) pipelines for the authenticated tenant, newest updates first.
139
+
140
+ Raises:
141
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
142
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
143
+
144
+ Returns:
145
+ Response[ErrorResponse | ExtractionPipelineListResponse]
146
+ """
147
+
148
+
149
+ kwargs = _get_kwargs(
150
+
151
+ )
152
+
153
+ response = await client.get_async_httpx_client().request(
154
+ **kwargs
155
+ )
156
+
157
+ return _build_response(client=client, response=response)
158
+
159
+ async def asyncio(
160
+ *,
161
+ client: AuthenticatedClient | Client,
162
+
163
+ ) -> ErrorResponse | ExtractionPipelineListResponse | None:
164
+ """ List saved extraction pipelines
165
+
166
+ Return active (non-deleted) pipelines for the authenticated tenant, newest updates first.
167
+
168
+ Raises:
169
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
170
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
171
+
172
+ Returns:
173
+ ErrorResponse | ExtractionPipelineListResponse
174
+ """
175
+
176
+
177
+ return (await asyncio_detailed(
178
+ client=client,
179
+
180
+ )).parsed
@@ -0,0 +1,258 @@
1
+ from http import HTTPStatus
2
+ from typing import Any, cast
3
+ from urllib.parse import quote
4
+
5
+ import httpx
6
+
7
+ from ...client import AuthenticatedClient, Client
8
+ from ...types import Response, UNSET
9
+ from ... import errors
10
+
11
+ from ...models.error_response import ErrorResponse
12
+ from ...models.extraction_pipeline import ExtractionPipeline
13
+ from ...models.update_extraction_pipeline_request import UpdateExtractionPipelineRequest
14
+ from typing import cast
15
+
16
+
17
+
18
+ def _get_kwargs(
19
+ id: str,
20
+ *,
21
+ body: UpdateExtractionPipelineRequest,
22
+
23
+ ) -> dict[str, Any]:
24
+ headers: dict[str, Any] = {}
25
+
26
+
27
+
28
+
29
+
30
+
31
+ _kwargs: dict[str, Any] = {
32
+ "method": "patch",
33
+ "url": "/pipelines/{id}".format(id=quote(str(id), safe=""),),
34
+ }
35
+
36
+ _kwargs["json"] = body.to_dict()
37
+
38
+
39
+ headers["Content-Type"] = "application/json"
40
+
41
+ _kwargs["headers"] = headers
42
+ return _kwargs
43
+
44
+
45
+
46
+ def _parse_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> ErrorResponse | ExtractionPipeline | None:
47
+ if response.status_code == 200:
48
+ response_200 = ExtractionPipeline.from_dict(response.json())
49
+
50
+
51
+
52
+ return response_200
53
+
54
+ if response.status_code == 400:
55
+ response_400 = ErrorResponse.from_dict(response.json())
56
+
57
+
58
+
59
+ return response_400
60
+
61
+ if response.status_code == 401:
62
+ response_401 = ErrorResponse.from_dict(response.json())
63
+
64
+
65
+
66
+ return response_401
67
+
68
+ if response.status_code == 403:
69
+ response_403 = ErrorResponse.from_dict(response.json())
70
+
71
+
72
+
73
+ return response_403
74
+
75
+ if response.status_code == 404:
76
+ response_404 = ErrorResponse.from_dict(response.json())
77
+
78
+
79
+
80
+ return response_404
81
+
82
+ if response.status_code == 409:
83
+ response_409 = ErrorResponse.from_dict(response.json())
84
+
85
+
86
+
87
+ return response_409
88
+
89
+ if response.status_code == 422:
90
+ response_422 = ErrorResponse.from_dict(response.json())
91
+
92
+
93
+
94
+ return response_422
95
+
96
+ if response.status_code == 429:
97
+ response_429 = ErrorResponse.from_dict(response.json())
98
+
99
+
100
+
101
+ return response_429
102
+
103
+ if client.raise_on_unexpected_status:
104
+ raise errors.UnexpectedStatus(response.status_code, response.content)
105
+ else:
106
+ return None
107
+
108
+
109
+ def _build_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> Response[ErrorResponse | ExtractionPipeline]:
110
+ return Response(
111
+ status_code=HTTPStatus(response.status_code),
112
+ content=response.content,
113
+ headers=response.headers,
114
+ parsed=_parse_response(client=client, response=response),
115
+ )
116
+
117
+
118
+ def sync_detailed(
119
+ id: str,
120
+ *,
121
+ client: AuthenticatedClient | Client,
122
+ body: UpdateExtractionPipelineRequest,
123
+
124
+ ) -> Response[ErrorResponse | ExtractionPipeline]:
125
+ """ Update a saved extraction pipeline
126
+
127
+ Patch one or more fields, re-validate the effective merged config, and atomically
128
+ increment `version`. In-flight and completed jobs keep their admit-time snapshots.
129
+
130
+ Args:
131
+ id (str): Prefixed public id for a saved extraction pipeline (`oppl_…`).
132
+ body (UpdateExtractionPipelineRequest): Partial update; at least one field required.
133
+ Successful updates bump `version`.
134
+
135
+ Raises:
136
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
137
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
138
+
139
+ Returns:
140
+ Response[ErrorResponse | ExtractionPipeline]
141
+ """
142
+
143
+
144
+ kwargs = _get_kwargs(
145
+ id=id,
146
+ body=body,
147
+
148
+ )
149
+
150
+ response = client.get_httpx_client().request(
151
+ **kwargs,
152
+ )
153
+
154
+ return _build_response(client=client, response=response)
155
+
156
+ def sync(
157
+ id: str,
158
+ *,
159
+ client: AuthenticatedClient | Client,
160
+ body: UpdateExtractionPipelineRequest,
161
+
162
+ ) -> ErrorResponse | ExtractionPipeline | None:
163
+ """ Update a saved extraction pipeline
164
+
165
+ Patch one or more fields, re-validate the effective merged config, and atomically
166
+ increment `version`. In-flight and completed jobs keep their admit-time snapshots.
167
+
168
+ Args:
169
+ id (str): Prefixed public id for a saved extraction pipeline (`oppl_…`).
170
+ body (UpdateExtractionPipelineRequest): Partial update; at least one field required.
171
+ Successful updates bump `version`.
172
+
173
+ Raises:
174
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
175
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
176
+
177
+ Returns:
178
+ ErrorResponse | ExtractionPipeline
179
+ """
180
+
181
+
182
+ return sync_detailed(
183
+ id=id,
184
+ client=client,
185
+ body=body,
186
+
187
+ ).parsed
188
+
189
+ async def asyncio_detailed(
190
+ id: str,
191
+ *,
192
+ client: AuthenticatedClient | Client,
193
+ body: UpdateExtractionPipelineRequest,
194
+
195
+ ) -> Response[ErrorResponse | ExtractionPipeline]:
196
+ """ Update a saved extraction pipeline
197
+
198
+ Patch one or more fields, re-validate the effective merged config, and atomically
199
+ increment `version`. In-flight and completed jobs keep their admit-time snapshots.
200
+
201
+ Args:
202
+ id (str): Prefixed public id for a saved extraction pipeline (`oppl_…`).
203
+ body (UpdateExtractionPipelineRequest): Partial update; at least one field required.
204
+ Successful updates bump `version`.
205
+
206
+ Raises:
207
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
208
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
209
+
210
+ Returns:
211
+ Response[ErrorResponse | ExtractionPipeline]
212
+ """
213
+
214
+
215
+ kwargs = _get_kwargs(
216
+ id=id,
217
+ body=body,
218
+
219
+ )
220
+
221
+ response = await client.get_async_httpx_client().request(
222
+ **kwargs
223
+ )
224
+
225
+ return _build_response(client=client, response=response)
226
+
227
+ async def asyncio(
228
+ id: str,
229
+ *,
230
+ client: AuthenticatedClient | Client,
231
+ body: UpdateExtractionPipelineRequest,
232
+
233
+ ) -> ErrorResponse | ExtractionPipeline | None:
234
+ """ Update a saved extraction pipeline
235
+
236
+ Patch one or more fields, re-validate the effective merged config, and atomically
237
+ increment `version`. In-flight and completed jobs keep their admit-time snapshots.
238
+
239
+ Args:
240
+ id (str): Prefixed public id for a saved extraction pipeline (`oppl_…`).
241
+ body (UpdateExtractionPipelineRequest): Partial update; at least one field required.
242
+ Successful updates bump `version`.
243
+
244
+ Raises:
245
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
246
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
247
+
248
+ Returns:
249
+ ErrorResponse | ExtractionPipeline
250
+ """
251
+
252
+
253
+ return (await asyncio_detailed(
254
+ id=id,
255
+ client=client,
256
+ body=body,
257
+
258
+ )).parsed
@@ -0,0 +1,270 @@
1
+ import ssl
2
+ from typing import Any
3
+
4
+ from attrs import define, field, evolve
5
+ import httpx
6
+
7
+
8
+
9
+
10
+
11
+ @define
12
+ class Client:
13
+ """A class for keeping track of data related to the API
14
+
15
+ The following are accepted as keyword arguments and will be used to construct httpx Clients internally:
16
+
17
+ ``base_url``: The base URL for the API, all requests are made to a relative path to this URL
18
+
19
+ ``cookies``: A dictionary of cookies to be sent with every request
20
+
21
+ ``headers``: A dictionary of headers to be sent with every request
22
+
23
+ ``timeout``: The maximum amount of a time a request can take. API functions will raise
24
+ httpx.TimeoutException if this is exceeded.
25
+
26
+ ``verify_ssl``: Whether or not to verify the SSL certificate of the API server. This should be True in production,
27
+ but can be set to False for testing purposes.
28
+
29
+ ``follow_redirects``: Whether or not to follow redirects. Default value is False.
30
+
31
+ ``httpx_args``: A dictionary of additional arguments to be passed to the ``httpx.Client`` and ``httpx.AsyncClient`` constructor.
32
+
33
+
34
+ Attributes:
35
+ raise_on_unexpected_status: Whether or not to raise an errors.UnexpectedStatus if the API returns a
36
+ status code that was not documented in the source OpenAPI document. Can also be provided as a keyword
37
+ argument to the constructor.
38
+ """
39
+ raise_on_unexpected_status: bool = field(default=False, kw_only=True)
40
+ _base_url: str = field(alias="base_url")
41
+ _cookies: dict[str, str] = field(factory=dict, kw_only=True, alias="cookies")
42
+ _headers: dict[str, str] = field(factory=dict, kw_only=True, alias="headers")
43
+ _timeout: httpx.Timeout | None = field(default=None, kw_only=True, alias="timeout")
44
+ _verify_ssl: str | bool | ssl.SSLContext = field(default=True, kw_only=True, alias="verify_ssl")
45
+ _follow_redirects: bool = field(default=False, kw_only=True, alias="follow_redirects")
46
+ _httpx_args: dict[str, Any] = field(factory=dict, kw_only=True, alias="httpx_args")
47
+ _client: httpx.Client | None = field(default=None, init=False)
48
+ _async_client: httpx.AsyncClient | None = field(default=None, init=False)
49
+
50
+ def with_headers(self, headers: dict[str, str]) -> "Client":
51
+ """Get a new client matching this one with additional headers"""
52
+ if self._client is not None:
53
+ self._client.headers.update(headers)
54
+ if self._async_client is not None:
55
+ self._async_client.headers.update(headers)
56
+ return evolve(self, headers={**self._headers, **headers})
57
+
58
+ def with_cookies(self, cookies: dict[str, str]) -> "Client":
59
+ """Get a new client matching this one with additional cookies"""
60
+ if self._client is not None:
61
+ self._client.cookies.update(cookies)
62
+ if self._async_client is not None:
63
+ self._async_client.cookies.update(cookies)
64
+ return evolve(self, cookies={**self._cookies, **cookies})
65
+
66
+ def with_timeout(self, timeout: httpx.Timeout) -> "Client":
67
+ """Get a new client matching this one with a new timeout configuration"""
68
+ if self._client is not None:
69
+ self._client.timeout = timeout
70
+ if self._async_client is not None:
71
+ self._async_client.timeout = timeout
72
+ return evolve(self, timeout=timeout)
73
+
74
+ def set_httpx_client(self, client: httpx.Client) -> "Client":
75
+ """Manually set the underlying httpx.Client
76
+
77
+ **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
78
+ """
79
+ self._client = client
80
+ return self
81
+
82
+ def get_httpx_client(self) -> httpx.Client:
83
+ """Get the underlying httpx.Client, constructing a new one if not previously set"""
84
+ if self._client is None:
85
+ self._client = httpx.Client(
86
+ base_url=self._base_url,
87
+ cookies=self._cookies,
88
+ headers=self._headers,
89
+ timeout=self._timeout,
90
+ verify=self._verify_ssl,
91
+ follow_redirects=self._follow_redirects,
92
+ **self._httpx_args,
93
+ )
94
+ return self._client
95
+
96
+ def __enter__(self) -> "Client":
97
+ """Enter a context manager for self.client—you cannot enter twice (see httpx docs)"""
98
+ self.get_httpx_client().__enter__()
99
+ return self
100
+
101
+ def __exit__(self, *args: Any, **kwargs: Any) -> None:
102
+ """Exit a context manager for internal httpx.Client (see httpx docs)"""
103
+ self.get_httpx_client().__exit__(*args, **kwargs)
104
+
105
+ def set_async_httpx_client(self, async_client: httpx.AsyncClient) -> "Client":
106
+ """Manually set the underlying httpx.AsyncClient
107
+
108
+ **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
109
+ """
110
+ self._async_client = async_client
111
+ return self
112
+
113
+ def get_async_httpx_client(self) -> httpx.AsyncClient:
114
+ """Get the underlying httpx.AsyncClient, constructing a new one if not previously set"""
115
+ if self._async_client is None:
116
+ self._async_client = httpx.AsyncClient(
117
+ base_url=self._base_url,
118
+ cookies=self._cookies,
119
+ headers=self._headers,
120
+ timeout=self._timeout,
121
+ verify=self._verify_ssl,
122
+ follow_redirects=self._follow_redirects,
123
+ **self._httpx_args,
124
+ )
125
+ return self._async_client
126
+
127
+ async def __aenter__(self) -> "Client":
128
+ """Enter a context manager for underlying httpx.AsyncClient—you cannot enter twice (see httpx docs)"""
129
+ await self.get_async_httpx_client().__aenter__()
130
+ return self
131
+
132
+ async def __aexit__(self, *args: Any, **kwargs: Any) -> None:
133
+ """Exit a context manager for underlying httpx.AsyncClient (see httpx docs)"""
134
+ await self.get_async_httpx_client().__aexit__(*args, **kwargs)
135
+
136
+
137
+ @define
138
+ class AuthenticatedClient:
139
+ """A Client which has been authenticated for use on secured endpoints
140
+
141
+ The following are accepted as keyword arguments and will be used to construct httpx Clients internally:
142
+
143
+ ``base_url``: The base URL for the API, all requests are made to a relative path to this URL
144
+
145
+ ``cookies``: A dictionary of cookies to be sent with every request
146
+
147
+ ``headers``: A dictionary of headers to be sent with every request
148
+
149
+ ``timeout``: The maximum amount of a time a request can take. API functions will raise
150
+ httpx.TimeoutException if this is exceeded.
151
+
152
+ ``verify_ssl``: Whether or not to verify the SSL certificate of the API server. This should be True in production,
153
+ but can be set to False for testing purposes.
154
+
155
+ ``follow_redirects``: Whether or not to follow redirects. Default value is False.
156
+
157
+ ``httpx_args``: A dictionary of additional arguments to be passed to the ``httpx.Client`` and ``httpx.AsyncClient`` constructor.
158
+
159
+
160
+ Attributes:
161
+ raise_on_unexpected_status: Whether or not to raise an errors.UnexpectedStatus if the API returns a
162
+ status code that was not documented in the source OpenAPI document. Can also be provided as a keyword
163
+ argument to the constructor.
164
+ token: The token to use for authentication
165
+ prefix: The prefix to use for the Authorization header
166
+ auth_header_name: The name of the Authorization header
167
+ """
168
+
169
+ raise_on_unexpected_status: bool = field(default=False, kw_only=True)
170
+ _base_url: str = field(alias="base_url")
171
+ _cookies: dict[str, str] = field(factory=dict, kw_only=True, alias="cookies")
172
+ _headers: dict[str, str] = field(factory=dict, kw_only=True, alias="headers")
173
+ _timeout: httpx.Timeout | None = field(default=None, kw_only=True, alias="timeout")
174
+ _verify_ssl: str | bool | ssl.SSLContext = field(default=True, kw_only=True, alias="verify_ssl")
175
+ _follow_redirects: bool = field(default=False, kw_only=True, alias="follow_redirects")
176
+ _httpx_args: dict[str, Any] = field(factory=dict, kw_only=True, alias="httpx_args")
177
+ _client: httpx.Client | None = field(default=None, init=False)
178
+ _async_client: httpx.AsyncClient | None = field(default=None, init=False)
179
+
180
+ token: str
181
+ prefix: str = "Bearer"
182
+ auth_header_name: str = "Authorization"
183
+
184
+ def with_headers(self, headers: dict[str, str]) -> "AuthenticatedClient":
185
+ """Get a new client matching this one with additional headers"""
186
+ if self._client is not None:
187
+ self._client.headers.update(headers)
188
+ if self._async_client is not None:
189
+ self._async_client.headers.update(headers)
190
+ return evolve(self, headers={**self._headers, **headers})
191
+
192
+ def with_cookies(self, cookies: dict[str, str]) -> "AuthenticatedClient":
193
+ """Get a new client matching this one with additional cookies"""
194
+ if self._client is not None:
195
+ self._client.cookies.update(cookies)
196
+ if self._async_client is not None:
197
+ self._async_client.cookies.update(cookies)
198
+ return evolve(self, cookies={**self._cookies, **cookies})
199
+
200
+ def with_timeout(self, timeout: httpx.Timeout) -> "AuthenticatedClient":
201
+ """Get a new client matching this one with a new timeout configuration"""
202
+ if self._client is not None:
203
+ self._client.timeout = timeout
204
+ if self._async_client is not None:
205
+ self._async_client.timeout = timeout
206
+ return evolve(self, timeout=timeout)
207
+
208
+ def set_httpx_client(self, client: httpx.Client) -> "AuthenticatedClient":
209
+ """Manually set the underlying httpx.Client
210
+
211
+ **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
212
+ """
213
+ self._client = client
214
+ return self
215
+
216
+ def get_httpx_client(self) -> httpx.Client:
217
+ """Get the underlying httpx.Client, constructing a new one if not previously set"""
218
+ if self._client is None:
219
+ self._headers[self.auth_header_name] = f"{self.prefix} {self.token}" if self.prefix else self.token
220
+ self._client = httpx.Client(
221
+ base_url=self._base_url,
222
+ cookies=self._cookies,
223
+ headers=self._headers,
224
+ timeout=self._timeout,
225
+ verify=self._verify_ssl,
226
+ follow_redirects=self._follow_redirects,
227
+ **self._httpx_args,
228
+ )
229
+ return self._client
230
+
231
+ def __enter__(self) -> "AuthenticatedClient":
232
+ """Enter a context manager for self.client—you cannot enter twice (see httpx docs)"""
233
+ self.get_httpx_client().__enter__()
234
+ return self
235
+
236
+ def __exit__(self, *args: Any, **kwargs: Any) -> None:
237
+ """Exit a context manager for internal httpx.Client (see httpx docs)"""
238
+ self.get_httpx_client().__exit__(*args, **kwargs)
239
+
240
+ def set_async_httpx_client(self, async_client: httpx.AsyncClient) -> "AuthenticatedClient":
241
+ """Manually set the underlying httpx.AsyncClient
242
+
243
+ **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
244
+ """
245
+ self._async_client = async_client
246
+ return self
247
+
248
+ def get_async_httpx_client(self) -> httpx.AsyncClient:
249
+ """Get the underlying httpx.AsyncClient, constructing a new one if not previously set"""
250
+ if self._async_client is None:
251
+ self._headers[self.auth_header_name] = f"{self.prefix} {self.token}" if self.prefix else self.token
252
+ self._async_client = httpx.AsyncClient(
253
+ base_url=self._base_url,
254
+ cookies=self._cookies,
255
+ headers=self._headers,
256
+ timeout=self._timeout,
257
+ verify=self._verify_ssl,
258
+ follow_redirects=self._follow_redirects,
259
+ **self._httpx_args,
260
+ )
261
+ return self._async_client
262
+
263
+ async def __aenter__(self) -> "AuthenticatedClient":
264
+ """Enter a context manager for underlying httpx.AsyncClient—you cannot enter twice (see httpx docs)"""
265
+ await self.get_async_httpx_client().__aenter__()
266
+ return self
267
+
268
+ async def __aexit__(self, *args: Any, **kwargs: Any) -> None:
269
+ """Exit a context manager for underlying httpx.AsyncClient (see httpx docs)"""
270
+ await self.get_async_httpx_client().__aexit__(*args, **kwargs)